tunacode-cli 0.0.39__py3-none-any.whl → 0.0.41__py3-none-any.whl

tunacode/utils/token_counter.py

@@ -1,23 +1,93 @@
-"""Simple token counting utility for estimating message sizes."""
+"""Token counting utility using tiktoken for accurate, offline token estimation."""
 
+import logging
+from functools import lru_cache
+from typing import Optional
 
-def estimate_tokens(text: str) -> int:
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Cache for tokenizer encodings
+_encoding_cache = {}
+
+
+@lru_cache(maxsize=8)
+def get_encoding(model_name: str):
+    """Get the appropriate tiktoken encoding for a model.
+
+    Args:
+        model_name: The model name in format "provider:model"
+
+    Returns:
+        A tiktoken encoding instance
+    """
+    try:
+        import tiktoken
+    except ImportError:
+        logger.warning("tiktoken not available, falling back to character estimation")
+        return None
+
+    # Extract the model part from "provider:model" format
+    if ":" in model_name:
+        provider, model = model_name.split(":", 1)
+    else:
+        provider, model = "unknown", model_name
+
+    # Map common models to their tiktoken encodings
+    if provider == "openai":
+        if "gpt-4" in model:
+            encoding_name = "cl100k_base"  # GPT-4 encoding
+        elif "gpt-3.5" in model:
+            encoding_name = "cl100k_base"  # GPT-3.5-turbo encoding
+        else:
+            encoding_name = "cl100k_base"  # Default for newer models
+    elif provider == "anthropic":
+        # Claude models use similar tokenization to GPT-4
+        encoding_name = "cl100k_base"
+    else:
+        # Default encoding for unknown models
+        encoding_name = "cl100k_base"
+
+    try:
+        return tiktoken.get_encoding(encoding_name)
+    except Exception as e:
+        logger.error(f"Error loading tiktoken encoding '{encoding_name}': {e}")
+        return None
+
+
+def estimate_tokens(text: str, model_name: Optional[str] = None) -> int:
     """
-    Estimate token count using a simple character-based approximation.
+    Estimate token count using tiktoken for accurate results.
 
-    This is a rough estimate: ~4 characters per token on average.
-    For more accurate counting, we would need tiktoken or similar.
+    Args:
+        text: The text to count tokens for.
+        model_name: Optional model name for model-specific tokenization.
+
+    Returns:
+        The estimated number of tokens.
     """
     if not text:
         return 0
 
-    # Simple approximation: ~4 characters per token
+    # Try tiktoken first if model is specified
+    if model_name:
+        encoding = get_encoding(model_name)
+        if encoding:
+            try:
+                return len(encoding.encode(text))
+            except Exception as e:
+                logger.error(f"Error counting tokens with tiktoken: {e}")
+
+    # Fallback to character-based estimation
     # This is roughly accurate for English text
     return len(text) // 4
 
 
 def format_token_count(count: int) -> str:
-    """Format token count for display."""
-    if count >= 1000:
+    """Format token count for display with full precision."""
+    if count >= 1_000_000:
+        return f"{count:,}"
+    elif count >= 1000:
         return f"{count:,}"
     return str(count)

{tunacode_cli-0.0.39.dist-info → tunacode_cli-0.0.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.0.39
+Version: 0.0.41
 Summary: Your agentic CLI developer.
 Author-email: larock22 <noreply@github.com>
 License-Expression: MIT
@@ -24,6 +24,7 @@ Requires-Dist: prompt_toolkit==3.0.51
 Requires-Dist: pydantic-ai[logfire]==0.2.6
 Requires-Dist: pygments==2.19.1
 Requires-Dist: rich==14.0.0
+Requires-Dist: tiktoken>=0.5.2
 Provides-Extra: dev
 Requires-Dist: build; extra == "dev"
 Requires-Dist: ruff; extra == "dev"
@@ -31,6 +32,7 @@ Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: textual-dev; extra == "dev"
+Requires-Dist: pre-commit; extra == "dev"
 Dynamic: license-file
 
 # TunaCode

{tunacode_cli-0.0.39.dist-info → tunacode_cli-0.0.41.dist-info}/RECORD

@@ -1,34 +1,33 @@
 tunacode/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tunacode/constants.py,sha256=fIAgJ0_fPxJgXWyaF-pWyVbZ7VK1gZnjVETFTVKB2lU,4074
-tunacode/context.py,sha256=6sterdRvPOyG3LU0nEAXpBsEPZbO3qtPyTlJBi-_VXE,2612
+tunacode/constants.py,sha256=gBTHjFJIlYIwg_85bzrFLBJFwJ10a6TUnyES7eBMZjI,4993
+tunacode/context.py,sha256=_gXVCyjU052jlyRAl9tklZSwl5U_zI_EIX8XN87VVWE,2786
 tunacode/exceptions.py,sha256=mTWXuWyr1k16CGLWN2tsthDGi7lbx1JK0ekIqogYDP8,3105
 tunacode/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tunacode/setup.py,sha256=XPt4eAK-qcIZQv64jGZ_ryxcImDwps9OmXjJfIS1xcs,1899
-tunacode/types.py,sha256=BciT-uxnQ44iC-4QiDY72OD23LOtqSyMOuK_N0ttlaA,7676
+tunacode/types.py,sha256=Czq7jYXHq7fZQtyqkCN5_7eEu1wyYUcB50C6v3sTWDw,8188
 tunacode/cli/__init__.py,sha256=zgs0UbAck8hfvhYsWhWOfBe5oK09ug2De1r4RuQZREA,55
-tunacode/cli/main.py,sha256=PIcFnfmIoI_pmK2y-zB_ouJbzR5fbSI7zsKQNPB_J8o,2406
-tunacode/cli/repl.py,sha256=iNaWh1pCybicGq8GKcoujr9BsN-t1IapGtSlYV8LXrI,17507
-tunacode/cli/textual_app.py,sha256=14-Nt0IIETmyHBrNn9uwSF3EwCcutwTp6gdoKgNm0sY,12593
-tunacode/cli/textual_bridge.py,sha256=LvqiTtF0hu3gNujzpKaW9h-m6xzEP3OH2M8KL2pCwRc,6333
-tunacode/cli/commands/__init__.py,sha256=YMrLz7szrmseJCRZGGX6_TyO3dJU8_QDCOFEhRAztzo,1634
+tunacode/cli/main.py,sha256=ypefhvSt9hXzNOv0WpR-PlkUOSGadvcFbUIRT13n9oo,2619
+tunacode/cli/repl.py,sha256=Sg9qDbT9lVs4brWiLhHhGH8mt_5EQf9MdwgvhS1KbuI,18333
+tunacode/cli/commands/__init__.py,sha256=zmE9JcJ9Qd2xJhgdS4jMDJOoZsrAZmL5MAFxbKkk7F8,1670
 tunacode/cli/commands/base.py,sha256=GxUuDsDSpz0iXryy8MrEw88UM3C3yxL__kDK1QhshoA,2517
-tunacode/cli/commands/registry.py,sha256=d2dSAKrz_c02zU2AKtWghqPmm-p7XG_L81yj9nI1qu8,8152
-tunacode/cli/commands/implementations/__init__.py,sha256=sRFG2aktjtWt-M0Co_GdeLjODiVjEFqHDcLyyVabj8M,917
+tunacode/cli/commands/registry.py,sha256=XVuLpp5S4Fw7GfIZfLrVZFo4jMLMNmYNpYN7xWgXyOk,8223
+tunacode/cli/commands/implementations/__init__.py,sha256=lMgLZRX9hnw-ftZu4ykqoJoHqkZ5Yu0lBvYuzHylm7Q,986
 tunacode/cli/commands/implementations/conversation.py,sha256=EsnsZB6yyVI_sbNNMvk37tCz3iAj4E85R9ev696qeqg,4683
-tunacode/cli/commands/implementations/debug.py,sha256=TdP72Dpd3Nq3lwwyj0qZEdbSjDDmRyFu-0t6ttPFNno,6769
+tunacode/cli/commands/implementations/debug.py,sha256=hWr9DOIS-kn8z89IJZ6HuRkyN1tOsnFZg5qlB8YPbG8,6763
 tunacode/cli/commands/implementations/development.py,sha256=kZRdVgReVmGU0uijFxtPio2RYkTrYMufOwgI1Aj1_NU,2729
 tunacode/cli/commands/implementations/model.py,sha256=uthx6IX9KwgwywNTDklkJpqCbaTX9h1_p-eVmqL73WQ,2245
 tunacode/cli/commands/implementations/system.py,sha256=2cGw5iCJO3aNhXTFF28CgAIyLgslvHmpfyL2ZHVB6oQ,7903
+tunacode/cli/commands/implementations/todo.py,sha256=Dtz5bgcuK2VXGPWEBBZQgnWUMYkRXNzTGf_qkVPLF2U,8125
 tunacode/configuration/__init__.py,sha256=MbVXy8bGu0yKehzgdgZ_mfWlYGvIdb1dY2Ly75nfuPE,17
-tunacode/configuration/defaults.py,sha256=lNeJUW1S8zj4-XTCkMP9UaDc-tHWXLff9K8t0uPA_oE,801
-tunacode/configuration/models.py,sha256=XPobkLM_TzKTuMIWhK-svJfGRGFT9r2LhKEM6rv6QHk,3756
+tunacode/configuration/defaults.py,sha256=lK_qf3BexmoQh7lbtxYG_ef0Kq3WyiLGOYmiVDO_amQ,840
+tunacode/configuration/models.py,sha256=buH8ZquvcYI3OQBDIZeJ08cu00rSCeNABtUwl3VQS0E,4103
 tunacode/configuration/settings.py,sha256=KoN0u6GG3Hh_TWt02D_wpRfbACYri3gCDTXHtJfHl2w,994
 tunacode/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tunacode/core/code_index.py,sha256=jgAx3lSWP_DwnyiP5Jkm1YvX4JJyI4teMzlNrJSpEOA,15661
-tunacode/core/state.py,sha256=d2xUa7sRYizVc1mkaQN4aU034VSeh9nxQadzpL-aRas,1810
+tunacode/core/state.py,sha256=hOLuvojJksd3GEezNT6jLPAsCcI-mZKldUUvlP4IKjw,3641
 tunacode/core/tool_handler.py,sha256=BPjR013OOO0cLXPdLeL2FDK0ixUwOYu59FfHdcdFhp4,2277
 tunacode/core/agents/__init__.py,sha256=UUJiPYb91arwziSpjd7vIk7XNGA_4HQbsOIbskSqevA,149
-tunacode/core/agents/main.py,sha256=BkAyqSr-xgENdyEmrPrx-E21ljclSU36EJh675Vhttk,41173
+tunacode/core/agents/main.py,sha256=xDHA9lqs089jNmTOdIX9W9BEOh28HbilZtyyOYqHNGQ,43101
 tunacode/core/agents/utils.py,sha256=VaNsPB2l1dAP-VlS_QLRKvCb4NW0pXNRoxkh12AGXAg,10744
 tunacode/core/background/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tunacode/core/background/manager.py,sha256=rJdl3eDLTQwjbT7VhxXcJbZopCNR3M8ZGMbmeVnwwMc,1126
@@ -36,11 +35,14 @@ tunacode/core/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 tunacode/core/setup/__init__.py,sha256=lzdpY6rIGf9DDlDBDGFvQZaSOQeFsNglHbkpq1-GtU8,376
 tunacode/core/setup/agent_setup.py,sha256=trELO8cPnWo36BBnYmXDEnDPdhBg0p-VLnx9A8hSSSQ,1401
 tunacode/core/setup/base.py,sha256=cbyT2-xK2mWgH4EO17VfM_OM2bj0kT895NW2jSXbe3c,968
-tunacode/core/setup/config_setup.py,sha256=xOQcjlFEL7HtGXH7wtpu83S6oA9K02wLYR1r-cWINpI,14449
+tunacode/core/setup/config_setup.py,sha256=ctf9GuN7niqCsRK5rbEzCbzsN0MrAOeWzKtwOnyJJmY,14654
 tunacode/core/setup/coordinator.py,sha256=oVTN2xIeJERXitVJpkIk9tDGLs1D1bxIRmaogJwZJFI,2049
 tunacode/core/setup/environment_setup.py,sha256=n3IrObKEynHZSwtUJ1FddMg2C4sHz7ca42awemImV8s,2225
 tunacode/core/setup/git_safety_setup.py,sha256=CRIqrQt0QUJQRS344njty_iCqTorrDhHlXRuET7w0Tk,6714
-tunacode/prompts/system.md,sha256=YgcHMlksWHzi_R8XKGN7B7vz1rsqUZKHYq05Dlqewt0,13202
+tunacode/core/token_usage/api_response_parser.py,sha256=CTtqGaFaxpkzkW3TEbe00QJzyRULpWN1EQxIYMleseg,1622
+tunacode/core/token_usage/cost_calculator.py,sha256=oQPMphGhqIt7NKdOg1o5Zbo59_nwFWmRJMQ30ViiCWs,1835
+tunacode/core/token_usage/usage_tracker.py,sha256=uKCpdZgmDfLauwsawSCifMu0kJE3lAnK7Sjd-0KYUgA,3894
+tunacode/prompts/system.md,sha256=hXpjZ8Yiv2Acr2_6EmC2uOklP8FbmvyYR9oais-1KLk,16290
 tunacode/services/__init__.py,sha256=w_E8QK6RnvKSvU866eDe8BCRV26rAm4d3R-Yg06OWCU,19
 tunacode/services/mcp.py,sha256=R48X73KQjQ9vwhBrtbWHSBl-4K99QXmbIhh5J_1Gezo,3046
 tunacode/tools/__init__.py,sha256=ECBuUWWF1JjHW42CCceaPKgVTQyuljbz3RlhuA2fe2s,314
@@ -52,6 +54,7 @@ tunacode/tools/list_dir.py,sha256=1kNqzYCNlcA5rqXIEVqcjQy6QxlLZLj5AG6YIECfwio,72
 tunacode/tools/read_file.py,sha256=z2omev9xzj4-0GG9mRssD13rj-Aa1c-pszFi2Z7Hxvk,3268
 tunacode/tools/read_file_async_poc.py,sha256=2v2ckLQlwahgPGWGdE2c5Es37B35Y7zWdseZwT46E1E,6453
 tunacode/tools/run_command.py,sha256=7UvXjFQI1Av4vceXx48MbQCTrsFNj4PlygTAAhNDYIA,4376
+tunacode/tools/todo.py,sha256=bVbohgwKqvvTe8efxXrMZDQU8vdk4E3jF9Cj38dRq7k,12727
 tunacode/tools/update_file.py,sha256=bW1MhTzRjBDjJzqQ6A1yCVEbkr1oIqtEC8uqcg_rfY4,3957
 tunacode/tools/write_file.py,sha256=prL6u8XOi9ZyPU-YNlG9YMLbSLrDJXDRuDX73ncXh-k,2699
 tunacode/ui/__init__.py,sha256=aRNE2pS50nFAX6y--rSGMNYwhz905g14gRd6g4BolYU,13
@@ -59,10 +62,10 @@ tunacode/ui/completers.py,sha256=Jx1zyCESwdm_4ZopvCBtb0bCJF-bRy8aBWG2yhPQtDc,487
 tunacode/ui/console.py,sha256=YXNFlnV7n4wyaIy-VohzIMJJ71C7fzgcjuLheNIO-QU,2079
 tunacode/ui/constants.py,sha256=A76B_KpM8jCuBYRg4cPmhi8_j6LLyWttO7_jjv47r3w,421
 tunacode/ui/decorators.py,sha256=e2KM-_pI5EKHa2M045IjUe4rPkTboxaKHXJT0K3461g,1914
-tunacode/ui/input.py,sha256=6LlEwKIXYXusNDI2PD0DDjRymQgu5mf2v06TsHbUln0,2957
+tunacode/ui/input.py,sha256=E_zAJqNYoAVFA-j4xE9Qgs22y-GrdSZNqiseX-Or0ho,2955
 tunacode/ui/keybindings.py,sha256=h0MlD73CW_3i2dQzb9EFSPkqy0raZ_isgjxUiA9u6ts,691
 tunacode/ui/lexers.py,sha256=tmg4ic1enyTRLzanN5QPP7D_0n12YjX_8ZhsffzhXA4,1340
-tunacode/ui/output.py,sha256=kjVklUZumGwjV8LeB7aX1WjcwAbURnuudZZY1t4qsN4,4499
+tunacode/ui/output.py,sha256=9BnjBa9iCFqMK7FleGyAjVIi0FVIlO2ku5RXPj2Gn2s,5549
 tunacode/ui/panels.py,sha256=IZpiWBb7jVXaycH5BPAnqTCs2-_ccJYq2V55MxkVHzQ,8199
 tunacode/ui/prompt_manager.py,sha256=U2cntB34vm-YwOj3gzFRUK362zccrz8pigQfpxr5sv8,4650
 tunacode/ui/tool_ui.py,sha256=S5-k1HwRlSqiQ8shGQ_QYGXQbuzb6Pg7u3CTqZwffdQ,6533
@@ -73,15 +76,16 @@ tunacode/utils/bm25.py,sha256=yq7KFWP3g_zIsjUV7l2hFPXYCzXyNQUInLU7u4qsc_4,1909
 tunacode/utils/diff_utils.py,sha256=V9QqQ0q4MfabVTnWptF3IXDp3estnfOKcJtDe_Sj14I,2372
 tunacode/utils/file_utils.py,sha256=AXiAJ_idtlmXEi9pMvwtfPy9Ys3yK-F4K7qb_NpwonU,923
 tunacode/utils/import_cache.py,sha256=q_xjJbtju05YbFopLDSkIo1hOtCx3DOTl3GQE5FFDgs,295
+tunacode/utils/message_utils.py,sha256=kM6VSS2Dudjplie009khHgmIRjDoBUzv6tvHcYNDAAE,586
 tunacode/utils/ripgrep.py,sha256=AXUs2FFt0A7KBV996deS8wreIlUzKOlAHJmwrcAr4No,583
 tunacode/utils/security.py,sha256=e_zo9VmcOKFjgFMr9GOBIFhAmND4PBlJZgY7zqnsGjI,6548
 tunacode/utils/system.py,sha256=FSoibTIH0eybs4oNzbYyufIiV6gb77QaeY2yGqW39AY,11381
-tunacode/utils/text_utils.py,sha256=IiRviMqz5uoAbid8emkRXxgvQz6KE27ZeQom-qh9ymI,2984
-tunacode/utils/token_counter.py,sha256=nGCWwrHHFbKywqeDCEuJnADCkfJuzysWiB6cCltJOKI,648
+tunacode/utils/text_utils.py,sha256=6YBD9QfkDO44-6jxnwRWIpmfIifPG-NqMzy_O2NAouc,7277
+tunacode/utils/token_counter.py,sha256=l5KemYLfsypAtAF_YrDtVKFtBEghydS_MA8c-8mpPvM,2721
 tunacode/utils/user_configuration.py,sha256=Ilz8dpGVJDBE2iLWHAPT0xR8D51VRKV3kIbsAz8Bboc,3275
-tunacode_cli-0.0.39.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-tunacode_cli-0.0.39.dist-info/METADATA,sha256=WZEtECuHPPI7pnDuhnxR5K4LwFT-2ReaMPFeG_GKF5g,5064
-tunacode_cli-0.0.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tunacode_cli-0.0.39.dist-info/entry_points.txt,sha256=hbkytikj4dGu6rizPuAd_DGUPBGF191RTnhr9wdhORY,51
-tunacode_cli-0.0.39.dist-info/top_level.txt,sha256=lKy2P6BWNi5XSA4DHFvyjQ14V26lDZctwdmhEJrxQbU,9
-tunacode_cli-0.0.39.dist-info/RECORD,,
+tunacode_cli-0.0.41.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+tunacode_cli-0.0.41.dist-info/METADATA,sha256=kkUdWvJ3JjXnCCF_Jr1zdgi9lSYEo8CJuibfVNau_U0,5137
+tunacode_cli-0.0.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tunacode_cli-0.0.41.dist-info/entry_points.txt,sha256=hbkytikj4dGu6rizPuAd_DGUPBGF191RTnhr9wdhORY,51
+tunacode_cli-0.0.41.dist-info/top_level.txt,sha256=lKy2P6BWNi5XSA4DHFvyjQ14V26lDZctwdmhEJrxQbU,9
+tunacode_cli-0.0.41.dist-info/RECORD,,

tunacode/cli/textual_app.py

@@ -1,420 +0,0 @@
-"""
-Modern Textual-based TUI for TunaCode.
-
-Provides a rich, interactive terminal user interface with:
-- Split-pane layout with chat history and input
-- Sidebar with model info and commands
-- Modern dialog boxes for tool confirmations
-- Real-time status updates
-"""
-
-import asyncio
-from typing import Optional
-
-from textual import on
-from textual.app import App, ComposeResult
-from textual.binding import Binding
-from textual.containers import Container, Horizontal, Vertical, VerticalScroll
-from textual.message import Message
-from textual.widgets import Button, Footer, Header, Static, TextArea
-
-from tunacode.core.state import StateManager
-from tunacode.setup import setup
-from tunacode.utils.system import check_for_updates
-
-
-class ChatMessage(Static):
-    """A single chat message widget."""
-
-    def __init__(self, sender: str, content: str, message_type: str = "user"):
-        super().__init__()
-        self.sender = sender
-        self.content = content
-        self.message_type = message_type
-
-    def compose(self) -> ComposeResult:
-        """Compose the chat message."""
-        if self.message_type == "user":
-            yield Static(f"[bold cyan]❯ You[/bold cyan]\n{self.content}", classes="user-message")
-        elif self.message_type == "agent":
-            yield Static(
-                f"[bold green]🤖 TunaCode[/bold green]\n{self.content}", classes="agent-message"
-            )
-        elif self.message_type == "system":
-            yield Static(
-                f"[bold yellow]⚠️ System[/bold yellow]\n{self.content}", classes="system-message"
-            )
-        elif self.message_type == "tool":
-            yield Static(
-                f"[bold magenta]🔧 Tool[/bold magenta]\n{self.content}", classes="tool-message"
-            )
-
-
-class Sidebar(Container):
-    """Sidebar with model info and commands."""
-
-    def __init__(self, state_manager: StateManager):
-        super().__init__()
-        self.state_manager = state_manager
-
-    def compose(self) -> ComposeResult:
-        """Compose the sidebar."""
-        yield Static("[bold]TunaCode[/bold]", classes="sidebar-title")
-        yield Static(f"Model: {self.state_manager.session.current_model}", id="current-model")
-        yield Static("", classes="spacer")
-
-        yield Static("[bold]Commands[/bold]", classes="section-title")
-        yield Static("/help - Show help", classes="command-item")
-        yield Static("/clear - Clear chat", classes="command-item")
-        yield Static("/model - Switch model", classes="command-item")
-        yield Static("/yolo - Toggle confirmations", classes="command-item")
-        yield Static("", classes="spacer")
-
-        yield Static("[bold]Status[/bold]", classes="section-title")
-        yield Static("● Ready", id="status", classes="status-ready")
-
-
-class ChatHistory(VerticalScroll):
-    """Scrollable chat history container."""
-
-    def add_message(self, sender: str, content: str, message_type: str = "user") -> None:
-        """Add a new message to the chat history."""
-        message = ChatMessage(sender, content, message_type)
-        self.mount(message)
-        self.scroll_end(animate=True)
-
-
-class InputArea(Container):
-    """Input area with text area and send button."""
-
-    class SendMessage(Message):
-        """Message sent when user submits input."""
-
-        def __init__(self, content: str) -> None:
-            self.content = content
-            super().__init__()
-
-    def compose(self) -> ComposeResult:
-        """Compose the input area."""
-        with Horizontal():
-            yield TextArea(id="message-input")
-            yield Button("Send", id="send-button", variant="primary")
-
-    @on(Button.Pressed, "#send-button")
-    def send_message(self) -> None:
-        """Send the current message."""
-        text_area = self.query_one("#message-input", TextArea)
-        content = text_area.text.strip()
-        if content:
-            self.post_message(self.SendMessage(content))
-            text_area.clear()
-
-    def on_key(self, event) -> None:
-        """Handle key events."""
-        if event.key == "ctrl+enter":
-            self.send_message()
-
-
-class TunaCodeApp(App):
-    """Main TunaCode Textual application."""
-
-    CSS = """
-    Sidebar {
-        width: 30;
-        background: $surface;
-        border: thick $primary;
-        padding: 1;
-    }
-
-    .sidebar-title {
-        text-align: center;
-        color: $primary;
-        margin-bottom: 1;
-    }
-
-    .section-title {
-        color: $accent;
-        margin: 1 0;
-    }
-
-    .command-item {
-        color: $text-muted;
-        margin-left: 1;
-    }
-
-    .status-ready {
-        color: $success;
-    }
-
-    .status-busy {
-        color: $warning;
-    }
-
-    .status-error {
-        color: $error;
-    }
-
-    ChatHistory {
-        border: thick $primary;
-        padding: 1;
-        height: 1fr;
-    }
-
-    .user-message {
-        background: $surface;
-        border-left: thick $primary;
-        padding: 1;
-        margin: 1 0;
-    }
-
-    .agent-message {
-        background: $surface;
-        border-left: thick $success;
-        padding: 1;
-        margin: 1 0;
-    }
-
-    .system-message {
-        background: $surface;
-        border-left: thick $warning;
-        padding: 1;
-        margin: 1 0;
-    }
-
-    .tool-message {
-        background: $surface;
-        border-left: thick $accent;
-        padding: 1;
-        margin: 1 0;
-    }
-
-    InputArea {
-        height: 5;
-        padding: 1;
-    }
-
-    #message-input {
-        height: 3;
-    }
-
-    #send-button {
-        width: 10;
-        margin-left: 1;
-    }
-    """
-
-    BINDINGS = [
-        Binding("ctrl+c", "quit", "Quit"),
-        Binding("ctrl+l", "clear_chat", "Clear"),
-        Binding("f1", "help", "Help"),
-        Binding("f2", "model_info", "Model"),
-    ]
-
-    def __init__(self, state_manager: StateManager):
-        super().__init__()
-        self.state_manager = state_manager
-        self.chat_history: Optional[ChatHistory] = None
-        self.sidebar: Optional[Sidebar] = None
-        self.input_area: Optional[InputArea] = None
-
-    def compose(self) -> ComposeResult:
-        """Compose the main application layout."""
-        yield Header()
-
-        with Horizontal():
-            self.sidebar = Sidebar(self.state_manager)
-            yield self.sidebar
-
-            with Vertical():
-                self.chat_history = ChatHistory()
-                yield self.chat_history
-
-                self.input_area = InputArea()
-                yield self.input_area
-
-        yield Footer()
-
-    def on_mount(self) -> None:
-        """Called when the app is mounted."""
-        # Add welcome messages
-        self.chat_history.add_message(
-            "System", "Welcome to TunaCode v0.11 - Your AI-powered development assistant", "system"
-        )
-        self.chat_history.add_message(
-            "System", f"Current model: {self.state_manager.session.current_model}", "system"
-        )
-        self.chat_history.add_message(
-            "System",
-            "⚠️ IMPORTANT: Always use git branches before making major changes\n"
-            "Type '/help' for available commands",
-            "system",
-        )
-
-    @on(InputArea.SendMessage)
-    async def handle_message(self, message: InputArea.SendMessage) -> None:
-        """Handle incoming messages from the input area."""
-        content = message.content
-
-        # Add user message to chat
-        self.chat_history.add_message("You", content, "user")
-
-        # Update status
-        status_widget = self.sidebar.query_one("#status", Static)
-        status_widget.update("● Processing...")
-        status_widget.classes = "status-busy"
-
-        if content.startswith("/"):
-            await self.handle_command(content)
-        else:
-            await self.handle_user_input(content)
-
-        # Reset status
-        status_widget.update("● Ready")
-        status_widget.classes = "status-ready"
-
-    async def handle_command(self, command: str) -> None:
-        """Handle slash commands."""
-        if command == "/help":
-            help_text = """Available Commands:
-
-/help - Show this help message
-/clear - Clear chat history
-/model - Show current model info
-/yolo - Toggle confirmation prompts
-/quit - Exit the application
-
-Keyboard Shortcuts:
-Ctrl+C - Quit
-Ctrl+L - Clear chat
-F1 - Help
-F2 - Model info
-Ctrl+Enter - Send message"""
-            self.chat_history.add_message("System", help_text, "system")
-
-        elif command == "/clear":
-            await self.action_clear_chat()
-
-        elif command == "/model":
-            model_info = f"Current model: {self.state_manager.session.current_model}"
-            self.chat_history.add_message("System", model_info, "system")
-
-        elif command == "/yolo":
-            # Toggle yolo mode
-            current_state = getattr(self.state_manager.session, "yolo_mode", False)
-            self.state_manager.session.yolo_mode = not current_state
-            new_state = "enabled" if not current_state else "disabled"
-            self.chat_history.add_message("System", f"Confirmation prompts {new_state}", "system")
-
-        elif command == "/quit":
-            await self.action_quit()
-
-        else:
-            self.chat_history.add_message("System", f"Unknown command: {command}", "system")
-
-    async def handle_user_input(self, text: str) -> None:
-        """Handle regular user input."""
-        try:
-            # Use the bridge to process the input
-            if not hasattr(self, "_bridge"):
-                from tunacode.cli.textual_bridge import TextualAgentBridge
-
-                self._bridge = TextualAgentBridge(self.state_manager, self._bridge_message_callback)
-
-            # Process the request
-            response = await self._bridge.process_user_input(text)
-
-            # Add the agent's response to chat
-            self.chat_history.add_message("TunaCode", response, "agent")
-
-        except Exception as e:
-            self.chat_history.add_message("System", f"Error: {str(e)}", "system")
-
-    async def _bridge_message_callback(self, message_type: str, content: str) -> None:
-        """Callback for bridge to send messages to the UI."""
-        self.chat_history.add_message("System", content, message_type)
-
-    def action_clear_chat(self) -> None:
-        """Clear the chat history."""
-        self.chat_history.remove_children()
-        self.chat_history.add_message("System", "Chat cleared", "system")
-
-    def action_help(self) -> None:
-        """Show help information."""
-        self.handle_command("/help")
-
-    def action_model_info(self) -> None:
-        """Show model information."""
-        self.handle_command("/model")
-
-
-async def run_textual_app(state_manager: StateManager) -> None:
-    """Run the Textual application."""
-    app = TunaCodeApp(state_manager)
-    await app.run_async()
-
-
-def main():
-    """Main entry point for the Textual app."""
-    import sys
-
-    # Handle command line arguments
-    version_flag = "--version" in sys.argv or "-v" in sys.argv
-    if version_flag:
-        from tunacode.constants import APP_VERSION
-
-        print(f"TunaCode v{APP_VERSION}")
-        return
-
-    # Initialize state manager
-    state_manager = StateManager()
-    # Show banner
-    print("🐟 TunaCode - Modern AI Development Assistant")
-    print("=" * 50)
-
-    # Check for updates
-    has_update, latest_version = check_for_updates()
-    if has_update:
-        print(f"📦 Update available: v{latest_version}")
-        print("Run: pip install --upgrade tunacode-cli")
-        print()
-
-    # Parse CLI arguments for configuration
-    cli_config = {}
-    args = sys.argv[1:]
-    i = 0
-    while i < len(args):
-        if args[i] == "--model" and i + 1 < len(args):
-            cli_config["model"] = args[i + 1]
-            i += 2
-        elif args[i] == "--key" and i + 1 < len(args):
-            cli_config["key"] = args[i + 1]
-            i += 2
-        elif args[i] == "--baseurl" and i + 1 < len(args):
-            cli_config["baseurl"] = args[i + 1]
-            i += 2
-        elif args[i] == "--setup":
-            cli_config["setup"] = True
-            i += 1
-        else:
-            i += 1
-
-    async def run_app():
-        try:
-            # Run setup
-            run_setup = cli_config.get("setup", False)
-            await setup(run_setup, state_manager, cli_config)
-
-            # Run the Textual app
-            await run_textual_app(state_manager)
-
-        except KeyboardInterrupt:
-            print("\n👋 Goodbye!")
-        except Exception as e:
-            print(f"❌ Error: {e}")
-
-    # Run the async app
-    asyncio.run(run_app())
-
-
-if __name__ == "__main__":
-    main()