Update main.py

main.py

@@ -1,35 +1,48 @@
-import google.generativeai as palm
-import pandas as pd
+from flask import Flask,request,make_response
 import os
-import io
-from flask import Flask, request, jsonify
-from flask_cors import CORS, cross_origin
+import logging
+from dotenv import load_dotenv
+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
-import firebase_admin
-from firebase_admin import credentials, firestore, auth
+from langchain_google_genai import GoogleGenerativeAI
+from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
 import requests
-import pandas as pd
-from datetime import datetime
-import os
-from pandasai.llm import GoogleGemini
-from pandasai import SmartDataframe, SmartDatalake
+import base64
+from pandasai.helpers import path
+import uuid
+import pandasai as pai
 
-from pandasai.responses.response_parser import  ResponseParser
-import matplotlib.pyplot as plt
-from wordcloud import WordCloud
-import random
+# load env data
+load_dotenv()
 
-from dotenv import load_dotenv
-import json
 
+'''
+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
+'''
 
-from dotenv import load_dotenv
 
-load_dotenv()
 
+guid = uuid.uuid4()
+new_filename = f"{guid}"
+user_defined_path = os.path.join("/plots/", new_filename)
 
-app = Flask(__name__)
-cors = CORS(app)
 
 class FlaskResponse(ResponseParser):
   def __init__(self, context) -> None:
@@ -55,62 +68,115 @@ class FlaskResponse(ResponseParser):
   def format_other(self, result):
       return str(result['value'])
 
-gemini_api_key = os.environ['Gemini']
 
-@app.route("/", methods=["GET"])
-def home():
-   
-    return "Hello Qx!"
+# messenger object
+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()
 
+g_api_key = os.environ["google_api_key"]
 
-llm = GoogleGemini(api_key=gemini_api_key)
-        
-# Initialize Firebase app
-if not firebase_admin._apps:
-   
-    cred = credentials.Certificate("quant-app-99d09-firebase-adminsdk-6prb1-37f34e1c91.json")
-    firebase_admin.initialize_app(cred)
-
-db = firestore.client()
-
-
-
-
-@app.route("/predict", methods=["POST"])
-@cross_origin()
-def bot():
-
-
-    user_id = request.json.get("user_id")
-    user_question = request.json.get("user_question")
-    load_dotenv()
-    inventory_ref = db.collection("test_users").document(user_id).collection('inventory')
 
 
-    sales_ref = db.collection("test_users").document(user_id).collection('sales')
-
-    inventory_list = []
-    for doc in inventory_ref.stream():
-        a = doc.to_dict()
-        inventory_list.append(a)
-
-    sales_list = []
-    for doc in sales_ref.stream():
-        a = doc.to_dict()
-        sales_list.append(a)
-
-    inventory_df = pd.DataFrame(inventory_list)
-    sales_df = pd.DataFrame(sales_list)
-
-    lake = SmartDatalake([inventory_df, sales_df], config={"llm":llm, "response_parser":FlaskResponse, "enable_cache": False, "save_logs":False})
-    response = lake.chat(user_question)
-    print(user_question)
-
-    resp = str(response)
-    
-    return jsonify(resp)
-
+app = Flask(__name__)
 
-if __name__ == "__main__":
-    app.run(debug=True,host="0.0.0.0", port=7860)
+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):
+  llm = GoogleGemini(api_key=g_api_key)
+  pandas_agent = Agent(df,config={"llm":llm, "save_charts_path": user_defined_path,"save_charts": True,"response_parser":FlaskResponse, "enable_cache": False, "conversational":True})
+  answer = pandas_agent.chat(prompt)
+  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":
+    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)
+
+      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?"
+          messenger.send_message(message=f"{response}",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 == "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))
+        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)