WebashalarForML commited on
Commit
77797bf
1 Parent(s): fa159ef

Update utility/utils.py

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