Update main.py

main.py

@@ -57,10 +57,11 @@ def messenger_button(recipient_phone, message, header='Transaction Confirmation'
         },
     )
 
-def messenger_reply_button(recipient_phone, message):
+def messenger_reply_button(recipient_phone, message, header = 'Transaction Confirmation'):
     messenger.send_reply_button(
         recipient_id=f"{recipient_phone}",
         button={
+            "header": f"{header}",
             "type": "button",
             "body": {
                 "text": f"{message}"
@@ -90,12 +91,39 @@ def respond(query_str: str):
     response = "hello, I don't have a brain yet"
     return response
 
-def process_user_msg(message):
+def persist_temporary_transaction(transactions, mobile):
+    """
+    Persists the transaction data temporarily in Firestore.
+    """
+    intent = transactions[0]['intent'].lower()
+    temp_ref = db.collection("users").document(mobile).collection("temp_transactions").document(intent)
+    data = {
+        "transactions": transactions,
+        "status": "pending",
+        "created_at": datetime.now().isoformat()
+    }
+    temp_ref.set(data)
+    return True
+    
+
+def process_user_msg(message, mobile):
     response = str(generateResponse(message))
     parsed_trans_data = parse_multiple_transactions(response)
     # return None, parsed_trans_data, response
     logging.info(f"\nAnswer: {response}\n")
-    return response, parsed_trans_data
+    intent = parsed_trans_data[0]['intent'].lower()
+    if intent == 'read':
+        response2 = str(read_datalake(mobile, message))
+        # parsed_trans_data = ""
+        msg = response2
+        messenger.send_message(f"Raw Response: {response}, \n \n Parsed Response: {parsed_trans_data}, \n \n Final Response: {msg}", recipient_id=mobile)
+    else: 
+        # Persist transaction data temporarily for Create, Update, or Delete operations
+        persist_temporary_transaction(parsed_trans_data, mobile)
+        # Give user the chance to confirm/cancel transaction before processing other intents
+        messenger_reply_button(mobile, f"Raw Response: {response}, \n \n Parsed Response: {parsed_trans_data}")
+
+    return True 
     
 
 def process_intent(response, parsed_trans_data, mobile):
@@ -122,10 +150,10 @@ def process_intent(response, parsed_trans_data, mobile):
     elif intent == 'delete':
         pass
             
-    elif intent == 'read':
-        response = str(read_datalake(mobile, message))
-        parsed_trans_data = ""
-        firestore_msg = response
+    # elif intent == 'read':
+    #     response = str(read_datalake(mobile, message))
+    #     parsed_trans_data = ""
+    #     firestore_msg = response
                     
     else:
         firestore_msg = f'The detected intent, {intent}, is not currently supported!'
@@ -161,11 +189,7 @@ def hook():
                     
                     messenger.send_message(message=f"{response}", recipient_id=mobile)
                 else:
-                    # firestore_msg1, parsed_trans_data1, response1 = process_user_msg(message, mobile)
-                    response, parsed_trans_data = process_user_msg(message)
-                    messenger_reply_button(mobile, f"Raw Response: {response}, \n \n Parsed Response: {parsed_trans_data}")
-
-                    # messenger.send_message(f"Raw Response: {response1}, \n \n Parsed Response: {parsed_trans_data1}, \n \n Final Response: {firestore_msg1}", recipient_id=mobile)
+                    process_user_msg(message, mobile)
 
             elif message_type == "audio":
                 audio = messenger.get_audio(data)
@@ -178,10 +202,10 @@ def hook():
                 transcribed_message = transcript.text
             
                 # logging.info(f"\nAudio: {audio}\n")
-                response, parsed_trans_data = process_user_msg(message)
-                messenger_reply_button(mobile, f"Raw Response: {response}, \n \n Parsed Response: {parsed_trans_data}")
+                process_user_msg(transcribed_message, mobile)
                 # firestore_msg1, parsed_trans_data1, response1 = process_user_msg(transcribed_message, mobile)
                 # messenger.send_message(message=f"Raw Response: {response1}, \n \n Parsed Response: {parsed_trans_data1}, \n \n Final Response: {firestore_msg1}", recipient_id=mobile)
+                
             elif message_type == "interactive":
                 message_response = messenger.get_interactive_response(data)
                 interactive_type = message_response.get("type")