Update utils/mistral.py

utils/mistral.py

@@ -102,13 +102,12 @@ def Model_ProfessionalDetails_Output(resume, client):
     #for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True):#, temperature=0.35):
     #    response += message.choices[0].delta.content
         
-    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35)
-    print("This is without stream data ",data.choices[0].message.content)
+    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35)   
     response = data.choices[0].message.content
     
     try:
         clean_response = Data_Cleaner(response)
-        print("This is clean_response data ",clean_response)
+        print("This is clean_response data----> ",clean_response)
         parsed_response = json.loads(clean_response)
     except json.JSONDecodeError as e:
         logging.error(f"JSON Decode Error: {e}")
@@ -142,13 +141,12 @@ def Model_EducationalDetails_Output(resume, client):
     #response = ""
     #for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True):#, temperature=0.35):
     #    response += message.choices[0].delta.content
-    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35)
-    print("This is without stream data ",data.choices[0].message.content)
+    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35)    
     response = data.choices[0].message.content
     
     try:
         clean_response = Data_Cleaner(response)
-        print("This is clean_response data ",clean_response)
+        print("This is clean_response data----> ",clean_response)
         parsed_response = json.loads(clean_response)
     except json.JSONDecodeError as e:
         logging.error(f"JSON Decode Error: {e}")
@@ -182,17 +180,16 @@ def Model_PersonalDetails_Output(resume, client):
     # Response
     #response = ""
     #for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=True):#, temperature=0.35):
-    #    response += message.choices[0].delta.content
+    #response += message.choices[0].delta.content
 
-    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35)
-    print("This is without stream data ",data.choices[0].message.content)
+    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35)    
     response = data.choices[0].message.content
     
     # Handle cases where the response might have formatting issues
     try:
         #print('The Og response:-->',response)
         clean_response=Data_Cleaner(response)
-        print("This is clean_response data ",clean_response)
+        print("This is clean_response data----> ",clean_response)
         parsed_response = json.loads(clean_response)
         
     except json.JSONDecodeError as e:
@@ -202,7 +199,6 @@ def Model_PersonalDetails_Output(resume, client):
 
     return parsed_response
 
-
 # # Fallback to SpaCy if Mistral fails
 
 # Add regex pattern for LinkedIn and GitHub links
@@ -354,7 +350,6 @@ def is_valid_contact(contact):
     # Check if the contact matches any of the patterns
         return any(re.match(pattern, contact) for pattern in patterns) is not None
 
-
 def validate_contact_email(personal_data):
     contact = personal_data.get('contact', [])
     email = personal_data.get('email', [])
@@ -467,8 +462,7 @@ def process_resume_data(file_path):
                 ]
             }
         }
-        
-       
+              
 
         #Appending the list if any available as a text 
         if result['personal']['other_links'] is not None:
@@ -476,10 +470,10 @@ def process_resume_data(file_path):
             
         #Added the validator for details, Validate contact and email
         #valid_contact, invalid_contact, valid_email, invalid_email = validate_contact_email(result['personal'])
-        #result['personal']['valid_contact'] = valid_contact
-        #result['personal']['invalid_contact'] = invalid_contact
-        #result['personal']['valid_email'] = valid_email
-        #result['personal']['invalid_email'] = invalid_email
+        #result['personal']['valid_contact'] = None
+        result['personal']['invalid_contact'] = None
+        #result['personal']['valid_email'] = None
+        result['personal']['invalid_email'] = None
         
         #Appending the Educational Details if any available as a text