chat-console 0.1.1__py3-none-any.whl

app/ui/styles.py

@@ -0,0 +1,275 @@
+from rich.style import Style
+from rich.theme import Theme
+from textual.widget import Widget
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal, Vertical
+from textual.css.query import NoMatches
+
+# Define color palette
+COLORS = {
+    "dark": {
+        "background": "#0C0C0C",
+        "foreground": "#33FF33",
+        "user_msg": "#00FFFF",
+        "assistant_msg": "#33FF33",
+        "system_msg": "#FF8C00",
+        "highlight": "#FFD700",
+        "selection": "#1A1A1A",
+        "border": "#33FF33",
+        "error": "#FF0000",
+        "success": "#33FF33",
+    },
+    "light": {
+        "background": "#F0F0F0",
+        "foreground": "#000000",
+        "user_msg": "#0000FF",
+        "assistant_msg": "#008000",
+        "system_msg": "#800080",
+        "highlight": "#0078D7",
+        "selection": "#ADD6FF",
+        "border": "#D0D0D0",
+        "error": "#D32F2F",
+        "success": "#388E3C",
+    }
+}
+
+def get_theme(theme_name="dark"):
+    """Get Rich theme based on theme name"""
+    colors = COLORS.get(theme_name, COLORS["dark"])
+    
+    return Theme({
+        "user": Style(color=colors["user_msg"], bold=True),
+        "assistant": Style(color=colors["assistant_msg"]),
+        "system": Style(color=colors["system_msg"], italic=True),
+        "highlight": Style(color=colors["highlight"], bold=True),
+        "selection": Style(bgcolor=colors["selection"]),
+        "border": Style(color=colors["border"]),
+        "error": Style(color=colors["error"], bold=True),
+        "success": Style(color=colors["success"]),
+        "prompt": Style(color=colors["highlight"]),
+        "heading": Style(color=colors["highlight"], bold=True),
+        "dim": Style(color=colors["border"]),
+        "code": Style(bgcolor="#2D2D2D", color="#D4D4D4"),
+        "code.syntax": Style(color="#569CD6"),
+        "link": Style(color=colors["highlight"], underline=True),
+    })
+
+# Textual CSS for the application
+CSS = """
+/* Base styles */
+Screen {
+    background: $surface;
+    color: $text;
+}
+
+/* Chat message styles */
+.message {
+    width: 100%;
+    padding: 0 1;
+    margin: 0;
+}
+
+.message-content {
+    width: 100%;
+    text-align: left;
+    padding: 0;
+}
+
+/* Code blocks */
+.code-block {
+    background: $surface-darken-3;
+    color: $text-muted;
+    border: solid $primary-darken-3;
+    margin: 1 2;
+    padding: 1;
+    overflow: auto;
+}
+
+/* Input area */
+#input-container {
+    height: auto;
+    background: $surface;
+    border-top: solid $primary-darken-2;
+    padding: 0;
+}
+
+#message-input {
+    background: $surface-darken-1;
+    color: $text;
+    border: solid $primary-darken-2;
+    min-height: 2;
+    padding: 0 1;
+}
+
+#message-input:focus {
+    border: tall $primary;
+}
+
+/* Action buttons */
+.action-button {
+    background: $primary;
+    color: $text;
+    border: none;
+    min-width: 10;
+    margin-left: 1;
+}
+
+.action-button:hover {
+    background: $primary-lighten-1;
+}
+
+/* Sidebar */
+#sidebar {
+    width: 25%;
+    min-width: 18;
+    background: $surface-darken-1;
+    border-right: solid $primary-darken-2 1;
+}
+
+/* Chat list */
+.chat-item {
+    padding: 0 1;
+    height: 2;
+    border-bottom: solid $primary-darken-3 1;
+}
+
+.chat-item:hover {
+    background: $primary-darken-2;
+}
+
+.chat-item.selected {
+    background: $primary-darken-1;
+    border-left: wide $primary;
+}
+
+.chat-title {
+    width: 100%;
+    content-align: center middle;
+    text-align: left;
+}
+
+.chat-model {
+    color: $text-muted;
+    text-align: right;
+}
+
+.chat-date {
+    color: $text-muted;
+    text-align: right;
+}
+
+/* Search input */
+#search-input {
+    width: 100%;
+    border: solid $primary-darken-2 1;
+    margin: 0 1;
+    height: 2;
+}
+
+#search-input:focus {
+    border: solid $primary;
+}
+
+/* Model selector */
+#model-selector {
+    width: 100%;
+    height: 2;
+    margin: 0 1;
+    background: $surface-darken-1;
+    border: solid $primary-darken-2 1;
+}
+
+/* Style selector */
+#style-selector {
+    width: 100%;
+    height: 2;
+    margin: 0 1;
+    background: $surface-darken-1;
+    border: solid $primary-darken-2 1;
+}
+
+/* Header */
+#app-header {
+    width: 100%;
+    height: 2;
+    background: $surface-darken-2;
+    color: $text;
+    content-align: center middle;
+    text-align: center;
+    border-bottom: solid $primary-darken-2 1;
+}
+
+/* Loading indicator */
+#loading-indicator {
+    background: $surface-darken-1;
+    color: $text;
+    padding: 0 1;
+    height: auto;
+    width: 100%;
+    border-top: solid $primary-darken-2 1;
+    display: none;
+}
+
+/* Settings modal */
+.modal {
+    background: $surface;
+    border: solid $primary;
+    padding: 1;
+    height: auto;
+    min-width: 40;
+    max-width: 60;
+}
+
+.modal-title {
+    background: $primary;
+    color: $text;
+    width: 100%;
+    height: 3;
+    content-align: center middle;
+    text-align: center;
+}
+
+.form-label {
+    width: 100%;
+    padding: 1 0;
+}
+
+.form-input {
+    width: 100%;
+    background: $surface-darken-1;
+    border: solid $primary-darken-2;
+    height: 3;
+    margin-bottom: 1;
+}
+
+.form-input:focus {
+    border: solid $primary;
+}
+
+.button-container {
+    width: 100%;
+    height: 3;
+    align: right middle;
+}
+
+.button {
+    background: $primary-darken-1;
+    color: $text;
+    min-width: 6;
+    margin-left: 1;
+    border: solid $primary 1;
+}
+
+.button.cancel {
+    background: $error;
+}
+
+/* Tags */
+.tag {
+    background: $primary-darken-1;
+    color: $text;
+    padding: 0 1;
+    margin: 0 1 0 0;
+    border: solid $border;
+}
+"""

app/utils.py

@@ -0,0 +1,202 @@
+from datetime import datetime
+import re
+import asyncio
+import time
+from typing import List, Dict, Any, Optional, Generator, Awaitable, Callable
+import textwrap
+import threading
+from rich.text import Text
+from rich.markdown import Markdown
+from rich.syntax import Syntax
+from rich.panel import Panel
+from rich.console import Console
+
+from .models import Message, Conversation
+from .database import ChatDatabase
+from .api.base import BaseModelClient
+
+def generate_conversation_title(messages: List[Message], model: str) -> str:
+    """Generate a title for a conversation based on its content"""
+    # Find the first user message
+    first_user_message = None
+    for msg in messages:
+        if msg.role == "user":
+            first_user_message = msg
+            break
+            
+    if first_user_message is None:
+        return f"New conversation ({datetime.now().strftime('%Y-%m-%d %H:%M')})"
+    
+    # Use first line of the first user message (up to 40 chars)
+    content = first_user_message.content.strip()
+    
+    # Get first line
+    first_line = content.split('\n')[0]
+    
+    # Truncate if needed
+    if len(first_line) > 40:
+        title = first_line[:37] + "..."
+    else:
+        title = first_line
+        
+    return title
+
+def format_code_blocks(text: str) -> str:
+    """Ensure code blocks have proper formatting"""
+    # Make sure code blocks are properly formatted with triple backticks
+    pattern = r"```(\w*)\n(.*?)\n```"
+    
+    def code_replace(match):
+        lang = match.group(1)
+        code = match.group(2)
+        # Ensure code has proper indentation
+        code_lines = code.split('\n')
+        code = '\n'.join([line.rstrip() for line in code_lines])
+        return f"```{lang}\n{code}\n```"
+        
+    return re.sub(pattern, code_replace, text, flags=re.DOTALL)
+
+def extract_code_blocks(text: str) -> List[Dict[str, str]]:
+    """Extract code blocks from text content"""
+    blocks = []
+    pattern = r"```(\w*)\n(.*?)\n```"
+    matches = re.finditer(pattern, text, re.DOTALL)
+    
+    for match in matches:
+        lang = match.group(1) or "text"
+        code = match.group(2).strip()
+        blocks.append({
+            "language": lang,
+            "code": code,
+            "start": match.start(),
+            "end": match.end()
+        })
+    
+    return blocks
+
+def format_text(text: str, highlight_code: bool = True) -> Text:
+    """Format text with optional code highlighting"""
+    result = Text()
+    
+    if not highlight_code:
+        return Text(text)
+    
+    # Split by code blocks
+    parts = re.split(r'(```\w*\n.*?\n```)', text, flags=re.DOTALL)
+    
+    for part in parts:
+        if part.startswith('```'):
+            # Handle code block
+            match = re.match(r'```(\w*)\n(.*?)\n```', part, re.DOTALL)
+            if match:
+                lang = match.group(1) or "text"
+                code = match.group(2).strip()
+                syntax = Syntax(
+                    code,
+                    lang,
+                    theme="monokai",
+                    line_numbers=True,
+                    word_wrap=True,
+                    indent_guides=True
+                )
+                result.append("\n")
+                result.append(syntax)
+                result.append("\n")
+        else:
+            # Handle regular text
+            if part.strip():
+                result.append(Text(part.strip()))
+                result.append("\n")
+    
+    return result
+
+def create_new_conversation(db: ChatDatabase, model: str, style: str = "default") -> Conversation:
+    """Create a new conversation in the database"""
+    title = f"New conversation ({datetime.now().strftime('%Y-%m-%d %H:%M')})"
+    conversation_id = db.create_conversation(title, model, style)
+    
+    # Get full conversation object
+    conversation_data = db.get_conversation(conversation_id)
+    return Conversation.from_dict(conversation_data)
+
+def update_conversation_title(db: ChatDatabase, conversation: Conversation) -> None:
+    """Update the title of a conversation based on its content"""
+    if not conversation.messages:
+        return
+        
+    title = generate_conversation_title(conversation.messages, conversation.model)
+    db.update_conversation(conversation.id, title=title)
+    conversation.title = title
+
+def add_message_to_conversation(
+    db: ChatDatabase, 
+    conversation: Conversation, 
+    role: str, 
+    content: str
+) -> Message:
+    """Add a message to a conversation in the database"""
+    message_id = db.add_message(conversation.id, role, content)
+    
+    # Create message object
+    message = Message(
+        id=message_id,
+        conversation_id=conversation.id,
+        role=role,
+        content=content,
+        timestamp=datetime.now().isoformat()
+    )
+    
+    # Add to conversation
+    conversation.messages.append(message)
+    
+    # Update conversation title if it's the default
+    if conversation.title.startswith("New conversation"):
+        update_conversation_title(db, conversation)
+        
+    return message
+
+def run_in_thread(func: Callable, *args, **kwargs) -> threading.Thread:
+    """Run a function in a separate thread"""
+    thread = threading.Thread(target=func, args=args, kwargs=kwargs)
+    thread.daemon = True
+    thread.start()
+    return thread
+
+async def generate_streaming_response(
+    messages: List[Dict[str, str]],
+    model: str,
+    style: str,
+    client: BaseModelClient,
+    callback: Callable[[str], Awaitable[None]]
+) -> str:
+    """Generate a streaming response and call the callback for each chunk"""
+    full_response = ""
+    
+    try:
+        # Get the async generator from the client
+        stream = client.generate_stream(messages, model, style)
+        # Iterate over the generator properly
+        async for chunk in stream:
+            if chunk:  # Only process non-empty chunks
+                full_response += chunk
+                # Update UI and ensure event loop processes it
+                await callback(chunk)
+                # Small delay to prevent overwhelming the event loop
+                await asyncio.sleep(0.01)
+    except Exception as e:
+        error_msg = f"\n\nError generating response: {str(e)}"
+        full_response += error_msg
+        await callback(error_msg)
+        
+    return full_response
+
+def get_elapsed_time(start_time: float) -> str:
+    """Get the elapsed time as a formatted string"""
+    elapsed = time.time() - start_time
+    
+    if elapsed < 60:
+        return f"{elapsed:.1f}s"
+    else:
+        minutes = int(elapsed // 60)
+        seconds = elapsed % 60
+        return f"{minutes}m {seconds:.1f}s"

chat_console-0.1.1.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

chat_console-0.1.1.dist-info/METADATA

@@ -0,0 +1,111 @@
+Metadata-Version: 2.2
+Name: chat-console
+Version: 0.1.1
+Summary: A command-line interface for chatting with LLMs, storing chats and (future) rag interactions
+Home-page: https://github.com/wazacraftrfid/chat-console
+Author: Johnathan Greenaway
+Author-email: john@fimbriata.dev
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: textual>=0.11.1
+Requires-Dist: typer>=0.7.0
+Requires-Dist: requests>=2.28.1
+Requires-Dist: anthropic>=0.5.0
+Requires-Dist: openai>=0.27.0
+Requires-Dist: python-dotenv>=0.21.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Chat CLI
+
+A comprehensive command-line interface for chatting with various AI language models. This application allows you to interact with different LLM providers through an intuitive terminal-based interface.
+
+## Features
+
+- Interactive terminal UI with Textual library
+- Support for multiple AI models:
+  - OpenAI models (GPT-3.5, GPT-4)
+  - Anthropic models (Claude 3 Opus, Sonnet, Haiku)
+- Conversation history with search functionality
+- Customizable response styles (concise, detailed, technical, friendly)
+- Code syntax highlighting
+- Markdown rendering
+
+## Installation
+
+1. Clone this repository:
+   ```
+   git clone https://github.com/yourusername/chat-cli.git
+   cd chat-cli
+   ```
+
+2. Install the required dependencies:
+   ```
+   pip install -r requirements.txt
+   ```
+
+3. Set up your API keys:
+   
+   Create a `.env` file in the project root directory with your API keys:
+   ```
+   OPENAI_API_KEY=your_openai_api_key_here
+   ANTHROPIC_API_KEY=your_anthropic_api_key_here
+   ```
+
+## Usage
+
+Run the application:
+```
+chat-cli
+```
+
+### Keyboard Shortcuts
+
+- `q` - Quit the application
+- `n` - Start a new conversation
+- `s` - Toggle sidebar
+- `f` - Focus search box
+- `Escape` - Cancel current generation
+- `Ctrl+C` - Quit the application
+
+### Configuration
+
+The application creates a configuration file at `~/.chatcli/config.json` on first run. You can edit this file to:
+
+- Change the default model
+- Modify available models
+- Add or edit response styles
+- Change the theme
+- Adjust other settings
+
+## Data Storage
+
+Conversation history is stored in a SQLite database at `~/.chatcli/chat_history.db`.
+
+## Development
+
+The application is structured as follows:
+
+- `main.py` - Main application entry point
+- `app/` - Application modules
+  - `api/` - LLM provider API client implementations
+  - `ui/` - User interface components
+  - `config.py` - Configuration management
+  - `database.py` - Database operations
+  - `models.py` - Data models
+  - `utils.py` - Utility functions
+
+## License
+
+MIT

chat_console-0.1.1.dist-info/RECORD

@@ -0,0 +1,23 @@
+app/__init__.py,sha256=u5X4kPcpqZ12ZLnhwwOCScNvftaknDTrb0DMXqR_iLc,130
+app/config.py,sha256=PLEic_jwfWvWJxDfQMbKSbJ4ULrcmDhVe0apqegMO_g,3571
+app/database.py,sha256=nt8CVuDpy6zw8mOYqDcfUmNw611t7Ln7pz22M0b6-MI,9967
+app/main.py,sha256=B_E8ovyGX1xLXDJF_nkjQcbX_AP5hLDih3jqJXdHMMY,19848
+app/models.py,sha256=4-y9Lytay2exWPFi0FDlVeRL3K2-I7E-jBqNzTfokqY,2644
+app/utils.py,sha256=oUpQpqrxvvQn0S0lMCSwDC1Rx0PHpoAIRDySohYV5Oo,6586
+app/api/__init__.py,sha256=A8UL84ldYlv8l7O-yKzraVFcfww86SgWfpl4p7R03-w,62
+app/api/anthropic.py,sha256=leWSnCfqKnxHB5k3l_oVty4km3q18dodJkPAxwvhEt0,4211
+app/api/base.py,sha256=-Lx6nfgvEPjrAnQXuCgG-zr8soD1AibTtP15gVD3O48,3138
+app/api/ollama.py,sha256=naD5-WVCthZ-0s4iBo_bYV1hRMcuczly-lghmB2_loQ,5033
+app/api/openai.py,sha256=70NITI4upGld_xpaCZLoMd0ObSeVdhtiyUfY9hYHlhE,3420
+app/ui/__init__.py,sha256=RndfbQ1Tv47qdSiuQzvWP96lPS547SDaGE-BgOtiP_w,55
+app/ui/chat_interface.py,sha256=5gSOa7zT9bWujkPYctB8gVm4yypnkmKHcY1VtaKcEQs,11126
+app/ui/chat_list.py,sha256=WQTYVNSSXlx_gQal3YqILZZKL9UiTjmNMIDX2I9pAMM,11205
+app/ui/model_selector.py,sha256=Rv0i2VjLL2-cp4Pn_uMnAnAIV7Zk9gBX1XoWKBzkxHg,10367
+app/ui/search.py,sha256=b-m14kG3ovqW1-i0qDQ8KnAqFJbi5b1FLM9dOnbTyIs,9763
+app/ui/styles.py,sha256=eVDBTpBGnQ-mg5SeLi6i74ZjhCpItxAwWh1IelD09GY,5445
+chat_console-0.1.1.dist-info/LICENSE,sha256=srHZ3fvcAuZY1LHxE7P6XWju2njRCHyK6h_ftEbzxSE,1057
+chat_console-0.1.1.dist-info/METADATA,sha256=Nhshut-tQIN9whzOfxVBNYVAjx6wG90KoyzLgpNeFlg,2899
+chat_console-0.1.1.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+chat_console-0.1.1.dist-info/entry_points.txt,sha256=kkVdEc22U9PAi2AeruoKklfkng_a_aHAP6VRVwrAD7c,67
+chat_console-0.1.1.dist-info/top_level.txt,sha256=io9g7LCbfmTG1SFKgEOGXmCFB9uMP2H5lerm0HiHWQE,4
+chat_console-0.1.1.dist-info/RECORD,,

chat_console-0.1.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (76.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

chat_console-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+c-c = app.main:main
+chat-console = app.main:main

chat_console-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+app