Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+from openai import OpenAI
 from smolagents import DuckDuckGoSearchTool
 import re
 import time
@@ -60,11 +60,11 @@ def respond(
     max_tokens,
     temperature,
     top_p,
-    hf_token,
+    openrouter_key,
 ):
-    client = InferenceClient(
-        provider="hf-inference",
-        api_key=hf_token
+    client = OpenAI(
+        base_url="https://openrouter.ai/api/v1",
+        api_key=openrouter_key,
     )
     
     messages = [{"role": "system", "content": system_message}]
@@ -86,15 +86,19 @@ def respond(
             
             try:
                 completion = client.chat.completions.create(
-                    model="Qwen/QwQ-32B",
+                    model="qwen/qwq-32b:free",
                     messages=messages,
                     max_tokens=max_tokens,
                     temperature=temperature,
                     top_p=top_p,
-                    stream=True
+                    stream=True,
+                    extra_headers={
+                        "HTTP-Referer": "https://your-domain.com",
+                        "X-Title": "Web Research Agent"
+                    }
                 )
             except Exception as e:
-                yield f"⚠️ API Error: {str(e)}\n\nPlease check your HF token and model access."
+                yield f"⚠️ API Error: {str(e)}\n\nPlease check your OpenRouter API key."
                 return
 
             response = ""
@@ -136,7 +140,7 @@ demo = gr.ChatInterface(
         gr.Slider(minimum=1000, maximum=15000, value=6000, step=500, label="Max Tokens"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.5, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.85, step=0.05, label="Top-p"),
-        gr.Textbox(label="HF API Token", type="password")
+        gr.Textbox(label="OpenRouter API Key", type="password")
     ],
     title="Web Research Agent 🤖",
     description="Advanced AI assistant with web search capabilities",