Update app.py

app.py

@@ -1,162 +1,76 @@
 import os
 import io
-import sys
 import gradio as gr
 from pathlib import Path
 from rich.console import Console
 
-# Import your custom modules
+# Import your modules
 from cli_app.command_handler import handle_command
 import cli_app.command_handler as handler
-from utils.logger import logger
 from vector_store.instruction_embedder import embed_analyze_instructions
 
-# Global console for local CLI use
-console = Console()
-
 def terminal_logic(command):
-    """
-    Captures Rich console output and detects if an image was generated.
-    Returns: (text_output, image_path)
-    """
+    if not command: return "", None
     output_buffer = io.StringIO()
-    temp_console = Console(
-        file=output_buffer, 
-        force_terminal=True, 
-        width=100, 
-        color_system=None, 
-        legacy_windows=False
-    )
+    temp_console = Console(file=output_buffer, force_terminal=True, width=100, color_system=None)
     
     old_console = getattr(handler, 'console', None)
     handler.console = temp_console 
 
     try:
-        if command.lower() == "exit":
-            return "Session ended. Refresh to restart.", None
-
-        # Execute logic
         result = handle_command(command)
         captured = output_buffer.getvalue()
+        combined = f"{captured}\n{str(result) if result else ''}".strip()
         
-        # --- PLOT DETECTION LOGIC ---
-        # Look for a path in the result or captured text (e.g., "output/plot.png")
-        # Adjust the 'output' string to match where your tool saves PNGs
         image_path = None
-        if ".png" in captured or (result and ".png" in result):
-            # Simple heuristic: find a word ending in .png
-            words = (captured + (result or "")).split()
-            for word in words:
-                if word.endswith(".png") and os.path.exists(word):
-                    image_path = word
+        if ".png" in combined:
+            for word in combined.split():
+                clean = word.strip(".,![]'\"")
+                if clean.endswith(".png") and os.path.exists(clean):
+                    image_path = clean
                     break
-
-        full_text = f"{captured}\n{result if result else ''}".strip()
-        return full_text or "Command executed.", image_path
-
+        return f"```\n{combined}\n
+```", image_path
     except Exception as e:
-        logger.error(f"Space Terminal Error: {e}")
         return f"Error: {str(e)}", None
     finally:
-        if old_console:
-            handler.console = old_console
+        handler.console = old_console
 
 def run_space_interface():
-    # 1. Initialize Vector Store instructions
     embed_analyze_instructions()
-
-    # 2. AUTO-HYDRATE DATA (Detects Parquet and Pickles)
-    # Looking in data/datasets/ because that's where your sync folder puts them
+    
+    # Auto-hydrate
     data_dir = Path("data/datasets")
-    found_data = False
-
     if data_dir.exists():
         for file in os.listdir(data_dir):
             if file.endswith((".parquet", ".pkl")) and "embeddings" not in file:
-                try:
-                    # Simulates 'load data/datasets/filename'
-                    handler.handle_command(f"load {data_dir / file}")
-                    found_data = True
-                except:
-                    pass
+                try: handler.handle_command(f"load {data_dir / file}")
+                except: continue
 
-    # 3. Generate Help Table for Welcome Screen
-    help_buffer = io.StringIO()
-    help_console = Console(file=help_buffer, force_terminal=True, width=100, color_system=None)
-    
-    old_console = getattr(handler, 'console', None)
-    handler.console = help_console
-    handler.handle_command("help")
-    welcome_table = help_buffer.getvalue()
-    handler.console = old_console
-
-    status_msg = "✅ Datasets Ready" if found_data else "⚠️ No Datasets Found. Use 'load <path>'"
-    welcome_text = f"WELCOME TO EDA EXPLORER\n{status_msg}\n{'='*25}\n{welcome_table}"
-
-    # 4. Build the Modern Terminal UI
-    with gr.Blocks(title="EDA Explorer Terminal", theme="monochrome") as demo:
+    with gr.Blocks(theme="monochrome", analytics_enabled=False) as demo:
         gr.Markdown("# 📊 EDA Explorer Terminal")
         
         with gr.Row():
-            with gr.Column(scale=3):
-                output_box = gr.Code(
-                    value=welcome_text,
-                    label="Console Output", 
-                    language="markdown",
-                    lines=25
-                )
-            with gr.Column(scale=2):
-                plot_output = gr.Image(
-                    label="Visual Analysis", 
-                    type="filepath",
-                    interactive=False
-                )
+            # Markdown is significantly safer than gr.Code for preventing schema crashes
+            output_box = gr.Markdown(value="```\nTerminal Ready\n```")
+            plot_output = gr.Image(label="Visual Analysis", type="filepath")
         
-        input_box = gr.Textbox(
-            label="Command Line", 
-            placeholder="e.g., analyze titanic or histogram Age in titanic", 
-            autofocus=True
-        )
+        input_box = gr.Textbox(label="Command Line")
 
         gr.Examples(
-            examples=[
-                ["list"], 
-                ["analyze titanic"], 
-                ["histogram Age in titanic"], 
-                ["info Customer_Churn"]
-            ],
-            inputs=input_box,
-            label="Quick Actions"
+            examples=[["list"], ["NL: show top 10 rows in titanic"], ["NL: average Age in titanic"]],
+            inputs=input_box
         )
 
-        # Submit triggers the logic and updates BOTH the text and the image
+        # api_name=False is the crucial fix for the "bool is not iterable" crawler error
         input_box.submit(
             fn=terminal_logic, 
             inputs=input_box, 
-            outputs=[output_box, plot_output]
+            outputs=[output_box, plot_output],
+            api_name=False
         )
         
-    demo.launch(server_name="0.0.0.0", server_port=7860)
-
-def run_cli():
-    """Standard local terminal loop."""
-    embed_analyze_instructions() 
-    console.print("\n[bold cyan]EDA Explorer CLI[/bold cyan]")
-    while True:
-        try:
-            cmd = console.input("[bold yellow]> [/bold yellow]")
-            if cmd.lower() == "exit": break
-            result = handle_command(cmd)
-            if result: console.print(result)
-        except (KeyboardInterrupt, EOFError): break
-        except Exception as e: console.print(f"[red]Error:[/red] {e}")
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)
 
 if __name__ == "__main__":
-    if "SPACE_ID" in os.environ:
-        run_space_interface()
-    else:
-        run_cli()
-
-
-
-
+    run_space_interface()