Update app.py

app.py

@@ -90,13 +90,14 @@ def create_demo():
             # Clear current components
             response_boxes.clear()
             # Add new response boxes dynamically for only two models
+            columns = []
             for model_name in models:
                 with gr.Column() as response_column:
                     response_box = gr.Textbox(label=f"Response from {model_name}", interactive=False)
                     vote_button = gr.Button(f"Vote for {model_name}")
                     vote_count = gr.Number(value=0, label=f"Votes for {model_name}")
-                    response_boxes.append((model_name, response_column, response_box, vote_button, vote_count))
-            return [gr.update(visible=True) for _ in response_boxes]
+                    columns.append((model_name, response_column, response_box, vote_button, vote_count))
+            return columns
 
         # Add a button for generating responses
         submit_button = gr.Button("Generate Responses")
@@ -126,7 +127,7 @@ def create_demo():
         submit_button.click(
             generate_responses,
             inputs=[user_message, gr.State([]), system_message, max_tokens, temperature, top_p, selected_models],
-            outputs=[response[2] for response in response_boxes],
+            outputs=[response_box[2] for response_box in response_boxes],  # Only output the response boxes
         )
 
         # Link voting buttons to handle votes
@@ -139,9 +140,9 @@ def create_demo():
 
         # Update response boxes when models are selected
         selected_models.change(
-            generate_response_boxes,
+            lambda selected: generate_response_boxes(selected),
             inputs=[selected_models],
-            outputs=[gr.update(visible=True) for _ in response_boxes],
+            outputs=response_boxes,  # Update response boxes dynamically
         )
 
     return demo