Rename extract_and_store_supabase to extract_and_store_supabase.py

extract_and_store_supabase

@@ -1,19 +0,0 @@
-from supabase_models import Supabase_Client
-
-
-def store_message_data(message: Message):
-    attachments = message.attachments
-    if attachments:
-        for attachment in attachments:
-            extension = attachment.filename.split(".")[-1]
-            file_name = f"{message.id}_{attachment.attachment_id}.{extension}"
-            print(f"file_name: {file_name}")
-            supabase = Supabase_Client().instance
-            base64_data = attachment.data
-            try:
-                decoded_data = base64.urlsafe_b64decode(base64_data)
-                supabase.storage.from_('receipt_radar').upload(file_name, decoded_data)
-            except binascii.Error as e:
-                print(f"Error decoding base64 data: {e}")
-            except Exception as e:
-                print(f"Error uploading file: {e}")

extract_and_store_supabase.py

@@ -0,0 +1,17 @@
+from supabase_models import Supabase_Client
+
+
+def store_message_data(message_id:str , attachment_id:str):
+    if attachment_id and message_id:
+        extension = attachment.filename.split(".")[-1]
+        file_name = f"{message.id}_{attachment.attachment_id}.{extension}"
+        print(f"file_name: {file_name}")
+        supabase = Supabase_Client().instance
+        try:
+          response = supabase.storage.from_("receipt_radar").download(
+              file_name
+            )
+          base64_data = urlsafe_b64encode(response).decode('utf-8')
+          return base64_data
+        except Exception as e:
+            print(f"Error downloading or encoding file: {e}")