Update main.py

main.py

@@ -2,6 +2,7 @@ import os
 import shutil
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import JSONResponse
+from models import EvaluationRequest
 from pdf_processor import extract_answers_from_pdf, evaluate_student
 
 app = FastAPI()
@@ -14,10 +15,11 @@ if not os.path.exists(UPLOAD_DIR):
 @app.get("/")
 async def root():
     """
-    Root endpoint that provides a welcome message and basic instructions.
+    Root endpoint that provides a welcome message.
     """
-    return {"message": "Welcome to the PDF Processing API. Use '/extract/' to extract answers from a PDF or '/evaluate/' to evaluate student answers."}
-
+    return {
+        "message": "Welcome to the PDF Processing API. Use '/extract/' to extract answers from a PDF or '/evaluate/' to calculate marks using pre-extracted answers."
+    }
 
 @app.post("/extract/")
 async def extract_pdf(file: UploadFile = File(...)):
@@ -38,33 +40,16 @@ async def extract_pdf(file: UploadFile = File(...)):
             os.remove(file_location)
 
 @app.post("/evaluate/")
-async def evaluate_pdfs(answer_key_file: UploadFile = File(...), student_file: UploadFile = File(...)):
+async def evaluate(evaluation_request: EvaluationRequest):
     """
-    Endpoint to evaluate student answers by comparing the answer key and student's answer PDFs.
+    Endpoint to evaluate student answers.
+    Expects a JSON payload with the pre-extracted answer key and student answers.
     """
     try:
-        answer_key_path = os.path.join(UPLOAD_DIR, answer_key_file.filename)
-        student_path = os.path.join(UPLOAD_DIR, student_file.filename)
-        
-        with open(answer_key_path, "wb") as f:
-            shutil.copyfileobj(answer_key_file.file, f)
-        with open(student_path, "wb") as f:
-            shutil.copyfileobj(student_file.file, f)
-        
-        # Extract answers from both PDFs.
-        answer_key_result = extract_answers_from_pdf(answer_key_path)
-        student_result = extract_answers_from_pdf(student_path)
-        
-        # Evaluate the student answers.
-        evaluation = evaluate_student(answer_key_result, student_result)
+        evaluation = evaluate_student(evaluation_request.answer_key, evaluation_request.student)
         return JSONResponse(content=evaluation.model_dump())
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
-    finally:
-        if os.path.exists(answer_key_path):
-            os.remove(answer_key_path)
-        if os.path.exists(student_path):
-            os.remove(student_path)
 
 if __name__ == "__main__":
     import uvicorn