Switch to Docker (Python 3.10) for ultimate stability

Dockerfile

@@ -1,31 +1,30 @@
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
-
-ENV DEBIAN_FRONTEND=noninteractive
-ENV PYTHONUNBUFFERED=1
+# Usamos Python 3.10 que es el estándar más estable para IA
+FROM python:3.10-slim
 
+# Instalamos dependencias del sistema (FFMPEG para video y herramientas de compilación)
 RUN apt-get update && apt-get install -y \
-    git wget curl libgl1 libglib2.0-0 \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    git \
     && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /app
-
-# Instalar ComfyUI
-RUN git clone https://github.com/comfyanonymous/ComfyUI.git /app/ComfyUI
+# Creamos un usuario para Hugging Face (requerido por seguridad)
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:${PATH}"
 
-WORKDIR /app/ComfyUI
+WORKDIR /app
 
+# Copiamos e instalamos requerimientos
+COPY --chown=user requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
-RUN pip install --no-cache-dir \
-    xformers \
-    huggingface_hub
-
-# Crear directorios de modelos
-RUN mkdir -p models/checkpoints models/loras models/vae models/clip models/unet models/controlnet
 
-# Script de inicio
-COPY start.sh /app/start.sh
-RUN chmod +x /app/start.sh
+# Copiamos el código de la aplicación
+COPY --chown=user . .
 
+# Exponemos el puerto de Gradio
 EXPOSE 7860
 
-CMD ["/app/start.sh"]
+# Comando para arrancar
+CMD ["python", "app.py"]