from pathlib import Path


def run_chat(
    db_path: Path | None = None,
    read_only: bool = False,
    model: str | None = None,
    capabilities: list[str] | None = None,
) -> None:
    """Run the chat TUI.

    Args:
        db_path: Path to the LanceDB database. If None, uses default from config.
        read_only: Whether to open the database in read-only mode.
        model: Model to use for the chat.
        capabilities: Capabilities to enable ("rag", "analysis"). Defaults to ["rag"].
    """
    try:
        from haiku.rag.chat.app import ChatApp
    except ImportError as e:
        raise ImportError(
            "textual is not installed. Please install it with `pip install 'haiku.rag-slim[tui]'` or use the full haiku.rag package."
        ) from e

    from haiku.rag.config import get_config
    from haiku.rag.utils import get_model, parse_model_option

    config = get_config()
    if db_path is None:
        db_path = config.storage.data_dir / "haiku.rag.lancedb"

    if model:
        model_config = parse_model_option(model)
        config.qa.model = model_config
        config.analysis.model = model_config

    enabled = capabilities or ["rag"]
    capability_list = []

    if "rag" in enabled:
        from haiku.rag.capabilities.rag import create_capability

        capability_list.append(
            create_capability(db_path=db_path, config=config, defer_loading=False)
        )

    if "analysis" in enabled:
        from haiku.rag.capabilities.analysis import create_capability

        capability_list.append(
            create_capability(db_path=db_path, config=config, defer_loading=False)
        )

    app = ChatApp(
        db_path,
        capabilities=capability_list,
        read_only=read_only,
        model=model or get_model(config.qa.model, config),
    )
    app.run()