Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -5,8 +5,21 @@ import requests
 from supabase_models import Supabase_Client  
 from authenticate import get_access_token_v1
 
-def extract_structure_store_message(filename:str):
 
+
+def generate_custom_random_string():
+    # Length of the provided string (16 characters)
+    length = 16  
+    # Define the character set (hexadecimal characters: 0-9 and a-f)
+    characters = string.hexdigits.lower()
+    # Generate the random string
+    random_string = ''.join(random.choice(characters) for _ in range(length))
+    # Add "RAN" in front of the string
+    return "RAN" + random_string
+
+def extract_structure_store_message(filename:str,user_id:str):
+    message_id = generate_custom_random_string()
+    attachment_id = 'UPLOADED_MANUALLY'
     if attachment_id and message_id:
         project_id = os.getenv('PROJECT_ID')
         processor_id = os.getenv('PROCESSOR_ID')
@@ -44,19 +57,19 @@ def extract_structure_store_message(filename:str):
             )
             response_json = response.json()
             allowed_entities = [
-                        "due_date",
-                        "invoice_date",
-                        "total_amount",
-                        "total_tax_amount",
-                        "receiver_name",
-                        "invoice_id",
+                        "credit_card_last_four_digits",
                         "currency",
-                        "receiver_address",
-                        "invoice_type",
+                        "end_date",
+                        "net_amount",
+                        "payment_type",
+                        "purchase_time",
+                        "receipt_date",
+                        "start_date",
+                        "supplier_address",
+                        "supplier_city",
                         "supplier_name",
-                        "payment_terms",
-                        "line_item",
-                        "line_item/description",
+                        "tip_amount",
+                        "total_amount",
                         "line_item/quantity",
                         "line_item/amount",
                         "line_item/unit_price"