Update personal_information.py

personal_information.py

@@ -29,15 +29,15 @@ def extract_location(text):
     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())
+            city_present = any(location_data.get('city', {}).values())
+            country_present = any(location_data.get('country', {}).values())
         except json.JSONDecodeError:
             print("Failed to parse JSON from response")
-            city, country = None, False
+            city_present, country_present = False, False
     else:
-        city, country = None, False
+        city_present, country_present = False, False
     
-    return city, country
+    return city_present, country_present
 
 def calculate_score(email_exists, phone_exists, city_exists, country_exists):
     score = 0
@@ -56,20 +56,18 @@ def analyze_personal_info(file_path):
     
     email = extract_email(text)
     phone = extract_phone(text)
-    city, country = extract_location(text)
+    city_present, country_present = 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)
+    score = calculate_score(email_exists, phone_exists, city_present, country_present)
     
     result = {
         "email": email_exists,
         "phone": phone_exists,
-        "city": city if city_exists else None,
-        "country": country_exists,
+        "city": city_present,
+        "country": country_present,
         "personal_info_score": score
     }