Update logging to include nlu_response object

app.py

@@ -68,16 +68,18 @@ async def evaluate_user_message_with_nlu_api(request: Request):
 
     # Handles if a student answer is already an integer or a float (ie., 8)
     if type(message_text) == int or type(message_text) == float:
-        # prepare_message_data_for_logging(message_data)
-        return JSONResponse(content={'type': 'integer', 'data': message_text})
+        nlu_response = {'type': 'integer', 'data': message_text, 'confidence': ''}
+        # prepare_message_data_for_logging(message_data, nlu_response)
+        return JSONResponse(content=nlu_response)
 
     # Removes whitespace and converts str to arr to handle multiple numbers
     message_text_arr = re.split(", |,| ", message_text.strip())
 
     # Handle if a student answer is a string of numbers (ie., "8,9, 10")
     if all(ele.isdigit() for ele in message_text_arr):
-        # prepare_message_data_for_logging(message_data)
-        return JSONResponse(content={'type': 'integer', 'data': ','.join(message_text_arr)})
+        nlu_response = {'type': 'integer', 'data': ','.join(message_text_arr), 'confidence': ''}
+        # prepare_message_data_for_logging(message_data, nlu_response)
+        return JSONResponse(content=nlu_response)
 
     student_response_arr = []
 
@@ -93,13 +95,13 @@ async def evaluate_user_message_with_nlu_api(request: Request):
         sentiment_api_resp = sentiment(message_text)
         # [{'label': 'POSITIVE', 'score': 0.991188645362854}]
         sent_data_dict = {'type': 'sentiment', 'data': sentiment_api_resp[0]['label']}
-        response_object = {'type': 'sentiment', 'data': 'negative'}
+        nlu_response = {'type': 'sentiment', 'data': 'negative', 'confidence': ''}
     else:
         if len(student_response_arr) > 1:
-            response_object = {'type': 'integer', 'data': ','.join(str(num) for num in student_response_arr)}
+            nlu_response = {'type': 'integer', 'data': ','.join(str(num) for num in student_response_arr), 'confidence': ''}
         else:
-            response_object = {'type': 'integer', 'data': student_response_arr[0]}
+            nlu_response = {'type': 'integer', 'data': student_response_arr[0], 'confidence': ''}
 
     # Uncomment to enable logging to Supabase
     # prepare_message_data_for_logging(message_data, nlu_response)
-    return JSONResponse(content=response_object)
+    return JSONResponse(content=nlu_response)

mathtext_fastapi/nlu.py

@@ -42,7 +42,7 @@ def get_or_create_supabase_entry(table_name, insert_data, check_variable=None):
 
 
 
-def prepare_message_data_for_logging(message_data):
+def prepare_message_data_for_logging(message_data, nlu_response):
     """ Builds the message data for each table and ensures it's logged to the database
 
     Input:
@@ -73,7 +73,8 @@ def prepare_message_data_for_logging(message_data):
         'channel_type': "whatsapp / turn.io",
         'message_inserted_at': message_data['message']['_vnd']['v1']['chat']['inserted_at'],
         'message_modified_at': message_data['message']['_vnd']['v1']['chat']['updated_at'],
-        'message_sent_at': format_datetime_in_isoformat(datetime.now())
+        'message_sent_at': format_datetime_in_isoformat(datetime.now()),
+        'nlu_response': nlu_response
         # Autogenerated fields: created_at, modified_at
     }
     message_data_log = log_message_data_through_supabase_api('message', message_data)