WebashalarForML commited on
Commit
696b013
1 Parent(s): ff65861

Update utility/utils.py

Browse files
Files changed (1) hide show
  1. utility/utils.py +10 -0
utility/utils.py CHANGED
@@ -476,4 +476,14 @@ def process_resume_data(LLMdata,cont_data,extracted_text):
476
  processed_data['email'].extend(LLMdata['Email'])
477
  processed_data['contact_number'].extend(LLMdata['Contact'])
478
  processed_data['Link'].extend(LLMdata['Link'])
 
 
 
 
 
 
 
 
 
 
479
  return processed_data
 
476
  processed_data['email'].extend(LLMdata['Email'])
477
  processed_data['contact_number'].extend(LLMdata['Contact'])
478
  processed_data['Link'].extend(LLMdata['Link'])
479
+
480
+ #to remove not found fields
481
+ # List of keys to check for 'Not found'
482
+ keys_to_check = ["name", "contact_number", "Designation", "email", "Location", "Link", "Company"]
483
+
484
+ # Replace 'Not found' with an empty list for each key
485
+ for key in keys_to_check:
486
+ if processed_data[key] == ['Not found'] or processed_data[key] == ['not found']:
487
+ processed_data[key] = []
488
+
489
  return processed_data