@mindstudio-ai/local-model-tunnel 0.2.0 → 0.3.1

package/dist/quickstart/QuickstartScreen.js

@@ -1,617 +0,0 @@
-import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
-import { useState, useEffect } from "react";
-import { Box, Text, useApp, useInput } from "ink";
-import TextInput from "ink-text-input";
-import Spinner from "ink-spinner";
-import * as path from "path";
-import * as os from "os";
-import { detectAllProviders } from "./detect.js";
-import { installOllama, installLMStudio, installStableDiffusion, installComfyUI, installComfyUICustomNodes, pullOllamaModel, startOllama, hasDefaultSdModel, getComfyUIModelStatus, getPythonVersion, isPythonVersionOk, } from "./installers.js";
-import { LogoString } from "../helpers.js";
-export function QuickstartScreen({ onExternalAction } = {}) {
-    const { exit } = useApp();
-    const [screen, setScreen] = useState("detecting");
-    const [providers, setProviders] = useState([]);
-    const [selectedIndex, setSelectedIndex] = useState(0);
-    const [installProgress, setInstallProgress] = useState(null);
-    const [logs, setLogs] = useState([]);
-    // Track what action was completed for context-specific done screen
-    const [completedAction, setCompletedAction] = useState(null);
-    // Path input state for Stable Diffusion
-    const defaultSdPath = path.join(os.homedir(), "sd-webui-forge-neo");
-    const [sdInstallPath, setSdInstallPath] = useState(defaultSdPath);
-    // Model download state for Ollama
-    const [modelName, setModelName] = useState("");
-    // Track if default SD model already exists
-    const [sdModelExists, setSdModelExists] = useState(false);
-    // ComfyUI state
-    const defaultComfyPath = path.join(os.homedir(), "ComfyUI");
-    const [comfyInstallPath, setComfyInstallPath] = useState(defaultComfyPath);
-    const [comfyModelStatus, setComfyModelStatus] = useState([]);
-    // Refresh provider detection and model status
-    const refreshProviders = async () => {
-        const detected = await detectAllProviders();
-        setProviders(detected);
-        const modelExists = await hasDefaultSdModel();
-        setSdModelExists(modelExists);
-        setComfyModelStatus(getComfyUIModelStatus());
-    };
-    // Navigate to a screen with a clean slate
-    const navigateTo = (target) => {
-        // Clear screen + scrollback buffer + reset cursor position
-        process.stdout.write("\x1B[2J\x1B[3J\x1B[H");
-        setInstallProgress(null);
-        setLogs([]);
-        setScreen(target);
-    };
-    // Detect providers on mount
-    useEffect(() => {
-        async function detect() {
-            await refreshProviders();
-            setScreen("menu");
-        }
-        detect();
-    }, []);
-    // Build menu items based on detected providers
-    const menuItems = [];
-    const ollama = providers.find((p) => p.id === "ollama");
-    const lmstudio = providers.find((p) => p.id === "lmstudio");
-    const sd = providers.find((p) => p.id === "stable-diffusion");
-    // --- Text providers (Ollama, LM Studio) ---
-    if (ollama) {
-        if (!ollama.installed) {
-            menuItems.push({
-                id: "install-ollama",
-                category: "text",
-                label: ollama.installable
-                    ? "Install Ollama (automatic)"
-                    : "Download Ollama (opens browser)",
-                action: async () => {
-                    if (ollama.installable) {
-                        // Use external action so Ink exits and sudo prompt is visible
-                        if (onExternalAction) {
-                            onExternalAction("install-ollama");
-                        }
-                        exit();
-                    }
-                    else {
-                        navigateTo("installing");
-                        await installOllama((progress) => {
-                            setInstallProgress(progress);
-                            if (progress.message) {
-                                setLogs((prev) => [...prev.slice(-10), progress.message]);
-                            }
-                        });
-                        setCompletedAction("install-ollama");
-                        navigateTo("done");
-                    }
-                },
-            });
-        }
-        else if (!ollama.running) {
-            menuItems.push({
-                id: "start-ollama",
-                category: "text",
-                label: "Start Ollama server",
-                action: async () => {
-                    navigateTo("installing");
-                    await startOllama((progress) => {
-                        setInstallProgress(progress);
-                    });
-                    setCompletedAction("start-ollama");
-                    navigateTo("done");
-                },
-            });
-        }
-        else {
-            // Ollama is installed and running - offer to download models and stop
-            menuItems.push({
-                id: "download-model",
-                category: "text",
-                label: "Download Ollama Models",
-                action: async () => {
-                    setModelName("");
-                    navigateTo("model-download");
-                },
-            });
-            menuItems.push({
-                id: "stop-ollama",
-                category: "text",
-                label: "Stop Ollama server",
-                action: async () => {
-                    if (onExternalAction) {
-                        onExternalAction("stop-ollama");
-                    }
-                    exit();
-                },
-            });
-        }
-    }
-    if (lmstudio && !lmstudio.installed) {
-        menuItems.push({
-            id: "install-lmstudio",
-            category: "text",
-            label: "Download LM Studio (opens browser)",
-            action: async () => {
-                navigateTo("installing");
-                await installLMStudio((progress) => {
-                    setInstallProgress(progress);
-                });
-                setCompletedAction("install-lmstudio");
-                navigateTo("done");
-            },
-        });
-    }
-    // --- Image providers (Stable Diffusion) ---
-    if (sd) {
-        if (sd.warning) {
-            menuItems.push({
-                id: "fix-python",
-                category: "image",
-                label: "Install Python 3.13 (required for Forge Neo)",
-                action: async () => {
-                    if (onExternalAction) {
-                        onExternalAction("fix-python");
-                    }
-                    exit();
-                },
-            });
-        }
-        if (!sd.installed) {
-            menuItems.push({
-                id: "install-sd",
-                category: "image",
-                label: sd.installable
-                    ? "Install Stable Diffusion Forge Neo"
-                    : "Stable Diffusion (requires git & python)",
-                disabled: !sd.installable,
-                action: async () => {
-                    navigateTo("path-input");
-                },
-            });
-        }
-        else if (!sd.running) {
-            menuItems.push({
-                id: "start-sd",
-                category: "image",
-                label: "Start Stable Diffusion server",
-                action: async () => {
-                    const pyInfo = await getPythonVersion();
-                    if (!pyInfo) {
-                        setInstallProgress({
-                            stage: "error",
-                            message: "Python not found",
-                            error: "Python is not installed. Forge Neo requires Python 3.13+.\nInstall from https://www.python.org/downloads/",
-                        });
-                        setCompletedAction("start-sd");
-                        navigateTo("done");
-                        return;
-                    }
-                    if (!isPythonVersionOk(pyInfo)) {
-                        setInstallProgress({
-                            stage: "error",
-                            message: `Python ${pyInfo.version} is too old`,
-                            error: `Forge Neo requires Python 3.13+. You have ${pyInfo.version}.\nUse "Install Python 3.13" from the setup menu for instructions.`,
-                        });
-                        setCompletedAction("start-sd");
-                        navigateTo("done");
-                        return;
-                    }
-                    if (onExternalAction) {
-                        onExternalAction("start-sd");
-                    }
-                    exit();
-                },
-            });
-        }
-        else {
-            menuItems.push({
-                id: "stop-sd",
-                category: "image",
-                label: "Stop Stable Diffusion server",
-                action: async () => {
-                    if (onExternalAction) {
-                        onExternalAction("stop-sd");
-                    }
-                    exit();
-                },
-            });
-        }
-        if (sd.installed && !sdModelExists) {
-            menuItems.push({
-                id: "download-sd-model",
-                category: "image",
-                label: "Download default SDXL model (~6.5 GB)",
-                action: async () => {
-                    if (onExternalAction) {
-                        onExternalAction("download-sd-model");
-                    }
-                    exit();
-                },
-            });
-        }
-    }
-    // Function to start SD installation with the chosen path
-    const startSdInstallation = async () => {
-        navigateTo("installing");
-        await installStableDiffusion((progress) => {
-            setInstallProgress(progress);
-            if (progress.message) {
-                setLogs((prev) => [...prev.slice(-10), progress.message]);
-            }
-        }, sdInstallPath);
-        setCompletedAction("install-sd");
-        navigateTo("done");
-    };
-    // Function to download an Ollama model
-    const startModelDownload = async () => {
-        if (!modelName.trim())
-            return;
-        navigateTo("installing");
-        await pullOllamaModel(modelName.trim(), (progress) => {
-            setInstallProgress(progress);
-            if (progress.message) {
-                setLogs((prev) => [...prev.slice(-10), progress.message]);
-            }
-        });
-        setCompletedAction("download-model");
-        navigateTo("done");
-    };
-    // --- Video providers (ComfyUI) ---
-    const comfyui = providers.find((p) => p.id === "comfyui");
-    if (comfyui) {
-        if (!comfyui.installed) {
-            menuItems.push({
-                id: "install-comfyui",
-                category: "video",
-                label: comfyui.installable
-                    ? "Install ComfyUI"
-                    : "ComfyUI (requires git & python)",
-                disabled: !comfyui.installable,
-                action: async () => {
-                    navigateTo("installing");
-                    setInstallProgress({
-                        stage: "start",
-                        message: "Installing ComfyUI...",
-                    });
-                    const success = await installComfyUI(comfyInstallPath, (progress) => {
-                        setInstallProgress(progress);
-                        if (progress.message) {
-                            setLogs((prev) => [...prev.slice(-10), progress.message]);
-                        }
-                    });
-                    if (success) {
-                        setInstallProgress({
-                            stage: "start",
-                            message: "Installing LTX-Video custom nodes...",
-                        });
-                        await installComfyUICustomNodes((progress) => {
-                            setInstallProgress(progress);
-                            if (progress.message) {
-                                setLogs((prev) => [...prev.slice(-10), progress.message]);
-                            }
-                        });
-                    }
-                    setCompletedAction("install-comfyui");
-                    navigateTo("done");
-                },
-            });
-        }
-        else if (!comfyui.running) {
-            menuItems.push({
-                id: "start-comfyui",
-                category: "video",
-                label: "Start ComfyUI server",
-                action: async () => {
-                    if (onExternalAction) {
-                        onExternalAction("start-comfyui");
-                    }
-                    exit();
-                },
-            });
-        }
-        else {
-            for (const model of comfyModelStatus) {
-                if (!model.installed) {
-                    menuItems.push({
-                        id: `download-comfyui-${model.id}`,
-                        category: "video",
-                        label: `Download ${model.label} (${model.totalSize})`,
-                        action: async () => {
-                            if (onExternalAction) {
-                                onExternalAction(`download-comfyui-model:${model.id}`);
-                            }
-                            exit();
-                        },
-                    });
-                }
-            }
-            menuItems.push({
-                id: "stop-comfyui",
-                category: "video",
-                label: "Stop ComfyUI server",
-                action: async () => {
-                    if (onExternalAction) {
-                        onExternalAction("stop-comfyui");
-                    }
-                    exit();
-                },
-            });
-        }
-    }
-    // --- General ---
-    menuItems.push({
-        id: "model-guide",
-        category: "general",
-        label: "How to Add Your Own Models",
-        action: async () => {
-            navigateTo("model-guide");
-        },
-    });
-    menuItems.push({
-        id: "exit",
-        category: "general",
-        label: "Exit",
-        action: async () => {
-            exit();
-        },
-    });
-    // Keyboard navigation
-    useInput((input, key) => {
-        if (screen === "menu") {
-            if (key.upArrow) {
-                setSelectedIndex((prev) => Math.max(0, prev - 1));
-            }
-            if (key.downArrow) {
-                setSelectedIndex((prev) => Math.min(menuItems.length - 1, prev + 1));
-            }
-            if (key.return) {
-                const item = menuItems[selectedIndex];
-                if (item && !item.disabled) {
-                    item.action();
-                }
-            }
-        }
-        // Allow escape to go back from path input
-        if (screen === "path-input" && key.escape) {
-            navigateTo("menu");
-            return;
-        }
-        // Allow escape to go back from model download
-        if (screen === "model-download" && key.escape) {
-            navigateTo("menu");
-            return;
-        }
-        // Allow escape or Enter to go back from model guide
-        if (screen === "model-guide" && (key.escape || key.return)) {
-            navigateTo("menu");
-            return;
-        }
-        // Allow Enter to exit from done screen (return to main menu)
-        if (screen === "done" && key.return) {
-            exit();
-            return;
-        }
-        // Global quit (but not during input screens, done screen, or guide)
-        if (screen !== "path-input" &&
-            screen !== "model-download" &&
-            screen !== "model-guide" &&
-            screen !== "done" &&
-            (input === "q" || key.escape)) {
-            exit();
-        }
-    });
-    // Detecting screen
-    if (screen === "detecting") {
-        return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsxs(Box, { marginTop: 1, children: [_jsx(Text, { color: "cyan", children: _jsx(Spinner, { type: "dots" }) }), _jsx(Text, { children: " Detecting installed providers..." })] })] }));
-    }
-    // Path input screen for Stable Diffusion
-    if (screen === "path-input") {
-        return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { bold: true, color: "white", children: "Install Stable Diffusion Forge Neo" }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, paddingY: 1, children: [_jsx(Text, { children: "Installation path:" }), _jsxs(Box, { marginTop: 1, children: [_jsx(Text, { color: "cyan", children: "> " }), _jsx(TextInput, { value: sdInstallPath, onChange: setSdInstallPath, onSubmit: () => {
-                                        startSdInstallation();
-                                    } })] })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "gray", children: "Enter: Confirm and install" }), _jsx(Text, { color: "gray", children: "Esc: Go back" })] })] }));
-    }
-    // Model download screen for Ollama
-    if (screen === "model-download") {
-        return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { bold: true, color: "white", children: "Download Ollama Model" }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { children: "Browse available models at:" }), _jsx(Text, { color: "cyan", bold: true, children: "https://ollama.com/library" })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "gray", children: "Popular models:" }), _jsx(Text, { color: "gray", children: " llama3.2, mistral, codellama, phi3, gemma2" })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, paddingY: 1, children: [_jsx(Text, { children: "Enter model name:" }), _jsxs(Box, { marginTop: 1, children: [_jsx(Text, { color: "cyan", children: "> " }), _jsx(TextInput, { value: modelName, onChange: setModelName, placeholder: "e.g. llama3.2", onSubmit: () => {
-                                        if (modelName.trim()) {
-                                            startModelDownload();
-                                        }
-                                    } })] })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "gray", children: "Enter: Download model" }), _jsx(Text, { color: "gray", children: "Esc: Go back" })] })] }));
-    }
-    // Model guide screen
-    if (screen === "model-guide") {
-        return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { bold: true, color: "white", children: "How to Add Your Own Models" }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, paddingY: 1, children: [_jsx(Text, { bold: true, children: "Text Models (Ollama)" }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsxs(Text, { color: "gray", children: ["Browse models at", " ", _jsx(Text, { color: "cyan", children: "https://ollama.com/library" })] }), _jsx(Text, { color: "gray", children: "Download via the setup menu or run:" }), _jsxs(Text, { color: "white", children: [" ollama pull ", "<model-name>"] })] })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, paddingY: 1, children: [_jsx(Text, { bold: true, children: "Image Models (Stable Diffusion)" }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsxs(Text, { color: "gray", children: ["Download ", _jsx(Text, { color: "white", children: ".safetensors" }), " files from", " ", _jsx(Text, { color: "cyan", children: "https://civitai.com" })] }), _jsxs(Text, { color: "gray", children: ["Filter by ", _jsx(Text, { color: "white", children: "SDXL 1.0" }), " for best compatibility."] })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "gray", children: "Place them in:" }), _jsxs(Text, { color: "white", children: [" ", sdInstallPath, "/models/Stable-diffusion/"] })] }), _jsx(Box, { marginTop: 1, flexDirection: "column", children: _jsx(Text, { color: "gray", children: "Restart the SD server to pick up new models." }) })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, paddingY: 1, children: [_jsx(Text, { bold: true, children: "Video Models (ComfyUI)" }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsxs(Text, { color: "gray", children: ["Download models from", " ", _jsx(Text, { color: "cyan", children: "https://huggingface.co" }), " ", "and place in the folders below."] }), _jsx(Text, { color: "gray", children: "Models marked with * can be auto-downloaded from the setup menu." })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { bold: true, color: "white", children: "LTX-Video" }), _jsxs(Text, { color: "gray", children: ["  * ltx-video-2b-v0.9.5.safetensors  ", _jsx(Text, { color: "white", children: "checkpoints/" }), "  ", _jsx(Text, { color: "gray", children: "(~6 GB)" })] }), _jsxs(Text, { color: "gray", children: ["  * t5xxl_fp16.safetensors             ", _jsx(Text, { color: "white", children: "text_encoders/" }), " ", _jsx(Text, { color: "gray", children: "(~10 GB)" })] })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { bold: true, color: "white", children: "Wan 2.1 Text-to-Video" }), _jsxs(Text, { color: "gray", children: ["  * wan2.1_t2v_1.3B_fp16.safetensors  ", _jsx(Text, { color: "white", children: "diffusion_models/" }), " ", _jsx(Text, { color: "gray", children: "(~2.6 GB)" })] }), _jsxs(Text, { color: "gray", children: ["    wan2.1_t2v_14B_fp16.safetensors   ", _jsx(Text, { color: "white", children: "diffusion_models/" }), " ", _jsx(Text, { color: "gray", children: "(~28 GB)" })] }), _jsxs(Text, { color: "gray", children: ["  * umt5_xxl_fp8_e4m3fn_scaled        ", _jsx(Text, { color: "white", children: "text_encoders/" }), "    ", _jsx(Text, { color: "gray", children: "(~5 GB, shared)" })] }), _jsxs(Text, { color: "gray", children: ["  * wan_2.1_vae.safetensors            ", _jsx(Text, { color: "white", children: "vae/" }), "              ", _jsx(Text, { color: "gray", children: "(~0.3 GB, shared)" })] })] }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsxs(Text, { color: "gray", children: ["All folders are relative to: ", _jsxs(Text, { color: "white", children: [comfyInstallPath, "/models/"] })] }), _jsx(Text, { color: "gray", children: "The 14B model uses the same text encoder and VAE as the 1.3B." }), _jsx(Text, { color: "gray", children: "Restart the ComfyUI server to pick up new models." })] })] }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "gray", children: "Press Esc or Enter to go back" }) })] }));
-    }
-    // Installing screen
-    if (screen === "installing") {
-        return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { bold: true, color: "white", children: "Loading..." }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [installProgress && (_jsxs(Box, { children: [!installProgress.complete && !installProgress.error && (_jsxs(Text, { color: "cyan", children: [_jsx(Spinner, { type: "dots" }), " "] })), installProgress.complete && _jsx(Text, { color: "green", children: "\u2713 " }), installProgress.error && _jsx(Text, { color: "red", children: "\u2717 " }), _jsx(Text, { children: installProgress.message })] })), installProgress?.error && (_jsxs(Text, { color: "red", children: ["Error: ", installProgress.error] }))] }), logs.length > 0 && (_jsx(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "gray", paddingX: 1, children: logs.slice(-5).map((log, i) => (_jsx(Text, { color: "gray", children: log }, i))) }))] }));
-    }
-    // Done screen - context-aware based on completed action
-    if (screen === "done") {
-        const getDoneMessage = () => {
-            switch (completedAction) {
-                case "start-ollama":
-                    return {
-                        title: "Ollama is now running!",
-                        description: "Ollama server is running in the background on port 11434.",
-                        nextSteps: [
-                            "Return to main menu to authenticate or start the tunnel.",
-                        ],
-                        note: "Ollama will keep running until you restart your computer or stop it manually.",
-                    };
-                case "stop-ollama":
-                    return {
-                        title: "Ollama server stopped!",
-                        description: "The Ollama server has been shut down.",
-                        nextSteps: [],
-                        note: null,
-                    };
-                case "download-model":
-                    return {
-                        title: "Model downloaded!",
-                        description: `The model "${modelName}" has been pulled successfully.`,
-                        nextSteps: [
-                            "Return to main menu to register models and start the tunnel.",
-                        ],
-                        note: null,
-                    };
-                case "install-ollama":
-                    return {
-                        title: "Ollama installed successfully!",
-                        description: "Ollama has been installed and the llama3.2 model has been pulled.",
-                        nextSteps: [
-                            "Return to main menu to authenticate and start the tunnel.",
-                        ],
-                        note: null,
-                    };
-                case "install-lmstudio":
-                    return {
-                        title: "LM Studio download started!",
-                        description: "The download page has been opened in your browser.",
-                        nextSteps: [
-                            "Complete the installation from the downloaded file.",
-                            "Launch LM Studio and download a model.",
-                            "Start the local server in LM Studio.",
-                            "Return to main menu to start the tunnel.",
-                        ],
-                        note: null,
-                    };
-                case "install-sd":
-                    return {
-                        title: "Stable Diffusion Forge Neo cloned!",
-                        description: `Repository cloned to: ${sdInstallPath}`,
-                        nextSteps: [
-                            'Use "Download default SDXL model" from the setup menu to get a model automatically.',
-                            'Or download from https://civitai.com/models (filter by "SDXL 1.0").',
-                            "Then start the server from the setup menu.",
-                        ],
-                        note: `To add your own models, download .safetensors files from https://civitai.com and place them in:\n  ${sdInstallPath}/models/Stable-diffusion/`,
-                    };
-                case "install-comfyui":
-                    return {
-                        title: "ComfyUI installed!",
-                        description: `Installed to: ${comfyInstallPath}`,
-                        nextSteps: [
-                            "Start the ComfyUI server from the setup menu.",
-                            "Download a video model (LTX-Video or Wan 2.1) once the server is running.",
-                            "Then start the tunnel to connect to MindStudio.",
-                        ],
-                        note: [
-                            "LTX-Video custom nodes have been installed for best compatibility.",
-                            "",
-                            "To add your own video models, download .safetensors files from https://huggingface.co and place them in:",
-                            `  Checkpoints:       ${comfyInstallPath}/models/checkpoints/`,
-                            `  Diffusion models:  ${comfyInstallPath}/models/diffusion_models/`,
-                            `  Text encoders:     ${comfyInstallPath}/models/text_encoders/`,
-                            `  VAE:               ${comfyInstallPath}/models/vae/`,
-                        ].join("\n"),
-                    };
-                case "start-sd":
-                    // If there was a pre-flight error (Python version), show it
-                    if (installProgress?.error) {
-                        return {
-                            title: installProgress.message,
-                            description: installProgress.error,
-                            nextSteps: [
-                                'Use "Install Python 3.13" from the setup menu for instructions.',
-                                `If you recently updated Python, delete the old venv: rm -rf ${sdInstallPath}/venv`,
-                            ],
-                            note: null,
-                        };
-                    }
-                    return {
-                        title: "Stable Diffusion server stopped.",
-                        description: "The server has been shut down.",
-                        nextSteps: [
-                            "Return to main menu to start it again or start the tunnel.",
-                        ],
-                        note: null,
-                    };
-                case "stop-sd":
-                    return {
-                        title: "Stable Diffusion server stopped!",
-                        description: "The server has been shut down.",
-                        nextSteps: ["Return to main menu to start it again if needed."],
-                        note: null,
-                    };
-                default:
-                    return {
-                        title: "Setup complete!",
-                        description: null,
-                        nextSteps: [
-                            "Return to main menu to authenticate and start the tunnel.",
-                        ],
-                        note: null,
-                    };
-            }
-        };
-        const msg = getDoneMessage();
-        return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "green", bold: true, children: msg.title }), msg.description && (_jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "white", children: msg.description }) })), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "white", bold: true, children: "Next steps:" }), msg.nextSteps.map((step, i) => (_jsxs(Text, { color: "gray", children: [" ", i + 1, ". ", step] }, i)))] }), msg.note && (_jsx(Box, { marginTop: 1, children: _jsxs(Text, { color: "yellow", children: ["Note: ", msg.note] }) })), _jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "cyan", children: "Press Enter to return to main menu" }) })] })] }));
-    }
-    // Menu screen
-    const allInstalled = providers.every((p) => p.installed);
-    const allRunning = providers
-        .filter((p) => p.installed)
-        .every((p) => p.running);
-    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { color: "cyan", children: LogoString }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { bold: true, color: "white", children: "Quickstart Setup" }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "gray", paddingX: 1, children: [_jsx(Text, { bold: true, children: "Provider Status" }), [
-                        { label: "Text", ids: ["ollama", "lmstudio"] },
-                        { label: "Image", ids: ["stable-diffusion"] },
-                        { label: "Video", ids: ["comfyui"] },
-                    ].map((group) => {
-                        const groupProviders = providers.filter((p) => group.ids.includes(p.id));
-                        if (groupProviders.length === 0)
-                            return null;
-                        return (_jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Text, { color: "gray", dimColor: true, children: group.label }), groupProviders.map((provider) => (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsx(Text, { color: provider.installed
-                                                        ? provider.running
-                                                            ? "green"
-                                                            : "yellow"
-                                                        : "red", children: provider.installed
-                                                        ? provider.running
-                                                            ? "●"
-                                                            : "○"
-                                                        : "✗" }), _jsxs(Text, { children: [" ", provider.name, " - "] }), _jsx(Text, { color: "gray", children: provider.running
-                                                        ? "Running"
-                                                        : provider.installed
-                                                            ? "Installed (not running)"
-                                                            : "Not installed" })] }), provider.warning && (_jsx(Box, { children: _jsxs(Text, { color: "yellow", children: [" \u26A0 ", provider.warning] }) }))] }, provider.id)))] }, group.label));
-                    })] }), process.platform === "darwin" && (_jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "yellow", children: "\u26A0 macOS: Stable Diffusion and ComfyUI may have compatibility issues due to limited CUDA support. GPU acceleration requires Apple Silicon with MPS." }) })), allInstalled && allRunning && (_jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "green", children: "\u2713 All providers are installed and running!" }) })), menuItems.length > 1 &&
-                (() => {
-                    const categoryOrder = [
-                        "text",
-                        "image",
-                        "video",
-                        "general",
-                    ];
-                    const categoryLabels = {
-                        text: "Text Generation",
-                        image: "Image Generation",
-                        video: "Video Generation",
-                        general: "",
-                    };
-                    const categoryIcons = {
-                        text: "💬",
-                        image: "🎨",
-                        video: "🎬",
-                        general: "",
-                    };
-                    // Only show categories that have items
-                    const activeCategories = categoryOrder.filter((cat) => menuItems.some((item) => item.category === cat));
-                    // Track the global index for selection
-                    let globalIndex = 0;
-                    return (_jsx(Box, { marginTop: 1, flexDirection: "column", children: activeCategories.map((cat, catIdx) => {
-                            const items = menuItems.filter((item) => item.category === cat);
-                            const startIdx = globalIndex;
-                            globalIndex += items.length;
-                            return (_jsxs(Box, { flexDirection: "column", marginTop: catIdx > 0 ? 1 : 0, children: [cat !== "general" && (_jsxs(Text, { bold: true, color: "gray", children: [categoryIcons[cat], " ", categoryLabels[cat]] })), items.map((item, i) => {
-                                        const idx = startIdx + i;
-                                        return (_jsxs(Box, { children: [_jsx(Text, { color: idx === selectedIndex ? "cyan" : "white", children: idx === selectedIndex ? " ❯ " : "   " }), _jsx(Text, { color: item.disabled
-                                                        ? "gray"
-                                                        : idx === selectedIndex
-                                                            ? "cyan"
-                                                            : "white", dimColor: item.disabled, children: item.label })] }, item.id));
-                                    })] }, cat));
-                        }) }));
-                })(), (sd?.installed || comfyui?.installed) && (_jsxs(Box, { marginTop: 1, flexDirection: "column", borderStyle: "round", borderColor: "gray", paddingX: 1, children: [_jsx(Text, { bold: true, color: "gray", children: "Add Your Own Models" }), sd?.installed && (_jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsxs(Text, { color: "gray", children: ["Image models (.safetensors) from", " ", _jsx(Text, { color: "cyan", children: "https://civitai.com" })] }), _jsxs(Text, { color: "gray", children: [" ", "Place in:", " ", _jsxs(Text, { color: "white", children: [sdInstallPath, "/models/Stable-diffusion/"] })] })] })), comfyui?.installed && (_jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsxs(Text, { color: "gray", children: ["Video models (.safetensors) from", " ", _jsx(Text, { color: "cyan", children: "https://huggingface.co" })] }), _jsxs(Text, { color: "gray", children: [" ", "Checkpoints:", " ", _jsxs(Text, { color: "white", children: [comfyInstallPath, "/models/checkpoints/"] })] }), _jsxs(Text, { color: "gray", children: [" ", "Diffusion models:", " ", _jsxs(Text, { color: "white", children: [comfyInstallPath, "/models/diffusion_models/"] })] })] }))] })), _jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "gray", children: "\u2191\u2193 Navigate \u2022 Enter Select \u2022 q Quit" }) })] }));
-}
-//# sourceMappingURL=QuickstartScreen.js.map