rearrange results better

app.py

@@ -10,7 +10,7 @@ import utils
 from prompts import preprocess, recruiting_assistant, matches, intro
 
 llm = ChatOpenAI(temperature=0.0, openai_api_key=os.environ["OPENAI"])
-
+MAX_SKILLS = 10
 
 def preprocess_resume(llm, resume) -> str:
     result = preprocess.preprocess_resume(llm, resume)
@@ -61,28 +61,38 @@ def postprocess_vancy(vacancies, resume):
                         skills_match, "skills_match_predicted"
                     )
                     print("getting matched skills ...")
-                    matched_skills = [
-                        element["content"]
-                        for element in skills_match_predicted
-                        if element["resume_index"] > 0 and element["vacancy_index"] > 0
-                    ]
+                    counter = 0
+                    matched_skills = []
+                    for element in skills_match_predicted:
+                        if element["resume_index"] > 0 and element["vacancy_index"] > 0:
+                            counter += 1
+                            if counter > MAX_SKILLS:
+                                break
+                            matched_skills.append(element["content"])
+
+                    # matched_skills = [
+                    #     element["content"]
+                    #     for element in skills_match_predicted
+                    #     if element["resume_index"] > 0 and element["vacancy_index"] > 0
+                    # ]
                     for element in matched_skills:
                         vacancy = vacancy.lower().replace(
                             element.lower(),
                             f'<span style="background-color: yellow;">{element}</span>',
                         )
+                    matches_score_tuples.append((vacancy, matched_skills, len(matched_skills)))
+                print("sorting matches based on the score.")
+                matches_score_tuples.sort(key=lambda x: x[-1], reverse=True)
+                print("constructing html table.")
+                for i, element in enumerate(matches_score_tuples, 1):
+                    vacancy, matched_skills, score = element
                     vacancy_formatted = vacancy.replace(".,", "<br/>")
-                    vacancy_formatted = f"VACATURE {i + 1}:<br/>{vacancy_formatted}"
+                    vacancy_formatted = f"VACATURE {i}:<br/>{vacancy_formatted}"
                     matches_html = "<br/> - ".join(matched_skills)
                     resume_matched_formatted = (
-                        f"Score: {len(matched_skills)}<br/>-{matches_html}"
+                        f"Score: {score}<br/>-{matches_html}"
                     )
-                    matches_score_tuples.append((vacancy_formatted, resume_matched_formatted, len(matched_skills)))
-                print("sorting matches based on the score.")
-                matches_score_tuples.sort(key=lambda x: x[-1], reverse=True)
-                print("constructing html table.")
-                for element in matches_score_tuples:
-                    html_table += f"<tr><td>{element[0]}</td><td>{element[1]}</td></tr>"
+                    html_table += f"<tr><td>{vacancy_formatted}</td><td>{resume_matched_formatted}</td></tr>"
             html_table += "</table>"
             return html_table