Update main.py

main.py

@@ -271,14 +271,28 @@ async def process_query(query: str,
     result = ask_openai(messages)
     print(result)
     if result == "done":
-        # Extract the waybill number
-        extract_message = [
-            {"role": "system", "content": "You are a helpful assistant for Curfox delivery system."},
-            {"role": "user", "content": f"""extract the waybill number from the user query example output is only : CA000001
-                                    user query : {query}""" }
-        ]
-        waybill_number = ask_openai(extract_message)
-        print("waybill number",waybill_number)
+        # # Extract the waybill number
+        # extract_message = [
+        #     {"role": "system", "content": "You are a helpful assistant for Curfox delivery system."},
+        #     {"role": "user", "content": f"""extract the waybill number from the user query example output is only number : CA000001
+        #                             user query : {query}""" }
+        # ]
+        # waybill_number = ask_openai(extract_message)
+        # print("waybill number",waybill_number)
+        import re
+                
+        # Regular expression to extract the waybill number (e.g., CA followed by digits)
+        pattern = r'\bCA\d{6}\b'
+        
+        # Search for the pattern in the query
+        match = re.search(pattern, query)
+        
+        # Extract the waybill number if found
+        if match:
+            waybill_number = match.group(0)
+            print(waybill_number)
+        else:
+            print("Waybill number not found.")
 
         # Fetch order ID and order details
         try: