Update src/streamlit_app.py

src/streamlit_app.py

@@ -137,18 +137,18 @@ if not comet_initialized:
 
 # --- Label Definitions ---
 labels = [
-"Email_Address", "Phone_Number", "Street_Address", "City", "State", "Zip_code", "Country",
-"Date_of_Birth", "Gender", "Marital_Status", "Full_Name",
-"Full_Time", "Part_Time", "Contract", "Temporary", "Terminated", "Active", "Retired",
-"Job_Title", "Employment_Type", "Start_Date", "End_Date", "Company_Name", "Organization_Name", "Role", "Position",
-"Performance_Review", "Performance_Rating", "Performance_Score",
-"Sick_Days", "Vacation_Days", "Leave_of_Absence", "Holidays",
-"Pension", "Retirement_Plan", "Bonus", "Stock_Options", "Health_Insurance","Retire date",
-"Pay_Rate", "Hourly_Wage", "Annual_Salary", "Overtime_Pay",
-"Tax", "Social_Security", "Deductions",
-"Job_Posting", "Job_Description", "Interview_Type", "Applicant", "Candidate", "Referral", "Job_board", "Recruiter",
+"Email_address", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country",
+"Date_of_birth", "Gender", "Marital_status", "Full_name",
+"Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired",
+"Job_title", "Employment_type", "Start_date", "End_date", "Company", "Organization", "Role", "Position",
+"Performance_review", "Performance_rating", "Performance_score",
+"Sick_days", "Vacation_days", "Leave_of_absence", "Holidays",
+"Pension", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance","Retire_date",
+"Pay_rate", "Hourly_wage", "Annual_salary", "Overtime_pay",
+"Tax", "Social_security", "Deductions",
+"Job_posting", "Job_description", "Interview_type", "Applicant", "Candidate", "Referral", "Job_board", "Recruiter",
 "Contract", "Offer_letter", "Agreement",
-"Training_Course", "Certification", "Skill"]
+"Training_course", "Certification", "Skill"]
 
 
 
@@ -157,19 +157,19 @@ labels = [
 category_mapping = {
 
     
-    "Contact Information": ["Email_Address", "Phone_Number", "Street_Address", "City", "State", "Zip_code", "Country"],
-    "Personal Details": ["Date of birth", "Gender", "Marital_Status", "Full_Name"],
-    "Employment Status": ["Full_Time", "Part_Time", "Contract", "Temporary", "Terminated", "Active", "Retired"],
-    "Employment Information" : ["Job_Title", "Employment_Type", "Start_Date", "End_Date", "Company_Name", "Organization_Name", "Role", "Position"],
+    "Contact Information": ["Email_address", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country"],
+    "Personal Details": ["Date_of_birth", "Gender", "Marital_status", "Full_name"],
+    "Employment Status": ["Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired"],
+    "Employment Information" : ["Job_title", "Employment_type", "Start_date", "End_date", "Company", "Organization", "Role", "Position"],
     
-    "Performance": ["Performance_Review", "Performance_Rating", "Performance_Score"],
-    "Attendance": ["Sick_Days", "Vacation_Days", "Leave_of_Absence", "Holidays"],
-    "Benefits": ["Pension", "Retirement_Plan", "Bonus", "Stock_Options", "Health_Insurance", "Retire date"],
-    "Compensation": ["Pay_Rate", "Hourly_Wage", "Annual_Salary", "Overtime_Pay"],
-    "Deductions": ["Tax", "Social_Security", "Deductions"],
-    "Recruitment & Sourcing": ["Job_Posting", "Job_Description", "Interview_Type", "Applicant", "Candidate", "Referral", "Job_board", "Recruiter"],
+    "Performance": ["Performance_review", "Performance_rating", "Performance_score"],
+    "Attendance": ["Sick_days", "Vacation_days", "Leave_of_absence", "Holidays"],
+    "Benefits": ["Pension", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance","Retire_date"],
+    "Compensation": ["Pay_rate", "Hourly_wage", "Annual_salary", "Overtime_pay"],
+    "Deductions": ["Tax", "Social_security", "Deductions"],
+    "Recruitment & Sourcing": ["Job_posting", "Job_description", "Interview_type", "Applicant", "Candidate", "Referral", "Job_board", "Recruiter"],
     "Legal & Compliance": ["Contract", "Offer_letter", "Agreement"],
-    "Professional_Development": ["Training_Course", "Certification", "Skill"]
+    "Professional_Development": ["Training_course", "Certification", "Skill"]
 }