closecode 0.1.0__py3-none-any.whl

closecode/ui/screens/main.py

@@ -0,0 +1,125 @@
+"""Close Code Main Screen — welcome view transitions to 4:1 chat layout."""
+
+from textual.app import ComposeResult
+from textual.screen import Screen
+from textual.widgets import Static, Input
+from textual.containers import Vertical, Horizontal, Center
+
+from closecode.ui.widgets.header import CloseCodeHeader
+from closecode.ui.widgets.chat import CloseCodeChat
+from closecode.ui.widgets.composer import CloseCodeComposer, ComposerSubmit
+from closecode.ui.widgets.statusbar import CloseCodeStatusBar
+from closecode.ui.widgets.infopanel import CloseCodeInfoPanel
+from closecode.agent import AgentState
+
+
+class MainScreen(Screen):
+    """Main application screen with two modes:
+
+    1. Welcome mode: centered greeting + input
+    2. Chat mode: 4:1 split (chat | info panel)
+
+    Transitions from welcome → chat on first message.
+    """
+
+    BINDINGS = [
+        ("ctrl+l", "clear_chat", "Clear"),
+        ("ctrl+n", "new_session", "New session"),
+        ("ctrl+c", "cancel", "Cancel"),
+    ]
+
+    def __init__(self):
+        super().__init__()
+        self._in_welcome = True
+
+    def compose(self) -> ComposeResult:
+        yield CloseCodeHeader()
+
+        # Welcome view (shown initially, hidden after first message)
+        with Center(id="welcome-view"):
+            with Vertical(id="welcome-container"):
+                yield Static("⬡", id="welcome-logo")
+                yield Static("Yo, How can I help you today?", id="welcome-greeting")
+                yield Static("", id="welcome-model-tag")
+                yield Input(
+                    placeholder="Ask me anything…",
+                    id="welcome-input",
+                )
+
+        # Chat view (hidden initially, shown after first message)
+        with Horizontal(id="chat-view"):
+            with Vertical(id="chat-column"):
+                yield CloseCodeChat()
+                yield CloseCodeComposer()
+            yield CloseCodeInfoPanel()
+
+        yield CloseCodeStatusBar()
+
+    def on_mount(self):
+        """Show welcome, hide chat."""
+        self.query_one("#chat-view").display = False
+        self.query_one("#welcome-input", Input).focus()
+
+    def on_input_submitted(self, event: Input.Submitted):
+        """Handle Enter on the welcome input — transition to chat mode."""
+        if event.input.id == "welcome-input":
+            text = event.value.strip()
+            if text:
+                self._transition_to_chat()
+                # Forward the message to the app as a ComposerSubmit
+                self.app.on_composer_submit(ComposerSubmit(text))
+
+    def _transition_to_chat(self):
+        """Switch from welcome view to chat view."""
+        if not self._in_welcome:
+            return
+        self._in_welcome = False
+        self.query_one("#welcome-view").display = False
+        self.query_one("#chat-view").display = True
+        try:
+            self.query_one(CloseCodeComposer).set_focus()
+        except Exception:
+            pass
+
+    def ensure_chat_mode(self):
+        """Ensure we're in chat mode (called when restoring sessions etc.)."""
+        self._transition_to_chat()
+
+    def update_state(self, state: AgentState, session=None):
+        """Update all widgets from agent state."""
+        try:
+            session_title = session.title if session else ""
+            self.query_one(CloseCodeStatusBar).update_state(state, session_title)
+            self.query_one(CloseCodeHeader).update_state(state)
+            # Update info panel
+            model_tag = ""
+            if hasattr(self.app, '_model_status_tag'):
+                model_tag = self.app._model_status_tag
+            self.query_one(CloseCodeInfoPanel).update_state(state, model_tag)
+        except Exception:
+            pass
+
+        # Update welcome model tag if still in welcome mode
+        if self._in_welcome and state.model:
+            try:
+                name = state.model
+                if "/" in name:
+                    name = name.split("/", 1)[1]
+                if ":free" in name:
+                    name = name.replace(":free", " · free")
+                self.query_one("#welcome-model-tag", Static).update(
+                    f"[#71717a]Model: [#00f5d4]{name}[/][/]"
+                )
+            except Exception:
+                pass
+
+    # ── Actions (delegated to app) ──────────────────────────────
+
+    def action_clear_chat(self):
+        self.app.action_clear_chat()
+
+    def action_new_session(self):
+        self.app.action_new_session()
+
+    def action_cancel(self):
+        self.app.action_cancel_generation()

closecode/ui/screens/onboarding.py

@@ -0,0 +1,101 @@
+"""Close Code Onboarding Screen — first-run setup."""
+
+from textual.app import ComposeResult
+from textual.screen import Screen
+from textual.widgets import Static, Input, Button
+from textual.containers import Vertical
+from textual.message import Message
+
+
+class OnboardingComplete(Message):
+    """Posted when onboarding is done."""
+    def __init__(self, api_url: str, api_key: str):
+        self.api_url = api_url
+        self.api_key = api_key
+        super().__init__()
+
+
+class OnboardingScreen(Screen):
+    """First-run screen: connect to LLMesh."""
+
+    BINDINGS = [("escape", "quit", "Exit")]
+
+    def __init__(self, default_url: str = "http://localhost:8087", default_key: str = ""):
+        super().__init__()
+        self._default_url = default_url
+        self._default_key = default_key
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="onboarding"):
+            with Vertical(id="onboarding-container"):
+                yield Static("⬡ Close Code", id="onboarding-title")
+                yield Static("Powered by LLMesh", id="onboarding-subtitle")
+                yield Static("")
+                yield Static(
+                    "Add your LLMesh API key to get started.\n"
+                    "Get yours from the LLMesh web dashboard.",
+                    classes="onboarding-label",
+                )
+                yield Static("")
+                yield Static("LLMesh Endpoint:", classes="onboarding-label")
+                yield Input(
+                    value=self._default_url,
+                    placeholder="http://localhost:8087",
+                    id="onboarding-url",
+                    classes="onboarding-input",
+                )
+                yield Static("API Key:", classes="onboarding-label")
+                yield Input(
+                    value=self._default_key,
+                    placeholder="Paste your LLMesh API key here",
+                    id="onboarding-key",
+                    password=True,
+                    classes="onboarding-input",
+                )
+                yield Static("")
+                yield Button("Add API Key & Connect", id="connect-btn", variant="primary")
+                yield Static("", id="onboarding-status")
+
+    def on_mount(self):
+        # Focus the key input if we already have a URL
+        if self._default_url and not self._default_key:
+            self.query_one("#onboarding-key", Input).focus()
+        else:
+            self.query_one("#onboarding-url", Input).focus()
+
+    def on_button_pressed(self, event: Button.Pressed):
+        if event.button.id == "connect-btn":
+            self._try_connect()
+
+    def on_key(self, event):
+        if event.key == "enter":
+            self._try_connect()
+
+    def _try_connect(self):
+        """Attempt to connect to LLMesh."""
+        api_url = self.query_one("#onboarding-url", Input).value.strip()
+        api_key = self.query_one("#onboarding-key", Input).value.strip()
+        status = self.query_one("#onboarding-status", Static)
+
+        if not api_url:
+            status.update("[red]Please enter a LLMesh endpoint[/]")
+            return
+
+        if not api_key:
+            status.update("[red]Please enter your API key[/]")
+            return
+
+        status.update("Connecting…")
+
+        # Post the connect event — the app will handle async verification
+        self.post_message(OnboardingComplete(api_url, api_key))
+
+    def show_status(self, text: str):
+        """Update status text from the app."""
+        try:
+            self.query_one("#onboarding-status", Static).update(text)
+        except Exception:
+            pass
+
+    def action_quit(self):
+        self.app.exit()

closecode/ui/theme.tcss

@@ -0,0 +1,374 @@
+/* Close Code — Textual CSS Theme
+ * #0a0a0b bg, #00f5d4 accent, bubble-style chat
+ */
+
+/* ── Global ─────────────────────────────────────────── */
+
+Screen {
+    background: #0a0a0b;
+    color: #fafafa;
+}
+
+/* ── Onboarding Screen ──────────────────────────────── */
+
+#onboarding {
+    align: center middle;
+}
+
+#onboarding-container {
+    width: 56;
+    height: auto;
+    max-height: 32;
+    border: round #27272a;
+    padding: 1 2;
+    background: #18181b;
+}
+
+#onboarding-title {
+    text-align: center;
+    color: #00f5d4;
+    text-style: bold;
+    margin-bottom: 1;
+}
+
+#onboarding-subtitle {
+    text-align: center;
+    color: #71717a;
+    margin-bottom: 1;
+}
+
+#onboarding-status {
+    margin-top: 1;
+    height: auto;
+    color: #a1a1aa;
+}
+
+.onboarding-label {
+    color: #a1a1aa;
+    margin-top: 1;
+}
+
+.onboarding-input {
+    margin-bottom: 1;
+    border: tall #27272a;
+    background: #0a0a0b;
+    color: #fafafa;
+}
+
+.onboarding-input:focus {
+    border: tall #00f5d4;
+}
+
+#connect-btn {
+    width: 100%;
+    margin-top: 1;
+    background: #00f5d4;
+    color: #0a0a0b;
+    text-style: bold;
+    border: none;
+}
+
+#connect-btn:hover {
+    background: #00d4b4;
+}
+
+/* ── Header ─────────────────────────────────────────── */
+
+CloseCodeHeader {
+    dock: top;
+    height: 1;
+    background: #18181b;
+    color: #fafafa;
+}
+
+#header-bar {
+    width: 100%;
+    height: 1;
+    layout: horizontal;
+}
+
+#header-title {
+    width: auto;
+    color: #00f5d4;
+    text-style: bold;
+    padding: 0 1;
+}
+
+#header-model {
+    width: auto;
+    color: #a1a1aa;
+}
+
+#header-spacer {
+    width: 1fr;
+}
+
+#header-connection {
+    width: auto;
+    padding: 0 1;
+    color: #71717a;
+}
+
+/* ── Welcome View ───────────────────────────────────── */
+
+#welcome-view {
+    height: 1fr;
+    width: 100%;
+    align: center middle;
+}
+
+#welcome-container {
+    width: 60;
+    height: auto;
+    align: center middle;
+    padding: 2 4;
+}
+
+#welcome-logo {
+    text-align: center;
+    color: #00f5d4;
+    text-style: bold;
+    width: 100%;
+    margin-bottom: 1;
+}
+
+#welcome-greeting {
+    text-align: center;
+    color: #fafafa;
+    text-style: bold;
+    width: 100%;
+    margin-bottom: 1;
+}
+
+#welcome-model-tag {
+    text-align: center;
+    width: 100%;
+    margin-bottom: 2;
+    color: #71717a;
+}
+
+#welcome-input {
+    width: 100%;
+    border: tall #27272a;
+    background: #18181b;
+    color: #fafafa;
+    padding: 0 1;
+}
+
+#welcome-input:focus {
+    border: tall #00f5d4;
+}
+
+/* ── Chat View (4:1 split) ──────────────────────────── */
+
+#chat-view {
+    height: 1fr;
+    width: 100%;
+    layout: horizontal;
+}
+
+#chat-column {
+    width: 4fr;
+    height: 100%;
+}
+
+/* ── Chat Area ──────────────────────────────────────── */
+
+CloseCodeChat {
+    height: 1fr;
+    padding: 0 1;
+    scrollbar-color: #27272a;
+    scrollbar-color-hover: #00f5d4;
+    scrollbar-color-active: #00f5d4;
+}
+
+/* Message rows */
+.msg-row {
+    width: 100%;
+    height: auto;
+    margin: 1 0 0 0;
+    layout: horizontal;
+}
+
+.msg-spacer {
+    width: 1fr;
+}
+
+/* User bubbles — right aligned */
+.msg-bubble-user {
+    width: auto;
+    max-width: 70%;
+    min-width: 20;
+    background: #1a3a3a;
+    color: #fafafa;
+    padding: 0 2;
+    border: round #00f5d4 40%;
+    margin-right: 1;
+}
+
+/* Assistant bubbles — left aligned */
+.msg-bubble-assistant {
+    width: auto;
+    max-width: 75%;
+    background: #18181b;
+    color: #fafafa;
+    padding: 1 2;
+    border: round #27272a;
+    margin-left: 0;
+}
+
+/* Avatar */
+.msg-avatar {
+    width: 3;
+    height: 1;
+    color: #00f5d4;
+    text-style: bold;
+    margin-top: 1;
+}
+
+/* Status messages */
+.msg-status {
+    color: #71717a;
+    text-align: center;
+    width: 100%;
+    margin: 0 0;
+}
+
+/* ── Info Panel (1 part of 4:1) ─────────────────────── */
+
+CloseCodeInfoPanel {
+    width: 1fr;
+    height: 100%;
+    background: #111113;
+    border-left: solid #27272a;
+    padding: 1;
+}
+
+#info-panel-inner {
+    width: 100%;
+    height: 100%;
+}
+
+.info-section-title {
+    color: #00f5d4;
+    text-style: bold;
+    margin-bottom: 1;
+}
+
+.info-stat {
+    color: #a1a1aa;
+    margin-bottom: 1;
+    padding-left: 1;
+}
+
+.info-divider {
+    color: #27272a;
+    margin: 1 0;
+}
+
+#info-tasks-list {
+    height: 1fr;
+    scrollbar-color: #27272a;
+}
+
+.info-task {
+    padding-left: 1;
+    margin-bottom: 0;
+}
+
+.info-task-done {
+    color: #71717a;
+}
+
+.info-task-pending {
+    color: #a1a1aa;
+}
+
+/* ── Composer ───────────────────────────────────────── */
+
+CloseCodeComposer {
+    dock: bottom;
+    height: auto;
+    min-height: 3;
+    max-height: 10;
+    padding: 0 1;
+    background: #18181b;
+    border-top: solid #27272a;
+}
+
+#composer-container {
+    height: auto;
+    width: 100%;
+}
+
+#composer-input {
+    height: auto;
+    min-height: 1;
+    max-height: 8;
+    width: 1fr;
+    border: tall #27272a;
+    background: #0a0a0b;
+    color: #fafafa;
+    padding: 0 1;
+}
+
+#composer-input:focus {
+    border: tall #00f5d4;
+}
+
+#composer-send {
+    width: 5;
+    min-width: 5;
+    height: 3;
+    margin-left: 1;
+    background: #00f5d4;
+    color: #0a0a0b;
+    border: none;
+}
+
+#composer-send:hover {
+    background: #00d4b4;
+}
+
+/* ── Status Bar ─────────────────────────────────────── */
+
+CloseCodeStatusBar {
+    dock: bottom;
+    height: 1;
+    background: #18181b;
+    border-top: solid #27272a;
+    color: #71717a;
+}
+
+#statusbar-content {
+    width: 100%;
+    height: 1;
+    layout: horizontal;
+    padding: 0 1;
+}
+
+.status-item {
+    width: auto;
+    padding: 0 1;
+    color: #71717a;
+}
+
+.status-sep {
+    color: #27272a;
+}
+
+.status-session {
+    color: #a1a1aa;
+}
+
+.status-model {
+    color: #fafafa;
+}
+
+.status-connected {
+    color: #22c55e;
+}
+
+.status-disconnected {
+    color: #ef4444;
+}

closecode/ui/widgets/__init__.py

@@ -0,0 +1 @@
+"""Close Code UI widgets."""

closecode/ui/widgets/chat.py

@@ -0,0 +1,105 @@
+"""Close Code Chat widget — scrollable conversation with bubble-style messages."""
+
+from textual.app import ComposeResult
+from textual.widget import Widget
+from textual.widgets import Static, Markdown
+from textual.containers import VerticalScroll, Horizontal
+
+
+class MessageBubble(Static):
+    """A single chat message bubble with role-based alignment."""
+    pass
+
+
+class CloseCodeChat(VerticalScroll):
+    """Scrollable conversation area with left/right aligned messages.
+
+    User messages → right side
+    Assistant messages → left side
+    """
+
+    def __init__(self):
+        super().__init__()
+        self._streaming_widget: Static | None = None
+        self._streaming_text: str = ""
+
+    def add_user_message(self, content: str):
+        """Add a user-aligned message (right side)."""
+        spacer = Static("", classes="msg-spacer")
+        bubble = Static(content, classes="msg-bubble msg-bubble-user")
+        row = Horizontal(spacer, bubble, classes="msg-row msg-row-user")
+        self.mount(row)
+        self.scroll_end(animate=False)
+
+    def add_assistant_start(self):
+        """Start a new assistant message (left side, for streaming)."""
+        label = Static("⬡", classes="msg-avatar msg-avatar-assistant")
+        widget = Static("", classes="msg-bubble msg-bubble-assistant")
+        spacer = Static("", classes="msg-spacer")
+        row = Horizontal(label, widget, spacer, classes="msg-row msg-row-assistant")
+        self.mount(row)
+        self._streaming_widget = widget
+        self._streaming_text = ""
+        self.scroll_end(animate=False)
+
+    def add_streaming_token(self, token: str):
+        """Append a token to the current streaming message."""
+        if self._streaming_widget is not None:
+            self._streaming_text += token
+            self._streaming_widget.update(self._streaming_text)
+            self.scroll_end(animate=False)
+
+    def finish_streaming(self, full_content: str = ""):
+        """Finish streaming — replace the Static with rendered Markdown."""
+        if self._streaming_widget is not None:
+            if full_content:
+                parent = self._streaming_widget.parent
+                self._streaming_widget.remove()
+                md = Markdown(full_content, classes="msg-bubble msg-bubble-assistant")
+                if parent:
+                    # Insert before the spacer (last child)
+                    children = list(parent.children)
+                    if children:
+                        parent.mount(md, before=children[-1])
+                    else:
+                        parent.mount(md)
+                else:
+                    self.mount(md)
+            self._streaming_widget = None
+            self._streaming_text = ""
+            self.scroll_end(animate=False)
+
+    def add_assistant_message(self, content: str):
+        """Add a complete assistant message (left side)."""
+        label = Static("⬡", classes="msg-avatar msg-avatar-assistant")
+        bubble = Markdown(content, classes="msg-bubble msg-bubble-assistant")
+        spacer = Static("", classes="msg-spacer")
+        row = Horizontal(label, bubble, spacer, classes="msg-row msg-row-assistant")
+        self.mount(row)
+        self.scroll_end(animate=False)
+
+    def add_status_message(self, message: str, icon: str = "●"):
+        """Add a centered status indicator."""
+        self.mount(Static(f" {icon} {message}", classes="msg-status"))
+        self.scroll_end(animate=False)
+
+    def add_usage_info(self, tokens: int = 0, model: str = "", latency_ms: float = 0):
+        """Add token usage info after a response."""
+        parts = []
+        if tokens:
+            parts.append(f"{tokens:,} tokens")
+        if latency_ms:
+            parts.append(f"{latency_ms:.0f}ms")
+        if model:
+            short = model.split("/", 1)[-1] if "/" in model else model
+            parts.append(short)
+        if parts:
+            self.mount(Static(f" · {' · '.join(parts)}", classes="msg-status"))
+            self.scroll_end(animate=False)
+
+    def clear_chat(self):
+        """Clear all messages."""
+        for child in list(self.children):
+            child.remove()
+        self._streaming_widget = None
+        self._streaming_text = ""