Update main.py

main.py

@@ -2,11 +2,9 @@ from flask import Flask, request, jsonify
 from langchain_community.llms import LlamaCpp
 from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModel
-
-# cosine_similarity
+from sklearn.metrics.pairwise import cosine_similarity
+from flask import Flask, request, jsonify
 import torch
-from torch.nn.functional import cosine_similarity
-import os 
 app = Flask(__name__)
 
 n_gpu_layers = 0
@@ -52,21 +50,28 @@ def health():
 @app.route('/compare', methods=['POST'])
 def compare():
     jobs_skills = request.json.get('jobs_skills')
-    #example: jobs_skills = ["<|assistant|> Sure! Here are the skills required for the Software Engineer position:\n\n1. Proficiency in Python programming language\n2. Experience with Java development\n3. Knowledge of web development technologies (e.g., HTML, CSS, JavaScript)\n4. Familiarity with machine learning algorithms and frameworks (e.g., TensorFlow, PyTorch)\n5. Strong problem-solving skills\n6. Effective communication abilities\n7. Agile methodology understanding\n8. Version control system expertise (e.g., Git)\n9. Software architecture design experience\n10. Testing methodologies knowledge (unit tests, integration tests, end-to-end tests)\n11. Continuous learning mindset for staying updated on technological advancements\n12. Collaboration skills for effective teamwork\n\nThese skills are essential for the Software Engineer role to contribute to the development of innovative software solutions, optimize performance, ensure code quality, and foster a collaborative work environment.", "<|assistant|> Certainly! Here are the skills required for the Data Scientist position:\n\n1. Proficiency in Python programming language\n2. Experience with data analysis and visualization tools (e.g., Pandas, Matplotlib)\n3. Knowledge of machine learning algorithms and statistical modeling techniques\n4. Strong problem-solving and analytical skills\n5. Effective communication abilities\n6. Agile methodology understanding\n7. Version control system expertise (e.g., Git)\n8. Data preprocessing and cleaning experience\n9. Model evaluation and optimization skills\n10. Continuous learning mindset for staying updated on data science advancements\n11. Collaboration skills for effective teamwork\n\nThese skills are essential for the Data Scientist role to analyze data, develop predictive models, optimize algorithms, and collaborate with cross-functional teams."]
-    employee_skills = request.json.get('employee_skills')# string
-    #example: employee_skills = "<|assistant|> Certainly! Based on the provided information, here are the relevant skills listed in my CV:\n\n1. Python programming language proficiency\n2. Java development expertise\n3. Web development experience (including front-end and back-end technologies)\n4. Proficient with machine learning algorithms and frameworks\n5. Strong problem-solving abilities\n6. Excellent communication skills, both written and verbal\n7. Agile methodology adherence\n8. Familiarity with version control systems (e.g., Git)\n9. Experience in designing scalable software architectures\n10. Proficient in testing methodologies such as unit tests, integration tests, and end-to-end tests\n11. Continuous learning mindset to stay updated on the latest technological advancements\n12. Strong collaboration skills for effective teamwork\n\nThese skills enable me to contribute significantly to an employer's objectives by developing innovative features that leverage cutting-edge technology, optimizing software performance through efficient coding practices and algorithmic improvements, enhancing code quality with rigorous testing methodologies, and fostering a collaborative work environment""
+    employee_skills = request.json.get('employee_skills')
+    
     if not isinstance(jobs_skills, list) or not all(isinstance(skill, str) for skill in jobs_skills):
         raise ValueError("jobs_skills must be a list of strings")
-    job_embeddings = model.encode(jobs_skills)
+    
+    # Encode job and employee skills
+    job_embeddings = [model.encode(skill) for skill in jobs_skills]
     employee_embeddings = model.encode(employee_skills)
-    sim = []
-    employee_embeddings_tensor = torch.from_numpy(employee_embeddings).unsqueeze(0)
-    for job_e in job_embeddings:
-        job_e_tensor = torch.from_numpy(job_e).unsqueeze(0)
-        sim.append(cosine_similarity(employee_embeddings_tensor, job_e_tensor,dim=1))
-    max_sim = max(sim)
-    index = sim.index(max_sim)
-    return jsonify({'job': jobs_skills[index]})
+    
+    # Calculate cosine similarity
+    similarities = []
+    employee_embedding_tensor = torch.tensor(employee_embeddings).unsqueeze(0)
+    for job_embedding in job_embeddings:
+        job_embedding_tensor = torch.tensor(job_embedding).unsqueeze(0)
+        similarity = cosine_similarity(employee_embedding_tensor, job_embedding_tensor)
+        similarities.append(similarity.item())
+    
+    # Find the job with highest similarity
+    max_similarity_index = similarities.index(max(similarities))
+    max_similarity_job = jobs_skills[max_similarity_index]
+    
+    return jsonify({'job': max_similarity_job, 'similarity': max(similarities)})
 
 if __name__ == '__main__':
-    app.run()
+    app.run())