Update main.py

main.py

@@ -9,6 +9,7 @@ from comparativesIdentification import comparatives_binding
 from earthquaqeIdentification import identify_earthquake_event
 
 
+
 def process_final_dict(final_dictionary):
   """
   Function to convert each one of the error codes from each component into a relevant code number to be handled by the SF
@@ -101,8 +102,6 @@ def natural_language_module(sentence):
     if comparative:
       final_dictionary.append(comparative)
 
-    print(final_dictionary)
-
     # identify the target date in the sentence
     date = dates_binding(sentence)
 
@@ -141,17 +140,30 @@ def natural_language_module(sentence):
     return "\n\n=== AN UNEXPECTED ERROR HAS OCCURED. PLEASE EXECUTE AGAIN THE SCRIPT OR COMMUNICATE WITH THE DEVELOPER TEAM === \n\n"
 
 
+def replace_zero_with_null(d):
+  """
+  This is a small helper function to convert the 0 references on the final json to be sent, into None, as needed by the SF
+  """
+  for k, v in d.items():
+    if isinstance(v, dict):
+      replace_zero_with_null(v)
+    elif v == 0 and k != "point":
+      d[k] = "null"
+  return d
+
+
 
 def process_json_sf(nl_json, sentence):
   """
   Function to conver the captured information an a relevant json format
   """
   try:
+
     sf_json_format = {
       "text": sentence,
       "page": "1",
       "nlp": {"event": nl_json['event'], "city": nl_json['city'][0], "country": nl_json['country'][0], "year": int(nl_json['date']['year']), "month": int(nl_json['date']['month']),
-              "day": int(nl_json['date']['day']), "magnitude": nl_json['Number'], "comparative": nl_json['comparative'], "point": False, "latitude": None,"lognitude": None}
+              "day": int(nl_json['date']['day']), "magnitude": nl_json['Number'], "comparative": nl_json['comparative'], "point": False, "latitude": "null", "lognitude": "null"}
     }
 
     return sf_json_format
@@ -169,5 +181,6 @@ def main(sentence):
 
   nl_data = natural_language_module(sentence)
   nl_json = process_json_sf(nl_data, sentence)
+  nl_json_with_null = replace_zero_with_null(nl_json)
 
-  return nl_json
+  return nl_json_with_null