Update app.py

app.py

@@ -24,7 +24,7 @@ MODEL_OPTIONS = [
 # History storage
 history = []
 
-def generate_comparisons_with_history(input_text, selected_models):
+def generate_comparisons_with_history(input_text, selected_models, history_state):
     global history
     results = {}
     for model in selected_models:
@@ -64,20 +64,34 @@ def generate_comparisons_with_history(input_text, selected_models):
     }
     history.append(history_entry)
     
-    return results, history
+    # Update the history state
+    history_state = history
+    
+    return results, history_state
+
+def clear_history():
+    global history
+    history = []
+    return history
 
 # Create Gradio interface with multiple model selection and history
-iface = gr.Interface(
-    fn=generate_comparisons_with_history,
-    inputs=[
-        gr.Textbox(lines=2, label="Input Text", placeholder="Enter your query here"),
-        gr.CheckboxGroup(choices=MODEL_OPTIONS, label="Select Models", value=[MODEL_OPTIONS[0]])
-    ],
-    outputs=[
-        gr.JSON(label="Model Comparisons"),
-        gr.JSON(label="History")
-    ],
-    title="Compare Outputs and Maintain History"
-)
+with gr.Blocks() as demo:
+    input_text = gr.Textbox(lines=2, label="Input Text", placeholder="Enter your query here")
+    selected_models = gr.CheckboxGroup(choices=MODEL_OPTIONS, label="Select Models", value=[MODEL_OPTIONS[0]])
+    
+    # Define output components with scrollable containers
+    output_comparisons = gr.JSON(label="Model Comparisons", elem_id="output-comparisons")
+    output_history = gr.JSON(label="History", elem_id="output-history")
+    
+    clear_history_button = gr.Button("Clear History")
+
+    # Add scrollbar to the output JSON areas with CSS
+    output_comparisons.style(height="300px", overflow="auto")
+    output_history.style(height="300px", overflow="auto")
+
+    clear_history_button.click(clear_history, outputs=output_history)
+    
+    demo.submit(generate_comparisons_with_history, inputs=[input_text, selected_models, gr.State()], outputs=[output_comparisons, output_history])
+
+demo.launch()
 
-iface.launch()