Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -38,18 +38,26 @@ async def extract_structure_store_message(filename:str,user_id:str):
                   filename
                 )
             base64_data = urlsafe_b64encode(response).decode('utf-8')
-
-            payload = {
-                    "skipHumanReview": True,
-                    "rawDocument": {
-                        "mimeType": f"application/{attachment_extension}",
-                        "content": base64_data
+            if attachment_extension == 'pdf':
+                payload = {
+                        "skipHumanReview": True,
+                        "rawDocument": {
+                            "mimeType": f"application/{attachment_extension}",
+                            "content": base64_data
+                        }
+                    }
+            elif attachment_extension == 'jpg' or attachment_extension == 'png':
+                payload = {
+                        "skipHumanReview": True,
+                        "rawDocument": {
+                            "mimeType": f"image/{attachment_extension}",
+                            "content": base64_data
+                        }
                     }
-                }
     
             access_token = get_access_token_v1()
             print(access_token)
-    
+            
             headers = {
                 'Authorization': f'Bearer {access_token}',
                 'Content-Type': 'application/json; charset=utf-8'