Update main.py

main.py

@@ -192,12 +192,16 @@ async def process_batch_job(dataset: Dict[str, Any]):
     try:
 
         openai_tasks = []
+
+        user_details = {}
         for ds in dataset.get('data'):
-            message_id = ds.get('message_id')
             user_id = ds.get('user_id')
+            
+            message_id = ds.get('message_id')
+            
             receipt_text = ds.get('receipt_text')
             email = ds.get('email')
-
+            
             text = adjust_prompt_tokens_v1(receipt_radar_prompt(receipt_text))
             
             task = {
@@ -242,8 +246,12 @@ async def process_batch_job(dataset: Dict[str, Any]):
             "batch_job_status": False,
             "batch_job_id" : batch_job.id
         }).execute()
-        
-        supabase.table("gmail_receipts_data").insert({"batch_id":batch_job.id}).execute()
+
+        for ds in dataset.get('data'):
+            user_id = ds.get('user_id')
+            message_id = ds.get('message_id')
+            email = ds.get('email')
+            supabase.table("gmail_receipts_data").update({"batch_id":batch_job.id}).eq('user_id',user_id).eq("message_id",message_id).eq("email",email).execute()
         print("Batch sent for processing")        
     except Exception as e:
         logger.error(f"Error processing batch job {batch_job_id}: {str(e)}")