Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 import requests
 import json
 import os
-import datetime
 
 # Retrieve the OpenRouter API Key from the Space secrets
-API_KEY = os.getenv("OpenRounter_API_KEY")
+API_KEY = os.getenv("OpenRouter_API_KEY")
 
 # Define available models for selection
 MODEL_OPTIONS = [
@@ -25,7 +24,7 @@ MODEL_OPTIONS = [
 # History storage
 history = []
 
-def generate_model_outputs_with_history(input_text, selected_models):
+def generate_comparisons_with_history(input_text, selected_models):
     global history
     results = {}
     for model in selected_models:
@@ -36,7 +35,7 @@ def generate_model_outputs_with_history(input_text, selected_models):
                 "Content-Type": "application/json"
             },
             data=json.dumps({
-                "model": model,
+                "model": model,  # Use the current model
                 "messages": [{"role": "user", "content": input_text}],
                 "top_p": 1,
                 "temperature": 1,
@@ -46,6 +45,7 @@ def generate_model_outputs_with_history(input_text, selected_models):
                 "top_k": 0,
             })
         )
+        
         # Parse the response
         if response.status_code == 200:
             try:
@@ -60,78 +60,24 @@ def generate_model_outputs_with_history(input_text, selected_models):
     history_entry = {
         "input": input_text,
         "selected_models": selected_models,
-        "outputs": results,
-        "timestamp": str(datetime.datetime.now())
+        "outputs": results
     }
     history.append(history_entry)
     
-    return results  # Return the results for the selected models
-
-# Create a dynamic number of outputs based on model selection
-def create_outputs(selected_models):
-    output_components = []
-    for model in selected_models:
-        output_components.append(
-            gr.Textbox(
-                label=f"Output from {model}",
-                interactive=False,
-                lines=5,  # Adjust lines as needed
-                max_lines=10,  # Max lines before scrolling
-                show_label=False,  # Hide label in the tabbed view
-                elem_id=f"output_{model}",  # Unique ID for styling
-                css=".output-window { overflow-y: auto; max-height: 200px; }"  # Styling for scrollable output
-            )
-        )
-    return output_components
-
-def clear_history():
-    global history
-    history = []
-    return "History Cleared!", []
-
-def export_history():
-    global history
-    # Save history to a file (e.g., JSON)
-    file_name = f"history_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
-    with open(file_name, 'w') as f:
-        json.dump(history, f, indent=4)
-    return f"History exported to {file_name}", []
-
-# Gradio interface with dynamic outputs and history
-with gr.Blocks() as demo:
-    with gr.Row():
-        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]])
-
-    output_placeholder = gr.State([])  # Placeholder for dynamic output components
-    history_placeholder = gr.State(history)  # Maintain history state
-
-    # Button to generate outputs and maintain history
-    generate_button = gr.Button("Generate Outputs")
-    
-    def generate_and_update(input_text, selected_models):
-        # Generate the output and dynamically update the outputs
-        results = generate_model_outputs_with_history(input_text, selected_models)
-        output_components = create_outputs(selected_models)
-        
-        # Update the output components with the actual model outputs
-        for i, model in enumerate(selected_models):
-            output_components[i].update(value=results.get(model, "No response"))
-        
-        return output_components, results  # Return dynamic outputs and results
-    
-    generate_button.click(
-        fn=generate_and_update,
-        inputs=[input_text, selected_models],
-        outputs=[output_placeholder, history_placeholder]
-    )
-
-    # Clear History button
-    clear_history_button = gr.Button("Clear History")
-    clear_history_button.click(fn=clear_history, outputs=[gr.Textbox(value="History Cleared!"), history_placeholder])
+    return results, history
 
-    # Export History button
-    export_history_button = gr.Button("Export History")
-    export_history_button.click(fn=export_history, outputs=[gr.Textbox(value="History Exported Successfully!")])
+# 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"
+)
 
-demo.launch()
+iface.launch()