fix post url

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, UploadFile
+from fastapi import FastAPI, HTTPException, UploadFile, File, Form, Depends
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, UploadFile 
 from typing import Union 
@@ -8,6 +8,7 @@ from modeles import bert, squeezebert, deberta, loadSqueeze
 from uploadFile import file_to_text
 from typing import List
 from transformers import pipeline
+from pydantic import BaseModel
 
 
 
@@ -21,7 +22,17 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+class SqueezeBERTRequest(BaseModel):
+    context: str
+    question: str
 
+class BERTRequest(BaseModel):
+    context: str
+    question: str
+
+class DeBERTaRequest(BaseModel):
+    context: str
+    question: str
 
 pipBert = pipeline('question-answering', model="ALOQAS/bert-large-uncased-finetuned-squad-v2", tokenizer="ALOQAS/bert-large-uncased-finetuned-squad-v2")
 pipDeberta = pipeline('question-answering', model="ALOQAS/deberta-large-finetuned-squad-v2", tokenizer="ALOQAS/deberta-large-finetuned-squad-v2")
@@ -33,12 +44,12 @@ async def root():
 
 
 @app.post("/uploadfile/")
-async def create_upload_file(files: List[UploadFile], texte: str, model: str):
+async def create_upload_file(files: List[UploadFile], question: str, model: str):
     res = []
     for file in files:
         fileToText = await file_to_text(file)
 
-        res.append({"model": model, "texte": texte, "filename": file.filename, "file_to_text": fileToText})
+        res.append({"model": model, "texte": question, "filename": file.filename, "file_to_text": fileToText})
 
     return res
 
@@ -54,14 +65,10 @@ async def create_upload_file(texte: str, model: str):
 
     return {"model": model, "texte": texte}
 
-# # Modèle Pydantic pour les requêtes SqueezeBERT
-# class SqueezeBERTRequest(BaseModel):
-#     context: str
-#     question: str
 @app.post("/squeezebert/")
-async def qasqueezebert(context: str, question: str):
+async def qasqueezebert(request: SqueezeBERTRequest):
     try:
-        squeezebert_answer = squeezebert(context, question, model, tokenizer)
+        squeezebert_answer = squeezebert(request.context, request.question, model, tokenizer)
         if squeezebert_answer:
             return squeezebert_answer
         else:
@@ -69,14 +76,10 @@ async def qasqueezebert(context: str, question: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
 
-# # Modèle Pydantic pour les requêtes BERT
-# class BERTRequest(BaseModel):
-#     context: str
-#     question: str
 @app.post("/bert/")
-async def qabert(context: str, question: str):
+async def qabert(request: BERTRequest):
     try:
-        bert_answer = bert(context, question, pipBert)
+        bert_answer = bert(request.context, request.question, pipBert)
         if bert_answer:
             return bert_answer
         else:
@@ -84,14 +87,10 @@ async def qabert(context: str, question: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
 
-# # Modèle Pydantic pour les requêtes DeBERTa
-# class DeBERTaRequest(BaseModel):
-#     context: str
-#     question: str
 @app.post("/deberta-v2/")
-async def qadeberta(context: str, question: str):
+async def qadeberta(request: DeBERTaRequest):
     try:
-        deberta_answer = deberta(context, question, pipDeberta)
+        deberta_answer = deberta(request.context, request.question, pipDeberta)
         if deberta_answer:
             return deberta_answer
         else:

requirements.txt

@@ -12,4 +12,5 @@ PyMuPDF
 chardet
 frontend
 typing
-torch
+torch
+pydantic