gcp_document_ai_test / extract_and_store_supabase
Omkar008's picture
Create extract_and_store_supabase
f126b71 verified
raw
history blame
818 Bytes
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}")