app.py

"""SoniCoder v2 — Entry point.

Uses FastAPI directly (not gradio.Server) so our custom index.html
is always served at / without Gradio overriding the root route.

Model is loaded BLOCKING before server starts — no lazy loading.
"""

from __future__ import annotations

import logging
import os

# Ensure workspace exists
from sonicoder.config import WORKSPACE_ROOT, CONFIG_DIR
WORKSPACE_ROOT.mkdir(parents=True, exist_ok=True)
CONFIG_DIR.mkdir(parents=True, exist_ok=True)

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
    datefmt="%H:%M:%S",
)
logger = logging.getLogger(__name__)


def main():
    logger.info("SoniCoder v2 starting...")

    # ── BLOCKING model load — model must be ready BEFORE server starts ──
    from sonicoder.model.loader import load_model
    from sonicoder.config import load_settings
    settings = load_settings()
    model_key = settings.default_model
    logger.info(f"Loading model {model_key} (blocking)...")
    try:
        load_model(model_key)
        logger.info("Model loaded successfully — ready to serve")
    except Exception as e:
        logger.error(f"Failed to load model: {e}")
        logger.info("Server will start anyway — model can be loaded later via API")

    # ── Set up MCP servers from config ─────────────────────────────────
    from sonicoder.mcp import get_mcp_manager
    if settings.mcp_servers:
        mcp = get_mcp_manager()
        mcp.configure(settings.mcp_servers)
        results = mcp.connect_all()
        for name, error in results.items():
            if error:
                logger.warning(f"MCP server '{name}' failed: {error}")
            else:
                logger.info(f"MCP server '{name}' connected")

    # ── Create the FastAPI server ──────────────────────────────────────
    from sonicoder.server import create_app
    app = create_app()

    logger.info("Launching on 0.0.0.0:7860...")
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")


if __name__ == "__main__":
    main()