dbagent-cli 0.3.8__tar.gz → 0.4.0__tar.gz

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.3.8
-Summary: Production-Grade AI Database Copilot with Responsive Zoom-Out & Wide Table Viewing (CLI)
+Version: 0.4.0
+Summary: Enterprise AI Database Copilot with Interactive Data Grid Window & High-Performance LLM (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/dbagent/__init__.py

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

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/dbagent/cli.py

@@ -60,6 +60,7 @@ from dbagent.ui.console import (
     open_interactive_scroller,
 )
 from dbagent.ui.viewer import SchemaViewer
+from dbagent.ui.grid_window import open_desktop_grid_window, open_browser_grid_window
 
 app = typer.Typer(
     name="db-agent",
@@ -636,6 +637,22 @@ def chat_command(
                     print_results_table(last_result.columns, last_result.rows, title="All Columns (Wide View)", show_all=True)
                     continue
 
+                if cmd_name in [":window", ":gui", ":grid", ":w"]:
+                    if not last_result or not last_result.rows:
+                        print_warning("No previous query results to display.")
+                        continue
+                    print_info("Opening desktop interactive window with horizontal & vertical scrollbars...")
+                    open_desktop_grid_window(last_result.columns, last_result.rows, title=f"DB-Agent Data Grid")
+                    continue
+
+                if cmd_name in [":web", ":browser", ":html"]:
+                    if not last_result or not last_result.rows:
+                        print_warning("No previous query results to display.")
+                        continue
+                    print_info("Opening browser interactive data table...")
+                    open_browser_grid_window(last_result.columns, last_result.rows, title=f"DB-Agent Data View")
+                    continue
+
                 if cmd_name in [":scroll", ":scroller", ":pager"]:
                     if not last_result or not last_result.rows:
                         print_warning("No previous query results to scroll.")

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/dbagent/llm/gemini_provider.py

@@ -1,6 +1,6 @@
 """
 Google Gemini Free Tier LLM Provider.
-Uses Gemini 2.5 Flash / Gemini 2.0 Flash / Gemini 1.5 Flash via direct HTTP API.
+Uses gemini-3.6-flash (current primary), gemini-3.7-flash, gemini-3.5-flash, gemini-flash-latest.
 """
 
 import json
@@ -14,13 +14,15 @@ class GeminiProvider(BaseLLMProvider):
     """Google Gemini LLM provider (Free Tier available)."""
 
     FALLBACK_MODELS = [
+        "gemini-3.6-flash",
+        "gemini-3.7-flash",
+        "gemini-3.5-flash",
+        "gemini-flash-latest",
+        "gemini-pro-latest",
         "gemini-2.5-flash",
-        "gemini-2.0-flash",
-        "gemini-1.5-flash",
-        "gemini-1.5-pro",
     ]
 
-    def __init__(self, api_key: Optional[str] = None, default_model: str = "gemini-2.5-flash"):
+    def __init__(self, api_key: Optional[str] = None, default_model: str = "gemini-3.6-flash"):
         self.api_key = api_key or os.getenv("GEMINI_API_KEY")
         self.default_model = default_model
 
@@ -82,7 +84,7 @@ class GeminiProvider(BaseLLMProvider):
 
             try:
                 if stream_callback:
-                    response = requests.post(endpoint, json=payload, stream=True, timeout=15)
+                    response = requests.post(endpoint, json=payload, stream=True, timeout=10)
                     if response.status_code == 200:
                         full_text = []
                         for line in response.iter_lines():
@@ -102,13 +104,13 @@ class GeminiProvider(BaseLLMProvider):
                                     except Exception:
                                         pass
                         return "".join(full_text)
-                    elif response.status_code in (404, 400):
-                        last_error = f"Model {target_model} not available ({response.status_code})"
+                    elif response.status_code in (404, 400, 503, 429):
+                        last_error = f"Model {target_model} status {response.status_code}"
                         continue
                     else:
                         raise RuntimeError(f"Gemini API error ({response.status_code}): {response.text}")
                 else:
-                    response = requests.post(endpoint, json=payload, timeout=15)
+                    response = requests.post(endpoint, json=payload, timeout=10)
                     if response.status_code == 200:
                         data = response.json()
                         candidates = data.get("candidates", [])
@@ -116,13 +118,13 @@ class GeminiProvider(BaseLLMProvider):
                             parts = candidates[0].get("content", {}).get("parts", [])
                             return "".join(p.get("text", "") for p in parts)
                         return ""
-                    elif response.status_code in (404, 400):
-                        last_error = f"Model {target_model} not available ({response.status_code})"
+                    elif response.status_code in (404, 400, 503, 429):
+                        last_error = f"Model {target_model} status {response.status_code}"
                         continue
                     else:
                         raise RuntimeError(f"Gemini API error ({response.status_code}): {response.text}")
             except requests.exceptions.Timeout:
-                last_error = f"Model {target_model} request timed out"
+                last_error = f"Model {target_model} timed out"
                 continue
             except Exception as e:
                 last_error = str(e)

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/dbagent/ui/console.py

@@ -113,7 +113,7 @@ def print_results_table(
 
     if total_pages > 1 and not show_all:
         console.print(
-            f"[dim]⚡ [bold]:page <1-{total_pages}>[/bold] to switch page | [bold]:wide[/bold] / zoom-out for all columns | [bold]:card[/bold] for vertical view | [bold]:scroll[/bold] for scroller[/dim]"
+            f"[dim]⚡ [bold]:window[/bold] (GUI scroll window) | [bold]:page <1-{total_pages}>[/bold] | [bold]:card[/bold] (vertical) | [bold]:wide[/bold] (zoom-out)[/dim]"
         )
     console.print(f"[bold green]Total rows:[/bold green] {len(rows)}\n")
 

dbagent_cli-0.4.0/dbagent/ui/grid_window.py

@@ -0,0 +1,245 @@
+"""
+Interactive Desktop and Browser Data Grid Viewers with Horizontal & Vertical Scrollbars.
+"""
+
+import os
+import tempfile
+import webbrowser
+import threading
+from typing import List, Dict, Any, Optional
+
+
+def open_desktop_grid_window(
+    columns: List[str],
+    rows: List[Dict[str, Any]],
+    title: str = "DB-Agent Interactive Data Viewer",
+) -> None:
+    """
+    Opens a native desktop GUI window with smooth horizontal and vertical scrollbars,
+    search/filter bar, and resizable columns. Runs in a separate thread.
+    """
+    if not columns or not rows:
+        return
+
+    def _launch_gui():
+        try:
+            import tkinter as tk
+            from tkinter import ttk, messagebox, filedialog
+            import csv
+
+            root = tk.Tk()
+            root.title(f"{title} - ({len(rows)} rows, {len(columns)} columns)")
+            root.geometry("1100x650")
+            root.minsize(700, 400)
+
+            # Modern Styling
+            style = ttk.Style()
+            try:
+                style.theme_use("clam")
+            except Exception:
+                pass
+
+            style.configure("Treeview.Heading", font=("Segoe UI", 10, "bold"), background="#2c3e50", foreground="white")
+            style.configure("Treeview", font=("Segoe UI", 9), rowheight=24)
+            style.map("Treeview", background=[("selected", "#3498db")])
+
+            # Top toolbar (Search & Export)
+            top_frame = ttk.Frame(root, padding=8)
+            top_frame.pack(fill=tk.X, side=tk.TOP)
+
+            ttk.Label(top_frame, text="🔍 Search / Filter:", font=("Segoe UI", 9, "bold")).pack(side=tk.LEFT, padx=(0, 5))
+            search_var = tk.StringVar()
+            search_entry = ttk.Entry(top_frame, textvariable=search_var, width=30)
+            search_entry.pack(side=tk.LEFT, padx=5)
+
+            status_lbl = ttk.Label(top_frame, text=f"Total: {len(rows)} records, {len(columns)} columns", font=("Segoe UI", 9))
+            status_lbl.pack(side=tk.LEFT, padx=15)
+
+            def _export_csv():
+                fpath = filedialog.asksaveasfilename(
+                    defaultextension=".csv",
+                    filetypes=[("CSV Files", "*.csv"), ("All Files", "*.*")],
+                    initialfile="query_results.csv",
+                )
+                if fpath:
+                    try:
+                        with open(fpath, "w", newline="", encoding="utf-8") as f:
+                            writer = csv.writer(f)
+                            writer.writerow(columns)
+                            for r in rows:
+                                writer.writerow([r.get(c, "") for c in columns])
+                        messagebox.showinfo("Export Success", f"Exported {len(rows)} rows to:\n{fpath}")
+                    except Exception as e:
+                        messagebox.showerror("Export Failed", str(e))
+
+            export_btn = ttk.Button(top_frame, text="💾 Export to CSV", command=_export_csv)
+            export_btn.pack(side=tk.RIGHT, padx=5)
+
+            # Center Data Grid Frame with Both Scrollbars
+            grid_frame = ttk.Frame(root)
+            grid_frame.pack(fill=tk.BOTH, expand=True, padx=8, pady=(0, 8))
+
+            # Vertical Scrollbar
+            v_scroll = ttk.Scrollbar(grid_frame, orient=tk.VERTICAL)
+            v_scroll.pack(side=tk.RIGHT, fill=tk.Y)
+
+            # Horizontal Scrollbar
+            h_scroll = ttk.Scrollbar(grid_frame, orient=tk.HORIZONTAL)
+            h_scroll.pack(side=tk.BOTTOM, fill=tk.X)
+
+            # Treeview Table
+            tree = ttk.Treeview(
+                grid_frame,
+                columns=columns,
+                show="headings",
+                yscrollcommand=v_scroll.set,
+                xscrollcommand=h_scroll.set,
+            )
+            v_scroll.config(command=tree.yview)
+            h_scroll.config(command=tree.xview)
+            tree.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
+
+            # Configure Columns
+            for col in columns:
+                tree.heading(col, text=col)
+                # Dynamic column width based on column name length
+                col_w = max(120, min(280, len(str(col)) * 12))
+                tree.column(col, width=col_w, minwidth=80, stretch=False)
+
+            def populate_table(filter_text=""):
+                tree.delete(*tree.get_children())
+                f_lower = filter_text.strip().lower()
+                count = 0
+                for r in rows:
+                    row_vals = [str(r.get(c, "")) if r.get(c) is not None else "NULL" for c in columns]
+                    if f_lower:
+                        if not any(f_lower in v.lower() for v in row_vals):
+                            continue
+                    tree.insert("", tk.END, values=row_vals)
+                    count += 1
+                status_lbl.config(text=f"Showing: {count} of {len(rows)} rows ({len(columns)} columns)")
+
+            populate_table()
+
+            def on_search_change(*args):
+                populate_table(search_var.get())
+
+            search_var.trace_add("write", on_search_change)
+
+            root.mainloop()
+        except Exception as e:
+            print(f"[DB-Agent GUI Error] Could not launch desktop window: {e}")
+
+    # Launch GUI in background thread so CLI prompt stays responsive
+    thread = threading.Thread(target=_launch_gui, daemon=True)
+    thread.start()
+
+
+def open_browser_grid_window(
+    columns: List[str],
+    rows: List[Dict[str, Any]],
+    title: str = "DB-Agent Interactive Data Grid",
+) -> None:
+    """
+    Generates a standalone, ultra-fast HTML data grid with frozen headers,
+    instant horizontal & vertical scrolling, search, and CSV export, then opens it in browser.
+    """
+    if not columns or not rows:
+        return
+
+    import json
+
+    cols_json = json.dumps([{"title": c, "field": c, "headerFilter": "input", "resizable": True} for c in columns])
+    data_json = json.dumps(rows, default=str)
+
+    html_content = f"""<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <title>{title}</title>
+  <link href="https://unpkg.com/tabulator-tables@5.5.0/dist/css/tabulator_midnight.min.css" rel="stylesheet">
+  <script type="text/javascript" src="https://unpkg.com/tabulator-tables@5.5.0/dist/js/tabulator.min.js"></script>
+  <style>
+    body {{
+      margin: 0;
+      padding: 16px;
+      font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
+      background: #0f172a;
+      color: #f8fafc;
+    }}
+    .header {{
+      display: flex;
+      justify-content: space-between;
+      align-items: center;
+      margin-bottom: 12px;
+    }}
+    h1 {{
+      margin: 0;
+      font-size: 1.3rem;
+      color: #38bdf8;
+    }}
+    .badge {{
+      background: #1e293b;
+      padding: 4px 10px;
+      border-radius: 6px;
+      font-size: 0.85rem;
+      color: #94a3b8;
+      border: 1px solid #334155;
+    }}
+    .btn {{
+      background: #0284c7;
+      color: white;
+      border: none;
+      padding: 6px 14px;
+      border-radius: 6px;
+      font-weight: 600;
+      cursor: pointer;
+    }}
+    .btn:hover {{
+      background: #0369a1;
+    }}
+    #data-table {{
+      height: calc(100vh - 85px);
+      background: #1e293b;
+      border-radius: 8px;
+      border: 1px solid #334155;
+    }}
+  </style>
+</head>
+<body>
+  <div class="header">
+    <div>
+      <h1>📊 {title}</h1>
+      <span class="badge">{len(rows)} Records &bull; {len(columns)} Columns</span>
+    </div>
+    <div>
+      <button class="btn" onclick="table.download('csv', 'data_export.csv')">💾 Download CSV</button>
+    </div>
+  </div>
+  <div id="data-table"></div>
+
+  <script>
+    const tableData = {data_json};
+    const tableCols = {cols_json};
+
+    const table = new Tabulator("#data-table", {{
+      data: tableData,
+      columns: tableCols,
+      layout: "fitData",
+      pagination: false,
+      renderVertical: "virtual",
+      renderHorizontal: "virtual",
+      movableColumns: true,
+      responsiveLayout: false,
+    }});
+  </script>
+</body>
+</html>
+"""
+
+    temp_dir = tempfile.gettempdir()
+    temp_path = os.path.join(temp_dir, "dbagent_data_grid.html")
+    with open(temp_path, "w", encoding="utf-8") as f:
+        f.write(html_content)
+
+    webbrowser.open(f"file://{temp_path}")

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/dbagent_cli.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.3.8
-Summary: Production-Grade AI Database Copilot with Responsive Zoom-Out & Wide Table Viewing (CLI)
+Version: 0.4.0
+Summary: Enterprise AI Database Copilot with Interactive Data Grid Window & High-Performance LLM (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/dbagent_cli.egg-info/SOURCES.txt

@@ -49,6 +49,7 @@ dbagent/sql/__init__.py
 dbagent/sql/optimizer.py
 dbagent/sql/validator.py
 dbagent/ui/console.py
+dbagent/ui/grid_window.py
 dbagent/ui/viewer.py
 dbagent/ui/visualizer.py
 dbagent_cli.egg-info/PKG-INFO

{dbagent_cli-0.3.8 → dbagent_cli-0.4.0}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.3.8"
-description = "Production-Grade AI Database Copilot with Responsive Zoom-Out & Wide Table Viewing (CLI)"
+version = "0.4.0"
+description = "Enterprise AI Database Copilot with Interactive Data Grid Window & High-Performance LLM (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"
 authors = [