Update main.py

main.py

@@ -4,7 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 import io
 import json
-from utils import get_text, translate_text
+from utils import get_text, translate_text, chat_text
 from json_flatten import flatten
 
 app = FastAPI(
@@ -130,4 +130,14 @@ async def translate_text_endpoint(text: str, target_language: str):
         response = json.loads(response)
         return JSONResponse(content=response)
     except Exception as e:
-        return JSONResponse(content={"error": str(e)}, status_code=400)
+        return JSONResponse(content={"error": str(e)}, status_code=400)
+
+class ChatRequest(BaseModel):
+    query: str
+    text: str
+    history: Optional[str] = None
+
+@app.post("/chat")
+def chat_endpoint(request: ChatRequest):
+    response_text = chat_text(request.query, request.text, request.history)
+    return {"reply": response_text}