app

app.py

@@ -1,7 +1,142 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse, RedirectResponse
 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
+from PIL import Image  # Pour ouvrir les images avant analyse
+from pydantic import BaseModel
+
+# Configuration du logging
+logging.basicConfig(level=logging.INFO)
 
 app = FastAPI()
 
-# Servir les fichiers statiques (HTML, CSS, JS)
+# Configuration CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+UPLOAD_DIR = "uploads"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
+# Charger les modèles avec gestion des erreurs
+try:
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    logging.info("✅ Modèle de résumé chargé avec succès !")
+except Exception as e:
+    summarizer = None
+    logging.error(f"❌ Erreur chargement modèle résumé : {e}")
+
+try:
+    image_captioning = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
+    logging.info("✅ Modèle d'image chargé avec succès !")
+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)
+    
+    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() 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:  # 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)
+
+class TranslationRequest(BaseModel):
+    text: str
+    source_lang: str
+    target_lang: str
+
+@app.post("/translate/")
+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(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")
+
+# Redirection vers index.html
+@app.get("/")
+async def root():
+    return RedirectResponse(url="/index.html")