speaksy 0.1.0__py3-none-any.whl

speaksy/__init__.py

@@ -0,0 +1,3 @@
+"""Speaksy - Voice typing for Linux."""
+
+__version__ = "0.1.0"

speaksy/__main__.py

@@ -0,0 +1,6 @@
+"""Entry point for python -m speaksy."""
+
+from speaksy.cli import main
+
+if __name__ == "__main__":
+    main()

speaksy/cli.py

@@ -0,0 +1,307 @@
+"""Interactive CLI for Speaksy."""
+
+import sys
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.prompt import Prompt, Confirm
+from rich.table import Table
+
+from speaksy import __version__
+from speaksy import config
+from speaksy import service
+from speaksy.setup_wizard import run_setup
+
+console = Console()
+
+
+def print_banner():
+    """Print the speaksy banner."""
+    banner = """
+  [bold cyan]╭────────────────────────────────────────╮[/bold cyan]
+  [bold cyan]│[/bold cyan]  [bold white]SPEAKSY[/bold white]                              [bold cyan]│[/bold cyan]
+  [bold cyan]│[/bold cyan]  [dim]talk it. type it. ship it.[/dim]           [bold cyan]│[/bold cyan]
+  [bold cyan]╰────────────────────────────────────────╯[/bold cyan]
+"""
+    console.print(banner)
+
+
+def get_status_display() -> str:
+    """Get the current status as a styled string."""
+    if not config.is_configured():
+        return "[cyan]waiting for setup[/cyan]"
+    elif service.is_running():
+        return "[green]vibing[/green]"
+    else:
+        return "[yellow]sleeping[/yellow]"
+
+
+def print_status_line():
+    """Print the current status line."""
+    status = get_status_display()
+    console.print(f"  [bold]Status:[/bold] {status}")
+
+    if config.is_configured():
+        ptt, toggle = config.get_hotkeys()
+        ptt_display = ptt.replace("Key.", "").replace("_", " ").title()
+        toggle_display = toggle.replace("Key.", "").upper()
+        console.print(f"  [bold]Hotkeys:[/bold] {ptt_display} (hold) | {toggle_display} (toggle)")
+    console.print()
+
+
+def print_commands():
+    """Print available commands."""
+    console.print("  [dim]Commands:[/dim]")
+    console.print("    [cyan]/setup[/cyan]   get your keys in here")
+    console.print("    [cyan]/start[/cyan]   let's gooo")
+    console.print("    [cyan]/stop[/cyan]    take a break")
+    console.print("    [cyan]/status[/cyan]  what's the vibe?")
+    console.print("    [cyan]/logs[/cyan]    receipts")
+    console.print("    [cyan]/config[/cyan]  tweak the drip")
+    console.print("    [cyan]/help[/cyan]    need backup?")
+    console.print("    [cyan]/quit[/cyan]    peace out")
+    console.print()
+
+
+def cmd_setup():
+    """Run the setup wizard."""
+    run_setup()
+
+
+def cmd_start():
+    """Start the service."""
+    if not config.is_configured():
+        console.print("[yellow]hold up, run /setup first[/yellow]")
+        return
+
+    if service.is_running():
+        console.print("[yellow]already vibing fam[/yellow]")
+        return
+
+    console.print("[dim]starting...[/dim]")
+    if service.start_service():
+        console.print("[green]lesss gooo! speaksy is now listening...[/green]")
+        ptt, toggle = config.get_hotkeys()
+        ptt_display = ptt.replace("Key.", "").replace("_", " ").title()
+        toggle_display = toggle.replace("Key.", "").upper()
+        console.print(f"   hold [cyan]{ptt_display}[/cyan] or tap [cyan]{toggle_display}[/cyan] to speak")
+    else:
+        console.print("[red]couldn't start the service[/red]")
+        console.print("[dim]check /logs for more info[/dim]")
+
+
+def cmd_stop():
+    """Stop the service."""
+    if not service.is_running():
+        console.print("[yellow]already sleeping[/yellow]")
+        return
+
+    console.print("[dim]stopping...[/dim]")
+    if service.stop_service():
+        console.print("[yellow]aight speaksy is taking a nap. run /start when you need me[/yellow]")
+    else:
+        console.print("[red]couldn't stop the service[/red]")
+
+
+def cmd_status():
+    """Show detailed status."""
+    console.print()
+    console.print("[bold]the vibe check:[/bold]")
+
+    status = service.get_status()
+
+    if status["running"]:
+        uptime = status.get("uptime", "unknown")
+        console.print(f"   [green]├─ service: running for {uptime}[/green]")
+    elif status["installed"]:
+        console.print("   [yellow]├─ service: stopped[/yellow]")
+    else:
+        console.print("   [red]├─ service: not installed[/red]")
+
+    if config.api_key_exists():
+        console.print("   [green]├─ api key: configured ✓[/green]")
+    else:
+        console.print("   [red]├─ api key: not set[/red]")
+
+    mode = config.get_privacy_mode()
+    if mode == "local":
+        console.print("   [cyan]└─ mode: local (privacy mode)[/cyan]")
+    else:
+        console.print("   [green]└─ mode: cloud (groq)[/green]")
+
+    console.print()
+
+
+def cmd_logs():
+    """Show recent logs."""
+    console.print()
+    console.print("[bold]receipts:[/bold]")
+    console.print()
+    logs = service.get_logs(lines=15)
+    console.print(f"[dim]{logs}[/dim]")
+    console.print()
+
+
+def cmd_config():
+    """Interactive config editor."""
+    console.print()
+    console.print("[bold]settings:[/bold]")
+    console.print()
+    console.print("  1. API key")
+    console.print("  2. Hotkeys")
+    console.print("  3. Privacy mode (cloud/local)")
+    console.print("  4. Text cleanup on/off")
+    console.print("  5. Back")
+    console.print()
+
+    choice = Prompt.ask("[cyan]pick one[/cyan]", choices=["1", "2", "3", "4", "5"], default="5")
+
+    if choice == "1":
+        console.print()
+        new_key = Prompt.ask("[cyan]new API key[/cyan]", password=True)
+        if new_key:
+            config.save_api_key(new_key)
+            console.print("[green]saved ✓[/green]")
+            if service.is_running():
+                service.restart_service()
+                console.print("[dim]service restarted[/dim]")
+
+    elif choice == "2":
+        console.print()
+        console.print("[dim]examples: Key.ctrl_r, Key.f8, Key.alt_l[/dim]")
+        ptt, toggle = config.get_hotkeys()
+
+        new_ptt = Prompt.ask(f"[cyan]push-to-talk[/cyan]", default=ptt)
+        new_toggle = Prompt.ask(f"[cyan]toggle[/cyan]", default=toggle)
+
+        config.set_hotkeys(new_ptt, new_toggle)
+        console.print("[green]saved ✓[/green]")
+
+        if service.is_running():
+            service.restart_service()
+            console.print("[dim]service restarted[/dim]")
+
+    elif choice == "3":
+        console.print()
+        current = config.get_privacy_mode()
+        console.print(f"[dim]currently: {current}[/dim]")
+        console.print()
+        console.print("  1. [green]cloud[/green] - Groq API, <1s response")
+        console.print("  2. [cyan]local[/cyan] - runs on your CPU, ~3-5s response")
+        console.print()
+
+        mode_choice = Prompt.ask("[cyan]pick[/cyan]", choices=["1", "2"], default="1" if current == "cloud" else "2")
+        new_mode = "cloud" if mode_choice == "1" else "local"
+        config.set_privacy_mode(new_mode)
+
+        if new_mode == "local":
+            console.print("[cyan]switched to local-only mode[/cyan]")
+            console.print("[dim]your voice never leaves your machine[/dim]")
+        else:
+            console.print("[green]switched to cloud mode[/green]")
+
+        if service.is_running():
+            service.restart_service()
+
+    elif choice == "4":
+        console.print()
+        current = config.get_cleanup_enabled()
+        console.print(f"[dim]currently: {'on' if current else 'off'}[/dim]")
+
+        new_state = Confirm.ask("[cyan]enable text cleanup?[/cyan]", default=current)
+        config.set_cleanup_enabled(new_state)
+        console.print(f"[green]cleanup {'enabled' if new_state else 'disabled'} ✓[/green]")
+
+        if service.is_running():
+            service.restart_service()
+
+    console.print()
+
+
+def cmd_help():
+    """Show help."""
+    console.print()
+    console.print("[bold]speaksy commands:[/bold]")
+    console.print()
+
+    table = Table(show_header=False, box=None, padding=(0, 2))
+    table.add_column("Command", style="cyan")
+    table.add_column("Description")
+
+    table.add_row("/setup", "Run the setup wizard (API key, hotkeys, etc.)")
+    table.add_row("/start", "Start the voice typing service")
+    table.add_row("/stop", "Stop the voice typing service")
+    table.add_row("/status", "Show detailed status info")
+    table.add_row("/logs", "View recent service logs")
+    table.add_row("/config", "Edit settings (hotkeys, privacy mode, etc.)")
+    table.add_row("/help", "Show this help message")
+    table.add_row("/quit", "Exit CLI (service keeps running)")
+
+    console.print(table)
+    console.print()
+    console.print("[dim]tip: service runs in background, so you can close this anytime[/dim]")
+    console.print()
+
+
+def run_repl():
+    """Run the interactive REPL."""
+    commands = {
+        "/setup": cmd_setup,
+        "/start": cmd_start,
+        "/stop": cmd_stop,
+        "/status": cmd_status,
+        "/logs": cmd_logs,
+        "/config": cmd_config,
+        "/help": cmd_help,
+        "/quit": lambda: None,
+        "/exit": lambda: None,
+        "/q": lambda: None,
+    }
+
+    while True:
+        try:
+            user_input = Prompt.ask("[bold magenta]speaksy>[/bold magenta]").strip().lower()
+
+            if not user_input:
+                continue
+
+            if user_input in ("/quit", "/exit", "/q"):
+                console.print("[dim]peace out ✌️[/dim]")
+                break
+
+            if user_input in commands:
+                commands[user_input]()
+            elif user_input.startswith("/"):
+                console.print(f"[red]unknown command: {user_input}[/red]")
+                console.print("[dim]type /help for available commands[/dim]")
+            else:
+                console.print("[dim]commands start with / (try /help)[/dim]")
+
+        except KeyboardInterrupt:
+            console.print()
+            console.print("[dim]peace out ✌️[/dim]")
+            break
+        except EOFError:
+            break
+
+
+def main():
+    """Main entry point."""
+    print_banner()
+    print_status_line()
+
+    # Auto-run setup if not configured
+    if not config.is_configured():
+        console.print("[yellow]looks like this is your first time![/yellow]")
+        console.print("[dim]let's get you set up...[/dim]")
+        console.print()
+        run_setup()
+        console.print()
+    else:
+        print_commands()
+
+    run_repl()
+
+
+if __name__ == "__main__":
+    main()

speaksy/config.py

@@ -0,0 +1,157 @@
+"""Configuration management for Speaksy."""
+
+import os
+from pathlib import Path
+
+import yaml
+
+# XDG config directory
+CONFIG_DIR = Path.home() / ".config" / "speaksy"
+CONFIG_FILE = CONFIG_DIR / "config.yaml"
+ENV_FILE = CONFIG_DIR / ".env"
+
+DEFAULT_CONFIG = {
+    "stt": {
+        "primary": "groq",
+        "groq_model": "whisper-large-v3-turbo",
+        "local_model": "base",
+        "local_device": "cpu",
+        "local_compute_type": "int8",
+        "language": "en",
+    },
+    "cleanup": {
+        "enabled": True,
+        "model": "llama-3.1-8b-instant",
+    },
+    "audio": {
+        "sample_rate": 16000,
+        "channels": 1,
+        "pre_buffer_seconds": 0.5,
+    },
+    "hotkeys": {
+        "push_to_talk": "Key.ctrl_r",
+        "toggle": "Key.f8",
+    },
+    "text_injection": {
+        "method": "clipboard",
+        "restore_clipboard": True,
+    },
+    "tray": {
+        "enabled": True,
+    },
+}
+
+
+def deep_merge(base: dict, override: dict) -> dict:
+    """Recursively merge override into base dict."""
+    result = base.copy()
+    for key, value in override.items():
+        if key in result and isinstance(result[key], dict) and isinstance(value, dict):
+            result[key] = deep_merge(result[key], value)
+        else:
+            result[key] = value
+    return result
+
+
+def ensure_config_dir():
+    """Create config directory if it doesn't exist."""
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def config_exists() -> bool:
+    """Check if config file exists."""
+    return CONFIG_FILE.exists()
+
+
+def load_config() -> dict:
+    """Load config from YAML, merge with defaults."""
+    config = DEFAULT_CONFIG.copy()
+    if CONFIG_FILE.exists():
+        with open(CONFIG_FILE) as f:
+            user_config = yaml.safe_load(f) or {}
+        config = deep_merge(DEFAULT_CONFIG, user_config)
+    return config
+
+
+def save_config(config: dict):
+    """Save config to YAML file."""
+    ensure_config_dir()
+    with open(CONFIG_FILE, "w") as f:
+        yaml.dump(config, f, default_flow_style=False, sort_keys=False)
+
+
+def get_api_key() -> str:
+    """Get API key from .env file."""
+    if ENV_FILE.exists():
+        with open(ENV_FILE) as f:
+            for line in f:
+                line = line.strip()
+                if line.startswith("GROQ_API_KEY="):
+                    return line.split("=", 1)[1].strip()
+    return os.getenv("GROQ_API_KEY", "")
+
+
+def save_api_key(api_key: str):
+    """Save API key to .env file."""
+    ensure_config_dir()
+    with open(ENV_FILE, "w") as f:
+        f.write(f"GROQ_API_KEY={api_key}\n")
+    # Set restrictive permissions
+    os.chmod(ENV_FILE, 0o600)
+
+
+def api_key_exists() -> bool:
+    """Check if API key is configured."""
+    return bool(get_api_key())
+
+
+def is_configured() -> bool:
+    """Check if speaksy is fully configured."""
+    return config_exists() and api_key_exists()
+
+
+def get_hotkeys() -> tuple:
+    """Get current hotkey settings."""
+    config = load_config()
+    hotkeys = config.get("hotkeys", {})
+    return (
+        hotkeys.get("push_to_talk", "Key.ctrl_r"),
+        hotkeys.get("toggle", "Key.f8"),
+    )
+
+
+def set_hotkeys(push_to_talk: str, toggle: str):
+    """Update hotkey settings."""
+    config = load_config()
+    config["hotkeys"]["push_to_talk"] = push_to_talk
+    config["hotkeys"]["toggle"] = toggle
+    save_config(config)
+
+
+def get_privacy_mode() -> str:
+    """Get current privacy mode (cloud or local)."""
+    config = load_config()
+    primary = config.get("stt", {}).get("primary", "groq")
+    return "local" if primary == "local" else "cloud"
+
+
+def set_privacy_mode(mode: str):
+    """Set privacy mode (cloud or local)."""
+    config = load_config()
+    config["stt"]["primary"] = "local" if mode == "local" else "groq"
+    save_config(config)
+
+
+def get_cleanup_enabled() -> bool:
+    """Check if text cleanup is enabled."""
+    config = load_config()
+    return config.get("cleanup", {}).get("enabled", True)
+
+
+def set_cleanup_enabled(enabled: bool):
+    """Enable or disable text cleanup."""
+    config = load_config()
+    if "cleanup" not in config:
+        config["cleanup"] = {}
+    config["cleanup"]["enabled"] = enabled
+    save_config(config)