Update app.py

app.py

@@ -33,6 +33,16 @@ WHATSAPP_PHONE_NUMBER_ID = os.getenv("WHATSAPP_PHONE_NUMBER_ID", "default_value"
 WHATSAPP_ACCESS_TOKEN = os.getenv("WHATSAPP_ACCESS_TOKEN", "default_value")
 MANAGEMENT_WHATSAPP_NUMBER = os.getenv("MANAGEMENT_WHATSAPP_NUMBER", "default_value")
 
+# Synthetic town prices for shipping costs
+TOWN_SHIPPING_COSTS = {
+    "lasu gate": 1000,  # N1,000 for LASU Gate
+    "ojo": 800,         # N800 for Ojo
+    "ajangbadi": 1200,  # N1,200 for Ajangbadi
+    "iba": 900,         # N900 for Iba
+    "okokomaiko": 1500, # N1,500 for Okokomaiko
+    "default": 1000     # Default shipping cost for unknown areas
+}
+
 # --- Database Setup ---
 Base = declarative_base()
 
@@ -255,16 +265,19 @@ async def log_order_tracking(order_id: str, status: str, message: str = None):
         await session.commit()
 
 # --- Advanced Internal Flow: Order Processing & Payment Integration ---
+def calculate_shipping_cost(address: str) -> int:
+    """
+    Calculate shipping cost based on the user's address.
+    """
+    address_lower = address.lower()
+    for area, cost in TOWN_SHIPPING_COSTS.items():
+        if area in address_lower:
+            return cost
+    return TOWN_SHIPPING_COSTS["default"]  # Default shipping cost for unknown areas
+
 def process_order_flow(user_id: str, message: str) -> str:
     """
-    Implements an FSM-based order flow that:
-      - Step 1: Expects the user to mention a dish name (optionally with quantity)
-      - Step 2: Asks for quantity if not provided
-      - Step 3: Requests the user's phone number if not already provided
-      - Step 4: Requests the delivery address and validates that it includes "OJO LGA, Lagos"
-      - Step 5: Asks if they would like to add any extras (e.g., sides or drinks)
-      - Step 6: Captures extras details if applicable
-      - Step 7: Asks for order confirmation and finalizes the order
+    Implements an FSM-based order flow with shipping cost calculation.
     """
     # Retrieve or initialize conversation state
     state = user_state.get(user_id)
@@ -361,17 +374,19 @@ def process_order_flow(user_id: str, message: str) -> str:
             if re.match(phone_pattern, message.strip()):
                 state.data["phone_number"] = message.strip()
                 state.step = 4
-                return "Thank you. Please provide your delivery address. (Make sure it includes 'OJO LGA, Lagos')"
+                return "Thank you. Please provide your delivery address."
             else:
                 return "Please enter a valid phone number (e.g., +234XXXXXXXXXX or 0XXXXXXXXXX)."
 
-        # --- Step 4: Requesting Delivery Address with Local Govt Check ---
+        # --- Step 4: Requesting Delivery Address ---
         if state.step == 4:
-            if "ojo lga, lagos" not in message.lower():
-                return "We only deliver to OJO LGA, Lagos. Please provide a valid delivery address that includes 'OJO LGA, Lagos'."
             state.data["address"] = message
+            # Calculate shipping cost based on the address
+            shipping_cost = calculate_shipping_cost(message)
+            state.data["shipping_cost"] = shipping_cost
             state.step = 5
-            return f"Thanks. Your delivery address is recorded as: {message}. Would you like to add any extras such as sides or drinks? (yes/no)"
+            return (f"Thanks. Your delivery address is recorded as: {message}.\n"
+                    f"Your delivery cost is N{shipping_cost}. Would you like to add any extras such as sides or drinks? (yes/no)")
 
         # --- Step 5: Asking if They Want Extras ---
         if state.step == 5:
@@ -385,8 +400,14 @@ def process_order_flow(user_id: str, message: str) -> str:
                 quantity = state.data.get("quantity", 1)
                 phone = state.data.get("phone_number", "")
                 address = state.data.get("address", "")
+                shipping_cost = state.data.get("shipping_cost", 0)
+                price_per_serving = 1500  # Fixed price per serving
+                total_price = (quantity * price_per_serving) + shipping_cost
                 summary = (f"Order Summary:\nDish: {dish}\nQuantity: {quantity}\n"
-                           f"Phone: {phone}\nAddress: {address}\nExtras: None\nConfirm order? (yes/no)")
+                           f"Phone: {phone}\nAddress: {address}\n"
+                           f"Shipping Cost: N{shipping_cost}\n"
+                           f"Total Price: N{total_price}\n"
+                           f"Extras: None\nConfirm order? (yes/no)")
                 return summary
             else:
                 return "Please respond with 'yes' or 'no'. Would you like to add any extras to your order? (yes/no)"
@@ -399,9 +420,15 @@ def process_order_flow(user_id: str, message: str) -> str:
             quantity = state.data.get("quantity", 1)
             phone = state.data.get("phone_number", "")
             address = state.data.get("address", "")
+            shipping_cost = state.data.get("shipping_cost", 0)
             extras = state.data.get("extras", "")
+            price_per_serving = 1500  # Fixed price per serving
+            total_price = (quantity * price_per_serving) + shipping_cost
             summary = (f"Order Summary:\nDish: {dish}\nQuantity: {quantity}\n"
-                       f"Phone: {phone}\nAddress: {address}\nExtras: {extras}\nConfirm order? (yes/no)")
+                       f"Phone: {phone}\nAddress: {address}\n"
+                       f"Shipping Cost: N{shipping_cost}\n"
+                       f"Total Price: N{total_price}\n"
+                       f"Extras: {extras}\nConfirm order? (yes/no)")
             return summary
 
         # --- Step 7: Order Confirmation & Finalization ---
@@ -409,12 +436,13 @@ def process_order_flow(user_id: str, message: str) -> str:
             if message.lower() in ["yes", "y"]:
                 order_id = f"ORD-{int(time.time())}"
                 state.data["order_id"] = order_id
-                price_per_serving = 1500  # fixed price per serving for demonstration
+                price_per_serving = 1500  # Fixed price per serving
                 quantity = state.data.get("quantity", 1)
-                total_price = quantity * price_per_serving
+                shipping_cost = state.data.get("shipping_cost", 0)
+                total_price = (quantity * price_per_serving) + shipping_cost
                 state.data["price"] = str(total_price)
 
-                # Save the order asynchronously (including delivery_address, phone, extras)
+                # Save the order asynchronously (including delivery_address, phone, extras, shipping_cost)
                 async def save_order():
                     async with async_session() as session:
                         order = Order(
@@ -424,7 +452,8 @@ def process_order_flow(user_id: str, message: str) -> str:
                             quantity=str(quantity),
                             price=str(total_price),
                             status="Pending Payment",
-                            delivery_address=state.data.get("address", "")
+                            delivery_address=state.data.get("address", ""),
+                            shipping_cost=str(shipping_cost)  # Save shipping cost
                         )
                         session.add(order)
                         await session.commit()