tzamuncode 0.1.0__py3-none-any.whl

tzamuncode/cli/main.py

@@ -0,0 +1,444 @@
+"""
+TzamunCode CLI - Main Entry Point
+"""
+
+import typer
+from rich.console import Console
+from rich.panel import Panel
+from rich.markdown import Markdown
+from rich.prompt import Confirm
+from typing import Optional
+import sys
+
+from ..models.ollama import OllamaClient
+from ..utils.file_ops import FileManager
+from .commands import chat_command, generate_command, edit_command, explain_command
+from .agentic_commands import agentic_command, project_command
+
+app = typer.Typer(
+    name="tzamuncode",
+    help="TzamunCode - AI Coding Assistant powered by local models 🚀",
+    add_completion=False,
+)
+
+console = Console()
+
+def version_callback(value: bool):
+    if value:
+        from .. import __version__
+        console.print(f"[bold blue]TzamunCode[/bold blue] version {__version__}")
+        console.print("Built with ❤️ in Saudi Arabia 🇸🇦")
+        raise typer.Exit()
+
+@app.callback(invoke_without_command=True)
+def main(
+    ctx: typer.Context,
+    version: Optional[bool] = typer.Option(
+        None,
+        "--version",
+        "-v",
+        callback=version_callback,
+        is_eager=True,
+        help="Show version and exit"
+    )
+):
+    """
+    TzamunCode - AI Coding Assistant
+    
+    Privacy-first AI coding assistant powered by local models.
+    """
+    # If no subcommand provided, show interactive backend selection
+    if ctx.invoked_subcommand is None:
+        from rich.prompt import Prompt
+        from .realtime_chat import RealtimeChat
+        from ..auth.auth_manager import AuthManager
+        
+        # Check authentication
+        auth = AuthManager()
+        if not auth.is_authenticated():
+            console.print("\n[bold yellow]🔐 Authentication Required[/bold yellow]\n")
+            console.print("[dim]Please login to use TzamunCode[/dim]\n")
+            console.print("Run: [cyan]tzamuncode login[/cyan]\n")
+            return
+        
+        console.print("\n[bold cyan]🚀 TzamunCode AI Assistant[/bold cyan]\n")
+        console.print("Select AI Backend:\n")
+        console.print("  [bold]1[/bold]. ⚡ [cyan]vLLM[/cyan] - Fast inference (Qwen 2.5 7B)")
+        console.print("  [bold]2[/bold]. 🦙 [green]Ollama[/green] - Powerful models (Qwen 2.5 32B)")
+        
+        choice = Prompt.ask("\nChoose backend", choices=["1", "2"], default="2")
+        
+        if choice == "1":
+            console.print("\n✅ Starting with [cyan]vLLM[/cyan] backend...\n")
+            chat = RealtimeChat(model="qwen2.5-7b-instruct", use_vllm=True)
+        else:
+            console.print("\n✅ Starting with [green]Ollama[/green] backend...\n")
+            chat = RealtimeChat(model="qwen2.5:32b", use_vllm=False)
+        
+        chat.run()
+
+@app.command()
+def chat(
+    model: str = typer.Option(
+        "qwen2.5:32b",
+        "--model",
+        "-m",
+        help="Model to use for chat"
+    ),
+    system: Optional[str] = typer.Option(
+        None,
+        "--system",
+        "-s",
+        help="System prompt"
+    ),
+    vllm: bool = typer.Option(
+        False,
+        "--vllm",
+        help="Use vLLM backend (faster)"
+    )
+):
+    """
+    Start an interactive chat session with TzamunCode AI.
+    
+    Features:
+    - Beautiful UI with welcome screen
+    - Type '/' to see all commands
+    - '/models' to switch models
+    - '/settings' to view settings
+    - '/help' for help
+    
+    Examples:
+        tzamuncode chat
+        tzamuncode chat --model deepseek-coder-v2
+        tzc chat -m qwen2.5:32b
+    """
+    from .realtime_chat import run_realtime_chat
+    from ..auth.auth_manager import AuthManager
+    
+    # Check authentication
+    auth = AuthManager()
+    if not auth.is_authenticated():
+        console.print("\n[bold yellow]🔐 Authentication Required[/bold yellow]\n")
+        console.print("[dim]Please login to use TzamunCode[/dim]\n")
+        console.print("Run: [cyan]tzamuncode login[/cyan]\n")
+        return
+    
+    if vllm and model == 'qwen2.5:32b':
+        model = 'qwen2.5-7b-instruct'
+    run_realtime_chat(model=model, system=system, use_vllm=vllm)
+
+@app.command()
+def generate(
+    prompt: str = typer.Argument(..., help="What to generate"),
+    model: str = typer.Option(
+        "qwen2.5:32b",
+        "--model",
+        "-m",
+        help="Model to use"
+    ),
+    output: Optional[str] = typer.Option(
+        None,
+        "--output",
+        "-o",
+        help="Output file or directory"
+    )
+):
+    """
+    Generate code based on a prompt.
+    
+    Examples:
+        tzamuncode generate "Create a Flask REST API"
+        tzamuncode generate "Add unit tests" --output tests/
+        tzc generate "Python web scraper" -o scraper.py
+    """
+    generate_command(prompt=prompt, model=model, output=output)
+
+@app.command()
+def edit(
+    file_path: str = typer.Argument(..., help="File to edit"),
+    instruction: str = typer.Argument(..., help="What to change"),
+    model: str = typer.Option(
+        "qwen2.5:32b",
+        "--model",
+        "-m",
+        help="Model to use"
+    ),
+    apply: bool = typer.Option(
+        False,
+        "--apply",
+        "-a",
+        help="Apply changes without confirmation"
+    )
+):
+    """
+    Edit a file using AI.
+    
+    Examples:
+        tzamuncode edit app.py "Add error handling"
+        tzamuncode edit . "Add type hints to all functions"
+        tzc edit main.py "Refactor to use async" --apply
+    """
+    edit_command(file_path=file_path, instruction=instruction, model=model, apply=apply)
+
+@app.command()
+def explain(
+    file_path: str = typer.Argument(..., help="File to explain"),
+    model: str = typer.Option(
+        "qwen2.5:32b",
+        "--model",
+        "-m",
+        help="Model to use"
+    ),
+    detailed: bool = typer.Option(
+        False,
+        "--detailed",
+        "-d",
+        help="Provide detailed explanation"
+    )
+):
+    """
+    Explain code in a file.
+    
+    Examples:
+        tzamuncode explain complex_function.py
+        tzamuncode explain auth.py --detailed
+        tzc explain main.py -d
+    """
+    explain_command(file_path=file_path, model=model, detailed=detailed)
+
+@app.command()
+def code(
+    instruction: str = typer.Argument(..., help="What to do"),
+    model: str = typer.Option(
+        "qwen2.5:32b",
+        "--model",
+        "-m",
+        help="Model to use"
+    ),
+    workspace: Optional[str] = typer.Option(
+        None,
+        "--workspace",
+        "-w",
+        help="Project directory (default: current)"
+    ),
+    apply: bool = typer.Option(
+        False,
+        "--apply",
+        "-a",
+        help="Auto-apply changes without confirmation"
+    )
+):
+    """
+    Agentic AI mode - AI autonomously accesses files and writes code.
+    
+    Similar to Claude Code, the AI will:
+    - Scan your project structure
+    - Read files autonomously
+    - Edit multiple files
+    - Execute git operations
+    - Plan and execute multi-step tasks
+    
+    Examples:
+        tzamuncode code "Add authentication to this Flask app"
+        tzamuncode code "Refactor all Python files to use async/await"
+        tzc code "Add unit tests for all functions" --apply
+    """
+    agentic_command(
+        instruction=instruction,
+        model=model,
+        workspace=workspace,
+        auto_apply=apply
+    )
+
+@app.command()
+def project(
+    workspace: Optional[str] = typer.Option(
+        None,
+        "--workspace",
+        "-w",
+        help="Project directory (default: current)"
+    )
+):
+    """
+    Analyze and understand project structure.
+    
+    Examples:
+        tzamuncode project
+        tzamuncode project --workspace /path/to/project
+        tzc project
+    """
+    project_command(workspace=workspace)
+
+@app.command()
+def login(
+    username: Optional[str] = typer.Option(
+        None,
+        "--username",
+        "-u",
+        help="TzamunAI username"
+    ),
+    api_key: Optional[str] = typer.Option(
+        None,
+        "--api-key",
+        "-k",
+        help="TzamunAI API key (alternative to username/password)"
+    )
+):
+    """
+    Login to TzamunAI (app.tzamun.ai)
+    
+    Examples:
+        tzamuncode login
+        tzamuncode login --username myuser
+        tzamuncode login --api-key YOUR_API_KEY
+    """
+    from ..auth.auth_manager import AuthManager
+    from rich.prompt import Prompt
+    import getpass
+    
+    auth = AuthManager()
+    
+    # Check if already logged in
+    if auth.is_authenticated():
+        user_info = auth.get_user_info()
+        console.print(f"\n[bold yellow]⚠[/bold yellow] Already logged in as [cyan]{user_info.get('username')}[/cyan]")
+        if not Confirm.ask("Logout and login again?"):
+            return
+        auth.logout()
+    
+    console.print("\n[bold cyan]🔐 Login to TzamunAI[/bold cyan]\n")
+    
+    # API Key login
+    if api_key:
+        console.print("[dim]Validating API key...[/dim]")
+        if auth.login_with_api_key(api_key):
+            console.print("\n[bold green]✓[/bold green] Successfully logged in with API key!\n")
+        else:
+            console.print("\n[bold red]✗[/bold red] Invalid API key\n")
+        return
+    
+    # Username/Password login
+    if not username:
+        username = Prompt.ask("[cyan]Username[/cyan]")
+    
+    password = getpass.getpass("Password: ")
+    
+    console.print("\n[dim]Authenticating...[/dim]")
+    
+    if auth.login_with_credentials(username, password):
+        console.print(f"\n[bold green]✓[/bold green] Successfully logged in as [cyan]{username}[/cyan]!\n")
+        console.print("[dim]Your session will expire in 30 days[/dim]\n")
+    else:
+        console.print("\n[bold red]✗[/bold red] Login failed. Check your credentials.\n")
+
+@app.command()
+def logout():
+    """
+    Logout from TzamunAI
+    
+    Examples:
+        tzamuncode logout
+    """
+    from ..auth.auth_manager import AuthManager
+    
+    auth = AuthManager()
+    
+    if not auth.is_authenticated():
+        console.print("\n[dim]Not logged in[/dim]\n")
+        return
+    
+    user_info = auth.get_user_info()
+    auth.logout()
+    
+    console.print(f"\n[bold green]✓[/bold green] Logged out [cyan]{user_info.get('username')}[/cyan]\n")
+
+@app.command()
+def whoami():
+    """
+    Show current authentication status
+    
+    Examples:
+        tzamuncode whoami
+    """
+    from ..auth.auth_manager import AuthManager
+    
+    auth = AuthManager()
+    
+    console.print()
+    
+    if not auth.is_authenticated():
+        console.print("[bold yellow]Not authenticated[/bold yellow]")
+        console.print("\n[dim]Login with:[/dim] [cyan]tzamuncode login[/cyan]\n")
+        return
+    
+    user_info = auth.get_user_info()
+    
+    table = Table(
+        title="[bold cyan]Authentication Status[/bold cyan]",
+        border_style="blue",
+        show_header=True,
+        header_style="bold white on blue"
+    )
+    table.add_column("Property", style="bold cyan", width=20)
+    table.add_column("Value", style="yellow")
+    
+    table.add_row("Status", "✓ Authenticated")
+    table.add_row("Username", user_info.get('username', 'N/A'))
+    table.add_row("Auth Type", user_info.get('type', 'N/A').replace('_', ' ').title())
+    
+    console.print(table)
+    console.print()
+
+@app.command()
+def info():
+    """
+    Show information about TzamunCode and available models.
+    """
+    from .. import __version__
+    from ..auth.auth_manager import AuthManager
+    
+    auth = AuthManager()
+    
+    console.print(Panel.fit(
+        f"""[bold blue]TzamunCode CLI[/bold blue] v{__version__}
+        
+[bold]AI Coding Assistant powered by local models[/bold]
+
+Built by [bold cyan]Tzamun Arabia IT Co.[/bold cyan] 🇸🇦
+
+[dim]Privacy-first • No cloud dependencies • Complete control[/dim]
+""",
+        border_style="blue"
+    ))
+    
+    # Show auth status
+    if auth.is_authenticated():
+        user_info = auth.get_user_info()
+        console.print(f"\n[bold green]✓[/bold green] Logged in as [cyan]{user_info.get('username')}[/cyan]")
+    else:
+        console.print("\n[dim]Not logged in • Use[/dim] [cyan]tzamuncode login[/cyan]")
+    
+    # Check Ollama connection
+    try:
+        client = OllamaClient()
+        models = client.list_models()
+        
+        console.print("\n[bold green]✓[/bold green] Connected to Ollama")
+        console.print(f"[bold]Available models:[/bold] {len(models)}")
+        
+        for model in models[:5]:  # Show first 5
+            console.print(f"  • {model}")
+        
+        if len(models) > 5:
+            console.print(f"  ... and {len(models) - 5} more")
+            
+    except Exception as e:
+        console.print(f"\n[bold red]✗[/bold red] Could not connect to Ollama: {e}")
+        console.print("[dim]Make sure Ollama is running: ollama serve[/dim]")
+
+def run():
+    """Entry point for the CLI"""
+    app()
+
+if __name__ == "__main__":
+    run()