Update extract_insights.py

extract_insights.py

@@ -9,6 +9,7 @@ import logging
 from datetime import datetime
 import os
 import tiktoken
+import re
 
 
 app = FastAPI()
@@ -158,6 +159,22 @@ def receipt_radar_prompt(raw_text:str)->str:
     """
     return system_prompt    
 
+
+
+def parse_number(num_str):
+    # Remove commas and strip any unnecessary whitespace
+    cleaned_str = re.sub(r'[^\d.]', '', num_str.strip())
+    try:
+        # Convert to float
+        return float(cleaned_str)
+    except ValueError:
+        # Handle the case where the conversion fails
+        print(f"Error: {num_str} is not a valid number.")
+        return None
+
+
+
+
 def adjust_prompt_tokens_v1(prompt: str) -> str:
     max_tokens = 127500
     encoding = tiktoken.encoding_for_model('gpt-4o-mini')
@@ -210,6 +227,7 @@ async def process_synchronous_job(user_id,email,message_id,raw_text):
         insert_data['email'] = email
         insert_data['user_id'] = user_id
         insert_data['message_id'] = message_id
+        insert_data['total_cost'] = parse_number(insert_data['total_cost'])
         print("Printing user_id")
         print(user_id)
 
@@ -228,5 +246,6 @@ async def process_synchronous_job(user_id,email,message_id,raw_text):
             .execute()
         )
 
+        print(update_status_response)
     except Exception as e:
         print(f"Error occurred during processing: {e}")