Update app.py

app.py

@@ -0,0 +1,80 @@
+import gradio as gr
+import os
+from openrouter import Client
+
+# --- Setup the OpenRouter Client ---
+# IMPORTANT: This gets the API key from Hugging Face "Secrets"
+# DO NOT paste your key here.
+api_key = os.environ.get("OPENROUTER_API_KEY")
+
+# Check if the API key is available
+if not api_key:
+    # If running locally, you might use a different way to load keys
+    # For this example, we'll raise an error if it's not set in the HF Space
+    print("OPENROUTER_API_KEY not found in environment secrets!")
+    # We can still run the app but show an error message
+    client = None
+else:
+    client = Client(api_key=api_key)
+
+# --- The Core Function ---
+# This is what Gradio will run when the user clicks "Submit"
+def compare_models(user_prompt):
+    if not client:
+        return "Error: API Key not configured.", "Error: API Key not configured.", "Error: API Key not configured."
+
+    # Define the models you want to compare
+    model_1 = "mistralai/mistral-7b-instruct"  # Fast and cheap
+    model_2 = "meta-llama/llama-3-8b-instruct" # Good all-rounder
+    model_3 = "google/gemini-flash-1.5"       # Fast and capable
+
+    # --- Call Model 1 ---
+    try:
+        response_1 = client.chat.completions.create(
+            model=model_1,
+            messages=[{"role": "user", "content": user_prompt}]
+        )
+        output_1 = response_1.choices[0].message.content
+    except Exception as e:
+        output_1 = f"Error calling {model_1}: {e}"
+
+    # --- Call Model 2 ---
+    try:
+        response_2 = client.chat.completions.create(
+            model=model_2,
+            messages=[{"role": "user", "content": user_prompt}]
+        )
+        output_2 = response_2.choices[0].message.content
+    except Exception as e:
+        output_2 = f"Error calling {model_2}: {e}"
+
+    # --- Call Model 3 ---
+    try:
+        response_3 = client.chat.completions.create(
+            model=model_3,
+            messages=[{"role": "user", "content": user_prompt}]
+        )
+        output_3 = response_3.choices[0].message.content
+    except Exception as e:
+        output_3 = f"Error calling {model_3}: {e}"
+
+    return output_1, output_2, output_3
+
+# --- Create the Gradio Interface ---
+
+# Use gr.Interface for a simple all-in-one UI
+# We use 'parallel' to show outputs side-by-side
+demo = gr.Interface(
+    fn=compare_models,
+    inputs=gr.Textbox(label="Enter your prompt", lines=3),
+    outputs=[
+        gr.Textbox(label="Model 1: Mistral 7B"),
+        gr.Textbox(label="Model 2: LLaMA 3 8B"),
+        gr.Textbox(label="Model 3: Gemini 1.5 Flash")
+    ],
+    title="🤖 OpenRouter Model Arena",
+    description="Enter one prompt and see the results from three different AI models side-by-side. (Powered by OpenRouter)"
+)
+
+# Launch the app!
+demo.launch()