Upload start.sh with huggingface_hub

start.sh

@@ -39,28 +39,24 @@ echo "Downloading models and PuLID code..."
 cd /app
 python3 download_models.py
 
-# Setup folder paths for ComfyUI
-cat > /tmp/folder_paths.json << EOF
-{
-    "input_path": "/tmp/comfyui_input",
-    "output_path": "/tmp/comfyui_output",
-    "temp_path": "/tmp/comfyui_temp",
-    "models_path": {
-        "checkpoints": ["/tmp/comfyui_models/checkpoints"],
-        "controlnet": ["/tmp/comfyui_models/controlnet"],
-        "clip_vision": ["/tmp/comfyui_models/clip_vision"],
-        "ipadapter": ["/tmp/comfyui_models/ipadapter"],
-        "pulid": ["/tmp/comfyui_models/pulid"],
-        "evaclip": ["/tmp/comfyui_models/evaclip"],
-        "insightface": ["/tmp/comfyui_models/insightface"]
-    },
-    "user_path": "/tmp/comfyui_user"
-}
-EOF
+# Setup folder paths for ComfyUI directly in Python code
+# Instead of relying on a YAML file that might have formatting issues
+echo "Setting up model paths in Python code..."
+cat > /tmp/setup_paths.py << EOF
+import os
+import folder_paths
 
-# Set environment variables to ensure we find all modules
-export PYTHONPATH="${PYTHONPATH}:/app/ComfyUI:/app/ComfyUI/custom_nodes"
-export COMFYUI_FOLDER_PATHS_FILE="/tmp/folder_paths.json"
+# Setup base folders (even if they already exist in folder_paths)
+folder_paths.folder_names_and_paths["checkpoints"] = (["/tmp/comfyui_models/checkpoints"], folder_paths.supported_pt_extensions)
+folder_paths.folder_names_and_paths["controlnet"] = (["/tmp/comfyui_models/controlnet"], folder_paths.supported_pt_extensions)
+folder_paths.folder_names_and_paths["clip_vision"] = (["/tmp/comfyui_models/clip_vision"], folder_paths.supported_pt_extensions)
+folder_paths.folder_names_and_paths["ipadapter"] = (["/tmp/comfyui_models/ipadapter"], folder_paths.supported_pt_extensions)
+folder_paths.folder_names_and_paths["pulid"] = (["/tmp/comfyui_models/pulid"], folder_paths.supported_pt_extensions)
+folder_paths.folder_names_and_paths["evaclip"] = (["/tmp/comfyui_models/evaclip"], folder_paths.supported_pt_extensions + [".pt"])
+folder_paths.folder_names_and_paths["insightface"] = (["/tmp/comfyui_models/insightface"], [".onnx"])
+
+print("Model paths set up successfully")
+EOF
 
 # Install additional Python packages (but don't fail if they can't be installed)
 echo "Installing required Python packages..."
@@ -284,7 +280,39 @@ fi
 # Start ComfyUI in the background with custom directories
 cd /app/ComfyUI
 echo "Starting ComfyUI server..."
-python3 main.py --listen 0.0.0.0 --port 8188 \
+
+# Create a simple startup script for ComfyUI that imports our path setup
+cat > /tmp/comfyui_starter.py << EOF
+# Import folder path setup first
+import sys
+import os
+
+# Add the temp directory to Python path so we can import setup_paths
+sys.path.append('/tmp')
+
+# Try to set up folder paths - ignore errors
+try:
+    import setup_paths
+except Exception as e:
+    print(f"Warning: Failed to import setup_paths: {e}")
+    print("Continuing anyway with default paths")
+
+# Now run the main ComfyUI script
+sys.path.append('/app/ComfyUI')
+
+# Set environment variables
+os.environ['PYTHONPATH'] = f"{os.environ.get('PYTHONPATH', '')}:/app/ComfyUI:/app/ComfyUI/custom_nodes"
+
+# Import and run ComfyUI
+try:
+    import main
+except Exception as e:
+    print(f"Error starting ComfyUI: {e}")
+    print("ComfyUI failed to start, but we'll continue with the Gradio interface")
+EOF
+
+# Run ComfyUI with our custom starter script
+python3 /tmp/comfyui_starter.py --listen 0.0.0.0 --port 8188 \
   --input-directory /tmp/comfyui_input \
   --output-directory /tmp/comfyui_output \
   --temp-directory /tmp/comfyui_temp \