Início / unix / Perguntas / 783854
Accepted
garrettlynchirl
Asked: 2024-09-21 22:38:44 +0800 CST

Script Bash para iniciar/parar o ComfyUI

  • 772

Estou tentando criar um script bash e um atalho .desktop que alternam (iniciam e param) o servidor ComfyUI. Este é meu script bash de trabalho atual para iniciar o servidor ComfyUI:

#!/bin/bash

# activate the venv
source ./AI/ComfyUI/venv/bin/activate

# start ComfyUI
python3 ./AI/ComfyUI/main.py

Tomando algumas dicas desta pergunta: Como posso verificar com um script Bash se um aplicativo está em execução?

  1. como substituo a entrada bash usada naquele script por um comando simples codificado para o script python? A tentativa abaixo está correta?
  2. já que ele usa venv, há algo extra que eu preciso fazer quando eu verificar a instância e depois fechá-la se ela estiver lá?

BASECMD=${1%%\ *}

#is this correct?
PID=$("python3 ./AI/ComfyUI/main.py" "$BASECMD")

if [ "$?" -eq "0" ]; then
   echo "at least one instance of "$BASECMD" found, killing all instances"

   # stop ComfyUI
   kill $PID
else
   echo "no running instances of "$BASECMD" found, starting one"

   # activate the venv
   source ./AI/ComfyUI/venv/bin/activate

   # start ComfyUI
   $1

fi

Último (pequeno) detalhe, para o atalho .desktop eu aponto para um ícone. Se possível, eu gostaria de alternar ícones para os estados de início/parada (como a imagem abaixo). Isso é possível com um .desktop?

insira a descrição da imagem aqui

bash

1 respostas

  1. Best Answer

    Isso deve funcionar para você. Basta modificar os caminhos para os locais apropriados.

    #!/bin/bash

pid_file="/tmp/comfyui.pid"
script_pid=""

update_icon() {
    if [ -z "$script_pid" ]; then
        # Not running, use inactive icon
        sed -i 's|^Icon=.*|Icon=/path/to/icon-inactive.png|' ~/Desktop/comfyui.desktop
    else
        # Running, use active icon
        sed -i 's|^Icon=.*|Icon=/path/to/icon-active.png|' ~/Desktop/comfyui.desktop
    fi
}

source /absolute/path/to/AI/ComfyUI/venv/bin/activate

# Check if file exists and read the PID
if [ -f "$pid_file" ]; then
    script_pid=$(cat "$pid_file")
fi

# Check if the process is running
if [ -n "$script_pid" ] && kill -0 "$script_pid" 2>/dev/null; then
    kill "$script_pid"
    wait "$script_pid"
    rm "$pid_file"      # Remove the PID file
    script_pid=""       # Clear the PID
else
    # If not running, start ComfyUI
    python3 /absolute/path/to/AI/ComfyUI/main.py &
    script_pid=$!
    echo "$script_pid" > "$pid_file"  # Save the PID to a file
fi

update_icon

deactivate

exit 0

    O ~/.local/share/applications/comfyui.desktoparquivo:

    [Desktop Entry]
Version=1.0
Name=ComfyUI
Exec=bash -c "/absolute/path/to/my_script.sh"
Icon=/path/to/icon-inactive.png
Type=Application
Terminal=false

    Copie-o para sua área de trabalho para que ele fique visível e você possa alterar livremente o arquivo .desktop sem afetar o arquivo padrão armazenado em ~/.local:

    cp ~/.local/share/applications/comfyui.desktop ~/Desktop/comfyui.desktop
    • 1

relate perguntas