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 generateResponse, parse_multiple_transactions, create_inventory, create_sale
+from utility import generateResponse, parse_multiple_transactions, create_inventory, create_sale, read_datalake
 from google.cloud import firestore
 
 
@@ -89,6 +89,42 @@ def respond(query_str: str):
     response = "hello, I don't have a brain yet"
     return response
 
+def process_user_msg(message, mobile):
+    response = str(generateResponse(message))
+    parsed_trans_data = parse_multiple_transactions(response)
+    logging.info(f"\nAnswer: {response}\n")
+
+    intent = parsed_trans_data[0]['intent'].lower()
+    trans_type = parsed_trans_data[0]['transaction_type'].lower()
+    
+    if intent == 'create':
+        if trans_type == 'purchase':
+            if create_inventory(mobile, parsed_trans_data):
+                firestore_msg = "Transaction recorded successfully!"
+            else:
+                firestore_msg = "Sorry, could not record transaction!"
+                
+        elif trans_type == 'sale':
+            if create_sale(mobile, parsed_trans_data):
+                firestore_msg = "Transaction recorded successfully!"
+            else:
+                firestore_msg = "Sorry, could not record transaction!"
+                
+    elif intent == 'update':
+        pass
+            
+    elif intent == 'delete':
+        pass
+            
+    elif intent == 'read':
+        response = str(read_datalake(mobile, message))
+        firestore_msg = response
+                    
+    else:
+        firestore_msg = f'The detected intent, {intent}, is not currently supported!'
+
+    return firestore_msg
+
 @app.route("/", methods=["GET", "POST"])
 def hook():
     if request.method == "GET":
@@ -112,6 +148,14 @@ def hook():
             
             if message_type == "text":
                 message = messenger.get_message(data)
+                # Handle greetings
+                if message.lower() in ("hi", "hello", "help", "how are you"):
+                    response = "Hi there! My name is SmartLedger. How can I help you today?"
+                    messenger.send_message(message=f"{response}", recipient_id=mobile)
+                else:
+                    firestore_msg = process_user_msg(message, mobile)
+
+                    messenger.send_message(message=f"{firestore_msg}", recipient_id=mobile)
 
             elif message_type == "audio":
                 audio = messenger.get_audio(data)
@@ -122,45 +166,17 @@ def hook():
                 print(audio_filename)
                 print(transcript.text)
                 message = transcript.text
-                messenger.send_message(message=f"Transcribed message:\n {message}", recipient_id=mobile)
                 logging.info(f"\nAudio: {audio}\n")
+                firestore_msg = process_user_msg(message, mobile)
+                messenger.send_message(message=f"{firestore_msg}", recipient_id=mobile)
 
-            else:
-                messenger.send_message(message="Please send me text or audio messages", recipient_id=mobile)
-                return None
 
-            # Handle greetings
-            if message.lower() in ("hi", "hello", "help", "how are you"):
-                response = "Hi there! My name is SmartLedger. How can I help you today?"
-                messenger.send_message(message=f"{response}", recipient_id=mobile)
             else:
-                response = str(generateResponse(message))
-                parsed_trans_data = parse_multiple_transactions(response)
-                logging.info(f"\nAnswer: {response}\n")
-
-                intent = parsed_trans_data[0]['intent'].lower()
-                trans_type = parsed_trans_data[0]['transaction_type'].lower()
-                if intent == 'create':
-                    if trans_type == 'purchase':
-                        if create_inventory(mobile, parsed_trans_data):
-                            firestore_msg = "Transaction recorded successfully!"
-                        else:
-                            firestore_msg = "Sorry, could not record transaction!"
-                    elif trans_type == 'sale':
-                        if create_sale(mobile, parsed_trans_data):
-                            firestore_msg = "Transaction recorded successfully!"
-                        else:
-                            firestore_msg = "Sorry, could not record transaction!"
-                elif intent == 'update':
-                    pass
-                elif intent == 'delete':
-                    pass
-                else:
-                    firestore_msg = f'The detected intent, {intent}, is not currently supported!'
+                messenger.send_message(message="Please send me text or audio messages", recipient_id=mobile)
 
-                messenger.send_message(message=f"{response},\n\n {firestore_msg}", recipient_id=mobile)
+            
 
     return "ok"
 
 if __name__ == '__main__':
-    app.run(debug=True, host="0.0.0.0", port=7860)
+    app.run(debug=True, host="0.0.0.0", port=7860)