WebashalarForML commited on
Commit
5b3d644
·
verified ·
1 Parent(s): fd1d87f

Update utility/utils.py

Browse files
Files changed (1) hide show
  1. utility/utils.py +16 -7
utility/utils.py CHANGED
@@ -405,15 +405,24 @@ def process_resume_data(LLMdata,cont_data,extracted_text):
405
 
406
  # Initialize the processed data dictionary
407
  processed_data = {
408
- "name": [LLMdata.get('Name', 'Not found')],
409
- "contact_number": [LLMdata.get('Contact', 'Not found')],
410
- "Designation":[LLMdata.get('Designation', 'Not found')],
411
- "email": [LLMdata.get("Email", 'Not found')],
412
- "Location": [LLMdata.get('Address', 'Not found')],
413
- "Link": [LLMdata.get('Link', 'Not found')],
414
- "Company":[LLMdata.get('Company', 'Not found')],
415
  "extracted_text": extracted_text
416
  }
 
 
 
 
 
 
 
 
 
417
  processed_data['email'].extend(cont_data.get("emails", []))
418
  processed_data['contact_number'].extend(cont_data.get("phone_numbers", []))
419
  processed_data['Link'].extend(cont_data.get("links_RE", []))
 
405
 
406
  # Initialize the processed data dictionary
407
  processed_data = {
408
+ "name": [],
409
+ "contact_number": [],
410
+ "Designation":[],
411
+ "email": [],
412
+ "Location": [],
413
+ "Link": [],
414
+ "Company":[],
415
  "extracted_text": extracted_text
416
  }
417
+ #LLM
418
+ processed_data['name'].extend(LLMdata.get('Name', []))
419
+ processed_data['contact_number'].extend(LLMdata.get('Contact', []))
420
+ processed_data['Designation'].extend(LLMdata.get('Designation', []))
421
+ processed_data['email'].extend(LLMdata.get("Email", []))
422
+ processed_data['Location'].extend(LLMdata.get('Address', []))
423
+ processed_data['Link'].extend(LLMdata.get('Link', []))
424
+ processed_data['Company'].extend(LLMdata.get('Company', []))
425
+ #Contact
426
  processed_data['email'].extend(cont_data.get("emails", []))
427
  processed_data['contact_number'].extend(cont_data.get("phone_numbers", []))
428
  processed_data['Link'].extend(cont_data.get("links_RE", []))