Update app.py

app.py

@@ -171,7 +171,7 @@ def create_paystack_payment_link(email: str, amount: int, reference: str) -> dic
         "email": email,
         "amount": amount,
         "reference": reference,
-        "callback_url": "https://custy-bot.vercel.app/payment_callback"
+        "callback_url": "https://fred808-botpy-808.hf.space/payment_callback"
     }
     try:
         response = requests.post(url, json=data, headers=headers, timeout=10)
@@ -826,47 +826,62 @@ async def payment_callback(request: Request):
     if request.method == "GET":
         params = request.query_params
         order_id = params.get("reference")
-        status = params.get("status", "Paid")
+        status = params.get("status", "failed")  # Default to failed
+        
         if not order_id:
             raise HTTPException(status_code=400, detail="Missing order reference in callback.")
+
         async with async_session() as session:
             result = await session.execute(
-                Order.__table__.select().where(Order.order_id == order_id)
+                select(Order).where(Order.order_id == order_id)
             )
-            order = result.scalar_one_or_none()
+            order = result.scalars().first()
+
             if order:
-                order.status = status
+                order.status = "Paid" if status == "success" else "Failed"
                 await session.commit()
+                
+                if status == "success":
+                    # Notify the user
+                    await asyncio.to_thread(send_whatsapp_message, order.user_id,
+                        f"✅ Payment successful! Your order (ID: {order_id}) is now being processed. 🚀"
+                    )
+                    
+                    # Redirect user to WhatsApp link
+                    return RedirectResponse(url="https://wa.link/am87s2")
+                else:
+                    return JSONResponse(content={"message": "Payment failed. Please try again."})
             else:
                 raise HTTPException(status_code=404, detail="Order not found.")
-        await log_order_tracking(order_id, "Payment Confirmed", f"Payment status updated to {status}.")
-        await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER,
-            f"Payment Update:\nOrder ID: {order_id} is now {status}."
-        )
-        redirect_url = f"https://wa.link/am87s2"
-        return RedirectResponse(url=redirect_url)
-    else:
+
+    else:  # Handle POST webhook (if needed)
         data = await request.json()
         order_id = data.get("reference")
-        new_status = data.get("status", "Paid")
+        new_status = data.get("status", "failed")
+
         if not order_id:
             raise HTTPException(status_code=400, detail="Missing order reference in callback.")
+
         async with async_session() as session:
             result = await session.execute(
-                Order.__table__.select().where(Order.order_id == order_id)
+                select(Order).where(Order.order_id == order_id)
             )
-            order = result.scalar_one_or_none()
+            order = result.scalars().first()
+
             if order:
-                order.status = new_status
+                order.status = "Paid" if new_status == "success" else "Failed"
                 await session.commit()
-                await log_order_tracking(order_id, "Payment Confirmed", f"Payment status updated to {new_status}.")
-                await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER,
-                    f"Payment Update:\nOrder ID: {order_id} is now {new_status}."
+
+                await asyncio.to_thread(send_whatsapp_message, order.user_id,
+                    f"✅ Payment successful! Your order (ID: {order_id}) is now being processed. 🚀"
+                    if new_status == "success" else
+                    f"❌ Payment failed for Order ID: {order_id}. Please retry."
                 )
                 return JSONResponse(content={"message": "Order updated successfully."})
             else:
                 raise HTTPException(status_code=404, detail="Order not found.")
 
+
 @app.get("/track_order/{order_id}")
 async def track_order(order_id: str):