Update app.py

app.py

@@ -1,14 +1,12 @@
-from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi import FastAPI, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse, FileResponse
-from pydantic import BaseModel
-import os
-import asyncio
-
-from chatbot import app as app_graph
 from langchain_core.messages import HumanMessage
+from chatbot import app as app_graph
 from tools import update_retriever
-from utils import TTS, STT
+from utils import STT, TTS
+import asyncio
+import os
 
 app = FastAPI()
 
@@ -24,54 +22,37 @@ UPLOAD_DIR = "uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 
-class TTSRequest(BaseModel):
-    text: str
-
-
 @app.get("/")
 def health():
-    return {"status": "API is running"}
+    return {"status": "running"}
 
 
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
-    file_path = os.path.join(UPLOAD_DIR, file.filename)
+    path = os.path.join(UPLOAD_DIR, file.filename)
 
-    with open(file_path, "wb") as f:
+    with open(path, "wb") as f:
         f.write(await file.read())
 
-    update_retriever(file_path)
+    update_retriever(path)
 
-    return {
-        "status": "success",
-        "filename": file.filename
-    }
+    return {"status": "uploaded", "file": file.filename}
 
 
 @app.post("/chat")
 async def chat(message: str, session_id: str = "default"):
 
-    async def event_generator():
+    async def stream():
         async for chunk in app_graph.astream(
             {"messages": [HumanMessage(content=message)]},
             config={"configurable": {"thread_id": session_id}},
-            stream_mode="messages"
+            stream_mode="messages",
         ):
-            if chunk:
-                msg = chunk[0] if isinstance(chunk, tuple) else chunk
-                if hasattr(msg, "content") and msg.content:
-                    yield f"data: {msg.content}\n\n"
-                    await asyncio.sleep(0.01)
-
-    return StreamingResponse(
-        event_generator(),
-        media_type="text/event-stream",
-        headers={
-            "Cache-Control": "no-cache",
-            "Connection": "keep-alive",
-            "X-Accel-Buffering": "no",
-        },
-    )
+            msg = chunk[0] if isinstance(chunk, tuple) else chunk
+            if hasattr(msg, "content") and msg.content:
+                yield msg.content + "\n"
+
+    return StreamingResponse(stream(), media_type="text/plain")
 
 
 @app.post("/stt")