Omkar008 commited on
Commit
6f8da54
·
verified ·
1 Parent(s): 9a0ee18

Update extract_and_store_supabase.py

Browse files
Files changed (1) hide show
  1. extract_and_store_supabase.py +8 -2
extract_and_store_supabase.py CHANGED
@@ -64,7 +64,13 @@ def extract_structure_store_message(user_id:str,message_id:str , attachment_id:s
64
  raw_text = response_json.get('document').get('text' , None)
65
  entities = response_json.get('document').get('entities' , None)
66
  document_entities['user_id'] = user_id
67
- document_entities['raw_text'] = raw_text
 
 
 
 
 
 
68
  print('Printing entities')
69
  print(entities)
70
  if entities is not None:
@@ -76,7 +82,7 @@ def extract_structure_store_message(user_id:str,message_id:str , attachment_id:s
76
  document_entities[ent.get('type')] = {"mention_text":mention_text,"normalizedValue":normalised_values}
77
  document_entities['email'] = email
78
  print(document_entities)
79
- insert_data_response = response = (
80
  supabase.table("document_ai_entities")
81
  .insert(document_entities)
82
  .execute()
 
64
  raw_text = response_json.get('document').get('text' , None)
65
  entities = response_json.get('document').get('entities' , None)
66
  document_entities['user_id'] = user_id
67
+ insert_data_ocr_text_response = (
68
+ supabase.table("gmail_receipts_data")
69
+ .insert({'receipt_text':raw_text})
70
+ .eq('user_id',user_id)
71
+ .eq('email',email)
72
+ .execute()
73
+ )
74
  print('Printing entities')
75
  print(entities)
76
  if entities is not None:
 
82
  document_entities[ent.get('type')] = {"mention_text":mention_text,"normalizedValue":normalised_values}
83
  document_entities['email'] = email
84
  print(document_entities)
85
+ insert_data_response = (
86
  supabase.table("document_ai_entities")
87
  .insert(document_entities)
88
  .execute()