Update start.sh

start.sh

@@ -1,104 +1,98 @@
 #!/bin/bash
 set -e
 
-# Configurar un límite alto de file descriptors para evitar colapsos por Playwright y subagentes
-ulimit -n 65536 2>/dev/null || true
-
 echo "=========================================================="
 echo "   INICIALIZANDO AGENT ZERO EN HUGGING FACE SPACES"
 echo "=========================================================="
 
-# Definición de rutas operacionales
+# Definición de rutas operacionales en volumen persistente
 DATA_USR="/data/usr"
 DATA_VENV_A0="/data/venv-a0"
 DATA_VENV_EXEC="/data/venv"
-REPO_DIR="/data/agent-zero-repo"
+DATA_REPO_DIR="/data/agent-zero-repo"
+DATA_HOME="/data/home"
+
+# Función para persistir directorios de manera limpia y compatible con actualizaciones
+persist_and_link() {
+    local CONTAINER_DIR="$1"
+    local PERSIST_DIR="$2"
+
+    echo "Procesando persistencia: $CONTAINER_DIR -> $PERSIST_DIR"
+
+    # Si no existe en el volumen /data, realizamos copia inicial
+    if; then
+        echo "  ➕ Copia inicial de $CONTAINER_DIR -> $PERSIST_DIR..."
+        mkdir -p "$(dirname "$PERSIST_DIR")"
+        if; then
+            cp -a "$CONTAINER_DIR" "$PERSIST_DIR"
+        else
+            mkdir -p "$PERSIST_DIR"
+        fi
+    else
+        # Si ya existe, mezclamos los nuevos archivos de la imagen sin clobber (no-overwrite)
+        # Esto previene que las actualizaciones de la imagen de Docker borren tus paquetes personalizados,
+        # pero permite que nuevas librerías oficiales añadidas por el desarrollador se importen de forma transparente.
+        echo "  🔄 Sincronizando actualizaciones del contenedor con volumen persistente..."
+        if; then
+            cp -an "$CONTAINER_DIR"/. "$PERSIST_DIR/" 2>/dev/null || true
+        fi
+    fi
 
-# Asegurar la propiedad de los archivos en tiempo de ejecución para el usuario UID 1000
-chown -R agent:agent /data 2>/dev/null || true
+    # Eliminar el directorio efímero y crear el enlace simbólico
+    rm -rf "$CONTAINER_DIR"
+    ln -sfn "$PERSIST_DIR" "$CONTAINER_DIR"
+}
 
-# 1. Inicialización quirúrgica del directorio de usuario /a0/usr
-if; then
-    echo "📦 Inicializando datos de usuario estructurados en /data/usr..."
-    mkdir -p "$DATA_USR"
-    if [ -d /a0/usr ]; then
-        cp -a /a0/usr/. "$DATA_USR/" 2>/dev/null || true
-    fi
-fi
+# 1. Persistir el directorio de usuario (contiene chats, configuración y proyectos)
+persist_and_link "/a0/usr" "$DATA_USR"
 
-# 2. Mitigación Crítica ENXIO: Eliminación de sockets UNIX residuales antes de actualizaciones/backups
-echo "🧹 Purgando sockets UNIX activos o huérfanos de escritorio en /data/usr para evitar fallos de compresión..."
+# 2. Mitigar el error ENXIO (No such device or address) en backups y actualizaciones
+# Eliminamos de forma preventiva sockets UNIX huérfanos generados por la interfaz de escritorio
+echo "🧹 Purgando sockets UNIX huérfanos de escritorio en /data/usr para evitar fallos de compresión..."
 find "$DATA_USR" -type s -delete 2>/dev/null || true
 
-# 3. Inicialización del entorno virtual de Framework persistente
-if; then
-    echo "📦 Copiando entorno virtual del framework (/opt/venv-a0) a /data/venv-a0..."
-    mkdir -p "$DATA_VENV_A0"
-    cp -a /opt/venv-a0/. "$DATA_VENV_A0/" 2>/dev/null || true
-fi
+# 3. Persistir entornos virtuales (para que las instalaciones de librerías del agente sean permanentes)
+persist_and_link "/opt/venv-a0" "$DATA_VENV_A0"
+persist_and_link "/opt/venv" "$DATA_VENV_EXEC"
 
-# 4. Inicialización del entorno virtual de Ejecución persistente (Habilidades del agente)
-if; then
-    echo "📦 Copiando entorno virtual del agente (/opt/venv) a /data/venv..."
-    mkdir -p "$DATA_VENV_EXEC"
-    cp -a /opt/venv/. "$DATA_VENV_EXEC/" 2>/dev/null || true
-fi
+# 4. Persistir el directorio HOME de usuario (guarda configs de git, historial de comandos, etc.)
+persist_and_link "/home/agent" "$DATA_HOME"
 
-# 5. Sincronización de Repositorio Resistente a Modificaciones Locales
+# 5. Sincronizar el repositorio de Agent Zero de forma segura contra conflictos de Git
 if; then
-    echo "🔄 Detectado repositorio existente de Agent Zero. Sincronizando parches del desarrollador..."
-    git config --global --add safe.directory "$REPO_DIR"
-    
-    # Resguardar posibles prompts locales o herramientas modificadas por el agente en caliente
-    echo "💾 Salvaguardando modificaciones locales mediante Auto-Stash..."
-    git -C "$REPO_DIR" stash push -u -m "a0-auto-stash-startup" 2>/dev/null || true
+    echo "📦 Clonando código base de Agent Zero por primera vez..."
+    git clone https://github.com/agent0ai/agent-zero.git "$DATA_REPO_DIR"
+else
+    echo "🔄 Actualizando repositorio de Agent Zero de forma segura (con Auto-Stash)..."
+    git config --global --add safe.directory "$DATA_REPO_DIR"
+    # Salvaguardar cualquier modificación local hecha por el agente en caliente para evitar dirty tree
+    git -C "$DATA_REPO_DIR" stash push -u -m "a0-auto-stash-startup" 2>/dev/null || true
     
-    # Intentar descargar de manera segura los cambios del núcleo upstream
-    if! git -C "$REPO_DIR" pull origin main; then
-        echo "⚠️ Advertencia: Conflicto severo en git pull. Realizando restablecimiento forzado para mantener estabilidad..."
-        git -C "$REPO_DIR" reset --hard origin/main
+    if! git -C "$DATA_REPO_DIR" pull origin main; then
+        echo "⚠️ Advertencia: Conflicto severo en git pull. Restableciendo repositorio para mantener estabilidad..."
+        git -C "$DATA_REPO_DIR" reset --hard origin/main
     fi
-    
-    # Devolver los prompts modificados localmente tras la actualización limpia
-    git -C "$REPO_DIR" stash pop 2>/dev/null || true
-else
-    echo "📦 Clonando código base oficial de Agent Zero en volumen persistente..."
-    rm -rf "$REPO_DIR"
-    git clone https://github.com/agent0ai/agent-zero.git "$REPO_DIR"
-    git config --global --add safe.directory "$REPO_DIR"
+    # Devolver las modificaciones tras la actualización limpia
+    git -C "$DATA_REPO_DIR" stash pop 2>/dev/null || true
 fi
 
-# Ajustar permisos globales del código descargado para UID 1000
-chown -R agent:agent "$REPO_DIR"
-
-# 6. Vinculación lógica inequívoca de rutas efímeras a volúmenes persistentes
-echo "🔗 Estructurando enlaces simbólicos quirúrgicos para desviar escrituras efímeras..."
-rm -rf /git/agent-zero /a0/usr /opt/venv-a0 /opt/venv
-
-ln -sfn "$REPO_DIR" /git/agent-zero
-ln -sfn "$DATA_USR" /a0/usr
-ln -sfn "$DATA_VENV_A0" /opt/venv-a0
-ln -sfn "$DATA_VENV_EXEC" /opt/venv
+# Reemplazar la carpeta de ejecución de código por la carpeta persistente y actualizada
+rm -rf /git/agent-zero
+ln -sfn "$DATA_REPO_DIR" /git/agent-zero
 
-# Asegurar la propiedad de los accesos directos simbólicos para evitar fallas lógicas de resolución
-chown -h agent:agent /git/agent-zero /a0/usr /opt/venv-a0 /opt/venv
-chown -R agent:agent /data/usr /data/venv-a0 /data/venv
+# Sincronizar el script del bot de Telegram para asegurar que use el venv persistente
+if [ -f /git/agent-zero/plugins/_telegram_integration/helpers/dependencies.py ]; then
+    sed -i 's|sys.executable,|"/opt/venv-a0/bin/python",|' \
+    /git/agent-zero/plugins/_telegram_integration/helpers/dependencies.py 2>/dev/null || true
+fi
 
 # Diagnóstico de almacenamiento
-echo "📊 Diagnóstico de ocupación física en volumen persistente:"
+echo "📊 Estado de ocupación en volumen persistente /data:"
 du -sh /data/ 2>/dev/null || echo "No se pudo calcular la ocupación de disco"
 echo "=========================================================="
 
-# 7. Ejecución de la UI web bajo identidad estricta no privilegiada
+# 6. Lanzar la interfaz web de Agent Zero directamente (sin setpriv redundante)
 cd /git/agent-zero
-export HOME=/home/agent USER=agent LOGNAME=agent
-
-PYTHON_CMD="/opt/venv-a0/bin/python"
-if; then
-    echo "❌ Error Catastrófico: Intérprete de Python no ejecutable en /opt/venv-a0"
-    exit 1
-fi
-
-echo "🚀 Iniciando servidor de interfaz de usuario de Agent Zero en puerto $PORT..."
-exec setpriv --reuid=1000 --regid=1000 --clear-groups \
-    "$PYTHON_CMD" run_ui.py --host 0.0.0.0 --port "$PORT" --dockerized=True
+PORT=${PORT:-7860}
+echo "🚀 Iniciando servidor de Agent Zero en el puerto $PORT..."
+exec /opt/venv-a0/bin/python run_ui.py --host 0.0.0.0 --port "$PORT" --dockerized=True