"""
Indic Heritage Studio v2 — Application entry point.

Launches the Gradio web UI exposing all six multimodal pipelines:
  1. Text → Heritage-styled image (SDXL + DreamShaper-XL + per-style LoRA)
  2. Image + style → Stylized image (IP-Adapter XL)
  3. Image → 4-second video (Stable Video Diffusion)
  4. ControlNet (Canny/Depth/OpenPose composition conditioning)
  5. Inpainting (mask + restyle)
  6. Batch processing (multi-GPU parallel)

Usage:
    python app.py
    # then open http://localhost:7860
"""
from __future__ import annotations

import logging
import os
import sys
from pathlib import Path

from dotenv import load_dotenv

load_dotenv()

logging.basicConfig(
    level=os.getenv("LOG_LEVEL", "INFO"),
    format="%(asctime)s | %(levelname)-7s | %(name)s | %(message)s",
    datefmt="%H:%M:%S",
)
log = logging.getLogger("indic-heritage-studio")


def _preflight() -> None:
    """Run sanity checks before starting the UI."""
    root = Path(__file__).resolve().parent
    if str(root) not in sys.path:
        sys.path.insert(0, str(root))

    if not os.getenv("AMD_MODEL_API_KEY"):
        log.warning(
            "AMD_MODEL_API_KEY not set — agent layer will be disabled. "
            "Core generation still works."
        )

    try:
        import torch
        if not torch.cuda.is_available():
            log.warning(
                "torch.cuda.is_available() is False — running on CPU. "
                "Generation will be very slow."
            )
        else:
            n = torch.cuda.device_count()
            log.info("GPU ready: %d device(s) detected — %s",
                     n, torch.cuda.get_device_name(0))
            if n > 1:
                log.info("Multi-GPU mode: %d GPUs available, total %.0f GB VRAM",
                         n, sum(torch.cuda.get_device_properties(i).total_memory
                                for i in range(n)) / 1e9)
    except ImportError:
        log.error("PyTorch not installed. Run: pip install -r requirements.txt")
        sys.exit(1)


def main() -> None:
    _preflight()
    from ui.gradio_app import build_ui

    log.info("Starting Indic Heritage Studio v2 …")
    demo = build_ui()
    demo.launch(
        server_name=os.getenv("GRADIO_SERVER_NAME", "0.0.0.0"),
        server_port=int(os.getenv("GRADIO_SERVER_PORT", "7860")),
        share=False,
        show_error=True,
        inbrowser=False,
        max_threads=8,
    )


if __name__ == "__main__":
    main()