Update app.py

app.py

@@ -2,33 +2,37 @@ import requests
 import os
 import gradio as gr
 
-API_TOKEN = os.getenv("HF_API_TOKEN")
-if not API_TOKEN:
-    raise ValueError("HF_API_TOKEN not found. Please set it in the Space Secrets.")
+API_KEY = os.getenv("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")
+if not API_KEY:
+    raise ValueError("OPENROUTER_API_KEY not found. Please set it in your environment variables.")
 
-MODEL_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
+# OpenRouter API endpoint
+MODEL_URL = "https://openrouter.ai/api/v1/chat/completions"
 
 HEADERS = {
-    "Authorization": f"Bearer {API_TOKEN}",
-    "Content-Type": "application/json"
+    "Authorization": f"Bearer {API_KEY}",
+    "Content-Type": "application/json",
+    "HTTP-Referer": "https://your-app-url.com",  # Optional but recommended
+    "X-Title": "LocoBot"  # Optional: your app name
 }
 
-def query_deepseek(prompt):
-    """Send a prompt to DeepSeek-R1 and get a response."""
+def query_gpt35(prompt):
+    """Send a prompt to GPT-3.5-Turbo via OpenRouter and get a response."""
     payload = {
-        "inputs": prompt,
-        "parameters": {
-            "max_new_tokens": 100, 
-            "temperature": 0.6,    
-            "top_p": 0.9           
-        }
+        "model": "openai/gpt-3.5-turbo",
+        "messages": [
+            {"role": "system", "content": "You are LocoBot, a helpful assistant."},
+            {"role": "user", "content": prompt}
+        ],
+        "max_tokens": 200,
+        "temperature": 0.7
     }
     
     try:
         response = requests.post(MODEL_URL, headers=HEADERS, json=payload)
-        response.raise_for_status() 
+        response.raise_for_status()
         result = response.json()
-        return result[0]["generated_text"].strip()
+        return result["choices"][0]["message"]["content"].strip()
     except requests.exceptions.RequestException as e:
         return f"Error: Could not connect to the API - {str(e)}"
     except (KeyError, IndexError):
@@ -36,33 +40,28 @@ def query_deepseek(prompt):
 
 def chat_interface(user_input, history):
     """Handle chatbot interaction with Gradio."""
-    prompt = f"User: {user_input}\nAssistant: "
-    response = query_deepseek(prompt)
-    return response
+    return query_gpt35(user_input)
 
 css = """
 button {
-    background-color: #007bff !important;  /* Blue background */
-    color: white !important;              /* White text */
+    background-color: #007bff !important;
+    color: white !important;
     border: none !important;
     padding: 10px 20px !important;
     border-radius: 5px !important;
 }
 button:hover {
-    background-color: #0056b3 !important; /* Darker blue on hover */
+    background-color: #0056b3 !important;
 }
 """
 
-
 interface = gr.ChatInterface(
     fn=chat_interface,
     title="LocoBot : Your AI Companion",
-    description="Engage in smart, dynamic conversations with LocoBot",
+    description="Engage in smart, dynamic conversations with LocoBot (powered by GPT-3.5 Turbo via OpenRouter)",
     theme="default",
     css=css
 )
 
 if __name__ == "__main__":
     interface.launch(server_name="0.0.0.0", server_port=7860)
-
-