Update helper.py

helper.py

@@ -296,9 +296,10 @@ def identify_locations(sentence):
               locations_dict[loc_type] = []
           locations_dict[loc_type].append(location)
       
+      
       # conditions for multiple references
       # it is mandatory that a country will exist
-      if locations_dict['country']:
+      if 'country' in locations_dict:
 
           # if a city exists
           if 'city' in locations_dict:
@@ -333,7 +334,7 @@ def identify_locations(sentence):
       
       # error if no country is referred
       else:
-         return (0, "LOCATION", "no_country")
+          return (0, "LOCATION", "no_country")
 
     except: 
       # handle the exception if any errors occur while identifying a country/city