Update main.py

main.py

@@ -6,21 +6,14 @@ from heyoo import WhatsApp
 import random
 import shutil
 from tempfile import NamedTemporaryFile
-import assemblyai as aai
-from pandasai.llm import GoogleGemini
-from pandasai import SmartDataframe
-from pandasai import Agent
-from pandasai.responses.response_parser import  ResponseParser
 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
-from pandasai.helpers import path
 import uuid
-import pandasai as pai
-import pymysql
+
 
 # load env data
 load_dotenv()
@@ -28,66 +21,9 @@ load_dotenv()
 
 import re
 
-def replace_country_code(mobile_number):
-    # Define a dictionary of country codes
-    country_codes = {
-        "27": "0",    # South Africa
-        "263": "07"   # Zimbabwe
-    }
-    
-    # Iterate over the country codes and replace if found
-    for code, replacement in country_codes.items():
-        pattern = r'^' + re.escape(code)
-        if re.match(pattern, mobile_number):
-            updated_number = re.sub(pattern, replacement, mobile_number)
-            return updated_number
-    
-    # Return the original number if no replacement is made
-    return mobile_number
-
-
-'''
-def modify_file_path(file_path):
-    guid = uuid.uuid4()
-    print(guid)
-    guid = str(uuid.uuid4())
-    new_filename = f"{guid}temp_chart.png"
-    new_file_path = os.path.join("/home/runner/Flasktestrairo/plots/", new_filename)
-    # Copy the file
-    shutil.copy2(file_path, new_file_path)
-    return new_file_path
-'''
-
-
-
-guid = uuid.uuid4()
-new_filename = f"{guid}"
-user_defined_path = os.path.join("/plots/", new_filename)
-
-
-class FlaskResponse(ResponseParser):
-  def __init__(self, context) -> None:
-      super().__init__(context)
-
-  def format_dataframe(self, result):
-      return result['value'].to_html()
-
-  def format_plot(self, result):
-        # Save the plot using savefig
-      try:
-          
-          img_path = result['value']
-      
-        
-      except ValueError:
-          img_path = str(result['value'])
-          print("value error!", img_path)
-        
-      print("response_class_path:", img_path)
-      return img_path
-
-  def format_other(self, result):
-      return str(result['value'])
+
+
+
 
 
 # messenger object
@@ -101,36 +37,13 @@ transcriber = aai.Transcriber()
 g_api_key = os.environ["google_api_key"]
 
 
-dbport = 3306  # Default port for MySQL
-
-# Connect to the db
-# Create a cursor object
-
-connection = pymysql.connect(
-    host=os.environ["dbhost"],
-    user=os.environ["dbuser"],
-    password=os.environ["dbpassword"],
-    database=os.environ["dbname"],
-    port=dbport
-)
-
-
-
 
-app = Flask(__name__)
-
-VERIFY_TOKEN = "30cca545-3838-48b2-80a7-9e43b1ae8ce4"
-
-#df = pd.read_excel("craig.xlsx")
-#df = pd.read_csv("small_business_data2.csv")
 
 
 llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=g_api_key, temperature=0.2)
 
-def generateResponse(prompt, df):
-  llm = GoogleGemini(api_key=g_api_key)
-  pandas_agent = Agent(df,config={"llm":llm, "save_charts_path": user_defined_path,"save_charts": False,"response_parser":FlaskResponse, "enable_cache": False, "conversational":True})
-  answer = pandas_agent.chat(prompt)
+def generateResponse(prompt):
+  answer = "I only understand images of receipts"
   return answer
 
 img_ID = "344744a88ad1098"
@@ -167,53 +80,16 @@ def hook():
       message_type = messenger.get_message_type(data)
       cursor = connection.cursor()
 
-# Execute a SELECT
-    
-      query = f"SELECT companyname FROM reg WHERE mobile = '{mobile_x}'"
-      try:
-          cursor.execute(query)
-      except pymysql.err.InterfaceError:
-          connection.ping(reconnect=True)
-          cursor.execute(query)
-
-# Fetch all rows
-
-      rows = cursor.fetchall()
-      if not rows:
-          response = "Your mobile number is not registered with Quantilytix"
-          print(response)
-          messenger.send_message(message=f"{response}, phone:{mobile}",recipient_id=mobile)
-          cursor.close()
-          connection.close()
-          
-      else:
-          print(rows[0][0])
-        
-
-          company_name = rows[0][0]
-          query2 = f"SELECT * FROM fin1 WHERE companyname = '{company_name}'"
-          cursor.execute(query2)
-
-          rows2 = cursor.fetchall()
-
-          column_names = [desc[0] for desc in cursor.description]
-
-          data_frame = pd.DataFrame(rows2, columns = column_names)
-
-          print("Dataframe!!!:", data_frame.head())
 
-# Close the cursor and connection
-          cursor.close()
-          connection.close()
 
       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 BuzyHelper. How can I help you today?"
+          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, data_frame))
+          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
@@ -234,24 +110,16 @@ def hook():
               # Handle cases where response is not a valid image path
               messenger.send_message(message=f"{response}", 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
-        logging.info(f"\nAudio: {audio}\n")
-        response = str(generateResponse(res, data_frame))
-        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)
+      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 text or audio messages",recipient_id=mobile)
+        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)