Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -5,14 +5,14 @@ import requests
 from supabase_models import Supabase_Client  
 
 
-def extract_structure_store_message(user_id:str,message_id:str , attachment_id:str):
+def extract_structure_store_message(user_id:str,message_id:str , attachment_id:str,attachment_extension:str):
 
     if attachment_id and message_id:
         project_id = os.getenv('PROJECT_ID')
         processor_id = os.getenv('PROCESSOR_ID')
         document_entities = {}
-        extension = attachment.filename.split(".")[-1]
-        file_name = f"{message_id}_{attachment_id}.{extension}"
+
+        file_name = f"{message_id}_{attachment_id}.{attachment_extension}"
         print(f"file_name: {file_name}")
         supabase = Supabase_Client().instance
         try: