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, create_transaction
+from utility import parse_multiple_transactions, create_inventory, create_sales
 from google.cloud import firestore
 
 
@@ -134,11 +134,20 @@ def hook():
           # messenger.send_message(message=f"{response}", recipient_id=mobile)
 
           intent = parsed_trans_data[0]['intent'].lower()
+          trans_type = parsed_trans_data[0]['transaction_type'].lower()
           if intent == 'create':
-                if create_transaction(mobile, parsed_trans_data):
-                    firestore_msg = "Transaction recorded successfully!"
-                else: 
-                    firestore_msg = "Sorry, could not record transaction!"
+                if trans_type == 'purchase':
+                    if create_inventory(user_phone, parsed_trans_data):
+                        firestore_msg = "Transaction recorded successfully!"
+                    else: 
+                        firestore_msg = "Sorry, could not record transaction!"
+
+                elif trans_type == 'sale':
+                    if create_sales(user_phone, 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