Update app.py

app.py

@@ -18,6 +18,8 @@ import logging
 import hmac
 import hashlib
 import json
+import uuid
+from sqlalchemy.exc import SQLAlchemyError
 from pydantic import BaseModel
 
 # Configure logging
@@ -277,6 +279,26 @@ async def enhanced_chatbot_handler(request: Request, bg: BackgroundTasks):
     bg.add_task(update_user_last_interaction, user_id)
     return JSONResponse(response)
 
+def create_paystack_payment_link(email: str, amount: int, reference: str) -> dict:
+    url = "https://api.paystack.co/transaction/initialize"
+    headers = {
+        "Authorization": f"Bearer {PAYSTACK_SECRET_KEY}",
+        "Content-Type": "application/json",
+    }
+    data = {
+        "email": email,
+        "amount": amount,  # Amount in kobo (i.e. 10000 = ₦100)
+        "reference": reference,
+        "callback_url": os.getenv("PAYMENT_REDIRECT_URL", "https://default-redirect.com")
+    }
+    try:
+        response = requests.post(url, json=data, headers=headers, timeout=10)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            return {"status": False, "message": "Failed to initialize payment."}
+    except Exception as e:
+        return {"status": False, "message": str(e)}
 
 @app.post("/ux/preferences")
 async def update_ux_preferences(request: Request):
@@ -536,11 +558,37 @@ async def create_delivery_order(order_req: DeliveryOrderRequest):
             session.add(new_order)
             await session.commit()
             await session.refresh(new_order)
-            return {"delivery_id": new_order.order_id}
+            
+            # Calculate the total amount (update this logic as needed)
+            # For example, you might calculate the amount based on package details or add a fixed delivery fee
+            total_amount = 10000  # Amount in kobo (₦100.00); replace with actual logic
+
+            # Retrieve user's email - use a default if not provided
+            email = getattr(order_req, "email", "[email protected]")
+            
+            # Initialize the payment with Paystack
+            payment_data = create_paystack_payment_link(email, total_amount, order_id)
+            
+            # Optional: Notify management of the new order (using a WhatsApp API call or similar)
+            if payment_data.get("status"):
+                payment_link = payment_data["data"]["authorization_url"]
+                # You could also schedule a background task to notify management
+                # e.g., asyncio.create_task(send_whatsapp_message(MANAGEMENT_WHATSAPP_NUMBER, f"New order {order_id} created."))
+                return {
+                    "delivery_id": new_order.order_id,
+                    "payment_link": payment_link,
+                    "message": "Order created successfully. Please complete payment using the link provided."
+                }
+            else:
+                return {
+                    "delivery_id": new_order.order_id,
+                    "message": "Order created, but we could not initialize payment. Please try again later."
+                }
         except SQLAlchemyError as e:
             await session.rollback()
             raise HTTPException(status_code=500, detail="Error creating delivery order.") from e
 
+
 # --------------------
 # ERROR HANDLING
 # --------------------