Update main.py

main.py

@@ -294,14 +294,15 @@ async def process_query(query: str,
             waybill_number = match.group(0)
             print(waybill_number)
         else:
-            print("Waybill number not found.")
+            print("Waybill number not correct.")
+            return "Waybill number not correct."
 
         # Fetch order ID and order details
         try:
             order_id = get_order_id(waybill_number)
             full_data = await get_all_order_data(order_id)
-        except HTTPException as e:
-            return {"error": str(e.detail)}
+        except Exception as e:
+            return {"error": str(e)}
 
         # Generate final response based on collected data
         response_message = [