Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -106,14 +106,21 @@ def extract_structure_store_message(user_id: str, message_id: str, attachment_id
                                     "normalizedValue": normalized_values
                                 })
 
-        document_entities['line_item_description'] = document_entities['line_item/description']
-        document_entities['line_item_quantity'] = document_entities['line_item/quantity']
-        document_entities['line_item_amount'] = document_entities['line_item/amount']
-        document_entities['line_item_unit_price'] = document_entities['line_item/unit_price']
-        document_entities.pop('line_item/description', None)
-        document_entities.pop('line_item/quantity', None)
-        document_entities.pop('line_item/amount', None)
-        document_entities.pop('line_item/unit_price', None)
+        if 'line_item/description' in document_entities:
+            document_entities['line_item_description'] = document_entities['line_item/description']
+            document_entities.pop('line_item/description', None)
+        
+        if 'line_item/quantity' in document_entities:
+            document_entities['line_item_quantity'] = document_entities['line_item/quantity']
+            document_entities.pop('line_item/quantity', None)
+        
+        if 'line_item/amount' in document_entities:
+            document_entities['line_item_amount'] = document_entities['line_item/amount']
+            document_entities.pop('line_item/amount', None)
+        
+        if 'line_item/unit_price' in document_entities:
+            document_entities['line_item_unit_price'] = document_entities['line_item/unit_price']
+            document_entities.pop('line_item/unit_price', None)
         document_entities['email'] = email
         document_entities['message_id'] = message_id
         print("Printing parsed json")