Update main.py

main.py

@@ -10,6 +10,7 @@ from utility import generateResponse, parse_multiple_transactions, create_invent
 from google.cloud import firestore
 import ast
 import base64
+import requests
 
 
 # load env data
@@ -119,7 +120,22 @@ def process_user_msg(message, mobile):
     intent = parsed_trans_data[0]['intent'].lower()
     if intent == 'read':
         response2 = str(read_datalake(mobile, message))
-        messenger.send_message(f"*Raw Response*:\n {response}, \n \n *Parsed Response*:\n {parsed_trans_data}, \n \n *Final Response*:\n {response2}", recipient_id=mobile)
+        # Check if response is a string and represents a valid image path
+          if isinstance(response2, str) and os.path.isfile(os.path.join(response2)):
+              image_path = os.path.join(response2)
+              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
+              response2 = requests.post(url, headers=headers, data={"image": base64_data})
+              url1= response2.json()["data"]["link"]
+              print(url1)
+            
+              messenger.send_image(image=url1, recipient_id=mobile)
+          else:
+              messenger.send_message(f"*Raw Response*:\n {response}, \n \n *Parsed Response*:\n {parsed_trans_data}, \n \n *Final Response*:\n {response2}", recipient_id=mobile)
     else: 
         # Persist transaction data temporarily for Create, Update, or Delete operations
         persist = persist_temporary_transaction(parsed_trans_data, mobile)