Update app.py

app.py

@@ -48,6 +48,7 @@ def respond(
 
     return responses
 
+
 # Build Gradio app
 def create_demo():
     with gr.Blocks() as demo:
@@ -59,6 +60,7 @@ def create_demo():
             """
         )
 
+        # Input Section
         with gr.Row():
             system_message = gr.Textbox(
                 value="You are a helpful assistant providing answers for technical and customer support queries.",
@@ -73,6 +75,7 @@ def create_demo():
                 minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
             )
 
+        # Model Selection Section
         with gr.Row():
             selected_models = gr.CheckboxGroup(
                 choices=list(model_options.keys()),
@@ -80,30 +83,24 @@ def create_demo():
                 value=["Llama-3.1-70B", "Qwen-2.5-1.5B-Instruct"],  # Default models
             )
 
-        submit_button = gr.Button("Generate Responses")
+        # Dynamic Response Section
+        response_boxes = []
+        vote_buttons = []
+        vote_counts = []
 
-        with gr.Row():
-            response_boxes = []
-            vote_buttons = []
-            vote_counts = []
-
-            # Dynamically create response sections for each model
-            for model_name in model_options.keys():
-                with gr.Column(visible=False) as column:  # Initially hide unused models
-                    response_box = gr.Textbox(label=f"Response from {model_name}")
+        def generate_response_boxes(models):
+            response_boxes.clear()
+            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, column, response_box, vote_button, vote_count))
+                    response_boxes.append((model_name, response_column, response_box, vote_button, vote_count))
+            return response_boxes
 
-        # Define visibility and update functions dynamically
-        def update_model_visibility(models):
-            for model_name, column, *_ in response_boxes:
-                column.visible = model_name in models
-
-        def handle_votes(vote_counts, model_name):
-            index = list(model_options.keys()).index(model_name)
-            vote_counts[index] += 1
-            return vote_counts
+        # 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(
@@ -114,37 +111,40 @@ def create_demo():
             )
             outputs = []
             for model_name, _, response_box, *_ in response_boxes:
-                if model_name in responses:
-                    outputs.append(responses[model_name])
-                else:
-                    outputs.append("")
+                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
+
+        # 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[2] for response in response_boxes],
         )
 
-        vote_state = gr.State([0] * len(model_options))  # Initialize one vote count per model
-        
+        # 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 state and display vote count
+                outputs=[vote_state, vote_count],
             )
 
-
-        # Update model visibility when the model selection changes
+        # Update response boxes when models are selected
         selected_models.change(
-            update_model_visibility,
+            generate_response_boxes,
             inputs=[selected_models],
             outputs=[response[1] for response in response_boxes],
         )
 
     return demo
 
+
 if __name__ == "__main__":
     demo = create_demo()
     demo.launch()