Update app.py

app.py

@@ -1,161 +1,161 @@
-# import os
-# import io
-# import sys
-# import gradio as gr
-# from pathlib import Path
-# from rich.console import Console
-
-# # Import your custom 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)
-#     """
-#     output_buffer = io.StringIO()
-#     temp_console = Console(
-#         file=output_buffer, 
-#         force_terminal=True, 
-#         width=100, 
-#         color_system=None, 
-#         legacy_windows=False
-#     )
+import os
+import io
+import sys
+import gradio as gr
+from pathlib import Path
+from rich.console import Console
+
+# Import your custom 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)
+    """
+    output_buffer = io.StringIO()
+    temp_console = Console(
+        file=output_buffer, 
+        force_terminal=True, 
+        width=100, 
+        color_system=None, 
+        legacy_windows=False
+    )
     
-#     old_console = getattr(handler, 'console', None)
-#     handler.console = temp_console 
+    old_console = getattr(handler, 'console', None)
+    handler.console = temp_console 
 
-#     try:
-#         if command.lower() == "exit":
-#             return "Session ended. Refresh to restart.", None
+    try:
+        if command.lower() == "exit":
+            return "Session ended. Refresh to restart.", None
 
-#         # Execute logic
-#         result = handle_command(command)
-#         captured = output_buffer.getvalue()
+        # Execute logic
+        result = handle_command(command)
+        captured = output_buffer.getvalue()
         
-#         # --- 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
-#                     break
-
-#         full_text = f"{captured}\n{result if result else ''}".strip()
-#         return full_text or "Command executed.", 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
-
-# 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
-#     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
-
-#     # 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)
+        # --- 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
+                    break
+
+        full_text = f"{captured}\n{result if result else ''}".strip()
+        return full_text or "Command executed.", 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
+
+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
+    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
+
+    # 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:
-#         gr.Markdown("# 📊 EDA Explorer Terminal")
+    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:
+        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
-#                 )
+        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
+                )
         
-#         input_box = gr.Textbox(
-#             label="Command Line", 
-#             placeholder="e.g., analyze titanic or histogram Age in titanic", 
-#             autofocus=True
-#         )
-
-#         gr.Examples(
-#             examples=[
-#                 ["list"], 
-#                 ["analyze titanic"], 
-#                 ["histogram Age in titanic"], 
-#                 ["info Customer_Churn"]
-#             ],
-#             inputs=input_box,
-#             label="Quick Actions"
-#         )
-
-#         # Submit triggers the logic and updates BOTH the text and the image
-#         input_box.submit(
-#             fn=terminal_logic, 
-#             inputs=input_box, 
-#             outputs=[output_box, plot_output]
-#         )
+        input_box = gr.Textbox(
+            label="Command Line", 
+            placeholder="e.g., analyze titanic or histogram Age in titanic", 
+            autofocus=True
+        )
+
+        gr.Examples(
+            examples=[
+                ["list"], 
+                ["analyze titanic"], 
+                ["histogram Age in titanic"], 
+                ["info Customer_Churn"]
+            ],
+            inputs=input_box,
+            label="Quick Actions"
+        )
+
+        # Submit triggers the logic and updates BOTH the text and the image
+        input_box.submit(
+            fn=terminal_logic, 
+            inputs=input_box, 
+            outputs=[output_box, plot_output]
+        )
         
-#     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}")
-
-# if __name__ == "__main__":
-#     if "SPACE_ID" in os.environ:
-#         run_space_interface()
-#     else:
-#         run_cli()
+    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}")
+
+if __name__ == "__main__":
+    if "SPACE_ID" in os.environ:
+        run_space_interface()
+    else:
+        run_cli()