FROM python:3.10-slim

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    wget \
    libgl1 \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# Set working directory
WORKDIR /app

# Install Python packages
RUN pip install --no-cache-dir \
    gradio \
    torch \
    torchvision \
    diffusers \
    transformers \
    accelerate \
    safetensors \
    pillow \
    spaces

# Copy application
COPY app.py /app/
COPY custom_nodes/ /app/custom_nodes/
COPY workflows/ /app/workflows/

# Create necessary directories
RUN mkdir -p /app/models

# Expose port
EXPOSE 7860

# Set environment for Zero GPU
ENV PYTHONUNBUFFERED=1
ENV HF_HOME=/app/models

# Run the app
CMD ["python", "app.py"]