Update Home.py

Home.py

@@ -138,22 +138,27 @@ if uploaded_file:
 
             except Exception as e:
                 st.error(f"❌ Failed to parse {filename}: {e}")
+    skill_categories = {
+                            "Programming Languages" : ["Python"],
+                            "Data Analysis & Visualisation": ["Pandas", "Numpy", "Excel", "Matplotlib", "Seaborn"],
+                            "Database Management": ["SQL", "Power BI"],
+                            "Deep Learning": ["ANN", "CNN", "RNN"],
+                            "Machine Learning": ["Scikit-learn", "OpenCV", "NLP", "Supervised learning", "Optuna", "Descriptive Statistics"],
+                            "Generative AI": ["Langchain", "LLMs"]
+                        }
 
     if parsed_resumes:
-        selected_skills = st.multiselect("Select required skills", sorted(unique_skills), key="skills_select")
-
-        if st.button("Evaluate Resumes"):
-            # Clear selected folder before saving new selection
-            if os.path.exists(SELECTED_FOLDER):
-                shutil.rmtree(SELECTED_FOLDER)
-            os.makedirs(SELECTED_FOLDER, exist_ok=True)
-
-            for resume in parsed_resumes:
-                if all(skill in resume["Skills"] for skill in selected_skills):
-                    st.success(f"✅ Selected: {resume['Name']}")
-                    target_path = os.path.join(SELECTED_FOLDER, os.path.basename(resume["file_path"]))
-                    shutil.copy(resume["file_path"], target_path)
-                    st.write(f"Copied {resume['file_path']} to {target_path}")
-                    st.write(f"Exists now? {os.path.exists(target_path)}")
-                else:
-                    st.warning(f"❌ Rejected: {resume['Name']}")
+    selected_categories = st.multiselect("Select required skill categories", list(skill_categories.keys()))
+
+    if st.button("Evaluate Resumes"):
+        required_keywords = set()
+        for category in selected_categories:
+            required_keywords.update(skill_categories[category])
+
+        for resume in parsed_resumes:
+            # Match: If any required keyword is in the resume skills
+            if any(req_skill.lower() in (skill.lower() for skill in resume["Skills"]) for req_skill in required_keywords):
+                st.success(f"✅ Selected: {resume['Name']}")
+                shutil.copy(resume["file_path"], os.path.join(SELECTED_FOLDER, os.path.basename(resume["file_path"])))
+            else:
+                st.warning(f"❌ Rejected: {resume['Name']}")