Add endpoint for conversation management of global and local states

app.py

@@ -17,6 +17,7 @@ from pydantic import BaseModel
 
 from mathtext_fastapi.logging import prepare_message_data_for_logging
 from mathtext_fastapi.conversation_manager import manage_conversation_response
+from mathtext_fastapi.v2_conversation_manager import manage_conversation_response
 from mathtext_fastapi.nlu import evaluate_message_with_nlu
 from mathtext_fastapi.nlu import run_intent_classification
 
@@ -56,7 +57,40 @@ def text2int_ep(content: Text = None):
     return JSONResponse(content=content)
 
 
-@app.post("/manager")
+@app.post("/v1/manager")
+async def programmatic_message_manager(request: Request):
+    """
+    Calls conversation management function to determine the next state
+
+    Input
+    request.body: dict - message data for the most recent user response
+    {
+        "author_id": "+47897891",
+        "contact_uuid": "j43hk26-2hjl-43jk-hnk2-k4ljl46j0ds09",
+        "author_type": "OWNER",
+        "message_body": "a test message",
+        "message_direction": "inbound",
+        "message_id": "ABJAK64jlk3-agjkl2QHFAFH",
+        "message_inserted_at": "2022-07-05T04:00:34.03352Z",
+        "message_updated_at": "2023-02-14T03:54:19.342950Z",
+    }
+
+    Output
+    context: dict - the information for the current state
+    {
+        "user": "47897891",
+        "state": "welcome-message-state",
+        "bot_message": "Welcome to Rori!",
+        "user_message": "",
+        "type": "ask"
+    }
+    """
+    data_dict = await request.json()
+    context = manage_conversation_response(data_dict)
+    return JSONResponse(context)
+
+
+@app.post("/v2/manager")
 async def programmatic_message_manager(request: Request):
     """
     Calls conversation management function to determine the next state

mathtext_fastapi/global_state_manager.py

@@ -18,3 +18,6 @@ class GlobalStateManager(object):
             transitions=GlobalStateManager.transitions,
             initial=initial_state
         )
+
+
+curriculum = GlobalStateManager()

mathtext_fastapi/v2_conversation_manager.py

@@ -0,0 +1,196 @@
+import base64
+import copy
+import dill
+import os
+import json
+import jsonpickle
+import pickle
+import random
+import requests
+import mathtext_fastapi.global_state_manager as gsm
+
+from dotenv import load_dotenv
+from mathtext_fastapi.nlu import evaluate_message_with_nlu
+from mathtext_fastapi.math_quiz_fsm import MathQuizFSM
+from mathtext_fastapi.math_subtraction_fsm import MathSubtractionFSM
+from supabase import create_client
+from transitions import Machine
+
+from scripts.quiz.generators import start_interactive_math
+from scripts.quiz.hints import generate_hint
+
+load_dotenv()
+
+SUPA = create_client(
+    os.environ.get('SUPABASE_URL'),
+    os.environ.get('SUPABASE_KEY')
+)
+
+
+def pickle_and_encode_state_machine(state_machine):
+    dump = pickle.dumps(state_machine)
+    dump_encoded = base64.b64encode(dump).decode('utf-8')
+    return dump_encoded
+
+
+def manage_math_quiz_fsm(user_message, contact_uuid, type):
+    fsm_check = SUPA.table('state_machines').select("*").eq(
+        "contact_uuid",
+        contact_uuid
+    ).execute()
+
+    # This doesn't allow for when one FSM is present and the other is empty
+    """
+    1
+    data=[] count=None
+    
+    2
+    data=[{'id': 29, 'contact_uuid': 'j43hk26-2hjl-43jk-hnk2-k4ljl46j0ds09', 'addition3': None, 'subtraction': None, 'addition': 
+
+    - but problem is there is no subtraction , but it's assuming there's a subtration
+
+    Cases
+    - make a completely new record
+    - update an existing record with an existing FSM
+    - update an existing record without an existing FSM
+    """
+    print("MATH QUIZ FSM ACTIVITY")
+    print("user_message")
+    print(user_message)
+    # Make a completely new entry
+    if fsm_check.data == []:
+        if type == 'addition':
+            math_quiz_state_machine = MathQuizFSM()
+        else:
+            math_quiz_state_machine = MathSubtractionFSM()
+        messages = [math_quiz_state_machine.response_text]
+        dump_encoded = pickle_and_encode_state_machine(math_quiz_state_machine)
+
+        SUPA.table('state_machines').insert({
+            'contact_uuid': contact_uuid,
+            f'{type}': dump_encoded
+        }).execute()
+    # Update an existing record with a new state machine
+    elif not fsm_check.data[0][type]:
+        if type == 'addition':
+            math_quiz_state_machine = MathQuizFSM()
+        else:
+            math_quiz_state_machine = MathSubtractionFSM()
+        messages = [math_quiz_state_machine.response_text]
+        dump_encoded = pickle_and_encode_state_machine(math_quiz_state_machine)
+
+        SUPA.table('state_machines').update({
+            f'{type}': dump_encoded
+        }).eq(
+            "contact_uuid", contact_uuid
+        ).execute()      
+    # Update an existing record with an existing state machine
+    elif fsm_check.data[0][type]:
+        undump_encoded = base64.b64decode(
+            fsm_check.data[0][type].encode('utf-8')
+        )
+        math_quiz_state_machine = pickle.loads(undump_encoded)
+
+        math_quiz_state_machine.student_answer = user_message
+        math_quiz_state_machine.correct_answer = str(math_quiz_state_machine.correct_answer)
+        messages = math_quiz_state_machine.validate_answer()
+        dump_encoded = pickle_and_encode_state_machine(math_quiz_state_machine)          
+        SUPA.table('state_machines').update({
+            f'{type}': dump_encoded
+        }).eq(
+            "contact_uuid", contact_uuid
+        ).execute()
+    return messages
+
+
+def retrieve_microlesson_content(context_data, user_message, microlesson, contact_uuid):
+    if context_data['local_state'] == 'addition-question-sequence' or \
+        user_message == 'add' or \
+        microlesson == 'addition':
+        messages = manage_math_quiz_fsm(user_message, contact_uuid, 'addition')
+
+        if user_message == 'exit':
+            state_label = 'exit'
+        else:
+            state_label = 'addition-question-sequence'
+
+        input_prompt = messages.pop()
+        message_package = {
+            'messages': messages,
+            'input_prompt': input_prompt,
+            'state': state_label
+        }
+    elif context_data['local_state'] == 'subtraction-question-sequence' or \
+        user_message == 'subtract' or \
+        microlesson == 'subtraction':
+        messages = manage_math_quiz_fsm(user_message, contact_uuid, 'subtraction')
+
+        if user_message == 'exit':
+            state_label = 'exit'
+        else:
+            state_label = 'subtraction-question-sequence'
+
+        input_prompt = messages.pop()
+
+        message_package = {
+            'messages': messages,
+            'input_prompt': input_prompt,
+            'state': state_label
+        }
+    print("MICROLESSON CONTENT RESPONSE")
+    print(message_package)
+    return message_package
+
+
+curriculum_lookup_table = {
+    'N1.1.1_G1': 'addition',
+    'N1.1.1_G2': 'subtraction'
+}
+
+
+def lookup_local_state(next_state):
+    microlesson = curriculum_lookup_table[next_state]
+    return microlesson
+
+
+def manage_conversation_response(data_json):
+    """ Calls functions necessary to determine message and context data """
+    print("V2 ENDPOINT")
+
+    user_message = ''
+
+    # nlu_response = evaluate_message_with_nlu(message_data)
+
+    context_data = {
+        'contact_uuid': 'abcdefg',
+        'current_state': 'N1.1.1_G2',
+        'user_message': '1',
+        'local_state': ''
+    }
+    print("STEP 1")
+    if not context_data['current_state']:
+        context_data['current_state'] = 'N1.1.1_G1'
+
+    curriculum_copy = copy.deepcopy(gsm.curriculum)
+
+    print("STEP 2")
+    if context_data['user_message'] == 'easier':
+        curriculum_copy.left()
+        next_state = curriculum_copy.state
+    elif context_data['user_message'] == 'harder':
+        curriculum_copy.right()
+        next_state = curriculum_copy.state
+    else:
+        next_state = context_data['current_state']
+
+    print("STEP 3")
+    microlesson = lookup_local_state(next_state)
+
+    print("microlesson")
+    print(microlesson)
+
+    microlesson_content = retrieve_microlesson_content(context_data, context_data['user_message'], microlesson, context_data['contact_uuid'])
+
+    print("STEP 4")
+    # combine microlesson content and context_data object
+    return context_data

scripts/make_request.py

@@ -69,12 +69,13 @@ def run_simulated_request(endpoint, sample_answer, context=None):
 # run_simulated_request('nlu', 'eight, nine, ten')
 # run_simulated_request('nlu', '8, 9, 10')
 # run_simulated_request('nlu', '8')
-run_simulated_request('nlu', "I don't know")
+# run_simulated_request('nlu', "I don't know")
 # run_simulated_request('nlu', "I don't know eight")
 # run_simulated_request('nlu', "I don't 9")
 # run_simulated_request('nlu', "0.2")
 # run_simulated_request('nlu', 'Today is a wonderful day')
 # run_simulated_request('nlu', 'IDK 5?')
+run_simulated_request('v2/manager', '')
 # run_simulated_request('manager', '')
 # run_simulated_request('manager', 'add')
 # run_simulated_request('manager', 'subtract')