Update app.py

app.py

@@ -7,6 +7,7 @@ app = FastAPI()
 
 
 from ranker import rank_resume 
+from embeddings import rank_jobs
 
 # Function to wrap the existing rank_resume
 def process_input(job_description, resumes):
@@ -18,6 +19,23 @@ def process_input(job_description, resumes):
     
     return rank_resume(job_description, resumes)[1]
 
+
+def process_input_suggestion(resume, job_descriptions):
+    # print("[JOB DESC]", job_description)
+    # print("[RESUMES]", resumes)
+    # resumes = [r for r in resumes if r and r.strip() != ""]  # Remove empty
+    # if not job_description.strip() or not resumes:
+    #     return "Please provide both resume and at least one job description."
+
+    return rank_jobs(job_descriptions, resume)[1]
+    
+    # results = zip(*rank_jobs(resumes, job_description))
+    # formatted_output = ""
+    # for i, (resume, score) in enumerate(results, 1):
+    #     formatted_output += f"Job #{i}:\nScore: {score:.2f}\nJob Description Snippet: {resume[:200]}...\n\n-------\n\n"
+    # return formatted_output
+
+
     
 app.get("/")
 def read_root():
@@ -28,9 +46,20 @@ class InputData(BaseModel):
     resumes: List[str]
     job_description: str
 
+class InputData2(BaseModel):
+    job_descriptions: List[str]
+    resume: str
+
 @app.post("/rank/")
 async def process_data(data: InputData):
     return dict(scores=process_input(data.job_description, data.resumes))
+
+
+@app.post("/suggest/")
+async def suggestion(data: InputData2):
+    return {
+        "scores":process_input_suggestion(data.resume, data.job_descriptions)
+    }