Update app.py

app.py

@@ -1,16 +1,15 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
-from tools import update_retriever
-from chatbot import app as app_graph, rebuild_graph
+from tools import create_rag_tool, update_retriever
+from chatbot import app as app_graph
 from langchain_core.messages import HumanMessage
 import os
 from fastapi.responses import StreamingResponse, FileResponse
+from langchain_core.messages import AIMessage
 from fastapi.middleware.cors import CORSMiddleware
+import asyncio
 from pydantic import BaseModel
 from utils import TTS, STT
 
-# =====================================================
-# APP SETUP
-# =====================================================
 
 app = FastAPI()
 
@@ -22,80 +21,48 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# =====================================================
-# MODELS
-# =====================================================
-
 class TTSRequest(BaseModel):
     text: str
 
 
-# ⚠️ HF requires persistent storage under /data
-UPLOAD_DIR = "/data/uploads"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
-
-# =====================================================
-# HEALTH CHECK
-# =====================================================
+UPLOAD_DIR = "uploads"
 
 @app.get("/")
 def health():
-    return {"Status": "The api is live and running"}
-
-
-# =====================================================
-# FILE UPLOAD (RAG)
-# =====================================================
+    return {'Status' : 'The api is live and running'}
 
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
+    os.makedirs(UPLOAD_DIR, exist_ok=True)
+
     file_path = os.path.join(UPLOAD_DIR, file.filename)
 
     with open(file_path, "wb") as f:
         f.write(await file.read())
 
-    # Update vector store
     update_retriever(file_path)
 
-    # 🔥 Rebuild LangGraph so RAG becomes active
-    rebuild_graph()
-
     return {
         "status": "success",
         "filename": file.filename
     }
 
 
-# =====================================================
-# CHAT ENDPOINT (STREAMING)
-# =====================================================
-
 @app.post("/chat")
 async def chat(message: str, session_id: str = "default"):
 
     async def event_generator():
-        buffer = ""
-
         async for chunk in app_graph.astream(
             {"messages": [HumanMessage(content=message)]},
             config={"configurable": {"thread_id": session_id}},
             stream_mode="messages"
         ):
-            if not chunk:
-                continue
-
-            msg = chunk[0] if isinstance(chunk, tuple) else chunk
-
-            if hasattr(msg, "content") and msg.content:
-                buffer += msg.content
-
-                # Flush every ~150 characters (prevents broken tokens)
-                if len(buffer) > 150:
-                    yield f"data: {buffer.strip()}\n\n"
-                    buffer = ""
-
-        if buffer:
-            yield f"data: {buffer.strip()}\n\n"
+            if len(chunk) >= 1:
+                message_chunk = chunk[0] if isinstance(chunk, tuple) else chunk
+                if hasattr(message_chunk, 'content') and message_chunk.content:
+                    data = str(message_chunk.content).replace("\n", "\\n")
+                    yield f"data: {data}\n\n"
+                    await asyncio.sleep(0.01)
 
     return StreamingResponse(
         event_generator(),
@@ -104,14 +71,9 @@ async def chat(message: str, session_id: str = "default"):
             "Cache-Control": "no-cache",
             "Connection": "keep-alive",
             "X-Accel-Buffering": "no",
-            "Access-Control-Allow-Origin": "*",
         },
     )
-
-
-# =====================================================
-# STT
-# =====================================================
+# ---------------- STT ---------------- #
 
 @app.post("/stt")
 async def transcribe_audio(file: UploadFile = File(...)):
@@ -121,10 +83,6 @@ async def transcribe_audio(file: UploadFile = File(...)):
         raise HTTPException(status_code=500, detail=str(e))
 
 
-# =====================================================
-# TTS
-# =====================================================
-
 @app.post("/tts")
 async def generate_tts(request: TTSRequest):
     try:
@@ -143,4 +101,4 @@ async def generate_tts(request: TTSRequest):
         )
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))