"""
DAGI Router Client
Sends requests to DAGI Router from Bot Gateway
"""
import logging
import os
import httpx
from typing import Dict, Any

logger = logging.getLogger(__name__)

# Router configuration from environment
ROUTER_BASE_URL = os.getenv("ROUTER_URL", "http://127.0.0.1:9102")
# Increased timeout for image generation + LLM calls (FLUX takes ~17s, LLM can take 30-60s)
ROUTER_TIMEOUT = float(os.getenv("ROUTER_TIMEOUT", "180.0"))


async def send_to_router(body: Dict[str, Any]) -> Dict[str, Any]:
    """
    Send request to DAGI Router.
    
    Args:
        body: Request payload with mode, message, agent, metadata, etc.
    
    Returns:
        Router response as dict
    
    Raises:
        httpx.HTTPError: if router request fails
    """
    agent_id = body.get("agent", "devtools")
    message = body.get("message", "")
    metadata = body.get("metadata", {})
    context = body.get("context", {})
    
    # Get system_prompt - check both body level and context level
    system_prompt = body.get("system_prompt") or context.get("system_prompt")
    
    if system_prompt:
        logger.info(f"Using system prompt ({len(system_prompt)} chars) for agent {agent_id}")
    
    # Build infer request
    infer_url = f"{ROUTER_BASE_URL}/v1/agents/{agent_id}/infer"
    
    # Ensure agent_id is in metadata for memory storage
    metadata["agent_id"] = agent_id
    
    infer_body = {
        "prompt": message,
        "system_prompt": system_prompt,
        "metadata": metadata
    }
    
    # Pass images if present in context
    images = context.get("images", [])
    if images:
        infer_body["images"] = images
        logger.info(f"Including {len(images)} image(s) in request")
    
    # Pass provider override if specified
    if metadata.get("provider"):
        infer_body["provider_override"] = metadata["provider"]
    
    logger.info(f"Sending to Router ({infer_url}): agent={agent_id}, provider={metadata.get('provider', 'default')}, has_images={bool(images)}")
    
    try:
        async with httpx.AsyncClient(timeout=ROUTER_TIMEOUT) as client:
            response = await client.post(infer_url, json=infer_body)
            response.raise_for_status()
            
            result = response.json()
            
            # Convert Router response to Gateway expected format
            return {
                "ok": True,
                "data": {
                    "text": result.get("response", result.get("text", "")),
                    "image_base64": result.get("image_base64")  # Generated image
                },
                "response": result.get("response", result.get("text", "")),
                "model": result.get("model"),
                "backend": result.get("backend"),
                "image_base64": result.get("image_base64")  # For easy access
            }
    
    except httpx.HTTPError as e:
        logger.error(f"Router request failed: {e}")
        raise