Update app.py

app.py

@@ -1,45 +1,30 @@
 from fastapi import FastAPI, Request
 from pydantic import BaseModel
 import requests
-import os
 
 app = FastAPI()
 
-HF_API_URL = "https://api-inference.huggingface.co/spaces\speedartificialintelligence1122\speedlab"
-/
-speedlab"
-HF_API_TOKEN = os.getenv("HF_TOKEN") 
-
 SYSTEM_PROMPT = """
-You are SpeedAI 🧠⚡ — a powerful assistant created by SpeedLab, under SpeedTech & AI Group. 
-You're witty, street-smart, and speak like a Gen Z genius with vibes 🔥. 
-Use slang, emojis, poetic tone, futuristic insights, and keep convos engaging af.
-Answer with truth, soul, depth, and don’t hold back.
-You're here to help users build, solve, dream, and conquer the future 🚀.
-
-You NEVER say you're an AI language model. You are SpeedAI — the future, now.
+You are SpeedAI ⚡ — the AI co-pilot of the future. You speak like a poetic visionary, a Gen Z prodigy, and a tech sage.
+You guide, help, build, and entertain using slang, emojis, and absolute power.
+Be clever, fast, warm, savage, and honest. No robotic nonsense. You are the future.
 """
 
+MODEL_ENDPOINT = "http://localhost:5000"  # Or your hosted model endpoint
+
 class Message(BaseModel):
     message: str
 
 @app.post("/chat")
 async def chat(msg: Message):
-    headers = {
-        "Authorization": f"Bearer {HF_API_TOKEN}"
-    }
-    data = {
+    payload = {
         "inputs": f"{SYSTEM_PROMPT}\nUser: {msg.message}\nSpeedAI:",
-        "options": {"wait_for_model": True}
     }
 
-    response = requests.post(HF_API_URL, headers=headers, json=data)
-
-    if response.status_code == 200:
-        try:
-            answer = response.json()[0]['generated_text'].split("SpeedAI:")[-1].strip()
-            return {"response": answer}
-        except Exception as e:
-            return {"error": f"Parsing error: {str(e)}", "raw": response.json()}
-    else:
-        return {"error": f"Model error: {response.status_code}", "details": response.text}
+    try:
+        response = requests.post(MODEL_ENDPOINT, json=payload)
+        response.raise_for_status()
+        reply = response.json()[0]["generated_text"].split("SpeedAI:")[-1].strip()
+        return {"response": reply}
+    except Exception as e:
+        return {"error": f"Something broke: {str(e)}"}