Update app.py

app.py

@@ -2,9 +2,10 @@ 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("OpenRouter_API_KEY")
+API_KEY = os.getenv("OpenRounter_API_KEY")
 
 # Define available models for selection
 MODEL_OPTIONS = [
@@ -24,7 +25,7 @@ MODEL_OPTIONS = [
 # History storage
 history = []
 
-def generate_comparisons_with_history(input_text, selected_models):
+def generate_model_outputs_with_history(input_text, selected_models):
     global history
     results = {}
     for model in selected_models:
@@ -35,7 +36,7 @@ def generate_comparisons_with_history(input_text, selected_models):
                 "Content-Type": "application/json"
             },
             data=json.dumps({
-                "model": model,  # Use the current model
+                "model": model,
                 "messages": [{"role": "user", "content": input_text}],
                 "top_p": 1,
                 "temperature": 1,
@@ -45,7 +46,6 @@ def generate_comparisons_with_history(input_text, selected_models):
                 "top_k": 0,
             })
         )
-        
         # Parse the response
         if response.status_code == 200:
             try:
@@ -60,24 +60,86 @@ def generate_comparisons_with_history(input_text, selected_models):
     history_entry = {
         "input": input_text,
         "selected_models": selected_models,
-        "outputs": results
+        "outputs": results,
+        "timestamp": str(datetime.datetime.now())
     }
     history.append(history_entry)
     
-    return results, history
+    return [results.get(model, "No output") for model in selected_models], 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"
-)
+# Create a dynamic number of outputs based on model selection
+def create_outputs(selected_models):
+    return [
+        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
+        )
+        for model in selected_models
+    ]
+
+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
+
+    def update_outputs_and_history(selected_models):
+        # Dynamically create outputs for each selected model
+        output_components = create_outputs(selected_models)
+        output_placeholder.set(output_components)
+        return output_components
+
+    # Button to generate outputs and maintain history
+    generate_button = gr.Button("Generate Outputs")
+    generate_button.click(
+        fn=generate_model_outputs_with_history,
+        inputs=[input_text, selected_models],
+        outputs=[update_outputs_and_history(selected_models), history_placeholder],
+    )
+
+    # Create tabs for each selected model and display scrollable outputs
+    with gr.TabbedInterface() as tabs:
+        with gr.Tab("History"):
+            gr.JSON(label="History", value=history_placeholder)
+        for model in MODEL_OPTIONS:
+            with gr.Tab(f"{model} Output"):
+                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
+                )
+
+    # 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])
+
+    # Export History button
+    export_history_button = gr.Button("Export History")
+    export_history_button.click(fn=export_history, outputs=[gr.Textbox(value="History Exported Successfully!")])
 
-iface.launch()
+demo.launch()