# 文件名  comfyui_v2.py
import os
import subprocess

import modal

image = (
    modal.Image.debian_slim(
        python_version="3.10"
    )
    .apt_install("git", "gcc", "libportaudio2", "ffmpeg", "wget")
    .run_commands("wget https://cdn.roasmax.cn/whls/comfy_cli-0.0.0-py3-none-any.whl")
    .pip_install("./comfy_cli==0.0.0")
    .run_commands(
        "comfy --skip-prompt install --fast-deps --nvidia --version 0.3.40"
    )
    .pip_install_from_pyproject(os.path.join(os.path.dirname(__file__), "pyproject.toml"))
    .run_commands("comfy node install https://github.com/yolain/ComfyUI-Easy-Use.git")
    .run_commands("comfy node install https://github.com/crystian/ComfyUI-Crystools.git")
    .run_commands("comfy node install https://github.com/pythongosssss/ComfyUI-Custom-Scripts.git")
    .run_commands("comfy node install https://github.com/kijai/ComfyUI-KJNodes.git")
    .run_commands("comfy node install https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git")
    .run_commands("comfy node install https://github.com/WASasquatch/was-node-suite-comfyui.git")
    .run_commands("comfy node install https://github.com/cubiq/ComfyUI_essentials.git")
    .run_commands("comfy node install https://github.com/jamesWalker55/comfyui-various.git")
    .run_commands("comfy node install https://gitea.bowongai.com/bowong/ComfyUI-CustomNode.git", force_build=True)
    .run_commands("comfy node install https://github.com/rgthree/rgthree-comfy.git")
    .run_commands("rm -rf /root/comfy/ComfyUI/models&&ln -s /models /root/comfy/ComfyUI/models")
    .run_commands("rm -rf /root/comfy/ComfyUI/input&&ln -s /models/input /root/comfy/ComfyUI/input")
    .run_commands("rm -rf /root/comfy/ComfyUI/output&&ln -s /models/output /root/comfy/ComfyUI/output")
)
app = modal.App(image=image)
custom_secret = modal.Secret.from_name("comfyui-custom-secret", environment_name="dev")
vol = modal.Volume.from_name("comfy_model", environment_name="dev", create_if_missing=True)


@app.function(
    min_containers=0,
    buffer_containers=0,
    max_containers=1,
    scaledown_window=600,
    secrets=[custom_secret],
    volumes={
        "/models": vol
    }
)
@modal.concurrent(
    max_inputs=10
)
@modal.web_server(8000, startup_timeout=120)
def ui_1():
    subprocess.Popen("comfy launch -- --cpu --listen 0.0.0.0 --port 8000", shell=True)

@app.function(
    min_containers=0,
    buffer_containers=0,
    max_containers=1,
    scaledown_window=600,
    secrets=[custom_secret],
    volumes={
        "/models": vol
    }
)
@modal.concurrent(
    max_inputs=10
)
@modal.web_server(8000, startup_timeout=120)
def ui_2():
    subprocess.Popen("comfy launch -- --cpu --listen 0.0.0.0 --port 8000", shell=True)

@app.function(
    min_containers=0,
    buffer_containers=0,
    max_containers=1,
    scaledown_window=600,
    secrets=[custom_secret],
    volumes={
        "/models": vol
    }
)
@modal.concurrent(
    max_inputs=10
)
@modal.web_server(8000, startup_timeout=120)
def ui_3():
    subprocess.Popen("comfy launch -- --cpu --listen 0.0.0.0 --port 8000", shell=True)

@app.function(
    min_containers=0,
    buffer_containers=0,
    max_containers=1,
    scaledown_window=600,
    secrets=[custom_secret],
    volumes={
        "/models": vol
    }
)
@modal.concurrent(
    max_inputs=10
)
@modal.web_server(8000, startup_timeout=120)
def ui_4():
    subprocess.Popen("comfy launch -- --cpu --listen 0.0.0.0 --port 8000", shell=True)