Create personal_information.py

personal_information.py

@@ -0,0 +1,76 @@
+import json
+import re
+from openai_utils import get_ai_response
+from ocr_extractor import process_file
+from cv_prompt import get_personal_info_prompt
+
+# Load the scoring data
+with open('personal_info_scores.json', 'r') as f:
+    score_data = json.load(f)
+
+def extract_email(text):
+    email_pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
+    emails = re.findall(email_pattern, text)
+    return emails[0] if emails else None
+
+def extract_phone(text):
+    phone_pattern = r'\b(?:\+?1[-.\s]?)?(?:\(\d{3}\)|\d{3})[-.\s]?\d{3}[-.\s]?\d{4}\b'
+    phones = re.findall(phone_pattern, text)
+    return phones[0] if phones else None
+
+def extract_location(text):
+    prompt = get_personal_info_prompt(text)
+    messages = [
+        {"role": "user", "content": prompt}
+    ]
+    
+    response = get_ai_response(messages)
+    
+    if response:
+        try:
+            location_data = json.loads(response)
+            city = list(location_data.get('city', {}).keys())[0] if location_data.get('city') else None
+            country = any(location_data.get('country', {}).values())
+        except json.JSONDecodeError:
+            print("Failed to parse JSON from response")
+            city, country = None, False
+    else:
+        city, country = None, False
+    
+    return city, country
+
+def calculate_score(email_exists, phone_exists, city_exists, country_exists):
+    score = 0
+    if email_exists:
+        score += score_data['email']
+    if phone_exists:
+        score += score_data['phone']
+    if city_exists:
+        score += score_data['city']
+    if country_exists:
+        score += score_data['country']
+    return score
+
+def analyze_personal_info(file_path):
+    text = process_file(file_path, 'ocr_weights.json')
+    
+    email = extract_email(text)
+    phone = extract_phone(text)
+    city, country = extract_location(text)
+    
+    email_exists = email is not None
+    phone_exists = phone is not None
+    city_exists = city is not None
+    country_exists = country
+    
+    score = calculate_score(email_exists, phone_exists, city_exists, country_exists)
+    
+    result = {
+        "email": email_exists,
+        "phone": phone_exists,
+        "city": city if city_exists else None,
+        "country": country_exists,
+        "personal_info_score": score
+    }
+    
+    return result