Fix HF Spaces build: add pre-requirements.txt and missing files

- Add pre-requirements.txt with PyTorch and build dependencies
- Create missing workflow JSON files for ComfyUI
- Add packages.txt for system dependencies
- Update directory structure with .gitkeep files
- Clean up requirements.txt to avoid duplicates
- Make app.py executable
- All features preserved

.gitignore

@@ -0,0 +1,51 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+env/
+venv/
+ENV/
+.venv
+
+# ComfyUI
+comfyui/ComfyUI/
+outputs/
+temp/
+*.safetensors
+*.ckpt
+*.pt
+*.pth
+*.bin
+
+# Models (large files)
+models/checkpoints/*
+models/loras/*
+models/vae/*
+!models/checkpoints/.gitkeep
+!models/loras/.gitkeep
+!models/vae/.gitkeep
+
+# Logs
+*.log
+*.out
+*.err
+
+# OS
+.DS_Store
+Thumbs.db
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# Temporary
+*.tmp
+*.temp
+*.cache
+
+# Hugging Face
+.huggingface/

README.md

@@ -4,9 +4,10 @@ emoji: 📸
 colorFrom: purple
 colorTo: pink
 sdk: gradio
-sdk_version: "4.44.0"
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
+license: apache-2.0
 ---
 
 # 🎨 Unified UGC Platform - Hugging Face Space

packages.txt

@@ -1,7 +1,8 @@
-git
-git-lfs
-wget
-curl
 ffmpeg
 libsm6
-libxext6
+libxext6
+libgl1-mesa-glx
+libglib2.0-0
+libgomp1
+git
+git-lfs

pre-requirements.txt

@@ -1,3 +1,23 @@
-git
-wget
-curl
+# Pre-requirements for Hugging Face Spaces
+# These packages need to be installed before other dependencies
+
+# PyTorch with CUDA support (Hugging Face Spaces uses CUDA by default)
+torch==2.1.0
+torchvision==0.16.0
+torchaudio==2.1.0
+
+# Essential build tools
+wheel>=0.40.0
+setuptools>=68.0.0
+pip>=23.0
+
+# Required for some ComfyUI dependencies
+cmake>=3.25.0
+ninja>=1.11.0
+
+# Image processing fundamentals
+pillow>=10.0.0
+numpy>=1.24.0
+
+# Required for xformers build
+triton>=2.1.0

requirements.txt

@@ -1,13 +1,9 @@
 # Core dependencies
 gradio>=4.44.0
-pillow>=10.0.0
-numpy>=1.24.0
 requests>=2.31.0
 huggingface_hub>=0.19.0
 
 # ComfyUI dependencies
-torch>=2.0.0
-torchvision>=0.15.0
 transformers>=4.30.0
 diffusers>=0.21.0
 accelerate>=0.20.0
@@ -41,7 +37,6 @@ albumentations>=1.3.0
 # Additional ComfyUI dependencies
 aiohttp>=3.8.0
 colorama>=0.4.6
-torchaudio>=2.0.0
 torchsde>=0.2.5
 soundfile>=0.12.1
 librosa>=0.10.0

setup.sh

@@ -1,7 +1,10 @@
 #!/bin/bash
 # Setup script for Hugging Face Spaces
 
+set -e  # Exit on error
+
 echo "🚀 Setting up Unified UGC Platform..."
+echo "Working directory: $(pwd)"
 
 # Create directory structure
 mkdir -p comfyui/ComfyUI
@@ -15,7 +18,7 @@ mkdir -p temp
 # Clone ComfyUI if not exists
 if [ ! -d "comfyui/ComfyUI/.git" ]; then
     echo "📦 Cloning ComfyUI..."
-    git clone https://github.com/comfyanonymous/ComfyUI.git comfyui/ComfyUI
+    git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git comfyui/ComfyUI
 fi
 
 # Install custom nodes for face enhancement
@@ -23,10 +26,14 @@ echo "📦 Installing custom nodes..."
 cd comfyui/ComfyUI/custom_nodes
 
 # FaceDetailer (from Impact Pack)
-git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack.git
+if [ ! -d "ComfyUI-Impact-Pack" ]; then
+    git clone --depth 1 https://github.com/ltdrdata/ComfyUI-Impact-Pack.git
+fi
 
 # Face Restore CF
-git clone https://github.com/mav-rik/facerestore_cf.git
+if [ ! -d "facerestore_cf" ]; then
+    git clone --depth 1 https://github.com/mav-rik/facerestore_cf.git
+fi
 
 # Install dependencies for custom nodes
 pip install ultralytics insightface onnxruntime
@@ -44,90 +51,10 @@ echo "📥 Downloading models..."
 # wget -c https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors -P models/checkpoints/
 # wget -c https://huggingface.co/strangerzonehf/Flux-Super-Realism-LoRA/resolve/main/super_realism.safetensors -P models/loras/
 
-# Create default workflows
-echo "📝 Creating default workflows..."
-
-# Organic Portrait Workflow
-cat > workflows/organic_portrait_workflow.json << 'EOF'
-{
-  "1": {
-    "class_type": "CheckpointLoaderSimple",
-    "inputs": {
-      "ckpt_name": "sd_xl_base_1.0.safetensors"
-    }
-  },
-  "2": {
-    "class_type": "CLIPTextEncode",
-    "inputs": {
-      "text": "portrait photo",
-      "clip": ["1", 1]
-    }
-  },
-  "3": {
-    "class_type": "KSampler",
-    "inputs": {
-      "seed": 0,
-      "steps": 28,
-      "cfg": 6.0,
-      "sampler_name": "ddim",
-      "scheduler": "normal",
-      "denoise": 1.0,
-      "model": ["1", 0],
-      "positive": ["2", 0],
-      "negative": ["2", 0],
-      "latent_image": ["4", 0]
-    }
-  },
-  "4": {
-    "class_type": "EmptyLatentImage",
-    "inputs": {
-      "width": 768,
-      "height": 1024,
-      "batch_size": 1
-    }
-  },
-  "5": {
-    "class_type": "VAEDecode",
-    "inputs": {
-      "samples": ["3", 0],
-      "vae": ["1", 2]
-    }
-  },
-  "6": {
-    "class_type": "SaveImage",
-    "inputs": {
-      "images": ["5", 0],
-      "filename_prefix": "ugc_output"
-    }
-  }
-}
-EOF
-
-# Face Enhancement Workflow
-cat > workflows/face_enhancement_workflow.json << 'EOF'
-{
-  "1": {
-    "class_type": "LoadImage",
-    "inputs": {
-      "image": "input.png"
-    }
-  },
-  "2": {
-    "class_type": "FaceDetailer",
-    "inputs": {
-      "image": ["1", 0],
-      "model": "GFPGAN",
-      "strength": 0.5
-    }
-  },
-  "3": {
-    "class_type": "SaveImage",
-    "inputs": {
-      "images": ["2", 0],
-      "filename_prefix": "face_enhanced"
-    }
-  }
-}
-EOF
+# Workflows are already created, just check they exist
+echo "📝 Checking workflows..."
+if [ ! -f "workflows/organic_portrait_workflow.json" ]; then
+    echo "⚠️  Missing workflow files!"
+fi
 
 echo "✅ Setup complete!"

workflows/face_enhancement_workflow.json

@@ -1,59 +1,34 @@
 {
-  "3": {
-    "inputs": {
-      "seed": -1,
-      "steps": 30,
-      "cfg": 6.0,
-      "sampler_name": "euler",
-      "scheduler": "normal",
-      "denoise": 1,
-      "model": ["4", 0],
-      "positive": ["6", 0],
-      "negative": ["7", 0],
-      "latent_image": ["5", 0]
-    },
-    "class_type": "KSampler"
-  },
-  "4": {
-    "inputs": {
-      "ckpt_name": "flux1-dev.safetensors"
-    },
-    "class_type": "CheckpointLoaderSimple"
-  },
-  "5": {
-    "inputs": {
-      "width": 768,
-      "height": 1024,
-      "batch_size": 1
-    },
-    "class_type": "EmptyLatentImage"
-  },
-  "6": {
+  "1": {
+    "class_type": "LoadImage",
     "inputs": {
-      "text": "professional photograph",
-      "clip": ["4", 1]
-    },
-    "class_type": "CLIPTextEncode"
+      "image": "input.png",
+      "upload": "image"
+    }
   },
-  "7": {
+  "2": {
+    "class_type": "FaceRestoreCFWithModel",
     "inputs": {
-      "text": "",
-      "clip": ["4", 1]
-    },
-    "class_type": "CLIPTextEncode"
+      "facerestore_model": "GFPGANv1.4.pth",
+      "image": ["1", 0],
+      "facedetection": "retinaface_resnet50",
+      "codeformer_fidelity": 0.5
+    }
   },
-  "8": {
+  "3": {
+    "class_type": "ImageBlend",
     "inputs": {
-      "samples": ["3", 0],
-      "vae": ["4", 2]
-    },
-    "class_type": "VAEDecode"
+      "image1": ["1", 0],
+      "image2": ["2", 0],
+      "blend_factor": 0.8,
+      "blend_mode": "normal"
+    }
   },
-  "9": {
+  "4": {
+    "class_type": "SaveImage",
     "inputs": {
-      "filename_prefix": "ComfyUI",
-      "images": ["8", 0]
-    },
-    "class_type": "SaveImage"
+      "images": ["3", 0],
+      "filename_prefix": "face_enhanced"
+    }
   }
 }

workflows/full_body_workflow.json

@@ -0,0 +1,59 @@
+{
+  "1": {
+    "class_type": "CheckpointLoaderSimple",
+    "inputs": {
+      "ckpt_name": "sd_xl_base_1.0.safetensors"
+    }
+  },
+  "2": {
+    "class_type": "CLIPTextEncode",
+    "inputs": {
+      "text": "full body portrait",
+      "clip": ["1", 1]
+    }
+  },
+  "3": {
+    "class_type": "CLIPTextEncode",
+    "inputs": {
+      "text": "blurry, low quality, distorted",
+      "clip": ["1", 1]
+    }
+  },
+  "4": {
+    "class_type": "EmptyLatentImage",
+    "inputs": {
+      "width": 576,
+      "height": 1024,
+      "batch_size": 1
+    }
+  },
+  "5": {
+    "class_type": "KSampler",
+    "inputs": {
+      "seed": 0,
+      "steps": 28,
+      "cfg": 6.0,
+      "sampler_name": "dpmpp_2m",
+      "scheduler": "karras",
+      "denoise": 1.0,
+      "model": ["1", 0],
+      "positive": ["2", 0],
+      "negative": ["3", 0],
+      "latent_image": ["4", 0]
+    }
+  },
+  "6": {
+    "class_type": "VAEDecode",
+    "inputs": {
+      "samples": ["5", 0],
+      "vae": ["1", 2]
+    }
+  },
+  "7": {
+    "class_type": "SaveImage",
+    "inputs": {
+      "images": ["6", 0],
+      "filename_prefix": "full_body"
+    }
+  }
+}

workflows/organic_portrait_workflow.json

@@ -0,0 +1,59 @@
+{
+  "1": {
+    "class_type": "CheckpointLoaderSimple",
+    "inputs": {
+      "ckpt_name": "sd_xl_base_1.0.safetensors"
+    }
+  },
+  "2": {
+    "class_type": "CLIPTextEncode",
+    "inputs": {
+      "text": "portrait photo",
+      "clip": ["1", 1]
+    }
+  },
+  "3": {
+    "class_type": "CLIPTextEncode",
+    "inputs": {
+      "text": "blurry, low quality, anime, cartoon",
+      "clip": ["1", 1]
+    }
+  },
+  "4": {
+    "class_type": "EmptyLatentImage",
+    "inputs": {
+      "width": 768,
+      "height": 1024,
+      "batch_size": 1
+    }
+  },
+  "5": {
+    "class_type": "KSampler",
+    "inputs": {
+      "seed": 0,
+      "steps": 28,
+      "cfg": 6.0,
+      "sampler_name": "ddim",
+      "scheduler": "normal",
+      "denoise": 1.0,
+      "model": ["1", 0],
+      "positive": ["2", 0],
+      "negative": ["3", 0],
+      "latent_image": ["4", 0]
+    }
+  },
+  "6": {
+    "class_type": "VAEDecode",
+    "inputs": {
+      "samples": ["5", 0],
+      "vae": ["1", 2]
+    }
+  },
+  "7": {
+    "class_type": "SaveImage",
+    "inputs": {
+      "images": ["6", 0],
+      "filename_prefix": "ugc_output"
+    }
+  }
+}

workflows/street_photo_workflow.json

@@ -0,0 +1,59 @@
+{
+  "1": {
+    "class_type": "CheckpointLoaderSimple",
+    "inputs": {
+      "ckpt_name": "sd_xl_base_1.0.safetensors"
+    }
+  },
+  "2": {
+    "class_type": "CLIPTextEncode",
+    "inputs": {
+      "text": "candid street photography",
+      "clip": ["1", 1]
+    }
+  },
+  "3": {
+    "class_type": "CLIPTextEncode",
+    "inputs": {
+      "text": "posed, studio lighting, professional",
+      "clip": ["1", 1]
+    }
+  },
+  "4": {
+    "class_type": "EmptyLatentImage",
+    "inputs": {
+      "width": 1024,
+      "height": 768,
+      "batch_size": 1
+    }
+  },
+  "5": {
+    "class_type": "KSampler",
+    "inputs": {
+      "seed": 0,
+      "steps": 28,
+      "cfg": 7.5,
+      "sampler_name": "euler_a",
+      "scheduler": "normal",
+      "denoise": 1.0,
+      "model": ["1", 0],
+      "positive": ["2", 0],
+      "negative": ["3", 0],
+      "latent_image": ["4", 0]
+    }
+  },
+  "6": {
+    "class_type": "VAEDecode",
+    "inputs": {
+      "samples": ["5", 0],
+      "vae": ["1", 2]
+    }
+  },
+  "7": {
+    "class_type": "SaveImage",
+    "inputs": {
+      "images": ["6", 0],
+      "filename_prefix": "street_photo"
+    }
+  }
+}