Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile, Form
+from fastapi import FastAPI, File, UploadFile, Form, BackgroundTasks
 from pydantic import BaseModel
 from typing import List
 from pathlib import Path
@@ -28,7 +28,7 @@ UPLOAD_DIR = Path("uploads")
 UPLOAD_DIR.mkdir(exist_ok=True)
 
 @app.post("/upload")
-async def upload_file(file: UploadFile = File(...), type: str = Form(...)):
+async def upload_file(file: UploadFile = File(...), type: str = Form(...), task: BackgroundTasks):
     # print(file)
     # file_path = Path(file.filename)
     # with file_path.open("wb") as buffer:
@@ -62,6 +62,7 @@ async def upload_file(file: UploadFile = File(...), type: str = Form(...)):
         resumes.append(result)
     elif type == "job":
         jobs.append(result)
+    task.add_task(process_scoring)
     return {
         "code":201,
         "message":"Request was successful.",
@@ -69,7 +70,6 @@ async def upload_file(file: UploadFile = File(...), type: str = Form(...)):
     }
 
 
-
 @app.get("/jobs")
 def get_jobs():
     return {
@@ -88,11 +88,7 @@ def get_resumes():
     }
 
 
-@app.get("/scoring")
-async def get_scoring():
-    # resume_ids = [x["id"] for x in resumes]
-    # job_ids = [x["id"] for x in jobs]
-
+async def process_scoring():
     score_resume_ids = [x["resume_id"] for x in scoring]
     score_job_ids = [x["job_id"] for x in scoring]
 
@@ -104,9 +100,16 @@ async def get_scoring():
                 scoring.append({
                     "resume_id":resume["id"],
                     "job_id":job["id"],
-                    "rank_score":rank_score,
-                    "suggestion_score":suggest_score
+                    "rank_score":rank_score[0],
+                    "suggestion_score":suggest_score[0]
                 })
+    
+@app.get("/scoring")
+async def get_scoring():
+    # resume_ids = [x["id"] for x in resumes]
+    # job_ids = [x["id"] for x in jobs]
+
+    scoring = await process_scoring()
         
     return {
         "code":200,