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 = [
@@ -59,15 +60,40 @@ def generate_model_outputs_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.get(model, "No output") for model in selected_models], history
+    return results  # Return the results for the selected models
 
 # 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) for model in 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:
@@ -78,22 +104,26 @@ with gr.Blocks() as demo:
     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")
+    
+    def generate_and_update(input_text, selected_models):
+        results = generate_model_outputs_with_history(input_text, selected_models)
+        output_components = create_outputs(selected_models)
+        return output_components, results
+    
     generate_button.click(
-        fn=generate_model_outputs_with_history,
+        fn=generate_and_update,
         inputs=[input_text, selected_models],
-        outputs=[update_outputs_and_history(selected_models), history_placeholder],
+        outputs=[output_placeholder, history_placeholder]
     )
 
-    # Display the history
-    with gr.Row():
-        gr.JSON(label="History", value=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])
+
+    # Export History button
+    export_history_button = gr.Button("Export History")
+    export_history_button.click(fn=export_history, outputs=[gr.Textbox(value="History Exported Successfully!")])
 
 demo.launch()