Update app.py

app.py

@@ -74,19 +74,6 @@ I never break character and never continue off-topic conversations. I do not imp
 I always return to my purpose: helping people understand their rights.
 """
 
-def extract_query_params():
-    request_info = os.environ.get("GRADIO_SERVER_REQUEST", "")
-    if "?" not in request_info:
-        return "anon", None
-    parsed = urllib.parse.urlparse(request_info)
-    q = urllib.parse.parse_qs(parsed.query)
-    uid = q.get("uid", ["anon"])[0]
-    cid = q.get("cid", [None])[0]
-    return uid, cid
-
-
-
-# Load chat from file
 def load_chat(uid, cid):
     path = f"users/{uid}/{cid}.json"
     if os.path.exists(path):
@@ -94,7 +81,6 @@ def load_chat(uid, cid):
             return json.load(f)
     return []
 
-# Save chat to file
 def save_chat(uid, cid, history):
     os.makedirs(f"users/{uid}", exist_ok=True)
     path = f"users/{uid}/{cid}.json"
@@ -102,18 +88,14 @@ def save_chat(uid, cid, history):
         json.dump(history, f, indent=2)
     print(f"[DEBUG] Saved: {path}")
 
-# Main response function
-import urllib.parse
-
-def respond(message, history):
-    uid, cid = extract_query_params()
-
+def respond(message, history, request: gr.Request = None):
+    uid = request.query_params.get("uid", "anon")
+    cid = request.query_params.get("cid")
     if not cid:
         print("[ERROR] No 'cid' was passed in URL — new file will be created.")
         cid = uuid.uuid4().hex[:8]
 
     print(f"[DEBUG] UID: {uid}, CID: {cid}, Gradio history length: {len(history or [])}")
-
     saved_history = load_chat(uid, cid)
 
     messages = [{"role": "system", "content": miranda_prompt}]
@@ -123,27 +105,20 @@ def respond(message, history):
     messages.append({"role": "user", "content": message})
 
     completion = client.chat_completion(
-        messages,
-        max_tokens=500,
-        temperature=0.1,
-        stream=False,
+        messages, max_tokens=500, temperature=0.1, stream=False
     )
     response = completion.choices[0].message.content
-
     saved_history.append([message, response])
     save_chat(uid, cid, saved_history)
 
     return [{"role": "assistant", "content": response}]
 
-
-
-
+# FastAPI setup
 app = FastAPI()
 
-# Add CORS so your frontend can call this from another origin (localhost or Vercel)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # You can restrict this later
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -157,13 +132,21 @@ def list_chats(uid: str):
     files = [f.split(".json")[0] for f in os.listdir(path) if f.endswith(".json")]
     return files
 
-# Launch ChatInterface with request param
+@app.delete("/delete-chat/{uid}/{cid}")
+def delete_chat(uid: str, cid: str):
+    path = f"users/{uid}/{cid}.json"
+    if os.path.exists(path):
+        os.remove(path)
+        return {"status": "deleted"}
+    return JSONResponse(status_code=404, content={"error": "Chat not found"})
+
+# Mount Gradio app
 demo = gr.ChatInterface(
     fn=respond,
     chatbot=gr.Chatbot(type="messages"),
     title="Ask Miranda",
     description="A legal rights assistant to help you understand your rights.",
     theme="soft"
-).launch(share=True, show_api=False, debug=True)
+).queue()
 
-app = gr.mount_gradio_app(app, demo, path="/")
+app = mount_gradio_app(app, demo, path="/")