Update app.py

app.py

@@ -1,34 +1,26 @@
 import gradio as gr
 import os
-from openrouter import Client
+from openai import OpenAI
 
-# --- 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)
+    client = OpenAI(
+        base_url="https://openrouter.ai/api/v1",
+        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
+    model_1 = "mistralai/mistral-7b-instruct"
+    model_2 = "meta-llama/llama-3-8b-instruct"
+    model_3 = "google/gemini-flash-1.5"
 
-    # --- Call Model 1 ---
     try:
         response_1 = client.chat.completions.create(
             model=model_1,
@@ -38,7 +30,6 @@ def compare_models(user_prompt):
     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,
@@ -48,7 +39,6 @@ def compare_models(user_prompt):
     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,
@@ -60,10 +50,6 @@ def compare_models(user_prompt):
 
     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),
@@ -76,5 +62,4 @@ demo = gr.Interface(
     description="Enter one prompt and see the results from three different AI models side-by-side. (Powered by OpenRouter)"
 )
 
-# Launch the app!
 demo.launch()