Update main.py

main.py

@@ -1,65 +1,38 @@
-from flask import Flask,request,make_response
-import os
-import logging
-from dotenv import load_dotenv
-from heyoo import WhatsApp
-import random
-import shutil
-from tempfile import NamedTemporaryFile
-from langchain_experimental.agents import create_pandas_dataframe_agent
-import pandas as pd
-from langchain_google_genai import GoogleGenerativeAI
-from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
-import requests
-import base64
-import uuid
-
-
-# load env data
-load_dotenv()
+# messenger object
 
 
-import re
 
 
 
 
 
 
-# messenger object
+import os
+import logging
+from heyoo import WhatsApp
+from dotenv import load_dotenv
+from flask import Flask, request, make_response
+
+# Initialize Flask App
+app = Flask(__name__)
+
+# Load .env file
+load_dotenv()
 messenger = WhatsApp(
   os.environ["whatsapp_token"],
   phone_number_id=os.environ["phone_number_id"] )
-
-aai.settings.api_key = os.environ["aai_key"]
-transcriber = aai.Transcriber()
+VERIFY_TOKEN = "30cca545-3838-48b2-80a7-9e43b1ae8ce4"
 
 g_api_key = os.environ["google_api_key"]
 
+# Logging
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+)
 
 
-
-
-llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=g_api_key, temperature=0.2)
-
-def generateResponse(prompt):
-  answer = "I only understand images of receipts"
-  return answer
-
-img_ID = "344744a88ad1098"
-img_secret = "3c542a40c215327045d7155bddfd8b8bc84aebbf"
-
-url = "https://api.imgur.com/3/image"
-headers = {"Authorization": f"Client-ID {img_ID}"}
-
-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":
+@app.get("/")
+async def verify_token():
     if request.args.get("hub.verify_token") == VERIFY_TOKEN:
         logging.info("Verified webhook")
         response = make_response(request.args.get("hub.challenge"), 200)
@@ -68,62 +41,81 @@ def hook():
     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)
-      mobile_x = replace_country_code(mobile)
-      message_type = messenger.get_message_type(data)
-      cursor = connection.cursor()
+@app.post("/")
+async def hook():
+    # Handle Webhook Subscriptions
+    data = request.get_json()
+    logging.info("Received webhook data: %s", data)
+    changed_field = messenger.changed_field(data)
+    if changed_field == "messages":
+        new_message = messenger.is_message(data)
+        if new_message:
+            mobile = messenger.get_mobile(data)
+            name = messenger.get_name(data)
+            message_type = messenger.get_message_type(data)
+            logging.info(
+                f"New Message; sender:{mobile} name:{name} type:{message_type}"
+            )
+            if message_type == "text":
+                message = messenger.get_message(data)
+                name = messenger.get_name(data)
+                logging.info("Message: %s", message)
+                await messenger.send_message(f"Hi {name}, nice to connect with you", mobile)
+
+            elif message_type == "interactive":
+                message_response = messenger.get_interactive_response(data)
+                interactive_type = message_response.get("type")
+                message_id = message_response[interactive_type]["id"]
+                message_text = message_response[interactive_type]["title"]
+                logging.info(f"Interactive Message; {message_id}: {message_text}")
+
+            elif message_type == "location":
+                message_location = messenger.get_location(data)
+                message_latitude = message_location["latitude"]
+                message_longitude = message_location["longitude"]
+                logging.info("Location: %s, %s", message_latitude, message_longitude)
+
+            elif message_type == "image":
+                image = messenger.get_image(data)
+                image_id, mime_type = image["id"], image["mime_type"]
+                image_url = await messenger.query_media_url(image_id)
+                image_filename = await messenger.download_media(image_url, mime_type)
+                logging.info(f"{mobile} sent image {image_filename}")
+
+            elif message_type == "video":
+                video = messenger.get_video(data)
+                video_id, mime_type = video["id"], video["mime_type"]
+                video_url = await messenger.query_media_url(video_id)
+                video_filename = await messenger.download_media(video_url, mime_type)
+                logging.info(f"{mobile} sent video {video_filename}")
+
+            elif message_type == "audio":
+                audio = messenger.get_audio(data)
+                audio_id, mime_type = audio["id"], audio["mime_type"]
+                audio_url = await messenger.query_media_url(audio_id)
+                audio_filename = await messenger.download_media(audio_url, mime_type)
+                logging.info(f"{mobile} sent audio {audio_filename}")
+
+            elif message_type == "document":
+                file = messenger.get_document(data)
+                file_id, mime_type = file["id"], file["mime_type"]
+                file_url = await messenger.query_media_url(file_id)
+                file_filename = await messenger.download_media(file_url, mime_type)
+                logging.info(f"{mobile} sent file {file_filename}")
+            else:
+                logging.info(f"{mobile} sent {message_type} ")
+                logging.info(data)
+        else:
+            delivery = messenger.get_delivery(data)
+            if delivery:
+                logging.info(f"Message : {delivery}")
+            else:
+                logging.info("No new message")
+    return "OK", 200
 
 
 
-      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 Petty Cash assistant. Upload an image of a receipt for me to record"
-          messenger.send_message(message=f"{response}, phone:{mobile}",recipient_id=mobile)
-        else:
-          response = str(generateResponse(message))
-          print("Response:", response)
-          logging.info(f"\nAnswer: {response}\n")
-          # Check if response is a string and represents a valid image path
-          if isinstance(response, str) and os.path.isfile(os.path.join(response)):
-              image_path = os.path.join(response)
-              print("My image path:", image_path)
-              with open(image_path, "rb") as file:
-                data = file.read()
-                base64_data = base64.b64encode(data)
-
-              # Upload image to Imgur and get URL
-              response = requests.post(url, headers=headers, data={"image": base64_data})
-              url1= response.json()["data"]["link"]
-              print(url1)
-            
-              messenger.send_image(image=url1, recipient_id=mobile)
-          else:
-              # Handle cases where response is not a valid image path
-              messenger.send_message(message=f"{response}", recipient_id=mobile)
-            
-      elif message_type == "image":
-        image = messenger.get_image(data)
-        image_id, mime_type = image["id"], image["mime_type"]
-        image_url = await messenger.query_media_url(image_id)
-        image_filename = await messenger.download_media(image_url, mime_type)
-        print(f"{mobile} sent image {image_filename}")
-        logging.info(f"{mobile} sent image {image_filename}")
-
-      else:
-        messenger.send_message(message="Please send me only images of receipts",recipient_id=mobile)
-
-    #else:messenger.send_message(message="Your mobile number is not registered with Quantilytix",recipient_id=mobile)
-
-  return "ok"
-
-if __name__ == '__main__':
+
+if __name__ == "__main__":
   app.run(debug=True,host="0.0.0.0", port=7860)