flowyml 1.6.0__py3-none-any.whl → 1.7.0__py3-none-any.whl

flowyml/cli/main.py

@@ -450,5 +450,381 @@ def logs(run_id: str, step: str, tail: int) -> None:
     click.echo("  [Log output would appear here]")
 
 
+# ============================================================================
+# Quick Commands: flowyml go / stop / status
+# ============================================================================
+
+
+@cli.command()
+@click.option("--host", default="localhost", help="Host to bind to")
+@click.option("--port", default=8080, type=int, help="Port to bind to")
+@click.option("--open-browser", "-o", is_flag=True, help="Open browser automatically")
+def go(host: str, port: int, open_browser: bool) -> None:
+    r"""🚀 Start flowyml - Initialize UI dashboard and show welcome message.
+
+    This is the quickest way to get started with flowyml. It starts the UI
+    dashboard server in the background and displays the URL to access it.
+
+    \b
+    Examples:
+        flowyml go              # Start on default port 8080
+        flowyml go -o           # Start and open browser
+        flowyml go --port 9000  # Start on custom port
+    """
+    import subprocess
+    import sys
+    import time
+    from flowyml.ui.utils import is_ui_running
+
+    try:
+        from rich.console import Console
+        from rich.panel import Panel
+        from rich.text import Text
+        from rich import box
+
+        console = Console()
+        rich_available = True
+    except ImportError:
+        rich_available = False
+
+    url = f"http://{host}:{port}"
+
+    # Check if already running
+    if is_ui_running(host, port):
+        if rich_available:
+            panel_content = Text()
+            panel_content.append("✅ ", style="green")
+            panel_content.append("flowyml is already running!\n\n", style="bold green")
+            panel_content.append("🌐 Dashboard: ", style="bold")
+            panel_content.append(url, style="cyan underline link " + url)
+            panel_content.append("\n\n", style="")
+            panel_content.append("Run ", style="dim")
+            panel_content.append("flowyml stop", style="bold yellow")
+            panel_content.append(" to stop the server.", style="dim")
+
+            console.print(
+                Panel(
+                    panel_content,
+                    title="[bold cyan]🌊 flowyml[/bold cyan]",
+                    border_style="cyan",
+                    box=box.DOUBLE,
+                ),
+            )
+        else:
+            click.echo("✅ flowyml is already running!")
+            click.echo(f"🌐 Dashboard: {url}")
+            click.echo("\nRun 'flowyml stop' to stop the server.")
+
+        if open_browser:
+            import webbrowser
+
+            webbrowser.open(url)
+        return
+
+    # Start the UI server as a background subprocess
+    if rich_available:
+        console.print("[bold cyan]🌊 flowyml[/bold cyan] - Starting up...\n")
+    else:
+        click.echo("🌊 flowyml - Starting up...")
+
+    try:
+        # Start uvicorn as a background process
+        # Using subprocess with nohup-like behavior
+        cmd = [
+            sys.executable,
+            "-m",
+            "uvicorn",
+            "flowyml.ui.backend.main:app",
+            "--host",
+            host,
+            "--port",
+            str(port),
+            "--log-level",
+            "warning",
+        ]
+
+        # Start as detached background process
+        if sys.platform == "win32":
+            # Windows: use CREATE_NEW_PROCESS_GROUP
+            process = subprocess.Popen(
+                cmd,
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                creationflags=subprocess.CREATE_NEW_PROCESS_GROUP | subprocess.DETACHED_PROCESS,
+            )
+        else:
+            # Unix: use start_new_session
+            process = subprocess.Popen(
+                cmd,
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                start_new_session=True,
+            )
+
+        # Wait for server to start (up to 8 seconds)
+        started = False
+        for _ in range(80):
+            time.sleep(0.1)
+            if is_ui_running(host, port):
+                started = True
+                break
+
+        if started:
+            # Save PID for later stop command
+            pid_file = Path.home() / ".flowyml" / "ui_server.pid"
+            pid_file.parent.mkdir(parents=True, exist_ok=True)
+            pid_file.write_text(f"{process.pid}\n{host}\n{port}")
+
+            if rich_available:
+                panel_content = Text()
+                panel_content.append("✅ ", style="green")
+                panel_content.append("flowyml is ready!\n\n", style="bold green")
+                panel_content.append("🌐 Dashboard: ", style="bold")
+                panel_content.append(url, style="cyan underline link " + url)
+                panel_content.append("\n\n", style="")
+                panel_content.append("📊 View pipelines: ", style="")
+                panel_content.append(f"{url}/pipelines", style="cyan")
+                panel_content.append("\n", style="")
+                panel_content.append("📜 View runs: ", style="")
+                panel_content.append(f"{url}/runs", style="cyan")
+                panel_content.append("\n\n", style="")
+                panel_content.append("Run ", style="dim")
+                panel_content.append("flowyml stop", style="bold yellow")
+                panel_content.append(" to stop the server.", style="dim")
+
+                console.print(
+                    Panel(
+                        panel_content,
+                        title="[bold cyan]🌊 flowyml[/bold cyan]",
+                        border_style="green",
+                        box=box.DOUBLE,
+                    ),
+                )
+
+                console.print()
+                console.print("[dim]Tip: The dashboard runs in the background. Your pipelines will[/dim]")
+                console.print("[dim]automatically show a clickable URL when they run.[/dim]")
+            else:
+                click.echo("✅ flowyml is ready!")
+                click.echo(f"🌐 Dashboard: {url}")
+                click.echo(f"📊 View pipelines: {url}/pipelines")
+                click.echo(f"📜 View runs: {url}/runs")
+                click.echo("\nRun 'flowyml stop' to stop the server.")
+                click.echo("\nTip: The dashboard runs in the background. Your pipelines will")
+                click.echo("automatically show a clickable URL when they run.")
+
+            if open_browser:
+                import webbrowser
+
+                webbrowser.open(url)
+        else:
+            # Server didn't start, kill the process
+            process.terminate()
+            raise RuntimeError("Server failed to start within timeout")
+
+    except Exception as e:
+        if rich_available:
+            panel_content = Text()
+            panel_content.append("❌ ", style="red")
+            panel_content.append("Failed to start flowyml UI server.\n\n", style="bold red")
+            panel_content.append(f"Error: {str(e)[:100]}\n\n", style="dim red")
+            panel_content.append("Possible issues:\n", style="")
+            panel_content.append(f"  • Port {port} might be in use\n", style="dim")
+            panel_content.append("  • Missing dependencies (uvicorn, fastapi)\n", style="dim")
+            panel_content.append("\n", style="")
+            panel_content.append("Try:\n", style="")
+            panel_content.append(f"  flowyml go --port {port + 1}", style="bold yellow")
+            panel_content.append("  (use different port)\n", style="dim")
+            panel_content.append("  flowyml ui start", style="bold yellow")
+            panel_content.append("  (for verbose output)", style="dim")
+
+            console.print(
+                Panel(
+                    panel_content,
+                    title="[bold red]Error[/bold red]",
+                    border_style="red",
+                    box=box.ROUNDED,
+                ),
+            )
+        else:
+            click.echo(f"❌ Failed to start flowyml UI server: {e}")
+            click.echo("Possible issues:")
+            click.echo(f"  • Port {port} might be in use")
+            click.echo("  • Missing dependencies (uvicorn, fastapi)")
+            click.echo(f"\nTry: flowyml go --port {port + 1}")
+            click.echo("Or run 'flowyml ui start' for verbose output.")
+
+
+@cli.command("stop")
+@click.option("--host", default="localhost", help="Host of the server")
+@click.option("--port", default=8080, type=int, help="Port of the server")
+def stop_server(host: str, port: int) -> None:
+    r"""🛑 Stop flowyml - Shutdown the UI dashboard server.
+
+    Stops the flowyml UI server if it's running.
+
+    \b
+    Examples:
+        flowyml stop              # Stop server on default port
+        flowyml stop --port 9000  # Stop server on custom port
+    """
+    import os
+    import signal
+    import time
+    from flowyml.ui.utils import is_ui_running
+
+    try:
+        from rich.console import Console
+        from rich.panel import Panel
+        from rich.text import Text
+        from rich import box
+
+        console = Console()
+        rich_available = True
+    except ImportError:
+        rich_available = False
+
+    pid_file = Path.home() / ".flowyml" / "ui_server.pid"
+
+    # First check if we have a PID file from 'flowyml go'
+    if pid_file.exists():
+        try:
+            content = pid_file.read_text().strip().split("\n")
+            pid = int(content[0])
+            # Note: saved_host and saved_port are in the file but we use the CLI args
+            # to allow stopping a server on a different port if needed
+
+            # Try to kill the process
+            try:
+                os.kill(pid, signal.SIGTERM)
+                time.sleep(0.5)
+
+                # Clean up PID file
+                pid_file.unlink(missing_ok=True)
+
+                if rich_available:
+                    console.print(f"[green]✅ flowyml server (PID {pid}) stopped successfully.[/green]")
+                else:
+                    click.echo(f"✅ flowyml server (PID {pid}) stopped successfully.")
+                return
+            except ProcessLookupError:
+                # Process already dead, clean up PID file
+                pid_file.unlink(missing_ok=True)
+            except PermissionError:
+                if rich_available:
+                    console.print(f"[red]❌ Permission denied to stop process {pid}[/red]")
+                else:
+                    click.echo(f"❌ Permission denied to stop process {pid}")
+                return
+        except (ValueError, IndexError):
+            # Invalid PID file, remove it
+            pid_file.unlink(missing_ok=True)
+
+    # Check if server is running
+    if not is_ui_running(host, port):
+        if rich_available:
+            console.print(f"[yellow]ℹ️  No flowyml server running on {host}:{port}[/yellow]")
+        else:
+            click.echo(f"ℹ️  No flowyml server running on {host}:{port}")
+        return
+
+    # Server is running but we don't have a PID file - must be from 'flowyml ui start'
+    if rich_available:
+        panel_content = Text()
+        panel_content.append("ℹ️  ", style="yellow")
+        panel_content.append("Server running but not started with 'flowyml go'.\n\n", style="")
+        panel_content.append("To stop it:\n", style="")
+        panel_content.append("  • If running in foreground: ", style="dim")
+        panel_content.append("Press Ctrl+C\n", style="bold")
+        panel_content.append("  • Find and kill: ", style="dim")
+        panel_content.append(f"pkill -f 'uvicorn.*:{port}'\n", style="bold")
+        panel_content.append("  • Or find PID: ", style="dim")
+        panel_content.append(f"lsof -i :{port}", style="bold")
+
+        console.print(
+            Panel(
+                panel_content,
+                title="[bold yellow]Manual Stop Required[/bold yellow]",
+                border_style="yellow",
+                box=box.ROUNDED,
+            ),
+        )
+    else:
+        click.echo("ℹ️  Server running but not started with 'flowyml go'.")
+        click.echo("To stop it:")
+        click.echo("  • If running in foreground: Press Ctrl+C")
+        click.echo(f"  • Find and kill: pkill -f 'uvicorn.*:{port}'")
+        click.echo(f"  • Or find PID: lsof -i :{port}")
+
+
+@cli.command("status")
+@click.option("--host", default="localhost", help="Host to check")
+@click.option("--port", default=8080, type=int, help="Port to check")
+def server_status(host: str, port: int) -> None:
+    r"""📊 Check flowyml status - Show if the UI server is running.
+
+    \b
+    Examples:
+        flowyml status              # Check default port
+        flowyml status --port 9000  # Check custom port
+    """
+    from flowyml.ui.utils import is_ui_running
+
+    try:
+        from rich.console import Console
+        from rich.panel import Panel
+        from rich.text import Text
+        from rich import box
+
+        console = Console()
+        rich_available = True
+    except ImportError:
+        rich_available = False
+
+    if is_ui_running(host, port):
+        url = f"http://{host}:{port}"
+        if rich_available:
+            panel_content = Text()
+            panel_content.append("✅ ", style="green")
+            panel_content.append("flowyml is running\n\n", style="bold green")
+            panel_content.append("🌐 Dashboard: ", style="bold")
+            panel_content.append(url, style="cyan underline link " + url)
+            panel_content.append("\n", style="")
+            panel_content.append("💚 Health: ", style="")
+            panel_content.append(f"{url}/api/health", style="dim")
+
+            console.print(
+                Panel(
+                    panel_content,
+                    title="[bold cyan]🌊 flowyml Status[/bold cyan]",
+                    border_style="green",
+                    box=box.ROUNDED,
+                ),
+            )
+        else:
+            click.echo("✅ flowyml is running")
+            click.echo(f"🌐 Dashboard: {url}")
+            click.echo(f"💚 Health: {url}/api/health")
+    else:
+        if rich_available:
+            panel_content = Text()
+            panel_content.append("❌ ", style="red")
+            panel_content.append(f"flowyml is not running on {host}:{port}\n\n", style="")
+            panel_content.append("Start with: ", style="dim")
+            panel_content.append("flowyml go", style="bold cyan")
+
+            console.print(
+                Panel(
+                    panel_content,
+                    title="[bold cyan]🌊 flowyml Status[/bold cyan]",
+                    border_style="red",
+                    box=box.ROUNDED,
+                ),
+            )
+        else:
+            click.echo(f"❌ flowyml is not running on {host}:{port}")
+            click.echo("Start with: flowyml go")
+
+
 if __name__ == "__main__":
     cli()

flowyml/core/display.py

@@ -21,7 +21,15 @@ except ImportError:
 class PipelineDisplay:
     """Beautiful CLI display for pipeline execution."""
 
-    def __init__(self, pipeline_name: str, steps: list[Any], dag: Any, verbose: bool = True):
+    def __init__(
+        self,
+        pipeline_name: str,
+        steps: list[Any],
+        dag: Any,
+        verbose: bool = True,
+        ui_url: str | None = None,
+        run_url: str | None = None,
+    ):
         """Initialize display system.
 
         Args:
@@ -29,11 +37,15 @@ class PipelineDisplay:
             steps: List of step objects
             dag: Pipeline DAG
             verbose: Whether to show detailed output
+            ui_url: Optional base URL for the UI dashboard
+            run_url: Optional URL to view this specific run in the UI
         """
         self.pipeline_name = pipeline_name
         self.steps = steps
         self.dag = dag
         self.verbose = verbose
+        self.ui_url = ui_url
+        self.run_url = run_url
         self.console = Console() if RICH_AVAILABLE else None
         self.step_status = {step.name: "pending" for step in steps}
         self.step_durations = {}
@@ -75,6 +87,9 @@ class PipelineDisplay:
             self.console.print(header)
             self.console.print()
 
+            # Show prominent UI URL if available (so users can click to follow execution)
+            self._show_ui_url_banner()
+
             # DAG visualization
             self._show_dag_rich()
         else:
@@ -83,8 +98,63 @@ class PipelineDisplay:
             print(f"🌊 flowyml Pipeline: {self.pipeline_name}")
             print("=" * 70)
             print()
+            # Show UI URL in simple mode too
+            self._show_ui_url_simple()
             self._show_dag_simple()
 
+    def _show_ui_url_banner(self) -> None:
+        """Show a prominent UI URL banner with clickable link (Rich mode)."""
+        if not RICH_AVAILABLE or not self.console:
+            return
+
+        if self.run_url:
+            # Create a prominent banner with the run URL
+            url_content = Text()
+            url_content.append("🌐 ", style="bold cyan")
+            url_content.append("Dashboard: ", style="bold white")
+            url_content.append(self.run_url, style="bold cyan underline link " + self.run_url)
+            url_content.append("\n", style="")
+            url_content.append("   ", style="")
+            url_content.append("↑ Click to follow pipeline execution in real-time", style="dim italic")
+
+            ui_panel = Panel(
+                url_content,
+                border_style="green",
+                box=box.DOUBLE,
+                title="[bold green]✨ Live Dashboard[/bold green]",
+                title_align="left",
+            )
+            self.console.print(ui_panel)
+            self.console.print()
+        elif self.ui_url:
+            # Show base UI URL if no specific run URL
+            url_content = Text()
+            url_content.append("🌐 ", style="bold cyan")
+            url_content.append("Dashboard: ", style="bold white")
+            url_content.append(self.ui_url, style="bold cyan underline link " + self.ui_url)
+
+            ui_panel = Panel(
+                url_content,
+                border_style="cyan",
+                box=box.ROUNDED,
+                title="[bold cyan]UI Available[/bold cyan]",
+                title_align="left",
+            )
+            self.console.print(ui_panel)
+            self.console.print()
+
+    def _show_ui_url_simple(self) -> None:
+        """Show UI URL in simple text mode."""
+        if self.run_url:
+            print("=" * 70)
+            print(f"🌐 Dashboard: {self.run_url}")
+            print("   ↑ Open this URL to follow pipeline execution in real-time")
+            print("=" * 70)
+            print()
+        elif self.ui_url:
+            print(f"🌐 UI Available: {self.ui_url}")
+            print()
+
     def _show_dag_rich(self) -> None:
         """Show DAG using rich."""
         if not self.dag:

flowyml/core/pipeline.py

@@ -451,22 +451,9 @@ class Pipeline:
         # Auto-start UI server if requested
         ui_url = None
         run_url = None
+        ui_start_failed = False
         if auto_start_ui:
-            try:
-                from flowyml.ui.server_manager import UIServerManager
-                from flowyml.ui.utils import get_ui_host_port
-
-                ui_manager = UIServerManager.get_instance()
-                # Use config values for host/port
-                host, port = get_ui_host_port()
-                if ui_manager.ensure_running(host=host, port=port, auto_start=True):
-                    ui_url = ui_manager.get_url()
-                    run_url = ui_manager.get_run_url(run_id)
-
-                    # UI URL will be shown in summary, no need to print here
-            except Exception:
-                # Silently fail if UI is not available
-                pass
+            ui_url, run_url, ui_start_failed = self._ensure_ui_server(run_id)
 
         # Determine stack for this run
         if stack is not None:
@@ -511,6 +498,8 @@ class Pipeline:
                 steps=self.steps,
                 dag=self.dag,
                 verbose=True,
+                ui_url=ui_url,  # Pass UI URL for prominent display at start
+                run_url=run_url,  # Pass run-specific URL for clickable link
             )
             display.show_header()
             display.show_execution_start()
@@ -594,6 +583,163 @@ class Pipeline:
             # Don't fail the run if definition saving fails
             print(f"Warning: Failed to save pipeline definition: {e}")
 
+    def _ensure_ui_server(self, run_id: str) -> tuple[str | None, str | None, bool]:
+        """Ensure UI server is running, start it if needed, or show guidance.
+
+        Args:
+            run_id: The run ID for generating the run URL
+
+        Returns:
+            Tuple of (ui_url, run_url, start_failed)
+            - ui_url: Base URL of the UI server if running
+            - run_url: URL to view this specific run if server is running
+            - start_failed: True if we tried to start and failed (show guidance)
+        """
+        import subprocess
+        import sys
+        import time
+        from pathlib import Path
+
+        try:
+            from flowyml.ui.utils import is_ui_running, get_ui_host_port
+        except ImportError:
+            return None, None, False
+
+        host, port = get_ui_host_port()
+        url = f"http://{host}:{port}"
+
+        # Check if already running
+        if is_ui_running(host, port):
+            return url, f"{url}/runs/{run_id}", False
+
+        # Try to start the UI server as a background subprocess
+        try:
+            # Check if uvicorn is available
+            try:
+                import uvicorn  # noqa: F401
+            except ImportError:
+                # uvicorn not installed, show guidance but don't fail
+                self._show_ui_guidance(host, port, reason="missing_deps")
+                return None, None, True
+
+            # Start uvicorn as a background process
+            cmd = [
+                sys.executable,
+                "-m",
+                "uvicorn",
+                "flowyml.ui.backend.main:app",
+                "--host",
+                host,
+                "--port",
+                str(port),
+                "--log-level",
+                "warning",
+            ]
+
+            # Start as detached background process
+            if sys.platform == "win32":
+                process = subprocess.Popen(
+                    cmd,
+                    stdout=subprocess.DEVNULL,
+                    stderr=subprocess.DEVNULL,
+                    creationflags=subprocess.CREATE_NEW_PROCESS_GROUP | subprocess.DETACHED_PROCESS,
+                )
+            else:
+                process = subprocess.Popen(
+                    cmd,
+                    stdout=subprocess.DEVNULL,
+                    stderr=subprocess.DEVNULL,
+                    start_new_session=True,
+                )
+
+            # Wait for server to start (up to 8 seconds)
+            started = False
+            for _ in range(80):
+                time.sleep(0.1)
+                if is_ui_running(host, port):
+                    started = True
+                    break
+
+            if started:
+                # Save PID for later stop command
+                pid_file = Path.home() / ".flowyml" / "ui_server.pid"
+                pid_file.parent.mkdir(parents=True, exist_ok=True)
+                pid_file.write_text(f"{process.pid}\n{host}\n{port}")
+
+                return url, f"{url}/runs/{run_id}", False
+            else:
+                # Server didn't start, kill the process and show guidance
+                process.terminate()
+                self._show_ui_guidance(host, port, reason="start_failed")
+                return None, None, True
+
+        except Exception:
+            # Show guidance on failure
+            self._show_ui_guidance(host, port, reason="error")
+            return None, None, True
+
+    def _show_ui_guidance(self, host: str, port: int, reason: str = "not_running") -> None:
+        """Show a helpful message guiding the user to start the UI server.
+
+        Args:
+            host: Host the server should run on
+            port: Port the server should run on
+            reason: Why we're showing guidance (not_running, missing_deps, start_failed, error)
+        """
+        try:
+            from rich.console import Console
+            from rich.panel import Panel
+            from rich.text import Text
+            from rich import box
+
+            console = Console()
+
+            content = Text()
+            content.append("💡 ", style="yellow")
+            content.append("Want to see your pipeline run in a live dashboard?\n\n", style="bold")
+
+            if reason == "missing_deps":
+                content.append("UI dependencies not installed. ", style="dim")
+                content.append("Install with:\n", style="")
+                content.append("  pip install uvicorn fastapi\n\n", style="bold cyan")
+
+            content.append("Start the dashboard with:\n", style="")
+            content.append("  flowyml go", style="bold green")
+
+            if port != 8080:
+                content.append(f" --port {port}", style="bold green")
+
+            content.append("\n\n", style="")
+            content.append("Then run your pipeline again to see it in the UI!", style="dim")
+
+            console.print()
+            console.print(
+                Panel(
+                    content,
+                    title="[bold cyan]🌐 Dashboard Available[/bold cyan]",
+                    border_style="yellow",
+                    box=box.ROUNDED,
+                ),
+            )
+            console.print()
+
+        except ImportError:
+            # Fallback to simple print
+            print()
+            print("=" * 60)
+            print("💡 Want to see your pipeline run in a live dashboard?")
+            print()
+            if reason == "missing_deps":
+                print("   UI dependencies not installed. Install with:")
+                print("     pip install uvicorn fastapi")
+                print()
+            print("   Start the dashboard with:")
+            print("     flowyml go" + (f" --port {port}" if port != 8080 else ""))
+            print()
+            print("   Then run your pipeline again to see it in the UI!")
+            print("=" * 60)
+            print()
+
     def _coerce_resource_config(self, resources: Any | None):
         """Convert resources input to ResourceConfig if necessary."""
         if resources is None:

flowyml/ui/server_manager.py

@@ -86,12 +86,15 @@ class UIServerManager:
         try:
             # Check if UI dependencies are available
             try:
-                import uvicorn
-
-                print(f"uvicorn {uvicorn.__version__}")
+                import uvicorn  # noqa: F401 - just check import
             except ImportError:
                 return False
 
+            # Capture host/port for closure
+            server_host = self._host
+            server_port = self._port
+            startup_error = {"error": None}
+
             # Start server in a daemon thread
             def run_server():
                 try:
@@ -100,13 +103,13 @@ class UIServerManager:
                     # Run uvicorn server (blocking call, but in daemon thread)
                     uvicorn.run(
                         "flowyml.ui.backend.main:app",
-                        host=host,
-                        port=port,
-                        log_level="warning",  # Reduce noise in background
+                        host=server_host,
+                        port=server_port,
+                        log_level="warning",  # Show startup issues
                         access_log=False,
                     )
-                except Exception:
-                    pass  # Server will be stopped
+                except Exception as e:
+                    startup_error["error"] = str(e)
 
             # Start in daemon thread
             self._server_thread = threading.Thread(
@@ -118,14 +121,19 @@ class UIServerManager:
             self._running = True
             self._started = True
 
-            # Wait a bit for server to start
-            max_wait = 5
+            # Wait a bit for server to start (up to 8 seconds)
+            max_wait = 8
             for _ in range(max_wait * 10):  # Check every 100ms
                 time.sleep(0.1)
-                if is_ui_running(host, port):
+                # Check if server started successfully
+                if is_ui_running(server_host, server_port):
                     return True
+                # Check if we have an error
+                if startup_error["error"]:
+                    self._running = False
+                    return False
 
-            # If we get here, server didn't start
+            # If we get here, server didn't start in time
             self._running = False
             return False
 

flowyml/ui/utils.py

@@ -78,12 +78,14 @@ def is_ui_running(host: str = "localhost", port: int = 8080) -> bool:
         conn = http.client.HTTPConnection(host, port, timeout=2)
         conn.request("GET", "/api/health")
         response = conn.getresponse()
-        conn.close()
 
         # Check if response is successful and from flowyml
+        # Note: must read data BEFORE closing connection
         if response.status == 200:
             data = response.read().decode("utf-8")
+            conn.close()
             return "flowyml" in data.lower() or "ok" in data.lower()
+        conn.close()
         return False
     except Exception:
         return False

{flowyml-1.6.0.dist-info → flowyml-1.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowyml
-Version: 1.6.0
+Version: 1.7.0
 Summary: Next-Generation ML Pipeline Framework
 License: Apache-2.0
 License-File: LICENSE

{flowyml-1.6.0.dist-info → flowyml-1.7.0.dist-info}/RECORD

@@ -11,7 +11,7 @@ flowyml/assets/report.py,sha256=CR1aI_08GereO-qsVwvy4JdG5_Du5rMcTfoqJ_1jmu8,9207
 flowyml/cli/__init__.py,sha256=bMA7grr-wiy3LeAjGFSeSG2WQwlXDnQKIeFP7X4-HhM,83
 flowyml/cli/experiment.py,sha256=ryPzfOlPKtCksjW9E7umJEVmE_5lNjAZqacbYX-azos,6284
 flowyml/cli/init.py,sha256=FAlk_xhiCYYI_HoiVBqPfiZVC4mGJmDWxqY7R77E7UY,6204
-flowyml/cli/main.py,sha256=nwRVRqGraZoDnk2L93bItShExOj7ETucACbPkXX4G2o,14934
+flowyml/cli/main.py,sha256=mltcYGsH-xdvGE5itDoORDyCjyuiNvnS9Ui9-J_qETU,29613
 flowyml/cli/models.py,sha256=fg5Ry3J_FaPpoHtnm4Ou-wNGMTrNS9ufYZhdonIZhKU,17772
 flowyml/cli/rich_utils.py,sha256=-jUZ71oNYVc9O3doCuw0Tx6EBuiNb-erIVXhOFJTEEc,2645
 flowyml/cli/run.py,sha256=bkpEhE9LOqP3baNE0_inDGw29eYfToyb-8pCUS8084A,2293
@@ -24,7 +24,7 @@ flowyml/core/cache.py,sha256=rIzIQ-GX2sVO7qAckpZMfmxDLKQU34XJKJp38zywIpk,6197
 flowyml/core/checkpoint.py,sha256=EBKAi0UqWkCOOiz5wyXY_SBDBWJH4POwH_E6-AI_h4s,4794
 flowyml/core/conditional.py,sha256=Huwb_dt6ZRr_OGjmA826IaXxwGQkUchGX8pnT5pHJ0g,11991
 flowyml/core/context.py,sha256=M0_K_REzIByJlF-2INCgHqDWFDKXmNQb-2HSmPt_WIY,4816
-flowyml/core/display.py,sha256=JviIXNwxQzGuO9EcSymBe6NmbkGYVmgnUW26tHYNqBI,19489
+flowyml/core/display.py,sha256=_zVGZAlqG_0n2NxVZPavbtMpFGsV2SKFhzY5s07YIms,22138
 flowyml/core/error_handling.py,sha256=TovzbOFzQYBHMMM8NjsR8WcbmbdtVjXuW8_fsdbgsPA,11952
 flowyml/core/execution_status.py,sha256=vlQOOzglpSea4z9_csqnPkFP27jJuXjAke6EQbPMq3g,1347
 flowyml/core/executor.py,sha256=ljBV472nJyp_06meeeHmD_aomm_JdJD6y03OiPvMtEA,19168
@@ -33,7 +33,7 @@ flowyml/core/hooks.py,sha256=UOqrNY1m3lmVql1FRls6AXDNAV3dxxZl-zO6ijEeRW8,4022
 flowyml/core/observability.py,sha256=NW05m8zki1TwKeRBBCtAP1UwguMcJasGz8HHgMVbjAU,7011
 flowyml/core/orchestrator.py,sha256=8duXRaVNJD6V7s7LBTRESJC7a8dBfQau-UQIk7UFo2Y,37733
 flowyml/core/parallel.py,sha256=KGstDu32i9FFKZV0bBRrm1kl3bKjcHmL2js48dVjWlk,12492
-flowyml/core/pipeline.py,sha256=IY-3zK4hCpfEdDpekIMFGoGmm2RJk7GFZ4wwwXTZSus,40692
+flowyml/core/pipeline.py,sha256=4Up-3Iz5_jy578l3-B8jvKkHYxLo6bmieut5jS6JNQ4,46076
 flowyml/core/project.py,sha256=7Cv_MUnXrD2e69Pzwo4ThXhGyjvP45De6CqL3Z2BtiA,8997
 flowyml/core/remote_orchestrator.py,sha256=LpHlNQslq14OliNZogBGf0fpu6c5n9T7f3BpftahM8Q,3527
 flowyml/core/resources.py,sha256=5TimjjSOEedALgNVVjRrvU-cHujKebyFjU4YxImvqZE,14402
@@ -190,8 +190,8 @@ flowyml/ui/frontend/src/utils/date.js,sha256=8tYLT-TspihDCbziiYCNaBjvMa5ez155kBx
 flowyml/ui/frontend/src/utils/downloads.js,sha256=2w3uSOiAktiCWAj56bTSZUx3eNA9QZt1qkVCzX3YrdY,305
 flowyml/ui/frontend/tailwind.config.js,sha256=__nxDJC93bzcg8Ro9uxt4c2DiErpUCJfi4B-zNRooYg,813
 flowyml/ui/frontend/vite.config.js,sha256=b4JAsNo2yU4wRdTKf7ppBKsaw6WW447LrS0V4AbXkbk,401
-flowyml/ui/server_manager.py,sha256=vUUt_vFIBPdVH9_wuphWZrKHj1lFo1T16CitqimJj1M,5864
-flowyml/ui/utils.py,sha256=Hi1YtCan56nYKeruE5zrlNWUHRw7RLnrZP6jNHfjigA,4434
+flowyml/ui/server_manager.py,sha256=Se37U4aXnzZsG4oh5RmKnGmmisXQ-8lOhaod0QbzSzc,6274
+flowyml/ui/utils.py,sha256=FeM5zga3Dbbn_HRKRRdjLrt0HbnCrxXk0NN7NqPeJII,4516
 flowyml/utils/__init__.py,sha256=eA_YZEZCnCvWRdcqH8IzwWIO-LSUI5-8sbA9mU6xBto,1490
 flowyml/utils/config.py,sha256=Oeywfo2vptI0-yF28AwmtIbf3fb-2SME-zbkfKfQgqs,10280
 flowyml/utils/debug.py,sha256=zcHZxGLbuSImLdcfn1V7CwfaDzc3SunXdV-pWR_UW90,6536
@@ -201,8 +201,8 @@ flowyml/utils/logging.py,sha256=PBJDFlGdp1mePS6A3g08dnGAB-v8jTlcNxEsYs9WSBo,1371
 flowyml/utils/performance.py,sha256=-ne9v9ddEltiKRPk-AerM1R3Gwwd_oCRKtNyHARWd4k,8655
 flowyml/utils/stack_config.py,sha256=STX1niArJzvu0YsqUQmrNJ0WTeMVW_setYNH36BlbVI,10826
 flowyml/utils/validation.py,sha256=mClumVro0bl_XXxT1zWPlRI6M_iZa3z2SZ0QUdmTOqs,10199
-flowyml-1.6.0.dist-info/METADATA,sha256=5S8xZHHAj9G9lAI8us_hQxUNUlDNLz2QfDz3NJhRGrw,15536
-flowyml-1.6.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-flowyml-1.6.0.dist-info/entry_points.txt,sha256=yuF-dOC4rbyJ2Aqi4CMRBxFhqIRoKO6Mhh6jfiQEVjI,48
-flowyml-1.6.0.dist-info/licenses/LICENSE,sha256=DRBRWOEjKZQBvy1WZwxyvp2NmnC1whW9Ef7v0Oo-p_g,626
-flowyml-1.6.0.dist-info/RECORD,,
+flowyml-1.7.0.dist-info/METADATA,sha256=ocS0ENtzA4OxxIyBctONIvO8VLiJnOaFjCww9WGuKcY,15536
+flowyml-1.7.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+flowyml-1.7.0.dist-info/entry_points.txt,sha256=yuF-dOC4rbyJ2Aqi4CMRBxFhqIRoKO6Mhh6jfiQEVjI,48
+flowyml-1.7.0.dist-info/licenses/LICENSE,sha256=DRBRWOEjKZQBvy1WZwxyvp2NmnC1whW9Ef7v0Oo-p_g,626
+flowyml-1.7.0.dist-info/RECORD,,