Update main.py

main.py

@@ -8,6 +8,8 @@ from utils import get_text, translate_text, chat_text
 from json_flatten import flatten
 from pydantic import BaseModel
 from typing import Optional
+from fastapi import FastAPI, HTTPException
+import json
 
 app = FastAPI(
     title="DOCUMANTICAI API",
@@ -111,7 +113,32 @@ async def list_models():
     return JSONResponse(content={"models": models})
 
 
-@app.get("/translate")
+class TranslateRequest(BaseModel):
+    text: str
+    target_language: str
+
+@app.post("/translate")
+async def translate_text_endpoint(request: TranslateRequest):
+    """
+    ### Endpoint Description:
+    Translate text to a specified language.
+
+    #### Request Body:
+    - `text`: The text to translate. (Required)
+    - `target_language`: The language to translate the text to. (Required)
+
+    #### Response:
+    - The translated text in the specified language.
+    """
+    try:
+        response = translate_text(request.text, request.target_language)
+        response = json.loads(response)
+        return JSONResponse(content=response)
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=400)
+
+
+# @app.get("/translate")
 async def translate_text_endpoint(text: str, target_language: str):
     """
     ### Endpoint Description:
@@ -139,12 +166,12 @@ class ChatRequest(BaseModel):
     text: str
     history: Optional[str] = None
 
-# @app.post("/chat")
+@app.post("/chat")
 def chat_endpoint(request: ChatRequest):
     response_text = chat_text(request.text, request.query, request.history)
     return {"reply": response_text}
 
-@app.get("/chat")
+# @app.get("/chat")
 def chat_endpoint(query: str, text: str, history: Optional[str] = None):
     response_text = chat_text(text, query, history)
     return {"reply": response_text}