Update app.py

app.py

@@ -21,7 +21,11 @@ MODEL_OPTIONS = [
     "liquid/lfm-40b:free"
 ]
 
-def generate_comparisons(input_text, selected_models):
+# History storage
+history = []
+
+def generate_comparisons_with_history(input_text, selected_models):
+    global history
     results = {}
     for model in selected_models:
         response = requests.post(
@@ -52,17 +56,28 @@ def generate_comparisons(input_text, selected_models):
         else:
             results[model] = f"Error: {response.status_code}, {response.text}"
     
-    return results
+    # Add input and results to history
+    history_entry = {
+        "input": input_text,
+        "selected_models": selected_models,
+        "outputs": results
+    }
+    history.append(history_entry)
+    
+    return results, history
 
-# Create Gradio interface with multiple model selection
+# Create Gradio interface with multiple model selection and history
 iface = gr.Interface(
-    fn=generate_comparisons,
+    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"),
-    title="Compare Outputs from Multiple Models"
+    outputs=[
+        gr.JSON(label="Model Comparisons"),
+        gr.JSON(label="History")
+    ],
+    title="Compare Outputs and Maintain History"
 )
 
 iface.launch()