Update app.py

app.py

@@ -1,32 +1,32 @@
 from fastapi import FastAPI
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-import gradio as gr
-import uvicorn
-import threading
 from tinydb import TinyDB, Query
+import uvicorn
+
+app = FastAPI()
 
 # Load model
 tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5")
 model = AutoModelForCausalLM.from_pretrained("microsoft/phi-1_5")
 model.eval()
 
-# Setup DB
+# Memory setup
 db = TinyDB("memory.json")
 User = Query()
 
-# API
-app = FastAPI()
-
 @app.get("/")
 async def chat(query: str = "", number: str = "", prompt: str = ""):
     if not query or not number:
-        return {"response": "Trigger needs both ?query= and &number= 🧠"}
+        return {"response": "Missing query or number."}
 
     record = db.get(User.number == number)
-
     if not record:
-        record = {"number": number, "history": [], "prompt": "You are Trigger, a clever, emotional AI boy created by 'I am him'. Speak like a witty, chill human friend."}
+        record = {
+            "number": number,
+            "history": [],
+            "prompt": "You are Trigger, a clever, emotional AI boy created by 'I am him'. Speak like a witty, chill human friend."
+        }
         db.insert(record)
 
     if prompt:
@@ -36,11 +36,11 @@ async def chat(query: str = "", number: str = "", prompt: str = ""):
     history = record["history"]
     system_prompt = record.get("prompt", "")
 
-    # Add new user input
+    # Append new input
     history.append(f"User: {query}")
     trimmed = history[-6:]
 
-    # Build prompt
+    # Construct final prompt
     full_prompt = system_prompt + "\n" + "\n".join(trimmed) + "\nTrigger:"
     inputs = tokenizer(full_prompt, return_tensors="pt", truncation=True)
 
@@ -57,19 +57,12 @@ async def chat(query: str = "", number: str = "", prompt: str = ""):
     decoded = tokenizer.decode(output[0], skip_special_tokens=True)
     reply = decoded.split("Trigger:")[-1].strip()
 
+    # Save memory
     history.append(f"Trigger: {reply}")
-    db.update({"history": history, "prompt": system_prompt, "number": number}, User.number == number)
+    db.update({"number": number, "history": history, "prompt": system_prompt}, User.number == number)
 
     return {"response": reply}
 
-# Run API
-def run_api():
-    uvicorn.run(app, host="0.0.0.0", port=7860)
-
-threading.Thread(target=run_api).start()
-
-# Gradio dummy
-def keep_alive():
-    return "Trigger’s up."
-
-gr.Interface(fn=keep_alive, inputs=[], outputs="text").launch()
+# Only runs locally, HF handles it differently
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)