Update extract_insights.py

extract_insights.py

@@ -53,7 +53,6 @@ def prompt_for_categorization(data):
     "Activewear": [],
     "Others": []
     }
-    "item_value": "1,590.00"
     },
     
     {
@@ -66,7 +65,6 @@ def prompt_for_categorization(data):
     "Activewear": [],
     "Others": []
     }
-    "item_value":"1,078.01"
     }
     }
     """
@@ -86,29 +84,30 @@ def prompt_for_categorization(data):
 async def extract_fashion_categories(user_id:str , email:str, message_id:str):
 
     
-    response = supabase.table("document_ai_entities").select("line_item_description , line_item_amount").eq("user_id",user_id).eq("email",email).eq("message_id",message_id).execute()
+    response = supabase.table("document_ai_entities").select("line_item_description").eq("user_id",user_id).eq("email",email).eq("message_id",message_id).execute()
     print("printing response.data")
     print(response.data)
 
     items = response.data[0].get('line_item_description',None)
     values = response.data[0].get('line_item_amount',None)
     print("items",items)
-    print("values",values)
-    combined_json = combine_json_arrays(items,values)
-    print(combined_json)
+    # print("values",values)
+    # combined_json = combine_json_arrays(items,values)
+    # print(combined_json)
     
-    # prompt = prompt_for_categorization(combined_json)
-    # completion = client.chat.completions.create(
-    # model="gpt-4o-mini",
-    # messages=[
-    #     {
-    #         "role": "user",
-    #         "content": prompt
-    #         }
-    #     ],
-    # response_format={ "type": "json_object" }
-    # )
-
+    prompt = prompt_for_categorization(combined_json)
+    completion = client.chat.completions.create(
+    model="gpt-4o-mini",
+    messages=[
+        {
+            "role": "user",
+            "content": prompt
+            }
+        ],
+    response_format={ "type": "json_object" }
+    )
+    print("Printing GPT response")
+    print(completion.choices[0].message)
     # response = (
     #     supabase.table("document_ai_entities")
     #     .update({"categorised_data": completion.choices[0].message})