main.py

"""ArrowCode entry point.

Launches the Textual app. If no config exists yet (or no API keys are
saved), the Setup screen is shown first; otherwise we boot straight into
the Dashboard.
"""
from __future__ import annotations

import asyncio
import os
import sys
from pathlib import Path

# Make the repo root importable no matter where the script is invoked from.
REPO_ROOT = Path(__file__).resolve().parent
if str(REPO_ROOT) not in sys.path:
    sys.path.insert(0, str(REPO_ROOT))

from textual.app import App, ComposeResult

from core.config import get_config
from ui.setup_screen import SetupScreen
from ui.dashboard import DashboardScreen


class ArrowCodeApp(App):
    """ArrowCode Textual application."""

    TITLE = "ArrowCode"
    SUB_TITLE = "24/7 autonomous multi-model engineering orchestrator"

    CSS = """
    Screen {
        background: #1a1b26;
        color: #c0caf5;
    }
    """

    def __init__(self) -> None:
        super().__init__()
        self.cfg = get_config()

    def on_mount(self) -> None:
        if not self.cfg.is_setup or not self.cfg.available_providers():
            # First launch: push Setup, then Dashboard after Save.
            self.push_screen(SetupScreen(), self._after_setup)
        else:
            self.push_screen(DashboardScreen())

    def _after_setup(self, _result=None) -> None:
        # SetupScreen pops itself; we just push the dashboard.
        self.push_screen(DashboardScreen())


def main() -> None:
    """Create the Textual app and run it."""
    # Some asyncio plumbing on certain Python builds.
    try:
        asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
    except Exception:
        pass

    app = ArrowCodeApp()
    app.run()


if __name__ == "__main__":
    main()