Update main.py

main.py

@@ -1,3 +1,18 @@
+import time
+
+print("\n\n ==== THE NATURAL LANGUAGE MODULE IS BEING LOADED. PLEASE WAIT ==== \n\n")
+start_time_load = time.time()
+
+from transformers import logging
+logging.set_verbosity_error()
+
+import warnings
+warnings.filterwarnings("ignore", category=UserWarning)
+
+import sys
+
+import requests
+
 from countriesIdentification import identify_locations
 from datesIdentification import dates_binding
 from magnitudeIdentification import magnitude_binding
@@ -73,65 +88,68 @@ def process_final_dict(final_dictionary):
 
 
 def natural_language_module(sentence):
-  
-
+  """
+  Function to execute the complete natural language module pipeline
+  """
   try:
     final_dictionary = []
-    
+
     # identify whether the sentence is referred on earthquake events
     earth_event = identify_earthquake_event(sentence)
-    
+
     if earth_event:
       final_dictionary.append(earth_event)
-    
+
     # identify the target country and city in the sentence
     location = identify_locations(sentence)
-    
+
     if location:
       final_dictionary.append(location)
-    
+
     # identify the target comparative in the sentence
     comparative = comparatives_binding(sentence)
-    
+
     if comparative:
       final_dictionary.append(comparative)
-    
+
+    print(final_dictionary)
+
     # identify the target date in the sentence
     date = dates_binding(sentence)
-    
+
     if isinstance(date, list):
-    
+
         date_dict = date[0]
         date_replc = date[1]
-    
+
         if date_dict:
           final_dictionary.append(date_dict[0])
-    
+
           # we also delete the date reference from the sentence so that there will
           # not be any confusion with it for the magnitude identification module
           if len(date_replc) == 1:
             sentence = sentence.replace(date_replc[0], " ")
-    
+
     # in case it is a tuple we add it as it is and we do not substitute something in the sentence
     elif isinstance(date, tuple):
       final_dictionary.append(date)
-    
+
     # identify the target magnitude number in the sentence
     magnitude = magnitude_binding(sentence)
-    
+
     if magnitude:
       final_dictionary.append(magnitude)
-    
+
     clean_final_dictionary = process_final_dict(final_dictionary)
-    
+
     result = {}
     for d in clean_final_dictionary:
       result.update(d)
-    
+
     return result
 
   except:
-      return "\n\n=== AN UNEXPECTED ERROR HAS OCCURED. PLEASE EXECUTE AGAIN THE SCRIPT OR COMMUNICATE WITH THE DEVELOPER TEAM === \n\n"
+    return "\n\n=== AN UNEXPECTED ERROR HAS OCCURED. PLEASE EXECUTE AGAIN THE SCRIPT OR COMMUNICATE WITH THE DEVELOPER TEAM === \n\n"
 
 
 
@@ -154,12 +172,55 @@ def process_json_sf(nl_json, sentence):
 
 
 
+def send_json_to_endpoint(json_output, username, password):
+  """
+  Function to send the produced json to a target endpoint
+  """
+  endpoint_url = "http://160.40.54.158:8087/SemanticFramework/api/retrieve"
+  headers = {'Content-type': 'application/json'}
+  auth = (username, password)
+
+  print(" === THE FINAL OUTPUT THAT IS SENT TO THE SF === \n")
+  print(json_output)
+
+  response = requests.post(endpoint_url, json=json_output, headers=headers, auth=auth)
+
+  if response.status_code == 200:
+    print("\n\n === THE RESPONSE RETRIEVED BY THE SF ===")
+    print("\n")
+    print(response.content)
+    print("\n\n")
+  else:
+    print("Error sending JSON to endpoint")
+    print("\n\n")
+
+
+
 def main(sentence):
   """
   Function to bind together all the info and be executed
   """
 
+  end_time_load = time.time()
+  elapsed_time_load = end_time_load - start_time_load
+
+  print(f"=== THE NATURAL LANGUAGE MODULE HAS BEEN LOADED. THE LOADING TIME WAS : {elapsed_time_load} seconds === \n\n")
+  print(" === THE NATURAL LANGUAGE MODULE HAS JUST STARTED PROCESSING THE INPUT SENTENCE === \n")
+
+  start_time = time.time()
+
   nl_data = natural_language_module(sentence)
+
+  print("\n")
+  print(" === FINAL OUTPUT === \n")
+  print(nl_data)
+
+  end_time = time.time()
+  elapsed_time = end_time - start_time
+
+  print("\n\n\n ==== THE NATURAL LANGUAGE MODULE HAS FINISHED ITS EXECUTION ==== \n\n\n")
+  print(f" ==== THE TIME OF SENTENCE PROCESSING WAS: {elapsed_time} seconds ==== \n\n")
+
   nl_json = process_json_sf(nl_data, sentence)
 
   return nl_json