Update app.py

app.py

@@ -113,57 +113,25 @@ 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)
 
-
-UPLOAD_DIR = "uploads"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
+class TranslationRequest(BaseModel):
+    text: str
+    source_lang: str
+    target_lang: str
 
 @app.post("/translate/")
-async def translate_file(file: UploadFile = File(...), target_lang: str = Form(...)):
-    if translator is None:
-        return JSONResponse(content={"error": "Modèle indisponible"}, status_code=500)
-
-    file_path = os.path.join(UPLOAD_DIR, file.filename)
-    
-    with open(file_path, "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
+async def translate_text(request: TranslationRequest):
+    if not request.text.strip():
+        return JSONResponse(content={"error": "Texte vide"}, status_code=400)
 
-    text = ""
-
-    # 📄 Gestion des fichiers TXT
-    if file.filename.endswith(".txt"):
-        with open(file_path, "r", encoding="utf-8") as f:
-            text = f.read()
-
-    # 📄 Gestion des fichiers PDF
-    elif file.filename.endswith(".pdf"):
-        try:
-            reader = PdfReader(file_path)
-            text = "\n".join([page.extract_text() or "" for page in reader.pages]).strip()
-        except Exception as e:
-            logging.error(f"❌ Erreur lecture PDF : {e}")
-            return JSONResponse(content={"error": "Impossible de lire le PDF"}, status_code=400)
-
-    # 📄 Gestion des fichiers DOCX
-    elif file.filename.endswith(".docx"):
-        try:
-            doc = docx.Document(file_path)
-            text = "\n".join([para.text for para in doc.paragraphs]).strip()
-        except Exception as e:
-            logging.error(f"❌ Erreur lecture DOCX : {e}")
-            return JSONResponse(content={"error": "Impossible de lire le fichier DOCX"}, status_code=400)
-
-    else:
-        return JSONResponse(content={"error": "Format non supporté"}, status_code=400)
-
-    if not text:
-        return JSONResponse(content={"error": "Fichier vide ou non lisible"}, status_code=400)
+    if translator is None:
+        return JSONResponse(content={"error": "Modèle de traduction non disponible"}, status_code=500)
 
     try:
-        formatted_text = f">>{target_lang}<< {text}"
-        translation = translator(formatted_text, max_length=512)[0]["translation_text"]
-        return {"translation": translation}
+        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:
-        return JSONResponse(content={"error": str(e)}, status_code=500)
+        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")
@@ -171,4 +139,4 @@ app.mount("/", StaticFiles(directory="static", html=True), name="static")
 # Redirection vers index.html
 @app.get("/")
 async def root():
-    return RedirectResponse(url="/index.html")
+    return RedirectResponse(url="/index.html")