dbagent-cli 0.8.0__tar.gz → 0.8.2__tar.gz

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.8.0
+Version: 0.8.2
 Summary: Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/dbagent/__init__.py

@@ -2,5 +2,5 @@
 DB-Agent: Universal Database Introspection and Script Generation AI Agent (CLI).
 """
 
-__version__ = "0.8.0"
+__version__ = "0.8.2"
 __author__ = "Santhosh Gupta"

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/dbagent/cli.py

@@ -437,9 +437,9 @@ def chat_command(
     provider: Optional[str] = typer.Option(None, "--provider", "-p", help="AI provider name"),
     model: Optional[str] = typer.Option(None, "--model", "-m", help="AI model name"),
     mode: str = typer.Option("CONFIRM", "--mode", help="Operation mode (SAFE, READ_ONLY, CONFIRM, SANDBOX, ADMIN)"),
+    inline: bool = typer.Option(False, "--inline", help="Run standard inline terminal prompt"),
 ):
     """Interactive multi-turn AI database copilot session."""
-    print_banner()
     db_url = resolve_db_url(db)
 
     with DynamicSpinner(DynamicSpinner.CONNECT_MESSAGES):
@@ -493,7 +493,18 @@ def chat_command(
         ":explain", ":history", ":auto", ":noauto", ":run", ":clear", ":export",
         ":exit", ":quit", ":q",
     ], ignore_case=True)
-    session = PromptSession(history=FileHistory(str(history_file)), completer=chat_completer)
+    chat_style = Style.from_dict({
+        "bottom-toolbar": "bg:#0f172a #94a3b8",
+        "bottom-toolbar.text": "#cbd5e1",
+        "bottom-toolbar.green": "#10b981 bold",
+        "bottom-toolbar.cyan": "#06b6d4 bold",
+        "bottom-toolbar.yellow": "#f59e0b bold",
+    })
+    session = PromptSession(
+        history=FileHistory(str(history_file)),
+        completer=chat_completer,
+        style=chat_style,
+    )
 
     auto_execute = True
     conversation_ctx = ConversationContext()
@@ -512,21 +523,44 @@ def chat_command(
         conversation_context=conversation_ctx,
     )
 
+    # Launch Full-Screen TUI (Fixed top header, middle scroller, static bottom chat bar)
+    if not inline and not os.getenv("DBAGENT_INLINE_CHAT"):
+        try:
+            from dbagent.ui.fullscreen_chat import FullScreenChatApp
+            app_tui = FullScreenChatApp(
+                connector=connector,
+                llm=llm,
+                pipeline=pipeline,
+                db_name=db_name,
+                dialect=dialect,
+                table_count=len(table_names),
+                config_mgr=config_mgr,
+                cache_mgr=cache_mgr,
+                model=model,
+                mode=current_mode,
+            )
+            app_tui.run()
+            connector.close()
+            return
+        except Exception:
+            pass
+
     def _get_bottom_toolbar():
         curr_model = model or getattr(llm, "default_model", "default")
         return HTML(
-            f" <b><ansigreen>● Connected:</ansigreen></b> <ansibrightwhite>{db_name}</ansibrightwhite> <ansidarkgray>({dialect} • {len(table_names)} tables)</ansidarkgray>  │  "
-            f"<b><ansicyan>🤖 {curr_model}</ansicyan></b> <ansidarkgray>({llm.name})</ansidarkgray>  │  "
-            f"<b><ansiyellow>🛡️ {current_mode.value}</ansiyellow></b>  │  "
-            f"<ansidarkgray><b>:help</b> for commands · <b>:models</b> to switch</ansidarkgray> "
+            f" <style class='bottom-toolbar.green'>● Connected:</style> <b>{db_name}</b> <i>({dialect} • {len(table_names)} tables)</i>  │  "
+            f"<style class='bottom-toolbar.cyan'>🤖 AI:</style> <b>{curr_model}</b>  │  "
+            f"<style class='bottom-toolbar.yellow'>🛡️ {current_mode.value}</style>  │  "
+            f"<b>:help</b> for commands · <b>:models</b> to switch"
         )
 
     while True:
         try:
             user_input = session.prompt(
-                HTML("<ansired><b>❯</b></ansired> "),
+                HTML("<ansicyan><b>❯</b></ansicyan> "),
                 bottom_toolbar=_get_bottom_toolbar,
-                placeholder=HTML("<ansidarkgray><i>Ask in plain English or paste SQL (e.g. \"list active users\")...</i></ansidarkgray>"),
+                rprompt=HTML(f"<ansidarkgray>db-agent v{__version__}</ansidarkgray>"),
+                placeholder=HTML("<ansidarkgray><i>Ask in plain English or paste direct SQL...</i></ansidarkgray>"),
             ).strip()
             if not user_input:
                 continue
@@ -542,31 +576,30 @@ def chat_command(
                     break
 
                 if cmd_name in [":help", ":h", ":?"]:
-                    console.print("\n[bold cyan]Interactive Chat Commands:[/bold cyan]")
-                    console.print("  [bold]:tables[/bold]                     -> List all tables in this database")
-                    console.print("  [bold]:table <name>[/bold]              -> Inspect columns, PKs, types & sample data")
-                    console.print("  [bold]:fk <name>[/bold]                 -> View mapped foreign keys & related tables")
-                    console.print("  [bold]:path <t1> <t2>[/bold]            -> Discover shortest join path between tables")
-                    console.print("  [bold]:join <t1> <t2>[/bold]            -> Explain join cardinality & syntax")
-                    console.print("  [bold]:doctor[/bold]                    -> Run database health diagnostics")
-                    console.print("  [bold]:mode <name>[/bold]               -> Switch mode (SAFE, READ_ONLY, CONFIRM, SANDBOX, ADMIN)")
-                    console.print("  [bold]:changes[/bold]                   -> List recent write ChangeSets")
-                    console.print("  [bold]:provider <name>[/bold]           -> Switch AI provider (gemini, ollama, groq, openrouter)")
-                    console.print("  [bold]:models[/bold]                     -> View list of available AI models & speed benchmarks")
-                    console.print("  [bold]:model <num|name>[/bold]          -> Switch active model (:model 1, :model 3.7, :model 3.6)")
-                    console.print("  [bold]:page <n>[/bold]                  -> View column page <n> for wide multi-column tables")
-                    console.print("  [bold]:card[/bold]                      -> View last result in expanded vertical card format (all columns)")
-                    console.print("  [bold]:wide[/bold]                      -> Force-show all columns in terminal (zoom out)")
-                    console.print("  [bold]:scroll[/bold]                    -> Open full-screen interactive horizontal/vertical scroller")
-                    console.print("  [bold]:window[/bold]                    -> Open desktop GUI window with scrollbars & search")
-                    console.print("  [bold]:web[/bold]                       -> Open browser-based interactive data grid")
-                    console.print("  [bold]:explain[/bold]                    -> Show generated SQL & query analysis for last turn")
-                    console.print("  [bold]:history[/bold]                    -> View history of queries in this session")
-                    console.print("  [bold]:auto[/bold] / [bold]:noauto[/bold]        -> Toggle automatic execution of read queries")
-                    console.print("  [bold]:run[/bold]                        -> Re-execute the last generated SQL query")
-                    console.print("  [bold]:clear[/bold]                      -> Clear conversation context")
-                    console.print("  [bold]:export <file>[/bold]              -> Save the last generated SQL to a file")
-                    console.print("  [bold]:exit[/bold]                       -> Exit chat session\n")
+                    console.print("\n[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
+                    console.print("[bold cyan]⌨️ DB-Agent Interactive Commands:[/bold cyan]\n")
+                    cmd_table = Table(box=None, padding=(0, 2), show_header=False)
+                    cmd_table.add_column("Command", style="bold cyan", width=22)
+                    cmd_table.add_column("Description", style="dim")
+                    cmd_table.add_row(":models, :model", "Switch active AI model (Claude 3.7, GPT-4o, DeepSeek, Gemini...)")
+                    cmd_table.add_row(":provider <name>", "Switch AI provider (claude, openai, deepseek, mistral, gemini)")
+                    cmd_table.add_row(":tables, :dt", "List all tables & schemas in this database")
+                    cmd_table.add_row(":table <name>", "Inspect columns, primary keys, data types & sample data")
+                    cmd_table.add_row(":fk <name>", "View foreign key relationships & dependency graph")
+                    cmd_table.add_row(":path <t1> <t2>", "Discover shortest SQL join path between two tables")
+                    cmd_table.add_row(":doctor", "Run database health, index, and query performance diagnostics")
+                    cmd_table.add_row(":mode <name>", "Switch safety mode (SAFE, CONFIRM, ADMIN, READ_ONLY)")
+                    cmd_table.add_row(":history", "View session query history, execution timing & row counts")
+                    cmd_table.add_row(":page <n>, :card", "Inspect wide multi-column query results")
+                    cmd_table.add_row(":window, :web", "Open interactive desktop or browser data grid")
+                    cmd_table.add_row(":explain", "Explain last generated SQL query with execution plan")
+                    cmd_table.add_row(":run", "Re-execute the last generated SQL query")
+                    cmd_table.add_row(":clear", "Clear conversation memory & redraw screen")
+                    cmd_table.add_row(":export <file>", "Save the last generated SQL to a file")
+                    cmd_table.add_row(":exit, :q", "Exit DB-Agent session")
+                    console.print(cmd_table)
+                    console.print("[dim]──────────────────────────────────────────────────────────────────────────────[/dim]")
+                    console.print("[dim]Enter to continue · Type any natural language question or direct SQL[/dim]\n")
                     continue
 
                 if cmd_name in [":tables", ":schema", ":dt"]:
@@ -727,46 +760,46 @@ def chat_command(
 
                     # If no arg passed or :models/:providers called, show rich model selection menu
                     if not cmd_arg or cmd_name in [":models", ":providers", ":ai"]:
-                        console.print(f"\n[bold cyan]Active AI Provider:[/bold cyan] [bold white]{llm.name}[/bold white]")
-                        console.print(f"[bold cyan]Active Model:[/bold cyan] [bold green]{curr_model}[/bold green]\n")
-                        if avail_models:
-                            console.print(f"[bold]Available Models for {llm.name}:[/bold]")
-                            model_descriptions = {
-                                # Gemini
-                                "gemini-flash-lite-latest": "⚡ Ultra-Fast (~0.8s) • Instant chat & queries",
-                                "gemini-3.6-flash": "⚡ High Quality (~3.5s) • Strong SQL reasoning & accuracy",
-                                "gemini-3.7-flash": "🧠 Deep Reasoning (~13s) • Latest flagship model",
-                                "gemini-3.1-flash-lite": "⚡ Fast Lite (~4.0s) • Low latency",
-                                "gemini-flash-latest": "⚡ Standard Flash (~7.0s)",
-                                # Claude
-                                "claude-3-7-sonnet-20250219": "🧠 Hybrid Reasoning Flagship • Deepest architecture understanding",
-                                "claude-3-5-sonnet-20241022": "⚡ High Intelligence • Expert SQL engineer & schema specialist",
-                                "claude-3-5-haiku-20241022": "⚡ Ultra-Fast (~1s) • High speed and cost efficient",
-                                "claude-3-opus-20240229": "Deep complex schema reasoning",
-                                # OpenAI
-                                "gpt-4o": "⚡ Omni Flagship • High speed multimodal & complex SQL reasoning",
-                                "gpt-4o-mini": "⚡ Ultra-Fast (~1s) • High speed and cost efficient",
-                                "gpt-4.5-preview": "🧠 Next-Generation Reasoning model",
-                                "o3-mini": "🧠 Deep STEM & Advanced Coding Reasoning",
-                                "o1": "🧠 Maximum Reasoning & Math Architecture",
-                                # DeepSeek
-                                "deepseek-chat": "⚡ DeepSeek-V3 Ultra-Fast • Strong code & SQL",
-                                "deepseek-reasoner": "🧠 DeepSeek-R1 Full Chain-of-Thought Reasoning",
-                                # Mistral
-                                "codestral-latest": "⚡ Specialized Code & SQL Architect (22B)",
-                                "mistral-large-latest": "🧠 Flagship Reasoning & Multi-language",
-                                "mistral-small-latest": "⚡ Fast & lightweight",
-                                # Groq
-                                "llama-3.3-70b-versatile": "⚡ Llama 3.3 70B (500+ tokens/sec on Groq LPUs)",
-                                "llama-3.1-8b-instant": "⚡ Ultra-Fast Llama 3.1 8B (800+ tokens/sec)",
-                                "qwen-2.5-coder-32b": "⚡ Qwen 2.5 Coder 32B specialized SQL engineer",
-                            }
-                            for idx, m_name in enumerate(avail_models, 1):
-                                is_active = " [bold green]✓ Active[/bold green]" if m_name == curr_model else ""
-                                desc = model_descriptions.get(m_name, "")
-                                desc_str = f" - [dim]{desc}[/dim]" if desc else ""
-                                console.print(f"  [bold cyan]{idx}[/bold cyan]. [bold white]{m_name}[/bold white]{desc_str}{is_active}")
-                            console.print("\n[dim]Quick Switch: [bold]:model <number|name>[/bold] (e.g. [bold]:model 1[/bold] or [bold]:model 3.7[/bold] or [bold]:model 4o[/bold] or [bold]:model r1[/bold])[/dim]\n")
+                        console.print(f"\n[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
+                        console.print(f"[bold cyan]Available Models for {llm.name}:[/bold cyan]\n")
+                        model_descriptions = {
+                            # Gemini
+                            "gemini-flash-lite-latest": "⚡ Ultra-Fast (~0.8s) · Instant chat & queries",
+                            "gemini-3.6-flash": "⚡ High Quality (~3.5s) · Strong SQL reasoning & accuracy",
+                            "gemini-3.7-flash": "🧠 Deep Reasoning (~13s) · Latest flagship model",
+                            "gemini-3.1-flash-lite": "⚡ Fast Lite (~4.0s) · Low latency",
+                            "gemini-flash-latest": "⚡ Standard Flash (~7.0s)",
+                            # Claude
+                            "claude-3-7-sonnet-20250219": "🧠 Hybrid Reasoning Flagship · Deepest architecture understanding",
+                            "claude-3-5-sonnet-20241022": "⚡ High Intelligence · Expert SQL engineer & schema specialist",
+                            "claude-3-5-haiku-20241022": "⚡ Ultra-Fast (~1s) · High speed and cost efficient",
+                            "claude-3-opus-20240229": "Deep complex schema reasoning",
+                            # OpenAI
+                            "gpt-4o": "⚡ Omni Flagship · High speed multimodal & complex SQL reasoning",
+                            "gpt-4o-mini": "⚡ Ultra-Fast (~1s) · High speed and cost efficient",
+                            "gpt-4.5-preview": "🧠 Next-Generation Reasoning model",
+                            "o3-mini": "🧠 Deep STEM & Advanced Coding Reasoning",
+                            "o1": "🧠 Maximum Reasoning & Math Architecture",
+                            # DeepSeek
+                            "deepseek-chat": "⚡ DeepSeek-V3 Ultra-Fast · Strong code & SQL",
+                            "deepseek-reasoner": "🧠 DeepSeek-R1 Full Chain-of-Thought Reasoning",
+                            # Mistral
+                            "codestral-latest": "⚡ Specialized Code & SQL Architect (22B)",
+                            "mistral-large-latest": "🧠 Flagship Reasoning & Multi-language",
+                            "mistral-small-latest": "⚡ Fast & lightweight",
+                            # Groq
+                            "llama-3.3-70b-versatile": "⚡ Llama 3.3 70B (500+ tokens/sec on Groq LPUs)",
+                            "llama-3.1-8b-instant": "⚡ Ultra-Fast Llama 3.1 8B (800+ tokens/sec)",
+                            "qwen-2.5-coder-32b": "⚡ Qwen 2.5 Coder 32B specialized SQL engineer",
+                        }
+                        for idx, m_name in enumerate(avail_models, 1):
+                            marker = "[bold cyan]❯[/bold cyan]" if m_name == curr_model else " "
+                            active_str = " [bold green]✓ Active[/bold green]" if m_name == curr_model else ""
+                            desc = model_descriptions.get(m_name, "")
+                            desc_str = f" · [dim]{desc}[/dim]" if desc else ""
+                            console.print(f" {marker} [bold cyan]{idx}.[/bold cyan] [bold white]{m_name}[/bold white]{active_str}{desc_str}")
+                        console.print("\n[dim]──────────────────────────────────────────────────────────────────────────────[/dim]")
+                        console.print("[dim]Quick Switch: [bold]:model <1-5|name>[/bold] (e.g. [bold]:model 1[/bold] or [bold]:model 3.7[/bold] or [bold]:model 4o[/bold])[/dim]\n")
                         continue
 
                     # Handle numeric shortcut or name shortcut
@@ -1867,6 +1900,98 @@ def use_command(
     print_success(f"Default AI provider switched to [bold cyan]{p}[/bold cyan]!")
 
 
+def interactive_entrypoint(
+    db: Optional[str] = None,
+    provider: Optional[str] = None,
+    model: Optional[str] = None,
+    mode: Optional[str] = None,
+):
+    """
+    Direct root interactive launcher for DB-Agent.
+    Prompts user to select or configure a database connection and transitions into chat.
+    """
+    if db:
+        chat_command(db=db, alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
+        return
+
+    saved_profiles = config_mgr.list_profiles()
+
+    console.print("")
+    console.print("[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
+    console.print("[bold cyan]⚡ DB-AGENT — Universal AI Database Copilot[/bold cyan]\n")
+    console.print(f"[dim white]Working Directory: {os.getcwd()}[/dim white]\n")
+    console.print("[bold white]Select a database connection to launch:[/bold white]\n")
+
+    options = []
+    # If saved connections exist
+    if saved_profiles:
+        for p_name, prof in saved_profiles.items():
+            p_url = prof.get("url", "")
+            dialect = p_url.split("://")[0] if "://" in p_url else "database"
+            options.append((p_name, f"[bold white]{p_name}[/bold white] [dim]({dialect})[/dim]"))
+
+    # Also add quick options
+    options.append(("__new__", "[bold cyan]+ Add a new database connection URL[/bold cyan]"))
+    options.append(("__sqlite__", "[bold white]Connect to local SQLite (:memory:)[/bold white]"))
+    options.append(("__exit__", "[dim]Exit[/dim]"))
+
+    for idx, (opt_id, opt_label) in enumerate(options, 1):
+        marker = "[bold cyan]❯[/bold cyan]" if idx == 1 else " "
+        console.print(f" {marker} [bold cyan]{idx}.[/bold cyan] {opt_label}")
+
+    console.print(f"\n[dim]Enter to confirm · Type number [1-{len(options)}] to select · Ctrl+C to exit[/dim]\n")
+
+    try:
+        raw_choice = Prompt.ask("[bold cyan]Select database[/bold cyan]", default="1")
+        raw_choice = raw_choice.strip()
+        if not raw_choice.isdigit() or int(raw_choice) < 1 or int(raw_choice) > len(options):
+            print_error("Invalid selection.")
+            return
+
+        selected_idx = int(raw_choice) - 1
+        sel_id, _ = options[selected_idx]
+
+        if sel_id == "__exit__":
+            return
+        elif sel_id == "__sqlite__":
+            chat_command(db="sqlite:///:memory:", alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
+        elif sel_id == "__new__":
+            new_url = Prompt.ask("Enter database connection URL (e.g. postgresql://user:pass@host:5432/db)")
+            if not new_url.strip():
+                return
+            new_name = Prompt.ask("Save profile name as (leave empty for temporary)", default="")
+            if new_name.strip():
+                config_mgr.save_profile(name=new_name.strip(), connection_url=new_url.strip(), description=f"Connection {new_name}")
+                print_success(f"Connection profile '[bold cyan]{new_name}[/bold cyan]' saved!")
+                chat_command(db=new_name.strip(), alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
+            else:
+                chat_command(db=new_url.strip(), alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
+        else:
+            # Saved connection name
+            chat_command(db=sel_id, alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
+    except (KeyboardInterrupt, EOFError):
+        console.print("\n[dim]Operation cancelled.[/dim]")
+        return
+
+
+@app.callback(invoke_without_command=True)
+def main_callback(
+    ctx: typer.Context,
+    db: Optional[str] = typer.Option(None, "--db", "-d", help="Database connection profile or URL"),
+    provider: Optional[str] = typer.Option(None, "--provider", "-p", help="AI provider name"),
+    model: Optional[str] = typer.Option(None, "--model", "-m", help="AI model name"),
+    mode: Optional[str] = typer.Option("CONFIRM", "--mode", help="Operation mode (SAFE, CONFIRM, ADMIN)"),
+    version: Optional[bool] = typer.Option(None, "--version", "-v", help="Show DB-Agent version"),
+):
+    """DB-Agent: Enterprise AI Database Copilot CLI."""
+    if version:
+        console.print(f"DB-Agent version [bold green]{__version__}[/bold green] by {__author__}")
+        raise typer.Exit(0)
+
+    if ctx.invoked_subcommand is None:
+        interactive_entrypoint(db=db, provider=provider, model=model, mode=mode)
+
+
 def main():
     """Top-level entry point with graceful Ctrl+C / interrupt handling."""
     try:

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/dbagent/ui/chat_screen.py

@@ -44,28 +44,28 @@ def render_chat_header(
     mode: str = "CONFIRM",
     recommendations: Optional[List[str]] = None,
 ) -> None:
-    """Render the exact Claude Code CLI split-card header panel."""
+    """Render DB-Agent's split-card welcome header in signature cyan & green theme."""
     model_str = ai_model or "gemini-3.6-flash"
 
     # Left Column: Greeting, Mascot, Connected Model & DB Path
     left_grid = Table.grid(padding=(0, 1))
     left_grid.add_column(justify="center", min_width=28)
-    left_grid.add_row("[bold white]Welcome back![/bold white]")
-    left_grid.add_row(f"[bold #D97757]{MASCOT_ART.strip()}[/bold #D97757]")
-    left_grid.add_row(f"[dim white]{model_str} · {ai_provider}[/dim white]")
-    left_grid.add_row(f"[dim]{db_name} ({dialect} • {table_count} tables)[/dim]")
+    left_grid.add_row("[bold white]Welcome to DB-Agent![/bold white]")
+    left_grid.add_row(f"[bold cyan]{MASCOT_ART.strip()}[/bold cyan]")
+    left_grid.add_row(f"[bold cyan]{model_str}[/bold cyan] [dim]({ai_provider})[/dim]")
+    left_grid.add_row(f"[bold green]🟢 Connected:[/bold green] [bold white]{db_name}[/bold white] [dim]({dialect} • {table_count} tables)[/dim]")
 
-    # Right Column: Tips for getting started + What's new in v0.8.0
+    # Right Column: Quick Tips + Powerful Features
     right_grid = Table.grid(padding=(0, 1))
     right_grid.add_column()
-    right_grid.add_row("[bold #D97757]Tips for getting started[/bold #D97757]")
+    right_grid.add_row("[bold cyan]💡 Quick Database Tips[/bold cyan]")
     right_grid.add_row("[dim]Ask in plain English or paste direct SQL for 0ms execution[/dim]")
     right_grid.add_row("[dim]Type :help to inspect tables (:tables), relations (:fk), or diagnostics (:doctor)[/dim]")
     right_grid.add_row("[dim]────────────────────────────────────────────────────────────────────────[/dim]")
-    right_grid.add_row("[bold #D97757]What's new in v0.8.0[/bold #D97757]")
-    right_grid.add_row("[dim]Added Universal AI: Claude 3.7 Sonnet, OpenAI GPT-4o, DeepSeek R1, Mistral, Gemini[/dim]")
-    right_grid.add_row("[dim]Added Zero-Latency Direct SQL Fast-Path with autonomous AI auto-repair on error[/dim]")
-    right_grid.add_row("[dim]Added Multi-turn conversational session memory graph & persistent connection pooling[/dim]")
+    right_grid.add_row(f"[bold green]🚀 Features in v{__version__}[/bold green]")
+    right_grid.add_row("[dim]Universal AI: Claude 3.7 Sonnet, OpenAI GPT-4o, DeepSeek R1, Mistral, Gemini[/dim]")
+    right_grid.add_row("[dim]Zero-Latency Direct SQL Fast-Path with autonomous AI auto-repair on error[/dim]")
+    right_grid.add_row("[dim]Multi-turn conversational session memory graph & persistent connection pooling[/dim]")
 
     layout_table = Table.grid(expand=True, padding=(0, 2))
     layout_table.add_column(ratio=2)
@@ -74,14 +74,14 @@ def render_chat_header(
 
     header_panel = Panel(
         layout_table,
-        title=f"[bold #D97757] DB-Agent v{__version__} [/bold #D97757]",
+        title=f"[bold cyan]⚡ DB-AGENT [/bold cyan][bold green]v{__version__}[/bold green]",
         title_align="left",
-        border_style="#D97757",
+        border_style="cyan",
         padding=(1, 1),
     )
 
     console.print(header_panel)
-    console.print(f"[bold #D97757]│[/bold #D97757] [bold #D97757]⚡ Claude 3.7 & Gemini 3.7 are live![/bold #D97757] [dim]Type [bold white]:models[/bold white] to switch models · [bold white]:doctor[/bold white] for diagnostics[/dim]\n")
+    console.print(f"[bold cyan]│[/bold cyan] [bold cyan]⚡ Multi-AI Engine Live![/bold cyan] [dim]Type [bold white]:models[/bold white] to switch AI models · [bold white]:doctor[/bold white] for diagnostics · Mode: [bold yellow]{mode}[/bold yellow][/dim]\n")
 
     if recommendations:
         console.print("[dim italic]💡 Try asking:[/dim italic]")
@@ -200,16 +200,20 @@ def render_assistant_turn(
 
 
 def render_session_history(query_history: List[PipelineResult]) -> None:
-    """Render a clean visual session history table."""
+    """Render a clean visual session history table in DB-Agent cyan/green style."""
     if not query_history:
         console.print("[dim]No query history in this session yet.[/dim]\n")
         return
 
-    table = Table(title="📜 Session Query History", border_style="cyan", show_lines=True)
+    console.print("")
+    console.print("[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
+    console.print("[bold cyan]📜 Session Query History:[/bold cyan]\n")
+
+    table = Table(box=None, padding=(0, 2), show_header=True, header_style="bold cyan")
     table.add_column("#", style="dim", width=4, justify="right")
-    table.add_column("Status", width=8, justify="center")
+    table.add_column("Status", width=10, justify="center")
     table.add_column("Type", width=8, justify="center")
-    table.add_column("Query / Message Preview", style="white", min_width=30)
+    table.add_column("Query / Message Preview", style="white", min_width=32)
     table.add_column("Rows", style="cyan", width=6, justify="right")
     table.add_column("Time", style="green", width=8, justify="right")
 
@@ -231,9 +235,9 @@ def render_session_history(query_history: List[PipelineResult]) -> None:
             preview = (r.error or r.sql or "")[:60]
             rows = "0"
 
-        time_str = f"{r.execution_time_ms / 1000:.1f}s" if r.execution_time_ms > 0 else "-"
+        time_str = f"{r.execution_time_ms / 1000:.2f}s" if r.execution_time_ms > 0 else "-"
         table.add_row(str(i), status, q_type, preview, rows, time_str)
 
-    console.print("")
     console.print(table)
-    console.print("")
+    console.print("[dim]──────────────────────────────────────────────────────────────────────────────[/dim]")
+    console.print("[dim]Enter to continue · :run to re-execute last query[/dim]\n")

dbagent_cli-0.8.2/dbagent/ui/fullscreen_chat.py

@@ -0,0 +1,392 @@
+"""
+Full-Screen TUI for DB-Agent.
+Features:
+- Static pinned top Header with live connection and AI status.
+- Middle scrollable conversation history with Rich-rendered tables, SQL syntax boxes, and markdown.
+- Static pinned bottom Chat Bar with prompt input and shortcut footer.
+"""
+
+import io
+import os
+import sys
+import time
+from typing import Optional, List, Dict, Any, Callable
+from prompt_toolkit.application import Application
+from prompt_toolkit.layout.containers import HSplit, VSplit, Window
+from prompt_toolkit.layout.controls import FormattedTextControl
+from prompt_toolkit.layout.layout import Layout
+from prompt_toolkit.widgets import TextArea, Frame
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.formatted_text import HTML, ANSI
+from prompt_toolkit.styles import Style
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.syntax import Syntax
+from rich.markdown import Markdown
+
+from dbagent import __version__
+from dbagent.agent.pipeline import QueryPipeline, PipelineResult
+from dbagent.connectors.base import BaseConnector
+from dbagent.llm.base import BaseLLMProvider
+from dbagent.llm.factory import get_llm_provider
+from dbagent.security.modes import OperationMode
+from dbagent.config import ConfigManager
+from dbagent.schema.cache import SchemaCacheManager
+from dbagent.schema.formatter import SchemaFormatter
+from dbagent.agent.recommender import QueryRecommender
+from dbagent.agent.doctor import DatabaseDoctor
+from dbagent.ui.chat_screen import MASCOT_ART, render_session_history
+
+
+class FullScreenChatApp:
+    """
+    Full-Screen interactive TUI with fixed Top Header, middle Scrollable History,
+    and fixed Bottom Chat Bar.
+    """
+
+    def __init__(
+        self,
+        connector: BaseConnector,
+        llm: BaseLLMProvider,
+        pipeline: QueryPipeline,
+        db_name: str,
+        dialect: str,
+        table_count: int,
+        config_mgr: ConfigManager,
+        cache_mgr: SchemaCacheManager,
+        model: Optional[str] = None,
+        mode: OperationMode = OperationMode.CONFIRM,
+    ):
+        self.connector = connector
+        self.llm = llm
+        self.pipeline = pipeline
+        self.db_name = db_name
+        self.dialect = dialect
+        self.table_count = table_count
+        self.config_mgr = config_mgr
+        self.cache_mgr = cache_mgr
+        self.model = model or getattr(llm, "default_model", "default")
+        self.mode = mode
+        self.query_history: List[PipelineResult] = []
+        self.last_result: Optional[PipelineResult] = None
+        self.auto_execute = True
+
+        # Rich console for buffer rendering
+        self.rich_io = io.StringIO()
+        self.rich_console = Console(
+            file=self.rich_io,
+            force_terminal=True,
+            color_system="truecolor",
+            width=110,
+        )
+
+        # Build UI layout components
+        self._setup_ui()
+
+    def _render_rich_to_ansi(self, renderable: Any) -> str:
+        """Helper to render any Rich renderable (Table, Panel, etc.) to ANSI string."""
+        self.rich_io.seek(0)
+        self.rich_io.truncate(0)
+        self.rich_console.print(renderable)
+        return self.rich_io.getvalue()
+
+    def _setup_ui(self) -> None:
+        """Create the prompt_toolkit UI containers and keybindings."""
+        kb = KeyBindings()
+
+        @kb.add("c-c")
+        @kb.add("c-q")
+        def _exit(event):
+            event.app.exit()
+
+        @kb.add("pageup")
+        def _pageup(event):
+            self.history_area.buffer.cursor_up(count=10)
+
+        @kb.add("pagedown")
+        def _pagedown(event):
+            self.history_area.buffer.cursor_down(count=10)
+
+        @kb.add("enter")
+        def _submit(event):
+            text = self.input_area.text.strip()
+            if not text:
+                return
+            self.input_area.text = ""
+            self._handle_input(text)
+
+        # 1. Top Fixed Header Window
+        def get_header_text():
+            return HTML(
+                f"<style bg='#0f172a' fg='#00ffff'><b> ⚡ DB-AGENT v{__version__} </b></style> "
+                f"<style bg='#0f172a' fg='#00ff88'><b>🟢 Connected:</b> {self.db_name} <i>({self.dialect} • {self.table_count} tables)</i></style>  │  "
+                f"<style bg='#0f172a' fg='#00ffff'><b>🤖 AI:</b> {self.llm.name} ({self.model})</style>  │  "
+                f"<style bg='#0f172a' fg='#fbbf24'><b>🛡️ {self.mode.value}</b></style>"
+            )
+
+        self.header_window = Window(
+            content=FormattedTextControl(get_header_text),
+            height=1,
+            style="bg:#0f172a",
+        )
+
+        # 2. Middle Scrollable Conversation History
+        self.history_area = TextArea(
+            text="",
+            read_only=True,
+            scrollbar=True,
+            wrap_lines=True,
+            style="bg:#020617 fg:#f8fafc",
+        )
+
+        # 3. Bottom Static Input Bar
+        self.input_area = TextArea(
+            height=1,
+            prompt="❯ ",
+            style="bg:#0f172a fg:#00ffff bold",
+            multiline=False,
+            wrap_lines=False,
+        )
+
+        self.input_frame = Frame(
+            body=self.input_area,
+            title="💬 Ask plain English or paste SQL",
+            style="bg:#0f172a fg:#00ffff",
+        )
+
+        # 4. Bottom Static Toolbar
+        def get_footer_text():
+            return HTML(
+                "<style bg='#0f172a' fg='#94a3b8'> <b>Enter</b> Send  │  "
+                "<b>PgUp/PgDn</b> Scroll History  │  "
+                "<b>:help</b> Commands  │  "
+                "<b>:models</b> Switch Model  │  "
+                "<b>:doctor</b> Health  │  "
+                "<b>:clear</b> Clear  │  "
+                "<b>:exit</b> Quit </style>"
+            )
+
+        self.footer_window = Window(
+            content=FormattedTextControl(get_footer_text),
+            height=1,
+            style="bg:#0f172a",
+        )
+
+        # Assemble FullScreen Layout
+        self.root_container = HSplit([
+            self.header_window,
+            self.history_area,
+            self.input_frame,
+            self.footer_window,
+        ])
+
+        self.custom_style = Style.from_dict({
+            "frame.label": "#00ffff bold",
+            "frame.border": "#00ffff",
+        })
+
+        self.app = Application(
+            layout=Layout(self.root_container, focused_element=self.input_area),
+            key_bindings=kb,
+            style=self.custom_style,
+            full_screen=True,
+            mouse_support=True,
+        )
+
+        # Print initial welcome banner into history
+        self._print_welcome()
+
+    def _append_text(self, text: str) -> None:
+        """Append text to the scrollable middle conversation buffer and auto-scroll to end."""
+        current = self.history_area.text
+        if current:
+            self.history_area.text = current + "\n" + text
+        else:
+            self.history_area.text = text
+        self.history_area.buffer.cursor_position = len(self.history_area.text)
+
+    def _print_welcome(self) -> None:
+        """Render initial split-card welcome screen into history area."""
+        left_grid = Table.grid(padding=(0, 1))
+        left_grid.add_column(justify="center", min_width=28)
+        left_grid.add_row("[bold white]Welcome to DB-Agent![/bold white]")
+        left_grid.add_row(f"[bold cyan]{MASCOT_ART.strip()}[/bold cyan]")
+        left_grid.add_row(f"[bold cyan]{self.model}[/bold cyan] [dim]({self.llm.name})[/dim]")
+        left_grid.add_row(f"[bold green]🟢 Connected:[/bold green] [bold white]{self.db_name}[/bold white] [dim]({self.dialect} • {self.table_count} tables)[/dim]")
+
+        right_grid = Table.grid(padding=(0, 1))
+        right_grid.add_column()
+        right_grid.add_row("[bold cyan]💡 Quick Database Tips[/bold cyan]")
+        right_grid.add_row("[dim]Ask in plain English or paste direct SQL for 0ms execution[/dim]")
+        right_grid.add_row("[dim]Type :help to inspect tables (:tables), relations (:fk), or diagnostics (:doctor)[/dim]")
+        right_grid.add_row("[dim]────────────────────────────────────────────────────────────────────────[/dim]")
+        right_grid.add_row(f"[bold green]🚀 Features in v{__version__}[/bold green]")
+        right_grid.add_row("[dim]Universal AI: Claude 3.7 Sonnet, OpenAI GPT-4o, DeepSeek R1, Mistral, Gemini[/dim]")
+        right_grid.add_row("[dim]Zero-Latency Direct SQL Fast-Path with autonomous AI auto-repair on error[/dim]")
+        right_grid.add_row("[dim]Multi-turn conversational session memory graph & persistent connection pooling[/dim]")
+
+        layout_table = Table.grid(expand=True, padding=(0, 2))
+        layout_table.add_column(ratio=2)
+        layout_table.add_column(ratio=3)
+        layout_table.add_row(left_grid, right_grid)
+
+        panel = Panel(
+            layout_table,
+            title=f"[bold cyan]⚡ DB-AGENT [/bold cyan][bold green]v{__version__}[/bold green]",
+            title_align="left",
+            border_style="cyan",
+            padding=(1, 1),
+        )
+
+        ansi_banner = self._render_rich_to_ansi(panel)
+        self._append_text(ansi_banner)
+
+        # Add sample recommendation prompts
+        try:
+            cached = self.cache_mgr.load_cached_schema(self.connector.connection_url)
+            if cached:
+                recs = QueryRecommender.generate_recommendations(cached, max_suggestions=3)
+                if recs:
+                    rec_lines = ["[dim italic]💡 Try asking:[/dim italic]"]
+                    for r in recs:
+                        rec_lines.append(f"  [dim]→[/dim] [cyan]{r}[/cyan]")
+                    self._append_text(self._render_rich_to_ansi("\n".join(rec_lines)))
+        except Exception:
+            pass
+
+    def _handle_input(self, user_input: str) -> None:
+        """Handle incoming prompt or REPL command."""
+        # 1. REPL Commands
+        if user_input.startswith(":"):
+            cmd_parts = user_input.split(maxsplit=1)
+            cmd_name = cmd_parts[0].lower()
+            cmd_arg = cmd_parts[1].strip() if len(cmd_parts) > 1 else ""
+
+            if cmd_name in [":exit", ":quit", ":q"]:
+                self.app.exit()
+                return
+
+            if cmd_name == ":clear":
+                self.history_area.text = ""
+                self._print_welcome()
+                return
+
+            if cmd_name in [":help", ":h", ":?"]:
+                cmd_table = Table(box=None, padding=(0, 2), show_header=False)
+                cmd_table.add_column("Command", style="bold cyan", width=22)
+                cmd_table.add_column("Description", style="dim")
+                cmd_table.add_row(":models, :model", "Switch active AI model (Claude 3.7, GPT-4o, DeepSeek, Gemini...)")
+                cmd_table.add_row(":provider <name>", "Switch AI provider (claude, openai, deepseek, mistral, gemini)")
+                cmd_table.add_row(":tables, :dt", "List all tables & schemas in this database")
+                cmd_table.add_row(":table <name>", "Inspect columns, primary keys, data types & sample data")
+                cmd_table.add_row(":fk <name>", "View foreign key relationships & dependency graph")
+                cmd_table.add_row(":path <t1> <t2>", "Discover shortest SQL join path between two tables")
+                cmd_table.add_row(":doctor", "Run database health, index, and query performance diagnostics")
+                cmd_table.add_row(":mode <name>", "Switch safety mode (SAFE, CONFIRM, ADMIN, READ_ONLY)")
+                cmd_table.add_row(":history", "View session query history, execution timing & row counts")
+                cmd_table.add_row(":explain", "Explain last generated SQL query with execution plan")
+                cmd_table.add_row(":run", "Re-execute the last generated SQL query")
+                cmd_table.add_row(":clear", "Clear conversation memory & redraw screen")
+                cmd_table.add_row(":exit, :q", "Exit DB-Agent session")
+                ansi_help = self._render_rich_to_ansi(Panel(cmd_table, title="[bold cyan]⌨️ Interactive Commands[/bold cyan]", border_style="cyan"))
+                self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n" + ansi_help)
+                return
+
+            if cmd_name in [":models", ":model"]:
+                avail = self.llm.list_models() if hasattr(self.llm, "list_models") else []
+                if not cmd_arg or cmd_name == ":models":
+                    lines = [f"[bold cyan]Available Models for {self.llm.name}:[/bold cyan]\n"]
+                    for idx, m in enumerate(avail, 1):
+                        marker = "❯" if m == self.model else " "
+                        active = " [bold green](Active)[/bold green]" if m == self.model else ""
+                        lines.append(f" {marker} [bold cyan]{idx}.[/bold cyan] [bold white]{m}[/bold white]{active}")
+                    lines.append("\n[dim]Switch with: :model <number|name> (e.g. :model 1 or :model 3.7)[/dim]")
+                    self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n" + self._render_rich_to_ansi("\n".join(lines)))
+                    return
+                else:
+                    # Switch model
+                    target = cmd_arg
+                    if target.isdigit() and avail:
+                        idx = int(target) - 1
+                        if 0 <= idx < len(avail):
+                            target = avail[idx]
+                    elif target in ["3.7", "37"]:
+                        target = "claude-3-7-sonnet-20250219" if self.llm.name == "claude" else "gemini-3.7-flash"
+                    elif target in ["3.6", "36"]:
+                        target = "gemini-3.6-flash"
+                    self.model = target
+                    self.pipeline.model = target
+                    self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n[bold green]✓ Switched model to {self.model}[/bold green]\n")
+                    return
+
+            if cmd_name == ":doctor":
+                doc = DatabaseDoctor(self.connector, self.llm)
+                diag = doc.diagnose()
+                self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n" + self._render_rich_to_ansi(diag.to_markdown()))
+                return
+
+            if cmd_name == ":history":
+                self._append_text(f"\n[cyan]❯ {user_input}[/cyan]")
+                render_session_history(self.query_history)
+                return
+
+            # Other commands
+            self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n[dim]Command executed.[/dim]")
+            return
+
+        # 2. Natural Language or Direct SQL Query
+        self._append_text(f"\n[bold blue]💬 You:[/bold blue] [bold white]{user_input}[/bold white]")
+
+        # Run via pipeline
+        t0 = time.time()
+        self.pipeline.model = self.model
+        self.pipeline.mode = self.mode
+        result = self.pipeline.run(user_prompt=user_input)
+
+        self.last_result = result
+        self.query_history.append(result)
+
+        timing_str = f"⚡ {result.execution_time_ms / 1000:.2f}s" if result.execution_time_ms > 0 else ""
+
+        # Chat response
+        if result.is_chat and result.message:
+            self._append_text(f"\n[bold cyan]🤖 DB-Agent ({self.model})[/bold cyan] [dim]{timing_str}[/dim]\n" + result.message)
+            return
+
+        # Error
+        if result.error:
+            self._append_text(f"\n[bold red]❌ Error ({timing_str}):[/bold red] {result.friendly_error or result.error}")
+            return
+
+        # Executed SQL results
+        if result.was_executed:
+            blocks = []
+            if result.is_direct_sql and result.retries == 0:
+                header_line = f"[bold cyan]🤖 DB-Agent[/bold cyan] [dim](Direct SQL • 0 tokens · {timing_str})[/dim]"
+            else:
+                header_line = f"[bold cyan]🤖 DB-Agent ({self.model})[/bold cyan] [dim]{timing_str}[/dim]"
+            blocks.append(header_line)
+
+            if result.sql:
+                syntax = Syntax(result.sql.strip(), "sql", theme="monokai", line_numbers=False)
+                blocks.append(self._render_rich_to_ansi(Panel(syntax, title="[dim]SQL Statement[/dim]", border_style="dim")))
+
+            if result.retries > 0:
+                blocks.append(f"[bold yellow]✓ Direct SQL error was automatically fixed by AI ({result.retries} retry)[/bold yellow]")
+
+            if result.has_results:
+                t = Table(box=None, padding=(0, 2), show_header=True, header_style="bold cyan")
+                for col in result.columns:
+                    t.add_column(str(col), style="white")
+                for row in result.rows[:50]:
+                    t.add_row(*[str(row.get(col, "")) for col in result.columns])
+                blocks.append(self._render_rich_to_ansi(t))
+                if len(result.rows) > 50:
+                    blocks.append(f"[dim]... and {len(result.rows) - 50} more rows[/dim]")
+
+            self._append_text("\n".join(blocks))
+
+    def run(self) -> None:
+        """Start the full-screen interactive TUI."""
+        self.app.run()

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/dbagent_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.8.0
+Version: 0.8.2
 Summary: Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/dbagent_cli.egg-info/SOURCES.txt

@@ -54,6 +54,7 @@ dbagent/sql/optimizer.py
 dbagent/sql/validator.py
 dbagent/ui/chat_screen.py
 dbagent/ui/console.py
+dbagent/ui/fullscreen_chat.py
 dbagent/ui/grid_window.py
 dbagent/ui/viewer.py
 dbagent/ui/visualizer.py

{dbagent_cli-0.8.0 → dbagent_cli-0.8.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.8.0"
+version = "0.8.2"
 description = "Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"