APPPP

app.py

@@ -1,15 +1,15 @@
+
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse, RedirectResponse
 from fastapi.staticfiles import StaticFiles
-from fastapi.middleware.cors import CORSMiddleware
 from transformers import pipeline
 import shutil
 import os
 import logging
+from fastapi.middleware.cors import CORSMiddleware
 from PyPDF2 import PdfReader
 import docx
-from PIL import Image
-import fitz  # PyMuPDF pour l'extraction de texte PDF
+from PIL import Image  # Pour ouvrir les images avant analyse
 
 # Configuration du logging
 logging.basicConfig(level=logging.INFO)
@@ -43,16 +43,11 @@ except Exception as e:
     image_captioning = None
     logging.error(f"❌ Erreur chargement modèle image : {e}")
 
-try:
-    translator = pipeline("translation", model="facebook/m2m100_418M")
-    logging.info("✅ Modèle de traduction chargé avec succès !")
-except Exception as e:
-    translator = None
-    logging.error(f"❌ Erreur chargement modèle traduction : {e}")
-
+# ✅ Déclare les routes AVANT le montage des fichiers statiques
 @app.post("/summarize/")
 async def summarize_document(file: UploadFile = File(...)):
     logging.info(f"📂 Requête reçue - Fichier : {file.filename}")
+
     file_path = os.path.join(UPLOAD_DIR, file.filename)
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
@@ -66,74 +61,54 @@ async def summarize_document(file: UploadFile = File(...)):
             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)
     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 de fichier non supporté"}, status_code=400)
 
     if not text:
+        logging.error("❌ Le fichier ne contient pas de texte lisible")
         return JSONResponse(content={"error": "Le fichier est vide ou non lisible"}, status_code=400)
 
+    # Tronquer le texte pour éviter l'erreur "IndexError: index out of range"
+    max_input_length = 1024  # Limite du modèle
+    text = text[:max_input_length]  # Tronquer le texte s'il est trop long
+
     try:
         summary = summarizer(text, max_length=150, min_length=50, do_sample=False)[0]["summary_text"]
         return JSONResponse(content={"summary": summary})
     except Exception as e:
+        logging.error(f"❌ Erreur lors du résumé : {e}")
         return JSONResponse(content={"error": "Échec du résumé. Texte trop long ou format invalide."}, status_code=500)
 
+
 @app.post("/interpret/")
 async def interpret_image(file: UploadFile = File(...)):
     logging.info(f"📂 Requête reçue - Image : {file.filename}")
+
     file_path = os.path.join(UPLOAD_DIR, file.filename)
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
-    
+
     try:
-        with Image.open(file_path) as img:
+        with Image.open(file_path) as img:  # Charger l'image correctement
             caption = image_captioning(img)[0]["generated_text"]
         return JSONResponse(content={"caption": caption})
     except Exception as e:
+        logging.error(f"❌ Erreur interprétation image : {e}")
         return JSONResponse(content={"error": "Échec de l'analyse de l'image"}, status_code=400)
 
-@app.post("/translate/")
-async def translate_document(file: UploadFile = File(...), target_lang: str = "fr"):
-    logging.info(f"📂 Requête reçue - Traduction : {file.filename}")
-
-    file_path = os.path.join(UPLOAD_DIR, file.filename)
-    with open(file_path, "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
-
-    try:
-        text = ""
-        if file.filename.endswith(".pdf"):
-            doc = fitz.open(file_path)  # Correction ici
-            text = "\n".join([page.get_text("text") for page in doc])  # Extraction correcte
-
-        elif file.filename.endswith(".docx"):
-            doc = docx.Document(file_path)  # Correction ici
-            text = "\n".join([para.text for para in doc.paragraphs])
-
-        else:
-            return JSONResponse(status_code=400, content={"error": "Format non supporté"})
-
-        if not text.strip():
-            return JSONResponse(status_code=400, content={"error": "Le document est vide ou non lisible"})
-
-        translated_text = translator(text, src_lang="en", target_lang=target_lang)  # Correction ici
-
-        return {"translated_text": translated_text[0]["translation_text"]}
-
-    except Exception as e:
-        logging.error(f"❌ Erreur lors de la traduction : {e}")
-        return JSONResponse(status_code=500, content={"error": str(e)})
-
-
+# ✅ Déplace ici le montage des fichiers statiques
 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")