Update app.py

app.py

@@ -1,46 +1,53 @@
 import requests
-import os
 import gradio as gr
 
-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.")
+# 🔑 Paste your OpenRouter API key here
+API_KEY = "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"
+
+if not API_KEY or API_KEY.startswith("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"):
+    raise ValueError("Please paste your actual OpenRouter API key in API_KEY variable.")
 
 # OpenRouter API endpoint
 MODEL_URL = "https://openrouter.ai/api/v1/chat/completions"
 
 HEADERS = {
     "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
+    "Content-Type": "application/json"
 }
 
-def query_gpt35(prompt):
-    """Send a prompt to GPT-3.5-Turbo via OpenRouter and get a response."""
+def query_gpt35(messages, max_tokens=500, temperature=0.7):
+    """Send a list of messages to OpenRouter GPT-3.5-Turbo."""
     payload = {
         "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
+        "messages": messages,
+        "max_tokens": max_tokens,
+        "temperature": temperature
     }
-    
+
     try:
-        response = requests.post(MODEL_URL, headers=HEADERS, json=payload)
-        response.raise_for_status()
-        result = response.json()
-        return result["choices"][0]["message"]["content"].strip()
+        resp = requests.post(MODEL_URL, headers=HEADERS, json=payload, timeout=30)
+        resp.raise_for_status()
+        data = resp.json()
+        return data["choices"][0]["message"]["content"].strip()
     except requests.exceptions.RequestException as e:
-        return f"Error: Could not connect to the API - {str(e)}"
+        return f"Error: Could not connect to the API - {e}"
     except (KeyError, IndexError):
         return "Error: Unexpected response format from API"
 
-def chat_interface(user_input, history):
-    """Handle chatbot interaction with Gradio."""
-    return query_gpt35(user_input)
+def respond(user_message, chat_history):
+    """Handles the chat flow in Gradio."""
+    messages = [
+        {"role": "system", "content": "You are LocoBot, a helpful assistant."}
+    ]
+    
+    if chat_history:
+        for user, bot in chat_history:
+            messages.append({"role": "user", "content": user})
+            messages.append({"role": "assistant", "content": bot})
+    
+    messages.append({"role": "user", "content": user_message})
+    assistant_reply = query_gpt35(messages)
+    return chat_history + [(user_message, assistant_reply)], ""
 
 css = """
 button {
@@ -55,13 +62,14 @@ button:hover {
 }
 """
 
-interface = gr.ChatInterface(
-    fn=chat_interface,
-    title="LocoBot : Your AI Companion",
-    description="Engage in smart, dynamic conversations with LocoBot (powered by GPT-3.5 Turbo via OpenRouter)",
-    theme="default",
-    css=css
-)
+with gr.Blocks(css=css) as demo:
+    gr.Markdown("## LocoBot — Your AI Companion (via OpenRouter GPT-3.5 Turbo)")
+    chatbot = gr.Chatbot()
+    message = gr.Textbox(placeholder="Type your message and press Enter")
+    send = gr.Button("Send")
+
+    send.click(fn=respond, inputs=[message, chatbot], outputs=[chatbot, message])
+    message.submit(fn=respond, inputs=[message, chatbot], outputs=[chatbot, message])
 
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)