Update main.py

main.py

@@ -65,6 +65,9 @@ async def process_batch_data(batch_id: str):
                 user_id = id[1]
                 email = id[2]
                 output = json.loads(resp.get('response').get('body').get('choices')[0].get('message').get('content'))
+                output['message_id'] = message_id
+                output['user_id'] = user_id
+                output['email'] = email
                 supa_data.append(output)
 
             print("Printing the the data")