Update app.py

app.py

@@ -48,14 +48,6 @@ def respond(
 
     return responses
 
-# Function to generate response boxes dynamically
-def generate_response_boxes(models):
-    response_boxes = []
-    for model_name in models:
-        response_box = gr.Textbox(label=f"Response from {model_name}", interactive=False)
-        response_boxes.append(response_box)
-    return response_boxes
-
 # Build Gradio app
 def create_demo():
     with gr.Blocks() as demo:
@@ -82,45 +74,34 @@ def create_demo():
                 minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
             )
 
-        # Model Selection Section (no maximum_choices)
-        with gr.Row():
-            selected_models = gr.CheckboxGroup(
-                choices=list(model_options.keys()),
-                label="Select two models to compare",
-                value=["Llama-3.1-70B", "Qwen-2.5-1.5B-Instruct"],  # Default models
-            )
+        # Model Selection Section
+        selected_models = gr.CheckboxGroup(
+            choices=list(model_options.keys()),
+            label="Select exactly two models to compare",
+            value=["Llama-3.1-70B", "Qwen-2.5-1.5B-Instruct"],  # Default models
+        )
 
         # Dynamic Response Section
-        response_boxes = []
+        response_box1 = gr.Textbox(label="Response from Model 1", interactive=False)
+        response_box2 = gr.Textbox(label="Response from Model 2", interactive=False)
 
+        # Function to generate responses
         def generate_responses(
-            message, history, system_message, max_tokens, temperature, top_p, selected_models
+            message, system_message, max_tokens, temperature, top_p, selected_models
         ):
             if len(selected_models) != 2:
-                return "Error: Please select exactly two models to compare."
+                return "Error: Please select exactly two models to compare.", ""
             responses = respond(
-                message, history, system_message, max_tokens, temperature, top_p, selected_models
+                message, [], system_message, max_tokens, temperature, top_p, selected_models
             )
-            outputs = []
-            for model_name in selected_models:
-                outputs.append(responses.get(model_name, ""))
-            return outputs
+            return responses.get(selected_models[0], ""), responses.get(selected_models[1], "")
 
         # Add a button for generating responses
         submit_button = gr.Button("Generate Responses")
-
-        # Link button click to generate responses
         submit_button.click(
             generate_responses,
-            inputs=[user_message, gr.State([]), system_message, max_tokens, temperature, top_p, selected_models],
-            outputs=response_boxes,  # Only output the response boxes
-        )
-
-        # Update response boxes when models are selected
-        selected_models.change(
-            lambda selected: generate_response_boxes(selected),
-            inputs=[selected_models],
-            outputs=response_boxes,  # Update response boxes dynamically
+            inputs=[user_message, system_message, max_tokens, temperature, top_p, selected_models],
+            outputs=[response_box1, response_box2],  # Link to response boxes
         )
 
     return demo