app.py

import logging

from agents import VERIFY_TOKEN, get_whatsapp_agent
from fastapi import FastAPI, HTTPException, Request
from fastapi.responses import PlainTextResponse

# Configure logging
logger = logging.getLogger(__name__)

# Initialize agent
agent = get_whatsapp_agent()

# Create FastAPI app
app = FastAPI()


@app.get("/webhook")
async def verify_webhook(request: Request):
    """Handle WhatsApp webhook verification"""
    mode = request.query_params.get("hub.mode")
    token = request.query_params.get("hub.verify_token")
    challenge = request.query_params.get("hub.challenge")

    if mode == "subscribe" and token == VERIFY_TOKEN:
        if not challenge:
            raise HTTPException(status_code=400, detail="No challenge received")
        return PlainTextResponse(content=challenge)

    raise HTTPException(status_code=403, detail="Invalid verify token or mode")


@app.post("/webhook")
async def handle_message(request: Request):
    """Handle incoming WhatsApp messages"""
    try:
        body = await request.json()

        # Validate webhook data
        if body.get("object") != "whatsapp_business_account":
            logger.warning(
                f"Received non-WhatsApp webhook object: {body.get('object')}"
            )
            return {"status": "ignored"}

        # Process messages
        for entry in body.get("entry", []):
            for change in entry.get("changes", []):
                messages = change.get("value", {}).get("messages", [])

                if not messages:
                    continue

                message = messages[0]
                if message.get("type") != "text":
                    continue

                # Extract message details
                phone_number = message["from"]
                message_text = message["text"]["body"]

                logger.info(f"Processing message from {phone_number}: {message_text}")

                # Generate and send response
                response = agent.run(message_text)
                agent.tools[0].send_text_message_sync(
                    recipient=phone_number, text=response.content
                )
                logger.info(f"Response sent to {phone_number}")

        return {"status": "ok"}

    except Exception as e:
        logger.error(f"Error processing webhook: {str(e)}")
        raise HTTPException(status_code=500, detail=str(e))


if __name__ == "__main__":
    import uvicorn

    logger.info("Starting WhatsApp Bot Server")
    uvicorn.run(app, host="0.0.0.0", port=8000)