updating the document ai code

extract_and_store_supabase.py

@@ -72,13 +72,30 @@ def extract_structure_store_message(user_id:str,message_id:str , attachment_id:s
             
             print('Printing entities')
             print(entities)
+            # if entities is not None:
+            #     for ent in entities:
+            #         if ent.get('type') is not None:
+            #             if ent.get('type') in allowed_entities:
+            #                 mention_text = ent.get('mentionText')
+            #                 normalised_values = ent.get('normalizedValue') if 'normalizedValue' in ent else None
+            #                 document_entities[ent.get('type')] = {"mention_text":mention_text,"normalizedValue":normalised_values}
             if entities is not None:
                 for ent in entities:
                     if ent.get('type') is not None:
-                        if ent.get('type') in allowed_entities:
+                        entity_type = ent.get('type')
+                        if entity_type in allowed_entities:
                             mention_text = ent.get('mentionText')
-                            normalised_values = ent.get('normalizedValue') if 'normalizedValue' in ent else None
-                            document_entities[ent.get('type')] = {"mention_text":mention_text,"normalizedValue":normalised_values}
+                            normalized_values = ent.get('normalizedValue') if 'normalizedValue' in ent else None
+    
+                            # Initialize a list for the entity type if not already present
+                            if entity_type not in document_entities:
+                                document_entities[entity_type] = []
+    
+                            # Append the entity data to the list
+                            document_entities[entity_type].append({
+                                "mention_text": mention_text,
+                                "normalizedValue": normalized_values
+                            })
             document_entities['email'] = email
             print(document_entities)
             insert_data_response =  (