fix: TEM-31, TEM-32, TEM-39, TEM-37 pip warm-up, RunPod REST API v2 in smoke tests, proxy-first checks

start.sh

@@ -276,8 +276,10 @@ else
     echo "Using existing ComfyUI installation"
 fi
 
-# Warm up pip so ComfyUI-Manager's 5s timeout check doesn't fail on cold start
-python -m pip --version > /dev/null 2>&1
+# Warm up pip so ComfyUI-Manager's 5s timeout check doesn't fail on cold start.
+# Log wall time — Manager fails if `python -m pip --version` takes >5s.
+echo "Warming up pip (Manager timeout is 5s)..."
+time python -m pip --version
 
 # Start ComfyUI — keep container alive if it crashes so SSH/Jupyter remain accessible
 cd $COMFYUI_DIR
@@ -292,11 +294,28 @@ fi
 echo "Starting ComfyUI with args: $FIXED_ARGS"
 python main.py $FIXED_ARGS &
 COMFY_PID=$!
-trap "kill $COMFY_PID 2>/dev/null" SIGTERM SIGINT
-wait $COMFY_PID || true
+
+# Distinguish a real ComfyUI crash from the pod being stopped/restarted/
+# terminated (RunPod sends SIGTERM to PID 1, which we forward to ComfyUI —
+# without the flag the crash banner would print on every normal shutdown).
+SHUTTING_DOWN=0
+trap 'SHUTTING_DOWN=1; kill $COMFY_PID 2>/dev/null' SIGTERM SIGINT
+
+COMFY_EXIT=0
+wait $COMFY_PID || COMFY_EXIT=$?
+
+if [ "$SHUTTING_DOWN" = "1" ]; then
+    echo "Pod is shutting down (stop/restart/terminate) — stopping ComfyUI, Jupyter and FileBrowser."
+    # Docker only signals PID 1; stop the nohup'd background services too so
+    # they exit cleanly instead of waiting for SIGKILL.
+    pkill -TERM -f "jupyter-lab" 2>/dev/null || true
+    pkill -TERM -x "filebrowser" 2>/dev/null || true
+    exit 0
+fi
 
 echo "============================================="
-echo "  ComfyUI crashed — check the logs above."
+echo "  ComfyUI exited unexpectedly (exit code $COMFY_EXIT)."
+echo "  Check the logs above for the error/traceback."
 echo "  SSH and JupyterLab are still available."
 echo "  To restart after fixing:"
 echo "    cd $COMFYUI_DIR && source .venv-cu128/bin/activate"