main.py

from flask import Flask,request,make_response
import os
import logging
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 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"] )

# Configure log level
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
)

aai.settings.api_key = os.environ["aai_key"]
transcriber = aai.Transcriber()


# Authenticate to Firesotre 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,
        button={
            "header": f"{header}",
            "body": f"{message}",
            "footer": f"{footer}",
            "action": {
                "button": f"{btn_name}",
                "sections": [
                    {
                        "title": "iBank",
                        "rows": [
                            {"id": "confirm", "title": "Record Transaction", "description": ""},
                            {
                                "id": "cancel",
                                "title": "Cancel Transaction",
                                "description": "",
                            },
                        ],
                    }
                ],
            },
        },
    )

def messenger_reply_button(recipient_phone, message):
    messenger.send_reply_button(
        recipient_id=f"{recipient_phone}",
        button={
            "type": "button",
            "body": {
                "text": f"{message}"
            },
            "action": {
                "buttons": [
                    {
                        "type": "reply",
                        "reply": {
                            "id": "confirm",
                            "title": "Record Transaction"
                        }
                    },
                    {
                        "type": "reply",
                        "reply": {
                            "id": "cancel",
                            "title": "Cancel"
                        }
                    }
                ]
            }
      },
    )

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":
    if request.args.get("hub.verify_token") == VERIFY_TOKEN:
        logging.info("Verified webhook")
        response = make_response(request.args.get("hub.challenge"), 200)
        response.mimetype = "text/plain"
        return response
    logging.error("Webhook Verification failed")
    return "Invalid verification token"

  # 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)
        
      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:
          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)

        
      elif message_type == "audio":
        audio = messenger.get_audio(data)
        audio_id, mime_type = audio["id"], audio["mime_type"]
        audio_url = messenger.query_media_url(audio_id)
        audio_filename = messenger.download_media(audio_url, mime_type)
        transcript =transcriber.transcribe(audio_filename)
        print(audio_filename)
        print(transcript.text)
        res = transcript.text
        messenger.send_message(message=res,recipient_id=mobile)
        logging.info(f"\nAudio: {audio}\n")
        response = str(generateResponse(res))
        if isinstance(response, str):
          messenger.send_message(message=f"{response}", recipient_id=mobile)
        # elif isinstance(response, str) and os.path.isfile(response):
        #   messenger.send_image(image_path=response, recipient_id=mobile)

      else:
        messenger.send_message(message="Please send me text or audio messages",recipient_id=mobile)

  return "ok"

if __name__ == '__main__':
  app.run(debug=True,host="0.0.0.0", port=7860)