Update app.py

app.py

@@ -9,7 +9,7 @@ from bs4 import BeautifulSoup
 from sqlalchemy import select
 
 from fastapi import FastAPI, Request, HTTPException, BackgroundTasks, UploadFile, File, Form
-from fastapi.responses import JSONResponse, StreamingResponse
+from fastapi.responses import JSONResponse, StreamingResponse, RedirectResponse
 
 import openai
 
@@ -28,6 +28,11 @@ DATABASE_URL = os.getenv("DATABASE_URL", "postgresql+asyncpg://postgres.lgbnxply
 NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "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")
 openai.api_key = os.getenv("OPENAI_API_KEY", "your_openai_api_key")
 
+# WhatsApp Business API credentials (Cloud API)
+WHATSAPP_PHONE_NUMBER_ID = os.getenv("WHATSAPP_PHONE_NUMBER_ID", "505165592686490")
+WHATSAPP_ACCESS_TOKEN = os.getenv("WHATSAPP_ACCESS_TOKEN", "EAANi3PCgUmQBO9HKRZA8afKesVL7trAlZAqsFlrebFZBEilHWKKBp4H8Xw2eb2GdRTUZCeOA9lsNmm6zFO19kFq7uJdTTZCq9HrJJoGgNvzc0iehNuLaVdHd8AhZConIoAwLY7MlaA1nc0OKJ0Lw5LhFvJN3ZCuJLP3NskPn42v6HPkFnbFdXeqfJx9iosk4sOQ")
+MANAGEMENT_WHATSAPP_NUMBER = os.getenv("MANAGEMENT_WHATSAPP_NUMBER", "2348035105411")  # e.g. "15551234567" (international format without prefix)
+
 # --- Database Setup ---
 Base = declarative_base()
 
@@ -78,7 +83,6 @@ async def init_db():
         await conn.run_sync(Base.metadata.create_all)
 
 # --- Global In-Memory Stores ---
-# Instead of a plain dict for conversation context, we use a dedicated class below.
 user_state = {}       # e.g., { user_id: ConversationState }
 conversation_context = {}  # { user_id: [ { "timestamp": ..., "role": "user"/"bot", "message": ... }, ... ] }
 proactive_timer = {}
@@ -166,6 +170,26 @@ def create_paystack_payment_link(email: str, amount: int, reference: str) -> dic
     except Exception as e:
         return {"status": False, "message": str(e)}
 
+# --- WhatsApp Business API Helper ---
+def send_whatsapp_message(recipient: str, message_body: str) -> dict:
+    """
+    Sends a WhatsApp text message using the WhatsApp Cloud API.
+    `recipient` should be in international format, e.g., "15551234567".
+    """
+    url = f"https://graph.facebook.com/v15.0/{WHATSAPP_PHONE_NUMBER_ID}/messages"
+    headers = {
+        "Authorization": f"Bearer {WHATSAPP_ACCESS_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "messaging_product": "whatsapp",
+        "to": recipient,
+        "type": "text",
+        "text": {"body": message_body}
+    }
+    response = requests.post(url, headers=headers, json=payload)
+    return response.json()
+
 # --- NVIDIA LLM Streaming Functions ---
 def stream_text_completion(prompt: str):
     from openai import OpenAI
@@ -316,6 +340,28 @@ def process_order_flow(user_id: str, message: str) -> str:
                         await session.commit()
                 asyncio.create_task(save_order())
 
+                # Notify management of the new order via WhatsApp
+                async def notify_management_order(order_details: dict):
+                    message_body = (
+                        f"New Order Received:\n"
+                        f"Order ID: {order_details['order_id']}\n"
+                        f"Dish: {order_details['dish']}\n"
+                        f"Quantity: {order_details['quantity']}\n"
+                        f"Total Price: {order_details['price']}\n"
+                        f"Delivery Address: {order_details.get('address', 'Not Provided')}\n"
+                        f"Status: Pending Payment"
+                    )
+                    await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER, message_body)
+                order_details = {
+                    "order_id": order_id,
+                    "dish": state.data["dish"],
+                    "quantity": state.data["quantity"],
+                    "price": state.data["price"],
+                    "address": state.data.get("address", ""),
+                    "status": "Pending Payment"
+                }
+                asyncio.create_task(notify_management_order(order_details))
+
                 email = "customer@example.com"  # Placeholder; retrieve from profile if available
                 payment_data = create_paystack_payment_link(email, total_price * 100, order_id)
                 dish_name = state.data.get("dish", "")
@@ -580,22 +626,54 @@ async def process_voice(file: UploadFile = File(...)):
     simulated_text = "Simulated speech-to-text conversion result."
     return {"transcription": simulated_text}
 
-@app.post("/payment_callback")
+# --- Payment Callback Endpoint with Payment Tracking and Redirection ---
+@app.api_route("/payment_callback", methods=["GET", "POST"])
 async def payment_callback(request: Request):
-    data = await request.json()
-    order_id = data.get("reference")
-    new_status = data.get("status", "Paid")
-    async with async_session() as session:
-        result = await session.execute(
-            Order.__table__.select().where(Order.order_id == order_id)
+    # GET: User redirection after payment
+    if request.method == "GET":
+        params = request.query_params
+        order_id = params.get("reference")
+        status = params.get("status", "Paid")
+        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)
+            )
+            order = result.scalar_one_or_none()
+            if order:
+                order.status = status
+                await session.commit()
+            else:
+                raise HTTPException(status_code=404, detail="Order not found.")
+        # Notify management via WhatsApp about the payment update
+        await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER,
+            f"Payment Update:\nOrder ID: {order_id} is now {status}."
         )
-        order = result.scalar_one_or_none()
-        if order:
-            order.status = new_status
-            await session.commit()
-            return JSONResponse(content={"message": "Order updated successfully."})
-        else:
-            raise HTTPException(status_code=404, detail="Order not found.")
+        # Redirect user back to the chat interface (adjust URL as needed)
+        redirect_url = f"https://yourdomain.com/chat?order_id={order_id}&status=success"
+        return RedirectResponse(url=redirect_url)
+    # POST: Server-to-server callback from Paystack
+    else:
+        data = await request.json()
+        order_id = data.get("reference")
+        new_status = data.get("status", "Paid")
+        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)
+            )
+            order = result.scalar_one_or_none()
+            if order:
+                order.status = new_status
+                await session.commit()
+                await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER,
+                    f"Payment Update:\nOrder ID: {order_id} is now {new_status}."
+                )
+                return JSONResponse(content={"message": "Order updated successfully."})
+            else:
+                raise HTTPException(status_code=404, detail="Order not found.")
 
 if __name__ == "__main__":
     import uvicorn