Update app.py

app.py

@@ -1,33 +1,41 @@
+import os
 import requests
 import gradio as gr
 
-# 🔑 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.")
+# Set this environment variable before running:
+# export OPENROUTER_API_KEY="your_openrouter_api_key_here"
+API_KEY = os.getenv("Open_Router")
+if not API_KEY:
+    raise ValueError("OPENROUTER_API_KEY not found. Please set it in your environment variables.")
 
 # OpenRouter API endpoint
 MODEL_URL = "https://openrouter.ai/api/v1/chat/completions"
 
 HEADERS = {
     "Authorization": f"Bearer {API_KEY}",
-    "Content-Type": "application/json"
+    "Content-Type": "application/json",
+    # Optional headers:
+    # "HTTP-Referer": "https://your-app-url.com",
+    # "X-Title": "LocoBot"
 }
 
 def query_gpt35(messages, max_tokens=500, temperature=0.7):
-    """Send a list of messages to OpenRouter GPT-3.5-Turbo."""
+    """
+    Send a list of messages (OpenAI-style) to OpenRouter and return assistant text.
+    messages: [{"role":"system"/"user"/"assistant", "content": "..."} , ...]
+    """
     payload = {
         "model": "openai/gpt-3.5-turbo",
         "messages": messages,
         "max_tokens": max_tokens,
-        "temperature": temperature
+        "temperature": temperature,
     }
 
     try:
         resp = requests.post(MODEL_URL, headers=HEADERS, json=payload, timeout=30)
         resp.raise_for_status()
         data = resp.json()
+        # OpenRouter's response format mirrors OpenAI ChatCompletion:
         return data["choices"][0]["message"]["content"].strip()
     except requests.exceptions.RequestException as e:
         return f"Error: Could not connect to the API - {e}"
@@ -35,39 +43,39 @@ def query_gpt35(messages, max_tokens=500, temperature=0.7):
         return "Error: Unexpected response format from API"
 
 def respond(user_message, chat_history):
-    """Handles the chat flow in Gradio."""
+    """
+    Gradio will pass chat_history as a list of (user, bot) tuples.
+    We convert that to a messages list, append the new user message, query the model,
+    then append the assistant reply and return updated history.
+    """
+    # Start with a system prompt to constrain behavior
     messages = [
-        {"role": "system", "content": "You are LocoBot, a helpful assistant."}
+        {"role": "system", "content": "You are LocoBot, a helpful assistant. Answer concisely and politely."}
     ]
-    
+
+    # Recreate prior conversation
     if chat_history:
         for user, bot in chat_history:
             messages.append({"role": "user", "content": user})
             messages.append({"role": "assistant", "content": bot})
-    
+
+    # Add current user message
     messages.append({"role": "user", "content": user_message})
+
     assistant_reply = query_gpt35(messages)
-    return chat_history + [(user_message, assistant_reply)], ""
 
-css = """
-button {
-    background-color: #007bff !important;
-    color: white !important;
-    border: none !important;
-    padding: 10px 20px !important;
-    border-radius: 5px !important;
-}
-button:hover {
-    background-color: #0056b3 !important;
-}
-"""
+    # Update history for Gradio (list of tuples)
+    new_history = chat_history + [(user_message, assistant_reply)]
+    return new_history, ""
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("## LocoBot — Your AI Companion (via OpenRouter GPT-3.5 Turbo)")
+# Build Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## LocoBot — Your AI Companion (via OpenRouter GPT-3.5)")
     chatbot = gr.Chatbot()
     message = gr.Textbox(placeholder="Type your message and press Enter")
     send = gr.Button("Send")
 
+    # Hook up interactions
     send.click(fn=respond, inputs=[message, chatbot], outputs=[chatbot, message])
     message.submit(fn=respond, inputs=[message, chatbot], outputs=[chatbot, message])