Update app.py

app.py

@@ -22,6 +22,7 @@ import uuid
 from sqlalchemy.exc import SQLAlchemyError
 from pydantic import BaseModel
 import os
+import random
 
 PAYSTACK_SECRET_KEY = os.getenv("PAYSTACK_SECRET_KEY", "your_default_secret_key")
 
@@ -266,43 +267,48 @@ async def on_startup():
 
 logger = logging.getLogger(__name__)
 
+
 @app.post("/delivery")
 async def create_delivery_order(order_req: DeliveryOrderRequest):
     try:
-        logger.info(f"Received order request: {order_req}")
-
-        # Generate a unique order ID
-        order_id = str(uuid.uuid4())
+        # Generate a unique order ID like "DEL-123456789"
+        order_id = f"DEL-{random.randint(100000000, 999999999)}"
 
-        # Extract delivery location from the request
+        # Extract delivery location
         delivery_address = order_req.delivery_address.lower()
-
-        # Determine the shipping cost based on the location
-        shipping_cost = TOWN_SHIPPING_COSTS.get("default", 1000)  # Default cost
+        shipping_cost = TOWN_SHIPPING_COSTS.get("default", 1000)
         for town, cost in TOWN_SHIPPING_COSTS.items():
             if town in delivery_address:
                 shipping_cost = cost
-                break  # Stop once a match is found
+                break
 
-        # Set a fixed item price (Modify this if price varies per package)
-        item_price = 5000  # ₦5000 for the package (adjust as needed)
-
-        # Calculate the total amount (item price + shipping cost)
+        # Set item price (adjustable)
+        item_price = 5000
         total_amount = item_price + shipping_cost
-
-        # Convert to kobo (since Paystack expects amount in kobo)
         total_amount_kobo = total_amount * 100  
 
-        # Retrieve user's email (use a default if not provided)
+        # Get user email or default
         email = getattr(order_req, "email", "customer@example.com")
 
         # Generate Paystack payment link
         payment_data = create_paystack_payment_link(email, total_amount_kobo, order_id)
 
-        logger.info(f"Generated payment data: {payment_data}")
-
         if payment_data.get("status"):
             payment_link = payment_data["data"]["authorization_url"]
+
+            # WhatsApp Notification
+            whatsapp_message = (
+                f"📦 *New Delivery Order!*\n"
+                f"🆔 Order ID: {order_id}\n"
+                f"📍 Pickup: {order_req.pickup_address}\n"
+                f"🏠 Delivery: {order_req.delivery_address}\n"
+                f"☎️ Contact: {order_req.contact_number}\n"
+                f"💰 Amount: ₦{total_amount}\n"
+                f"✅ Payment Link: {payment_link}\n"
+                f"🕒 Please confirm and process."
+            )
+            send_whatsapp_message(MANAGEMENT_WHATSAPP_NUMBER, whatsapp_message)
+
             return {
                 "order_id": order_id,
                 "total_amount": f"₦{total_amount}",
@@ -319,7 +325,6 @@ async def create_delivery_order(order_req: DeliveryOrderRequest):
             }
 
     except Exception as e:
-        logger.error(f"Error creating delivery order: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
 
 @app.post("/chatbot")