Update main.py

main.py

@@ -3,21 +3,26 @@ import uuid
 import logging
 from pathlib import Path
 from typing import List, Optional
-from fastapi import FastAPI, UploadFile, File, Form, HTTPException, Request
+from fastapi import FastAPI, UploadFile, File, HTTPException, Request
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
+from fastapi.responses import JSONResponse, FileResponse, HTMLResponse
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 import fitz  # PyMuPDF
-from PIL import Image
-import io
 import pandas as pd
 from docx import Document
 from pptx import Presentation
 
 # Configuration du logging
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler("app.log"),
+        logging.StreamHandler()
+    ]
+)
 logger = logging.getLogger(__name__)
 
 # Initialisation de l'application FastAPI
@@ -27,6 +32,7 @@ app = FastAPI()
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
+    allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
@@ -37,13 +43,13 @@ UPLOAD_FOLDER = BASE_DIR / "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 # Configuration des modèles Hugging Face
-HF_TOKEN = os.getenv("HF_TOKEN", "votre_token_ici")  # Remplacez par votre vrai token
+HF_TOKEN = os.getenv("HF_TOKEN", "votre_token_ici")
 client = InferenceClient(token=HF_TOKEN)
 
 MODELS = {
     "summary": "facebook/bart-large-cnn",
     "caption": "Salesforce/blip-image-captioning-large",
-    "qa": "mistralai/Mistral-7B-Instruct-v0.2"  # Modèle adapté et performant
+    "qa": "mistralai/Mistral-7B-Instruct-v0.2"
 }
 
 # Modèles Pydantic
@@ -65,7 +71,33 @@ class QARequest(BaseModel):
     file_id: Optional[str] = None
     question: str
 
-# Fonctions utilitaires (conservées identiques)
+# Middleware pour gérer les 404
+@app.middleware("http")
+async def catch_404(request: Request, call_next):
+    response = await call_next(request)
+    if response.status_code == 404:
+        return JSONResponse(
+            status_code=404,
+            content={"detail": "Endpoint non trouvé"},
+            headers={"X-Custom-Header": "404-handled"}
+        )
+    return response
+
+# Routes de base
+@app.get("/")
+async def read_root():
+    return {"message": "Bienvenue sur l'API de traitement de documents"}
+
+@app.get("/logs")
+async def get_logs():
+    """Endpoint pour récupérer les logs"""
+    try:
+        with open("app.log", "r") as f:
+            return {"logs": f.read()}
+    except FileNotFoundError:
+        raise HTTPException(404, "Aucun log disponible")
+
+# Fonctions utilitaires (identique à votre version originale)
 def extract_text_from_pdf(file_path: str) -> str:
     try:
         doc = fitz.open(file_path)
@@ -133,7 +165,7 @@ async def process_uploaded_file(file: UploadFile) -> FileInfo:
         extracted_text=text if text else None
     )
 
-# Endpoints
+# Endpoints fonctionnels
 @app.post("/api/upload")
 async def upload_files(files: List[UploadFile] = File(...)):
     try:
@@ -205,7 +237,6 @@ async def answer_question(request: QARequest):
                     with open(file_path, "r", encoding="utf-8") as f:
                         context = f.read()
         
-        # Format du prompt spécifique pour Mistral
         prompt = f"""<s>[INST] Vous êtes un assistant IA expert. Répondez en français.
 Contexte: {context[:3000]}
 Question: {request.question}
@@ -218,15 +249,22 @@ Fournissez une réponse précise et concise. [/INST]"""
             temperature=0.7
         )
         
-        # Nettoyage de la réponse
         clean_response = response.split("[/INST]")[-1].strip()
         return {"answer": clean_response}
     except Exception as e:
         logger.error(f"QA error: {e}")
         raise HTTPException(500, f"Erreur de réponse: {str(e)}")
 
-# ... (le reste de votre code reste inchangé)
+# Montage des fichiers statiques
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)
+    uvicorn.run(
+        "main:app",
+        host="0.0.0.0",
+        port=8000,
+        reload=True,
+        workers=4,
+        log_config="log.ini"
+    )