Update app.py

app.py

@@ -1,39 +1,25 @@
-from fastapi import FastAPI, UploadFile, File
-from models.summarization import summarize_text
-from models.qa import answer_question
-from models.visualization import generate_visualization
-from models.translation import translate_document
-from utils.document_parser import extract_text_from_pdf
-import uvicorn
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from transformers import pipeline
 
 app = FastAPI()
 
-@app.get("/")
-def home():
-    return {"message": "API is running on Hugging Face Spaces!"}
-
-@app.post("/summarize/")
-async def summarize(file: UploadFile = File(...)):
-    text = extract_text_from_pdf(file)
-    summary = summarize_text(text)
-    return {"summary": summary}
+# Charger le modèle de traduction
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
 
-@app.post("/qa/")
-async def qa(file: UploadFile = File(...), question: str = ""):
-    text = extract_text_from_pdf(file)
-    response = answer_question(text, question)
-    return {"answer": response}
-
-@app.post("/visualization/")
-async def visualization(file: UploadFile = File(...), request: str = ""):
-    code = generate_visualization(file, request)
-    return {"generated_code": code}
+class TranslationRequest(BaseModel):
+    text: str
 
 @app.post("/translate/")
-async def translate(file: UploadFile = File(...), target_language: str = "fr"):
-    text = extract_text_from_pdf(file)
-    translated_text = translate_document(text, target_language)
-    return {"translated_text": translated_text}
+async def translate(request: TranslationRequest):
+    if not request.text:
+        raise HTTPException(status_code=400, detail="Le texte ne peut pas être vide.")
+    try:
+        result = translator(request.text)
+        return {"translated_text": result[0]['translation_text']}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+@app.get("/")
+async def root():
+    return {"message": "Bienvenue sur l'API de traduction avec FastAPI 🚀"}