Update app.py

app.py

@@ -197,36 +197,37 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name, sy
 # Create the chat interface using Gradio Blocks
 with gr.Blocks() as demo:
     gr.Markdown("# Chat with Model")
+
+    gr.Markdown("## Context")
+    model_dropdown = gr.Dropdown(choices=list(models.keys()), label="Select Model", value=selected_model)
+    user_dropdown = gr.Dropdown(choices=user_names, label="Select User Name")
+    initial_story = stories[0]["title"] if stories else None
+    story_dropdown = gr.Dropdown(choices=[story["title"] for story in stories], label="Select Story", value=initial_story)
+    system_prompt_dropdown = gr.Dropdown(choices=system_prompts, label="Select System Prompt", value=system_prompts[0])
+    send_story_button = gr.Button("Send Story")
+
+    gr.Markdown("## Chat")
+    selected_story_textbox = gr.Textbox(label="Selected Story", lines=10, interactive=False)
+    with gr.Row():
+        with gr.Column(scale=1):
+            chatbot_input = gr.Textbox(placeholder="Type your message here...", label="User Input")
+            send_message_button = gr.Button("Send")
+        with gr.Column(scale=2):
+            chatbot_output = gr.Chatbot(label="Chat History")
     
-    with gr.Tab("Context"):
-        model_dropdown = gr.Dropdown(choices=list(models.keys()), label="Select Model", value=selected_model)
-        user_dropdown = gr.Dropdown(choices=user_names, label="Select User Name")
-        initial_story = stories[0]["title"] if stories else None
-        story_dropdown = gr.Dropdown(choices=[story["title"] for story in stories], label="Select Story", value=initial_story)
-        system_prompt_dropdown = gr.Dropdown(choices=system_prompts, label="Select System Prompt", value=system_prompts[0])
-        send_story_button = gr.Button("Send Story")
-    
-    with gr.Tab("Chat"):
-        selected_story_textbox = gr.Textbox(label="Selected Story", lines=10, interactive=False)
-        with gr.Row():
-            with gr.Column(scale=1):
-                chatbot_input = gr.Textbox(placeholder="Type your message here...", label="User Input")
-                send_message_button = gr.Button("Send")
-            with gr.Column(scale=2):
-                chatbot_output = gr.Chatbot(label="Chat History")
-    
-    with gr.Tab("Evaluation"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                score_input = gr.Slider(minimum=0, maximum=5, step=1, label="Score")
-                comment_input = gr.Textbox(placeholder="Add a comment...", label="Comment")
-                save_button = gr.Button("Save Score and Comment")
-        data_table = gr.DataFrame(headers=["User Input", "Chat Response", "Score", "Comment"])
+    gr.Markdown("## Evaluation")
+    with gr.Row():
+        with gr.Column(scale=1):
+            score_input = gr.Slider(minimum=0, maximum=5, step=1, label="Score")
+            comment_input = gr.Textbox(placeholder="Add a comment...", label="Comment")
+            save_button = gr.Button("Save Score and Comment")
+    data_table = gr.DataFrame(headers=["User Input", "Chat Response", "Score", "Comment"])
 
     chat_history_json = gr.JSON(value=[], visible=False)
+    interaction_counter_json = gr.JSON(value=0, visible=False)
     
-    send_story_button.click(fn=send_selected_story, inputs=[story_dropdown, model_dropdown, system_prompt_dropdown], outputs=[chatbot_output, chat_history_json, data_table, selected_story_textbox, gr.JSON(value=0)])
-    send_message_button.click(fn=interact, inputs=[chatbot_input, chat_history_json, gr.JSON(value=0)], outputs=[chatbot_input, chatbot_output, chat_history_json, gr.JSON(value=0)])
+    send_story_button.click(fn=send_selected_story, inputs=[story_dropdown, model_dropdown, system_prompt_dropdown], outputs=[chatbot_output, chat_history_json, data_table, selected_story_textbox, interaction_counter_json])
+    send_message_button.click(fn=interact, inputs=[chatbot_input, chat_history_json, interaction_counter_json], outputs=[chatbot_input, chatbot_output, chat_history_json, interaction_counter_json])
     save_button.click(fn=save_comment_score, inputs=[chatbot_output, score_input, comment_input, story_dropdown, user_dropdown, system_prompt_dropdown], outputs=[data_table, comment_input])
 
 demo.launch()