Spaces:
Build error
Build error
| from mathtext_fastapi.logging import prepare_message_data_for_logging | |
| from mathtext.sentiment import sentiment | |
| from mathtext.text2int import text2int | |
| import re | |
| def build_nlu_response_object(type, data, confidence): | |
| """ Turns nlu results into an object to send back to Turn.io | |
| Inputs | |
| - type: str - the type of nlu run (integer or sentiment-analysis) | |
| - data: str - the student message | |
| - confidence: - the nlu confidence score (sentiment) or '' (integer) | |
| """ | |
| return {'type': type, 'data': data, 'confidence': confidence} | |
| def test_for_float_or_int(message_data, message_text): | |
| nlu_response = {} | |
| if type(message_text) == int or type(message_text) == float: | |
| nlu_response = build_nlu_response_object('integer', message_text, '') | |
| prepare_message_data_for_logging(message_data, nlu_response) | |
| return nlu_response | |
| def test_for_number_sequence(message_text_arr, message_data, message_text): | |
| nlu_response = {} | |
| if all(ele.isdigit() for ele in message_text_arr): | |
| nlu_response = build_nlu_response_object( | |
| 'integer', | |
| ','.join(message_text_arr), | |
| '' | |
| ) | |
| prepare_message_data_for_logging(message_data, nlu_response) | |
| return nlu_response | |
| def run_text2int_on_each_list_item(message_text_arr): | |
| """ Attempts to convert each list item to an integer | |
| Input | |
| - message_text_arr: list - a set of text extracted from the student message | |
| Output | |
| - student_response_arr: list - a set of integers (32202 for error code) | |
| """ | |
| student_response_arr = [] | |
| for student_response in message_text_arr: | |
| int_api_resp = text2int(student_response.lower()) | |
| student_response_arr.append(int_api_resp) | |
| return student_response_arr | |
| def run_sentiment_analysis(message_text): | |
| # TODO: Add intent labelling here | |
| # TODO: Add logic to determine whether intent labeling or sentiment analysis is more appropriate (probably default to intent labeling) | |
| return sentiment(message_text) | |
| def evaluate_message_with_nlu(message_data): | |
| message_text = message_data['message_body'] | |
| message_text_arr = re.split(", |,| ", message_text.strip()) | |
| # TODO: Replace this with appropriate utility function (is_int, is_float, render_int_or_float) | |
| nlu_response = test_for_float_or_int(message_data, message_text) | |
| if len(nlu_response) > 0: | |
| return nlu_response | |
| # TODO: Replace this with appropriate utility function | |
| nlu_response = test_for_number_sequence(message_text_arr, message_data, message_text) | |
| if len(nlu_response) > 0: | |
| return nlu_response | |
| student_response_arr = run_text2int_on_each_list_item(message_text_arr) | |
| # '32202' is text2int's error code for non-integer student answers (ie., "I don't know") | |
| # If any part of the list is 32202, sentiment analysis will run | |
| # TODO: Need to replace this with logic that recognizes multiple intents (Maybe 36 = "sentiment analysis" & "integer") | |
| student_response_arr = run_text2int_on_each_list_item(message_text_arr) | |
| if 32202 in student_response_arr: | |
| sentiment_api_resp = sentiment(message_text) | |
| nlu_response = build_nlu_response_object( | |
| 'sentiment', | |
| sentiment_api_resp[0]['label'], | |
| sentiment_api_resp[0]['score'] | |
| ) | |
| else: | |
| if len(student_response_arr) > 1: | |
| nlu_response = build_nlu_response_object( | |
| 'integer', | |
| ','.join(str(num) for num in student_response_arr), | |
| '' | |
| ) | |
| else: | |
| nlu_response = build_nlu_response_object( | |
| 'integer', | |
| student_response_arr[0], | |
| '' | |
| ) | |
| prepare_message_data_for_logging(message_data, nlu_response) | |
| return nlu_response | |