Update app.py

app.py

@@ -693,68 +693,68 @@ async def process_order_flow(user_id: str, message: str) -> str:
 
     # 7) Step 7: Order Confirmation and Payment Link Generation
     if state and state.flow == "order" and state.step == 7:
-    if message.lower() in ["yes", "y"]:
-        order_id = f"ORD-{int(time.time())}"
-        state.data["order_id"] = order_id
-        # Calculate total price differently for multi-dish orders:
-        if "orders" in state.data:
-            total_price = 0
-            for dish, qty in state.data["orders"].items():
-                total_price += get_dish_price(dish) * qty
-            total_price += state.data.get("shipping_cost", 0)
-            dish_summary = ", ".join(state.data["orders"].keys())
-            quantity_total = sum(state.data["orders"].values())
-        else:
-            dish_summary = state.data.get("dish", "")
-            quantity_total = state.data.get("quantity", 1)
-            total_price = (quantity_total * get_dish_price(dish_summary)) + state.data.get("shipping_cost", 0)
-        state.data["price"] = str(total_price)
-
-        async def save_order():
-            async with async_session() as session:
-                order = Order(
-                    order_id=order_id,
-                    user_id=user_id,
-                    dish=dish_summary,
-                    quantity=str(quantity_total),
-                    price=str(total_price),
-                    status="Pending Payment",
-                    delivery_address=state.data.get("address", "")
-                )
-                session.add(order)
-                await session.commit()
-        asyncio.create_task(save_order())
-        asyncio.create_task(log_order_tracking(order_id, "Order Placed", "Order placed and awaiting payment."))
-
-        # Email notification and payment link code here (omitted for brevity)
-        email_for_paystack = "customer@example.com"
-        payment_data = create_paystack_payment_link(email_for_paystack, total_price * 100, order_id)
-        state.reset()
-        if user_id in user_state:
-            del user_state[user_id]
-        if payment_data.get("status"):
-            payment_link = payment_data["data"]["authorization_url"]
-            return (f"Thank you for your order of {quantity_total} serving(s) of {dish_summary}! "
-                    f"Your Order ID is {order_id}.\n\n"
-                    "Please complete your payment using one of the following options:\n"
-                    f"1. Pay online via our Paystack link: {payment_link}\n"
-                    "2. Alternatively, you can make a bank transfer to the following account:\n"
-                    "   Account Number: 1433042821\n"
-                    "   Bank: Access Bank\n"
-                    "   Account Name: Angelo Food Court 2\n\n"
-                    "If you choose the bank transfer option, please send a screenshot of your payment confirmation to this chatbot.\n\n"
-                    "You can track your order status using your Order ID.\n"
-                    "Is there anything else you'd like to order?")
-        else:
-            return (f"Your order has been placed with Order ID {order_id}, "
-                    "but we could not initialize online payment. Please try again later, or "
-                    "you may opt to pay via bank transfer to Account Number 1433042821, Access Bank, Angelo Food Court 2 "
-                    "and send your payment screenshot to this chatbot.")
-    elif message.lower() in ["no", "n"]:
-        state.reset()
-        if user_id in user_state:
-            del user_state[user_id]
-        return "Order canceled. Let me know if you'd like to try again."
+        if message.lower() in ["yes", "y"]:
+            order_id = f"ORD-{int(time.time())}"
+            state.data["order_id"] = order_id
+            # Calculate total price differently for multi-dish orders:
+            if "orders" in state.data:
+                total_price = 0
+                for dish, qty in state.data["orders"].items():
+                    total_price += get_dish_price(dish) * qty
+                total_price += state.data.get("shipping_cost", 0)
+                dish_summary = ", ".join(state.data["orders"].keys())
+                quantity_total = sum(state.data["orders"].values())
+            else:
+                dish_summary = state.data.get("dish", "")
+                quantity_total = state.data.get("quantity", 1)
+                total_price = (quantity_total * get_dish_price(dish_summary)) + state.data.get("shipping_cost", 0)
+            state.data["price"] = str(total_price)
+    
+            async def save_order():
+                async with async_session() as session:
+                    order = Order(
+                        order_id=order_id,
+                        user_id=user_id,
+                        dish=dish_summary,
+                        quantity=str(quantity_total),
+                        price=str(total_price),
+                        status="Pending Payment",
+                        delivery_address=state.data.get("address", "")
+                    )
+                    session.add(order)
+                    await session.commit()
+            asyncio.create_task(save_order())
+            asyncio.create_task(log_order_tracking(order_id, "Order Placed", "Order placed and awaiting payment."))
+    
+            # Email notification and payment link code here (omitted for brevity)
+            email_for_paystack = "customer@example.com"
+            payment_data = create_paystack_payment_link(email_for_paystack, total_price * 100, order_id)
+            state.reset()
+            if user_id in user_state:
+                del user_state[user_id]
+            if payment_data.get("status"):
+                payment_link = payment_data["data"]["authorization_url"]
+                return (f"Thank you for your order of {quantity_total} serving(s) of {dish_summary}! "
+                        f"Your Order ID is {order_id}.\n\n"
+                        "Please complete your payment using one of the following options:\n"
+                        f"1. Pay online via our Paystack link: {payment_link}\n"
+                        "2. Alternatively, you can make a bank transfer to the following account:\n"
+                        "   Account Number: 1433042821\n"
+                        "   Bank: Access Bank\n"
+                        "   Account Name: Angelo Food Court 2\n\n"
+                        "If you choose the bank transfer option, please send a screenshot of your payment confirmation to this chatbot.\n\n"
+                        "You can track your order status using your Order ID.\n"
+                        "Is there anything else you'd like to order?")
+            else:
+                return (f"Your order has been placed with Order ID {order_id}, "
+                        "but we could not initialize online payment. Please try again later, or "
+                        "you may opt to pay via bank transfer to Account Number 1433042821, Access Bank, Angelo Food Court 2 "
+                        "and send your payment screenshot to this chatbot.")
+        elif message.lower() in ["no", "n"]:
+            state.reset()
+            if user_id in user_state:
+                del user_state[user_id]
+            return "Order canceled. Let me know if you'd like to try again."
 
     # Final fallback
     return ""