sa-assistant 0.2.0__tar.gz → 0.2.1__tar.gz

{sa_assistant-0.2.0 → sa_assistant-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sa-assistant
-Version: 0.2.0
+Version: 0.2.1
 Summary: SA Assistant - AWS Solutions Architect Professional Agent with Multi-Agent Architecture
 Author-email: onesuit <wltks2155@gmail.com>
 License: MIT

{sa_assistant-0.2.0 → sa_assistant-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sa-assistant"
-version = "0.2.0"
+version = "0.2.1"
 requires-python = ">=3.10"
 description = "SA Assistant - AWS Solutions Architect Professional Agent with Multi-Agent Architecture"
 readme = "README.md"
@@ -73,7 +73,7 @@ where = ["src"]
 include = ["*"]
 
 [tool.setuptools.package-data]
-"*" = ["skills/**/*.md"]
+"*" = ["skills/**/*.md", "*.tcss"]
 
 [tool.ruff]
 line-length = 100

{sa_assistant-0.2.0 → sa_assistant-0.2.1}/src/sa_assistant.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sa-assistant
-Version: 0.2.0
+Version: 0.2.1
 Summary: SA Assistant - AWS Solutions Architect Professional Agent with Multi-Agent Architecture
 Author-email: onesuit <wltks2155@gmail.com>
 License: MIT

{sa_assistant-0.2.0 → sa_assistant-0.2.1}/src/sa_assistant.egg-info/SOURCES.txt

@@ -28,4 +28,5 @@ src/sa_assistant.egg-info/requires.txt
 src/sa_assistant.egg-info/top_level.txt
 src/tui/__init__.py
 src/tui/app.py
+src/tui/styles.tcss
 test/test_main.py

sa_assistant-0.2.1/src/tui/app.py

@@ -0,0 +1,386 @@
+"""SA Assistant TUI Application - OpenCode-style Terminal User Interface."""
+
+from datetime import datetime
+from pathlib import Path
+from typing import Optional
+
+from textual import work
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.containers import ScrollableContainer, Vertical, Horizontal
+from textual.widgets import Footer, Input, Static, RichLog, Collapsible
+from textual.worker import Worker
+from rich.text import Text
+from rich.markdown import Markdown
+
+from strands import Agent
+from strands.agent.conversation_manager import SlidingWindowConversationManager
+from strands_tools import file_read, file_write, shell, image_reader
+
+from model.load import load_sonnet
+from prompts.system_prompts import ORCHESTRATOR_PROMPT
+from agents.orchestrator import consult_guru, list_gurus
+from agents.specialists import consult_specialist, list_specialists, parallel_research
+from agentskills import discover_skills, generate_skills_prompt, create_skill_tool
+
+
+CSS_PATH = Path(__file__).parent / "styles.tcss"
+SKILLS_DIR = Path(__file__).parent.parent / "skills"
+VERSION = "0.2.1"
+
+
+class SessionHeader(Static):
+    """Top header showing session info in OpenCode style.
+
+    Format: # New session - {timestamp}    {tokens} {percentage} (${cost}) v{version}
+    """
+
+    def __init__(self):
+        super().__init__()
+        self.session_start = datetime.now()
+        self.session_id = self.session_start.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z"
+        self.token_count = 0
+        self.max_tokens = 200000  # Claude context window
+        self.cost = 0.0
+
+    def on_mount(self) -> None:
+        self._update_display()
+
+    def _update_display(self) -> None:
+        percentage = int((self.token_count / self.max_tokens) * 100) if self.max_tokens > 0 else 0
+        header_text = (
+            f"[bold]# New session[/bold] - {self.session_id}    "
+            f"[dim]{self.token_count:,}[/dim] {percentage}% "
+            f"[dim](${self.cost:.2f})[/dim] v{VERSION}"
+        )
+        self.update(header_text)
+
+    def update_stats(self, tokens: int = 0, cost: float = 0.0) -> None:
+        self.token_count += tokens
+        self.cost += cost
+        self._update_display()
+
+
+class AgentIndicator(Static):
+    """Bottom indicator showing current agent and model.
+
+    Format: ■ SA Assistant · Claude Sonnet
+    """
+
+    def __init__(self, agent_name: str = "SA Assistant", model_name: str = "Claude Sonnet"):
+        super().__init__()
+        self.agent_name = agent_name
+        self.model_name = model_name
+
+    def on_mount(self) -> None:
+        self._update_display()
+
+    def _update_display(self) -> None:
+        self.update(f"[#FF9900]■[/#FF9900] {self.agent_name} · [dim]{self.model_name}[/dim]")
+
+    def set_agent(self, name: str, model: str = None) -> None:
+        self.agent_name = name
+        if model:
+            self.model_name = model
+        self._update_display()
+
+
+class HotkeyBar(Static):
+    """Bottom hotkey bar showing available shortcuts.
+
+    Format: esc interrupt    ctrl+l clear    ctrl+t theme    ctrl+c quit
+    """
+
+    def on_mount(self) -> None:
+        hotkeys = (
+            "[dim]esc[/dim] interrupt    "
+            "[dim]ctrl+l[/dim] clear    "
+            "[dim]ctrl+t[/dim] theme    "
+            "[dim]ctrl+c[/dim] quit"
+        )
+        self.update(hotkeys)
+
+
+class MessagePanel(Vertical):
+    """A bordered panel for displaying a message with OpenCode styling.
+
+    Each message has a header line (# Title) and content area.
+    """
+
+    def __init__(
+        self,
+        title: str,
+        content: str = "",
+        panel_type: str = "default",
+        collapsible: bool = False,
+        collapsed: bool = False,
+    ):
+        super().__init__()
+        self.title = title
+        self._content = content
+        self.panel_type = panel_type
+        self.collapsible = collapsible
+        self.collapsed = collapsed
+        self._log: Optional[RichLog] = None
+        self.add_class(f"panel-{panel_type}")
+
+    def compose(self) -> ComposeResult:
+        # Title header
+        title_class = "panel-title"
+        if self.collapsible:
+            title_class += " collapsible-title"
+            collapse_indicator = "▶ " if self.collapsed else "▼ "
+            yield Static(
+                f"[bold]{collapse_indicator}# {self.title}[/bold]",
+                id="panel-header",
+                classes=title_class,
+            )
+        else:
+            yield Static(f"[bold]# {self.title}[/bold]", id="panel-header", classes=title_class)
+
+        # Content area
+        self._log = RichLog(highlight=True, markup=True, wrap=True, id="panel-content")
+        if not self.collapsed:
+            yield self._log
+
+    def on_mount(self) -> None:
+        if self._content and self._log:
+            self._log.write(self._content)
+
+    def append(self, text: str) -> None:
+        """Append text to the content area."""
+        if self._log:
+            self._log.write(text)
+
+    def update_content(self, text: str) -> None:
+        """Replace all content."""
+        if self._log:
+            self._log.clear()
+            self._log.write(text)
+
+    def toggle_collapse(self) -> None:
+        """Toggle collapsed state."""
+        if not self.collapsible:
+            return
+        self.collapsed = not self.collapsed
+        header = self.query_one("#panel-header", Static)
+        collapse_indicator = "▶ " if self.collapsed else "▼ "
+        header.update(f"[bold]{collapse_indicator}# {self.title}[/bold]")
+
+        if self.collapsed:
+            if self._log:
+                self._log.remove()
+                self._log = None
+        else:
+            self._log = RichLog(highlight=True, markup=True, wrap=True, id="panel-content")
+            self.mount(self._log)
+            if self._content:
+                self._log.write(self._content)
+
+
+class ChatContainer(ScrollableContainer):
+    """Main scrollable container for chat messages."""
+
+    def __init__(self):
+        super().__init__()
+        self._panels: list[MessagePanel] = []
+
+    def add_user_message(self, text: str) -> MessagePanel:
+        """Add a user message panel."""
+        panel = MessagePanel(
+            title="You",
+            content=text,
+            panel_type="user",
+            collapsible=False,
+        )
+        self._panels.append(panel)
+        self.mount(panel)
+        panel.scroll_visible()
+        return panel
+
+    def add_assistant_message(self, title: str = "SA Assistant") -> MessagePanel:
+        """Add an assistant message panel."""
+        panel = MessagePanel(
+            title=title,
+            panel_type="assistant",
+            collapsible=False,
+        )
+        self._panels.append(panel)
+        self.mount(panel)
+        panel.scroll_visible()
+        return panel
+
+    def add_tool_panel(self, tool_name: str, collapsed: bool = True) -> MessagePanel:
+        """Add a tool execution panel (collapsible)."""
+        emoji_map = {
+            "consult_guru": "🧙",
+            "consult_specialist": "🔬",
+            "file_read": "📄",
+            "file_write": "💾",
+            "shell": "🖥️",
+            "skill": "📚",
+            "image_reader": "🖼️",
+        }
+        emoji = emoji_map.get(tool_name, "🔧")
+        panel = MessagePanel(
+            title=f"{emoji} {tool_name}",
+            panel_type="tool",
+            collapsible=True,
+            collapsed=collapsed,
+        )
+        self._panels.append(panel)
+        self.mount(panel)
+        return panel
+
+    def add_thinking_panel(self) -> MessagePanel:
+        """Add a thinking/reasoning panel (collapsible)."""
+        panel = MessagePanel(
+            title="💭 Thinking",
+            panel_type="thinking",
+            collapsible=True,
+            collapsed=True,
+        )
+        self._panels.append(panel)
+        self.mount(panel)
+        return panel
+
+
+class SAAssistantApp(App):
+    """SA Assistant Textual TUI Application - OpenCode Style."""
+
+    CSS_PATH = "styles.tcss"
+    TITLE = "SA Assistant"
+
+    BINDINGS = [
+        Binding("escape", "cancel", "Interrupt", show=False),
+        Binding("ctrl+c", "quit", "Quit", show=False),
+        Binding("ctrl+l", "clear", "Clear", show=False),
+        Binding("ctrl+t", "toggle_dark", "Theme", show=False),
+    ]
+
+    def __init__(self):
+        super().__init__()
+        self._agent: Optional[Agent] = None
+        self._current_panel: Optional[MessagePanel] = None
+        self._response_buffer = ""
+        self._init_agent()
+
+    def _init_agent(self) -> None:
+        """Initialize the Strands agent."""
+        discovered_skills = discover_skills(SKILLS_DIR)
+        skills_prompt = generate_skills_prompt(discovered_skills)
+        full_prompt = ORCHESTRATOR_PROMPT + skills_prompt
+        skill_tool = create_skill_tool(discovered_skills, SKILLS_DIR) if discovered_skills else None
+
+        tools = [
+            consult_guru,
+            list_gurus,
+            consult_specialist,
+            list_specialists,
+            parallel_research,
+            file_read,
+            file_write,
+            image_reader,
+            shell,
+        ]
+        if skill_tool:
+            tools.append(skill_tool)
+
+        conversation_manager = SlidingWindowConversationManager(window_size=20)
+
+        self._agent = Agent(
+            model=load_sonnet(enable_thinking=False),
+            conversation_manager=conversation_manager,
+            system_prompt=full_prompt,
+            tools=tools,
+            callback_handler=self._streaming_callback,
+        )
+
+    def _streaming_callback(self, **kwargs) -> None:
+        """Handle streaming responses from the agent."""
+        if "data" in kwargs:
+            self._response_buffer += kwargs["data"]
+            if self._current_panel:
+                self.call_from_thread(self._current_panel.update_content, self._response_buffer)
+
+        if "current_tool_use" in kwargs:
+            tool_use = kwargs["current_tool_use"]
+            if isinstance(tool_use, dict):
+                tool_name = tool_use.get("name", "tool")
+                self.call_from_thread(self._add_tool_panel, tool_name)
+
+    def _add_tool_panel(self, tool_name: str) -> None:
+        """Add a tool execution panel to the chat."""
+        chat = self.query_one(ChatContainer)
+        chat.add_tool_panel(tool_name, collapsed=True)
+
+    def compose(self) -> ComposeResult:
+        """Compose the UI layout."""
+        yield SessionHeader()
+        yield ChatContainer()
+        yield AgentIndicator()
+        yield Input(placeholder="Type your message... (Enter to send)", id="input-field")
+        yield HotkeyBar()
+
+    def on_mount(self) -> None:
+        """Focus the input field on mount."""
+        self.query_one("#input-field", Input).focus()
+
+    async def on_input_submitted(self, event: Input.Submitted) -> None:
+        """Handle user input submission."""
+        user_input = event.value.strip()
+        if not user_input:
+            return
+
+        # Clear input
+        event.input.value = ""
+
+        # Handle exit commands
+        if user_input.lower() in ("exit", "quit", "종료"):
+            self.exit()
+            return
+
+        # Add user message to chat
+        chat = self.query_one(ChatContainer)
+        chat.add_user_message(user_input)
+
+        # Prepare for assistant response
+        self._current_panel = chat.add_assistant_message()
+        self._response_buffer = ""
+
+        # Run agent in background thread
+        self.run_agent(user_input)
+
+    @work(thread=True)
+    def run_agent(self, prompt: str) -> None:
+        """Run the agent in a background thread."""
+        try:
+            self._agent(prompt)
+        except Exception as e:
+            if self._current_panel:
+                self.call_from_thread(self._current_panel.update_content, f"[red]Error: {e}[/red]")
+
+    def action_cancel(self) -> None:
+        """Cancel running operations."""
+        self.workers.cancel_all()
+
+    def action_clear(self) -> None:
+        """Clear the chat container."""
+        chat = self.query_one(ChatContainer)
+        for panel in chat._panels:
+            panel.remove()
+        chat._panels.clear()
+
+    def action_toggle_dark(self) -> None:
+        """Toggle dark/light theme."""
+        self.dark = not self.dark
+
+
+def run_app() -> None:
+    """Entry point for the TUI application."""
+    app = SAAssistantApp()
+    app.run()
+
+
+if __name__ == "__main__":
+    run_app()

sa_assistant-0.2.1/src/tui/styles.tcss

@@ -0,0 +1,170 @@
+/* SA Assistant TUI - OpenCode Style Theme */
+
+$aws-orange: #FF9900;
+$aws-dark: #232F3E;
+$aws-blue: #1A73E8;
+$border-color: #3D4F5F;
+$panel-bg: #1E1E1E;
+
+/* ===== Screen ===== */
+Screen {
+    background: $surface;
+}
+
+/* ===== Session Header ===== */
+SessionHeader {
+    dock: top;
+    height: 3;
+    background: $surface-darken-1;
+    border-bottom: solid $border-color;
+    padding: 1 2;
+    color: $text;
+}
+
+/* ===== Agent Indicator ===== */
+AgentIndicator {
+    dock: bottom;
+    height: 3;
+    background: $surface-darken-2;
+    border-top: solid $border-color;
+    padding: 1 2;
+}
+
+/* ===== Hotkey Bar ===== */
+HotkeyBar {
+    dock: bottom;
+    height: 1;
+    background: $aws-dark;
+    padding: 0 2;
+    color: $text-muted;
+}
+
+/* ===== Input Field ===== */
+Input {
+    dock: bottom;
+    margin: 0 1;
+    border: tall $border-color;
+    background: $surface-darken-1;
+}
+
+Input:focus {
+    border: tall $aws-orange;
+}
+
+/* ===== Chat Container ===== */
+ChatContainer {
+    height: 1fr;
+    padding: 0 1;
+    overflow-y: auto;
+}
+
+/* ===== Message Panels ===== */
+MessagePanel {
+    margin: 1 0;
+    border: round $border-color;
+    background: $panel-bg;
+    height: auto;
+}
+
+/* Panel Title Header */
+.panel-title {
+    background: $surface-darken-2;
+    padding: 0 1;
+    height: 1;
+}
+
+.panel-title:hover {
+    background: $surface-darken-1;
+}
+
+.collapsible-title {
+    text-style: bold;
+}
+
+.collapsible-title:hover {
+    background: $primary-darken-2;
+    cursor: pointer;
+}
+
+/* Panel Content */
+#panel-content {
+    height: auto;
+    max-height: 40;
+    padding: 1 2;
+    background: $surface;
+}
+
+/* ===== Panel Type Variations ===== */
+
+/* User messages - green accent */
+.panel-user {
+    border: round green;
+}
+
+.panel-user .panel-title {
+    color: green;
+}
+
+/* Assistant messages - orange accent */
+.panel-assistant {
+    border: round $aws-orange;
+}
+
+.panel-assistant .panel-title {
+    color: $aws-orange;
+}
+
+/* Tool panels - cyan accent, dashed border */
+.panel-tool {
+    margin-left: 2;
+    border: dashed cyan;
+}
+
+.panel-tool .panel-title {
+    color: cyan;
+}
+
+.panel-tool #panel-content {
+    max-height: 20;
+}
+
+/* Thinking panels - dim purple */
+.panel-thinking {
+    margin-left: 2;
+    border: dashed magenta 50%;
+}
+
+.panel-thinking .panel-title {
+    color: magenta 50%;
+}
+
+.panel-thinking #panel-content {
+    max-height: 15;
+    color: $text-muted;
+}
+
+/* ===== RichLog Styling ===== */
+RichLog {
+    height: auto;
+    padding: 1 2;
+    background: $surface;
+    scrollbar-size: 1 1;
+}
+
+/* ===== Scrollbar ===== */
+ChatContainer::-webkit-scrollbar {
+    width: 1;
+}
+
+/* ===== Dark Mode Adjustments ===== */
+Screen.-dark-mode {
+    background: #0D1117;
+}
+
+Screen.-dark-mode MessagePanel {
+    background: #161B22;
+}
+
+Screen.-dark-mode .panel-title {
+    background: #21262D;
+}

sa_assistant-0.2.0/src/tui/app.py

@@ -1,269 +0,0 @@
-"""SA Assistant TUI Application - Main Textual App class."""
-
-from datetime import datetime
-from pathlib import Path
-from typing import Optional
-
-from textual import work
-from textual.app import App, ComposeResult
-from textual.binding import Binding
-from textual.containers import ScrollableContainer, Vertical
-from textual.widgets import Footer, Header, Input, Static, RichLog, Collapsible
-from textual.worker import Worker
-from rich.text import Text
-from rich.markdown import Markdown
-
-from strands import Agent
-from strands.agent.conversation_manager import SlidingWindowConversationManager
-from strands_tools import file_read, file_write, shell, image_reader
-
-from model.load import load_sonnet
-from prompts.system_prompts import ORCHESTRATOR_PROMPT
-from agents.orchestrator import consult_guru, list_gurus
-from agents.specialists import consult_specialist, list_specialists, parallel_research
-from agentskills import discover_skills, generate_skills_prompt, create_skill_tool
-
-
-CSS_PATH = Path(__file__).parent / "styles.tcss"
-SKILLS_DIR = Path(__file__).parent.parent / "skills"
-
-
-class SessionHeader(Static):
-    """Top header showing session info, tokens, cost."""
-
-    def __init__(self, session_id: str = None):
-        super().__init__()
-        self.session_start = datetime.now()
-        self.session_id = session_id or self.session_start.strftime("%Y-%m-%dT%H:%M:%S")
-        self.token_count = 0
-        self.cost = 0.0
-
-    def compose(self) -> ComposeResult:
-        yield Static(id="session-info")
-
-    def on_mount(self) -> None:
-        self._update_display()
-
-    def _update_display(self) -> None:
-        info = self.query_one("#session-info", Static)
-        info.update(
-            f"# Session - {self.session_id}    "
-            f"[dim]{self.token_count:,} tokens · ${self.cost:.2f}[/dim]"
-        )
-
-    def update_stats(self, tokens: int = 0, cost: float = 0.0) -> None:
-        self.token_count += tokens
-        self.cost += cost
-        self._update_display()
-
-
-class AgentIndicator(Static):
-    """Bottom indicator showing current agent and model."""
-
-    def __init__(self, agent_name: str = "SA Assistant", model_name: str = "Claude Sonnet"):
-        super().__init__()
-        self.agent_name = agent_name
-        self.model_name = model_name
-
-    def on_mount(self) -> None:
-        self._update_display()
-
-    def _update_display(self) -> None:
-        self.update(f"[cyan]■[/cyan] {self.agent_name} · [dim]{self.model_name}[/dim]")
-
-    def set_agent(self, name: str, model: str = None) -> None:
-        self.agent_name = name
-        if model:
-            self.model_name = model
-        self._update_display()
-
-
-class MessageSection(Collapsible):
-    """Collapsible section for a single message/action."""
-
-    def __init__(self, title: str, content: str = "", collapsed: bool = False):
-        super().__init__(title=title, collapsed=collapsed)
-        self._content = content
-        self._log: Optional[RichLog] = None
-
-    def compose(self) -> ComposeResult:
-        self._log = RichLog(highlight=True, markup=True, wrap=True)
-        yield self._log
-
-    def on_mount(self) -> None:
-        if self._content and self._log:
-            self._log.write(self._content)
-
-    def append(self, text: str) -> None:
-        if self._log:
-            self._log.write(text)
-
-    def update_content(self, text: str) -> None:
-        if self._log:
-            self._log.clear()
-            self._log.write(text)
-
-
-class ChatContainer(ScrollableContainer):
-    """Main scrollable container for chat messages."""
-
-    def __init__(self):
-        super().__init__()
-        self._sections: list[MessageSection] = []
-
-    def add_user_message(self, text: str) -> MessageSection:
-        section = MessageSection(title=f"[green]You[/green]", content=text, collapsed=False)
-        self._sections.append(section)
-        self.mount(section)
-        section.scroll_visible()
-        return section
-
-    def add_assistant_message(self, title: str = "SA Assistant") -> MessageSection:
-        section = MessageSection(title=f"[#FF9900]{title}[/#FF9900]", collapsed=False)
-        self._sections.append(section)
-        self.mount(section)
-        section.scroll_visible()
-        return section
-
-    def add_tool_section(self, tool_name: str, collapsed: bool = True) -> MessageSection:
-        emoji_map = {
-            "consult_guru": "🧙",
-            "consult_specialist": "🔬",
-            "file_read": "📄",
-            "file_write": "💾",
-            "shell": "🖥️",
-            "skill": "📚",
-        }
-        emoji = emoji_map.get(tool_name, "🔧")
-        section = MessageSection(title=f"[cyan]{emoji} {tool_name}[/cyan]", collapsed=collapsed)
-        self._sections.append(section)
-        self.mount(section)
-        return section
-
-
-class SAAssistantApp(App):
-    """SA Assistant Textual TUI Application."""
-
-    CSS_PATH = "styles.tcss"
-
-    TITLE = "SA Assistant"
-    SUB_TITLE = "AWS Solutions Architect Agent"
-
-    BINDINGS = [
-        Binding("escape", "cancel", "Cancel", show=True),
-        Binding("ctrl+c", "quit", "Quit", show=True),
-        Binding("ctrl+l", "clear", "Clear", show=True),
-        Binding("ctrl+t", "toggle_dark", "Theme", show=True),
-    ]
-
-    def __init__(self):
-        super().__init__()
-        self._agent: Optional[Agent] = None
-        self._current_section: Optional[MessageSection] = None
-        self._response_buffer = ""
-        self._init_agent()
-
-    def _init_agent(self) -> None:
-        discovered_skills = discover_skills(SKILLS_DIR)
-        skills_prompt = generate_skills_prompt(discovered_skills)
-        full_prompt = ORCHESTRATOR_PROMPT + skills_prompt
-        skill_tool = create_skill_tool(discovered_skills, SKILLS_DIR) if discovered_skills else None
-
-        tools = [
-            consult_guru,
-            list_gurus,
-            consult_specialist,
-            list_specialists,
-            parallel_research,
-            file_read,
-            file_write,
-            image_reader,
-            shell,
-        ]
-        if skill_tool:
-            tools.append(skill_tool)
-
-        conversation_manager = SlidingWindowConversationManager(window_size=20)
-
-        self._agent = Agent(
-            model=load_sonnet(enable_thinking=False),
-            conversation_manager=conversation_manager,
-            system_prompt=full_prompt,
-            tools=tools,
-            callback_handler=self._streaming_callback,
-        )
-
-    def _streaming_callback(self, **kwargs) -> None:
-        if "data" in kwargs:
-            self._response_buffer += kwargs["data"]
-            if self._current_section:
-                self.call_from_thread(self._current_section.update_content, self._response_buffer)
-
-        if "current_tool_use" in kwargs:
-            tool_use = kwargs["current_tool_use"]
-            if isinstance(tool_use, dict):
-                tool_name = tool_use.get("name", "tool")
-                self.call_from_thread(self._add_tool_section, tool_name)
-
-    def _add_tool_section(self, tool_name: str) -> None:
-        chat = self.query_one(ChatContainer)
-        chat.add_tool_section(tool_name, collapsed=True)
-
-    def compose(self) -> ComposeResult:
-        yield Header()
-        yield SessionHeader()
-        yield ChatContainer()
-        yield AgentIndicator()
-        yield Input(placeholder="Type your message... (Enter to send)")
-        yield Footer()
-
-    def on_mount(self) -> None:
-        self.query_one(Input).focus()
-
-    async def on_input_submitted(self, event: Input.Submitted) -> None:
-        user_input = event.value.strip()
-        if not user_input:
-            return
-
-        event.input.value = ""
-
-        if user_input.lower() in ("exit", "quit", "종료"):
-            self.exit()
-            return
-
-        chat = self.query_one(ChatContainer)
-        chat.add_user_message(user_input)
-
-        self._current_section = chat.add_assistant_message()
-        self._response_buffer = ""
-
-        self.run_agent(user_input)
-
-    @work(thread=True)
-    def run_agent(self, prompt: str) -> None:
-        try:
-            self._agent(prompt)
-        except Exception as e:
-            if self._current_section:
-                self.call_from_thread(
-                    self._current_section.update_content, f"[red]Error: {e}[/red]"
-                )
-
-    def action_cancel(self) -> None:
-        self.workers.cancel_all()
-
-    def action_clear(self) -> None:
-        chat = self.query_one(ChatContainer)
-        chat.remove_children()
-
-    def action_toggle_dark(self) -> None:
-        self.dark = not self.dark
-
-
-def run_app() -> None:
-    app = SAAssistantApp()
-    app.run()
-
-
-if __name__ == "__main__":
-    run_app()