Update app.py

app.py

@@ -3,39 +3,59 @@ import requests
 import json
 import os
 
-# Retrieve the Open Router API Key from the Space secrets
-API_KEY = os.getenv("OpenRounter_API_KEY")
+# Retrieve the OpenRouter API Key from the Space secrets
+API_KEY = os.getenv("OpenRouter_API_KEY")
 
-def generate_text(input_text):
-    response = requests.post(
-    url="https://openrouter.ai/api/v1/chat/completions",
-    headers={
-        "Authorization": f"Bearer {API_KEY}"
-    },
-    data=json.dumps({
-        "model": "openai/gpt-4o-mini-2024-07-18", # Optional
-        "messages": [{"role": "user", "content": input_text}],
-        "top_p": 1,
-        "temperature": 1,
-        "frequency_penalty": 0,
-        "presence_penalty": 0,
-        "repetition_penalty": 1,
-        "top_k": 0,
-        })
-    )
-    
-    # Handle errors
-    if response.status_code != 200:
-        return f"Error: {response.status_code}, {response.text}"
+# Define available models for selection
+MODEL_OPTIONS = [
+    "openai/gpt-4o-mini-2024-07-18",
+    "openai/gpt-4",
+    "anthropic/claude-2",
+    "cohere/command-xlarge-nightly"
+]
+
+def generate_comparisons(input_text, selected_models):
+    results = {}
+    for model in selected_models:
+        response = requests.post(
+            url="https://openrouter.ai/api/v1/chat/completions",
+            headers={
+                "Authorization": f"Bearer {API_KEY}",
+                "Content-Type": "application/json"
+            },
+            data=json.dumps({
+                "model": model,  # Use the current model
+                "messages": [{"role": "user", "content": input_text}],
+                "top_p": 1,
+                "temperature": 1,
+                "frequency_penalty": 0,
+                "presence_penalty": 0,
+                "repetition_penalty": 1,
+                "top_k": 0,
+            })
+        )
+        
+        # Parse the response
+        if response.status_code == 200:
+            try:
+                response_json = response.json()
+                results[model] = response_json.get("choices", [{}])[0].get("message", {}).get("content", "No content returned.")
+            except json.JSONDecodeError:
+                results[model] = "Error: Unable to parse response."
+        else:
+            results[model] = f"Error: {response.status_code}, {response.text}"
     
-    # Parse and return the content of the response
-    try:
-        response_json = response.json()
-        return response_json.get("choices", [{}])[0].get("message", {}).get("content", "No content returned.")
-    except json.JSONDecodeError:
-        return "Error: Unable to parse response."
+    return results
 
-# Create Gradio interface
-iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+# Create Gradio interface with multiple model selection
+iface = gr.Interface(
+    fn=generate_comparisons,
+    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"
+)
 
 iface.launch()