Update app.py

app.py

@@ -1,15 +1,14 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
-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 fastapi.responses import StreamingResponse, FileResponse
 from pydantic import BaseModel
-from utils import TTS, STT
+import os
+import asyncio
 
+from chatbot import app as app_graph
+from langchain_core.messages import HumanMessage
+from tools import update_retriever
+from utils import TTS, STT
 
 app = FastAPI()
 
@@ -21,20 +20,21 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+UPLOAD_DIR = "uploads"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
+
 class TTSRequest(BaseModel):
     text: str
 
 
-UPLOAD_DIR = "uploads"
-
 @app.get("/")
 def health():
-    return {'Status' : 'The api is live and running'}
+    return {"status": "API is 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:
@@ -57,11 +57,10 @@ async def chat(message: str, session_id: str = "default"):
             config={"configurable": {"thread_id": session_id}},
             stream_mode="messages"
         ):
-            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"
+            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(
@@ -73,32 +72,22 @@ async def chat(message: str, session_id: str = "default"):
             "X-Accel-Buffering": "no",
         },
     )
-# ---------------- STT ---------------- #
+
 
 @app.post("/stt")
 async def transcribe_audio(file: UploadFile = File(...)):
-    try:
-        return await STT(file)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    return await STT(file)
 
 
 @app.post("/tts")
 async def generate_tts(request: TTSRequest):
-    try:
-        if not request.text.strip():
-            raise HTTPException(status_code=400, detail="Text is empty")
-
-        audio_path = await TTS(text=request.text)
+    audio_path = await TTS(text=request.text)
 
-        if not os.path.exists(audio_path):
-            raise HTTPException(status_code=500, detail="Audio file not created")
+    if not os.path.exists(audio_path):
+        raise HTTPException(status_code=500, detail="Audio not generated")
 
-        return FileResponse(
-            path=audio_path,
-            media_type="audio/mpeg",
-            filename="speech.mp3"
-        )
-
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    return FileResponse(
+        path=audio_path,
+        media_type="audio/mpeg",
+        filename="speech.mp3"
+    )