APP

app.py

@@ -10,6 +10,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from PyPDF2 import PdfReader
 import docx
 from PIL import Image  # Pour ouvrir les images avant analyse
+from pydantic import BaseModel
 
 # Configuration du logging
 logging.basicConfig(level=logging.INFO)
@@ -113,22 +114,26 @@ async def interpret_image(file: UploadFile = File(...)):
         logging.error(f"❌ Erreur interprétation image : {e}")
         return JSONResponse(content={"error": "Échec de l'analyse de l'image"}, status_code=400)
 
+class TranslationRequest(BaseModel):
+    text: str
+    source_lang: str
+    target_lang: str
+
 @app.post("/translate/")
-async def translate_text(text: str, source_lang: str, target_lang: str):
-    if not text.strip():
+async def translate_text(request: TranslationRequest):
+    if not request.text.strip():
         return JSONResponse(content={"error": "Texte vide"}, status_code=400)
 
     if translator is None:
         return JSONResponse(content={"error": "Modèle de traduction non disponible"}, status_code=500)
 
     try:
-        translated_text = translator(text, src_lang=source_lang, tgt_lang=target_lang)[0]["translation_text"]
+        translated_text = translator(request.text, src_lang=request.source_lang, tgt_lang=request.target_lang)[0]["translation_text"]
         return JSONResponse(content={"translated_text": translated_text})
     except Exception as e:
         logging.error(f"❌ Erreur de traduction : {e}")
         return JSONResponse(content={"error": "Échec de la traduction"}, status_code=500)
 
-
 # ✅ Déplace ici le montage des fichiers statiques
 app.mount("/", StaticFiles(directory="static", html=True), name="static")