fix app.py for bert post

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI, HTTPException, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 from typing import Annotated
+from pydantic import BaseModel
 import uvicorn
 from fastapi import FastAPI, UploadFile, File 
 from typing import Union 
@@ -47,13 +48,22 @@ async def create_upload_file(texte: str, model: str):
 
     return {"model": model, "texte": texte}
 
+# # Modèle Pydantic pour les requêtes BERT
+# class BERTRequest(BaseModel):
+#     context: str
+#     question: str
+
 @app.post("/bert/")
-async def qabert(context, question):
-    bertAnswer = bert(context, question)
-    if bertAnswer:
-        return bertAnswer
-    else:
-        raise HTTPException(status_code=400, detail="Error while asking to bert")
+async def qabert(context: str, question: str):
+    try:
+        bert_answer = bert(context, question)
+        if bert_answer:
+            return bert_answer
+        else:
+            raise HTTPException(status_code=404, detail="No answer found")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
+
 
 def extract_data(file: UploadFile) -> Union[str, dict, list]:
     if file.filename.endswith(".txt"):