WebashalarForML commited on
Commit
c1723fc
1 Parent(s): 42a82b0

Update utility/utils.py

Browse files
Files changed (1) hide show
  1. utility/utils.py +10 -10
utility/utils.py CHANGED
@@ -470,17 +470,17 @@ def process_resume_data(LLMdata,cont_data,extracted_text):
470
  #LLM
471
 
472
  processed_data['name'].extend(LLMdata.get('Name', None))
473
- #processed_data['contact_number'].extend(LLMdata.get('Contact', None))
474
- processed_data['Designation'].extend(LLMdata.get('Designation', None))
475
- #processed_data['email'].extend(LLMdata.get("Email", None))
476
- processed_data['Location'].extend(LLMdata.get('Address', None))
477
- #processed_data['Link'].extend(LLMdata.get('Link', None))
478
- processed_data['Company'].extend(LLMdata.get('Company', None))
479
 
480
  #Contact
481
- #processed_data['email'].extend(cont_data.get("emails", None))
482
- #processed_data['contact_number'].extend(cont_data.get("phone_numbers", None))
483
- #processed_data['Link'].extend(cont_data.get("links_RE", None))
484
 
485
  #New_merge_data
486
  processed_data['email'].extend(LLMdata['Email'])
@@ -494,6 +494,6 @@ def process_resume_data(LLMdata,cont_data,extracted_text):
494
  # Replace 'Not found' with an empty list for each key
495
  for key in keys_to_check:
496
  if processed_data[key] == ['Not found'] or processed_data[key] == ['not found']:
497
- processed_data[key] = None
498
 
499
  return processed_data
 
470
  #LLM
471
 
472
  processed_data['name'].extend(LLMdata.get('Name', None))
473
+ #processed_data['contact_number'].extend(LLMdata.get('Contact', []))
474
+ processed_data['Designation'].extend(LLMdata.get('Designation', []))
475
+ #processed_data['email'].extend(LLMdata.get("Email", []))
476
+ processed_data['Location'].extend(LLMdata.get('Address', []))
477
+ #processed_data['Link'].extend(LLMdata.get('Link', []))
478
+ processed_data['Company'].extend(LLMdata.get('Company', []))
479
 
480
  #Contact
481
+ #processed_data['email'].extend(cont_data.get("emails", []))
482
+ #processed_data['contact_number'].extend(cont_data.get("phone_numbers", []))
483
+ #processed_data['Link'].extend(cont_data.get("links_RE", []))
484
 
485
  #New_merge_data
486
  processed_data['email'].extend(LLMdata['Email'])
 
494
  # Replace 'Not found' with an empty list for each key
495
  for key in keys_to_check:
496
  if processed_data[key] == ['Not found'] or processed_data[key] == ['not found']:
497
+ processed_data[key] = []
498
 
499
  return processed_data