Update app.py

app.py

@@ -48,6 +48,13 @@ 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():
@@ -86,24 +93,6 @@ def create_demo():
         # Dynamic Response Section
         response_boxes = []
 
-        def generate_response_boxes(models):
-            # 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}")
-                    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")
-        vote_state = gr.State([0] * len(model_options))  # Initialize votes for all models
-
-        # Generate responses
         def generate_responses(
             message, history, system_message, max_tokens, temperature, top_p, selected_models
         ):
@@ -113,31 +102,20 @@ def create_demo():
                 message, history, system_message, max_tokens, temperature, top_p, selected_models
             )
             outputs = []
-            for model_name, _, response_box, *_ in response_boxes:
+            for model_name in selected_models:
                 outputs.append(responses.get(model_name, ""))
             return outputs
 
-        # Handle votes
-        def handle_votes(votes, model_name):
-            index = list(model_options.keys()).index(model_name)
-            votes[index] += 1
-            return votes
+        # 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_box[2] for response_box in response_boxes],  # Only output the response boxes
+            outputs=response_boxes,  # Only output the response boxes
         )
 
-        # Link voting buttons to handle votes
-        for model_name, _, _, vote_button, vote_count in response_boxes:
-            vote_button.click(
-                lambda votes, name=model_name: handle_votes(votes, name),
-                inputs=[vote_state],
-                outputs=[vote_state, vote_count],
-            )
-
         # Update response boxes when models are selected
         selected_models.change(
             lambda selected: generate_response_boxes(selected),
@@ -147,7 +125,6 @@ def create_demo():
 
     return demo
 
-
 if __name__ == "__main__":
     demo = create_demo()
-    demo.launch()
+    demo.launch()