Update Dockerfile

Dockerfile

@@ -1,42 +1,41 @@
 FROM python:3.10-slim
 
-# Install system tools needed for the build
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential git git-lfs ffmpeg libsm6 libxext6 cmake libgl1 libglib2.0-0 curl \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# STEP 1 FROM GITHUB: Clone with recursive submodules
-# We use '.' at the end to force it into the current folder (/app)
-RUN git clone https://github.com/Seed3D/Puppeteer.git . --recursive
+# THE FIX: Clone to a temp folder, THEN move files to /app
+RUN git clone --depth 1 --recursive https://github.com/Seed3D/Puppeteer.git /tmp/puppeteer \
+    && cp -r /tmp/puppeteer/* /app/ \
+    && cp -r /tmp/puppeteer/.[!.]* /app/ \
+    && rm -rf /tmp/puppeteer
 
-# STEP 2 FROM GITHUB: Install Torch 
-# (Note: We use the 'cpu' link because HF Free Tier has no GPU, 
-# but the version 2.1.1 is exactly what they asked for)
-RUN pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cpu
+# Verify the file is actually there now
+RUN if [ ! -f "demo_rigging.sh" ]; then echo "CRITICAL: demo_rigging.sh still missing!" && exit 1; fi
 
-# STEP 3 FROM GITHUB: Install requirements
+# Official Steps (CPU Optimized)
+RUN pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install -r requirements.txt
-
-# STEP 4 FROM GITHUB: Install specific 3D dependencies
 RUN pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.1+cpu.html
 RUN pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cpu_pyt211/download.html
+RUN pip install gradio==5.23.1 tetgen mesh2sdf
 
-# --- MANDATORY FIXES FOR CLOUD DEPLOYMENT ---
-# These are the "missing pieces" that a local user would do manually
+# Weights (Missing from GitHub)
 RUN mkdir -p third_partys/Michelangelo/checkpoints/aligned_shape_latents/ && \
     curl -L -o third_partys/Michelangelo/checkpoints/aligned_shape_latents/shapevae-256.ckpt \
     https://huggingface.co/ZixiangZ/Michelangelo/resolve/main/aligned_shape_latents/shapevae-256.ckpt
 
-# Patch the code to run on CPU (sed replaces .cuda() with .to("cpu"))
+# CPU Patch
 RUN find . -type f -name "*.py" -print0 | xargs -0 sed -i 's/\.cuda()/.to("cpu")/g'
+RUN find . -type f -name "*.py" -print0 | xargs -0 sed -i "s/'cuda'/ 'cpu'/g"
 RUN chmod +x demo_rigging.sh
 
-# Environment Setup
 ENV PYTHONPATH="${PYTHONPATH}:/app:/app/skeleton:/app/skinning"
 ENV CUDA_VISIBLE_DEVICES=""
 
-# Copy your UI handler
+# Copy YOUR app.py over the one from the repo
 COPY app.py .
 CMD ["python", "app.py"]