Update main.py

main.py

@@ -68,5 +68,25 @@ def compare():
         similarity_scores.append({"job": jobs_skills[i], "similarity_score": similarity_score.item()})
 
     return jsonify(similarity_scores)
+@app.route('/compare_job', methods=['POST'])
+def compare_job():
+    job_skills = request.json.get('post')
+    employee_skills = request.json.get('employee_skills')
+
+    # Validation
+    if not isinstance(employee_skills, list) or not all(isinstance(skill, str) for skill in employee_skills):
+        raise ValueError("employee_skills must be a list of strings")
+
+    # Encoding skills into embeddings
+    job_embedding = model.encode(job_skills)
+    employee_embedding = model.encode(employee_skills)
+
+    # Computing cosine similarity between employee skills and the job
+    employee_embedding_tensor = torch.from_numpy(employee_embedding).unsqueeze(0)
+    job_embedding_tensor = torch.from_numpy(job_embedding).unsqueeze(0)
+    similarity_score = cosine_similarity(employee_embedding_tensor, job_embedding_tensor, dim=1)
+
+    return jsonify({"job": job_skills, "similarity_score": similarity_score.item()})
+
 if __name__ == '__main__':
     app.run()