Update main.py

main.py

@@ -1,4 +1,4 @@
-from flask import Flask,request,make_response
+from flask import Flask, request, make_response
 import os
 import logging
 from dotenv import load_dotenv
@@ -12,36 +12,28 @@ from google.cloud import firestore
 # load env data
 load_dotenv()
 
-
-
-
-
 # messenger object
 messenger = WhatsApp(
-  os.environ["whatsapp_token"],
-  phone_number_id=os.environ["phone_number_id"] )
-
+    os.environ["whatsapp_token"],
+    phone_number_id=os.environ["phone_number_id"]
+)
 
 aai.settings.api_key = os.environ["aai_key"]
 transcriber = aai.Transcriber()
 
-
-# Authenticate to Firesotre with the JSON account key
+# Authenticate to Firestore with the JSON account key
 db = firestore.Client.from_service_account_json("firestore-key.json")
 
-
 app = Flask(__name__)
 
 VERIFY_TOKEN = "30cca545-3838-48b2-80a7-9e43b1ae8ce4"
 
-
 client = openai.OpenAI(
     api_key=os.environ.get("sambanova_api_key"),
     base_url="https://api.sambanova.ai/v1",
 )
 
 # Interactive button
-# Returned message type: interactive
 def messenger_button(recipient_phone, message, header='Transaction Confirmation', footer='', btn_name='Confirm Details'):
     messenger.send_button(
         recipient_id=recipient_phone,
@@ -56,11 +48,7 @@ def messenger_button(recipient_phone, message, header='Transaction Confirmation'
                         "title": "iBank",
                         "rows": [
                             {"id": "confirm", "title": "Record Transaction", "description": ""},
-                            {
-                                "id": "cancel",
-                                "title": "Cancel Transaction",
-                                "description": "",
-                            },
+                            {"id": "cancel", "title": "Cancel Transaction", "description": ""},
                         ],
                     }
                 ],
@@ -94,14 +82,13 @@ def messenger_reply_button(recipient_phone, message):
                     }
                 ]
             }
-      },
+        },
     )
 
-def respond(query_str:str):
+def respond(query_str: str):
     response = "hello, I don't have a brain yet"
     return response
 
-
 @app.route("/", methods=["GET", "POST"])
 def hook():
     if request.method == "GET":
@@ -116,17 +103,16 @@ def hook():
     # get message update..
     data = request.get_json()
     changed_field = messenger.changed_field(data)
-    
+
     if changed_field == "messages":
         new_message = messenger.get_mobile(data)
         if new_message:
-          mobile = messenger.get_mobile(data)
-          message_type = messenger.get_message_type(data)
-          # messenger.send_message(message_type,  recipient_id=mobile)
-        
+            mobile = messenger.get_mobile(data)
+            message_type = messenger.get_message_type(data)
+            
             if message_type == "text":
                 message = messenger.get_message(data)
-            
+
             elif message_type == "audio":
                 audio = messenger.get_audio(data)
                 audio_id, mime_type = audio["id"], audio["mime_type"]
@@ -136,57 +122,45 @@ def hook():
                 print(audio_filename)
                 print(transcript.text)
                 message = transcript.text
-                messenger.send_message(message=f"Transcribed message:\n {message}", recipient_id=mobile) # Send transcribed text back to user
+                messenger.send_message(message=f"Transcribed message:\n {message}", recipient_id=mobile)
                 logging.info(f"\nAudio: {audio}\n")
-            
+
             else:
-                messenger.send_message(message="Please send me text or audio messages",recipient_id=mobile)
+                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)
+
+            # 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:
-                    response = str(generateResponse(message))
-                    # Parse AI generated response into a structured format
-                    parsed_trans_data = parse_multiple_transactions(response)
-                    
-                    # messenger.send_message(message=f"{response} \n\n {parsed_trans_data}", recipient_id=mobile)
-                    
-                    
-                    # print("Response:", response)
-                    logging.info(f"\nAnswer: {response}\n")
-                    
-                    # Send confirmation button
-                    # messenger_button(mobile, response)
-                    # messenger_reply_button(mobile, response)
-                    
-                    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': # 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)
+                    firestore_msg = f'The detected intent, {intent}, is not currently supported!'
+
+                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)