Update app.py

app.py

@@ -5,11 +5,12 @@ from langchain_core.messages import HumanMessage
 from chatbot import app as app_graph
 from tools import update_retriever
 from utils import STT, TTS
-import asyncio
 import os
+import asyncio
 
 app = FastAPI()
 
+# ---------------- CORS ---------------- #
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -24,25 +25,34 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 @app.get("/")
 def health():
-    return {"status": "running"}
+    return {"status": "API is running"}
 
 
+# =============================
+# Upload PDF
+# =============================
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
-    path = os.path.join(UPLOAD_DIR, file.filename)
+    file_path = os.path.join(UPLOAD_DIR, file.filename)
 
-    with open(path, "wb") as f:
+    with open(file_path, "wb") as f:
         f.write(await file.read())
 
-    update_retriever(path)
+    update_retriever(file_path)
 
-    return {"status": "uploaded", "file": file.filename}
+    return {
+        "status": "success",
+        "filename": file.filename
+    }
 
 
+# =============================
+# Chat Endpoint
+# =============================
 @app.post("/chat")
 async def chat(message: str, session_id: str = "default"):
 
-    async def stream():
+    async def event_stream():
         async for chunk in app_graph.astream(
             {"messages": [HumanMessage(content=message)]},
             config={"configurable": {"thread_id": session_id}},
@@ -52,15 +62,21 @@ async def chat(message: str, session_id: str = "default"):
             if hasattr(msg, "content") and msg.content:
                 yield msg.content + "\n"
 
-    return StreamingResponse(stream(), media_type="text/plain")
+    return StreamingResponse(event_stream(), media_type="text/plain")
 
 
+# =============================
+# STT
+# =============================
 @app.post("/stt")
 async def stt(file: UploadFile = File(...)):
     return await STT(file)
 
 
+# =============================
+# TTS
+# =============================
 @app.post("/tts")
 async def tts(text: str):
-    path = await TTS(text)
-    return FileResponse(path)
+    audio_path = await TTS(text)
+    return FileResponse(audio_path)