Update main.py

main.py

@@ -5,7 +5,7 @@ from dotenv import load_dotenv
 from heyoo import WhatsApp
 import assemblyai as aai
 import openai
-from utility import parse_multiple_transactions
+from utility import parse_multiple_transactions, create_transaction
 from google.cloud import firestore
 
 
@@ -131,8 +131,24 @@ def hook():
           print("Response:", response)
           logging.info(f"\nAnswer: {response}\n")
         # Handle cases where response is not a valid image path
-          messenger.send_message(message=f"{response}\n\n {parsed_trans_data}", recipient_id=mobile)
-            
+          # messenger.send_message(message=f"{response}", recipient_id=mobile)
+
+          intent = parsed_trans_data[0]['intent'].lower()
+          if intent == 'create':
+                if create_transaction(mobile, parsed_trans_data):
+                    firestore_msg = "Transaction recorded successfully!"
+                else: 
+                    firestore_msg = "Sorry, could not record transaction!"
+          elif intent == 'update': # handle update
+                pass
+          elif intent == 'delete': # handle delete
+                pass
+          else:
+                firestore_msg = f'The detected intent, {intent}, is not currently supported!'
+    
+          messenger.send_message(message=f"{response},\n\n {firestore_msg}", recipient_id=mobile)
+
+        
       # elif message_type == "audio":
       #   audio = messenger.get_audio(data)
       #   audio_id, mime_type = audio["id"], audio["mime_type"]