update

pages/leaderboard.py

@@ -32,7 +32,7 @@ def update_leaderboard(sort_by):
 
 def create_leaderboard():
     with gr.Blocks(css="#leaderboard table { width: 100%; } #leaderboard th, #leaderboard td { padding: 8px; text-align: left; }") as demo:
-        gr.Markdown("# 🏆 Chris-Project Summarization Arena Leaderboard")
+        gr.Markdown("# 🏆 Summarization Arena Leaderboard")
 
         with gr.Row():
             gr.Markdown("[Blog](placeholder) | [GitHub](placeholder) | [Paper](placeholder) | [Dataset](placeholder) | [Twitter](placeholder) | [Discord](placeholder)")

pages/summarization_playground.py

@@ -75,7 +75,7 @@ def update_input(example):
     return examples[example]
 
 def create_summarization_interface():
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme=gr.themes.Soft(spacing_size="sm",text_size="sm")) as demo:
         gr.Markdown("## This is a playground to test prompts for clinical dialogue summarizations")
 
         with gr.Row():
@@ -86,7 +86,24 @@ def create_summarization_interface():
         datapoint = random.choice(dataset)
         input_text = gr.Textbox(label="Input Dialogue", lines=10, placeholder="Enter text here...", value=datapoint['section_text'] + '\n\nDialogue:\n' + datapoint['dialogue'])
         submit_button = gr.Button("✨ Submit ✨")
-        output = gr.Markdown(line_breaks=True)
+
+        with gr.Row():
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("<div style='border: 4px solid white; padding: 3px; border-radius: 5px;width:100px;padding-top: 0.5px;padding-bottom: 10px;'><h3>Prompt 👥</h3></center></div>")
+                    prompt = gr.Textbox(label="Input", lines=6, placeholder = "Enter the Patient-Doctor conversation here.",elem_classes=["bordered-text"])
+                    context = gr.Textbox(label="Context", placeholder="Enter relevant context about the patient medical history.",elem_classes="bordered-text")
+                    token = gr.Textbox(label="Token",elem_classes="bordered-text")
+                with gr.Column():
+                    gr.Markdown("<div style='border: 4px solid white; padding: 2px; border-radius: 5px;width:130px;padding-bottom: 10px;'><b><h3>Parameters 📈</h3></center></b></div>")
+                    with gr.Column():
+                        topK = gr.Textbox(label="TopP",elem_classes="bordered-text")
+                        topP = gr.Textbox(label="TopK",elem_classes="bordered-text")
+                        temperature = gr.Textbox(label="Temperature",elem_classes="parameter-text")
+                        max_new_tokens = gr.Textbox(label="Max New Tokens",elem_classes="parameter-text")
+                        do_sample = gr.Dropdown(['Default','None'],label="Do Sample",elem_classes="parameter-text")
+                        return_text = gr.Dropdown(['Default','None'],label="Return Text",elem_classes="parameter-text")
+            output = gr.Markdown(line_breaks=True)
 
         example_dropdown.change(update_input, inputs=[example_dropdown], outputs=[input_text])
         submit_button.click(process_input, inputs=[input_text, model_dropdown, Template_text], outputs=[output])