APP

app.py

@@ -4,8 +4,13 @@ from fastapi.staticfiles import StaticFiles
 from transformers import pipeline
 import shutil
 import os
+import logging
 from fastapi.middleware.cors import CORSMiddleware
+from PyPDF2 import PdfReader
+import docx
 
+# Configuration du logging
+logging.basicConfig(level=logging.INFO)
 
 app = FastAPI()
 
@@ -17,8 +22,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-
 # Servir les fichiers statiques (HTML, CSS, JS)
 app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
@@ -38,17 +41,37 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 @app.post("/summarize/")
 async def summarize_document(file: UploadFile = File(...)):
-    """ Analyse et résume un document texte. """
+    """ Analyse et résume un document texte, PDF ou DOCX. """
+    logging.info(f"📂 Fichier reçu : {file.filename}")
+
     file_path = os.path.join(UPLOAD_DIR, file.filename)
-    
-    # Sauvegarde temporaire du fichier
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
-
-    # Lire le contenu du fichier (ex: PDF, DOCX) et l'envoyer au modèle
-    # Ici, je vais supposer que c'est un fichier texte brut pour simplifier
-    with open(file_path, "r", encoding="utf-8") as f:
-        text = f.read()
+    
+    text = ""
+    
+    if file.filename.endswith(".txt"):
+        with open(file_path, "r", encoding="utf-8") as f:
+            text = f.read()
+    elif file.filename.endswith(".pdf"):
+        try:
+            reader = PdfReader(file_path)
+            text = "\n".join([page.extract_text() for page in reader.pages if page.extract_text()])
+        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])
+        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.strip():
+        return JSONResponse(content={"error": "Le fichier ne contient pas de texte lisible"}, status_code=400)
 
     summary = summarizer(text, max_length=150, min_length=50, do_sample=False)[0]["summary_text"]
     
@@ -57,11 +80,16 @@ async def summarize_document(file: UploadFile = File(...)):
 @app.post("/interpret/")
 async def interpret_image(file: UploadFile = File(...)):
     """ Génère une légende pour une image. """
-    file_path = os.path.join(UPLOAD_DIR, file.filename)
+    logging.info(f"📂 Image reçue : {file.filename}")
 
+    file_path = os.path.join(UPLOAD_DIR, file.filename)
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
-
-    caption = image_captioning(file_path)[0]["generated_text"]
-
+    
+    try:
+        caption = image_captioning(file_path)[0]["generated_text"]
+    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)
+    
     return JSONResponse(content={"caption": caption})