.

Dockerfile

@@ -2,7 +2,7 @@
 FROM tiangolo/uvicorn-gunicorn-fastapi:python3.9
 
 
-RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
+RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 curl -y
 
 # Set the working directory to /code
 WORKDIR /code
@@ -16,11 +16,6 @@ RUN pip install pip==24.0
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-#RUN git clone https://github.com/da03/latent-diffusion.git
-
-# Install latent-diffusion in editable mode
-#RUN pip install -e ./latent-diffusion
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 # Switch to the "user" user
@@ -36,46 +31,106 @@ WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
 # Create a startup script for HF Spaces
-COPY --chown=user <<EOF $HOME/app/start_hf_spaces.sh
+RUN cat > start_hf_spaces.sh << 'EOF'
 #!/bin/bash
 set -e
 
 echo "🚀 Starting Neural OS for HF Spaces"
 echo "===================================="
+echo "📁 Current directory: $(pwd)"
+echo "📋 Files in current directory:"
+ls -la
+
+# Check if required files exist
+if [[ ! -f "dispatcher.py" ]]; then
+    echo "❌ Error: dispatcher.py not found"
+    exit 1
+fi
+
+if [[ ! -f "worker.py" ]]; then
+    echo "❌ Error: worker.py not found"
+    exit 1
+fi
+
+if [[ ! -f "static/index.html" ]]; then
+    echo "❌ Error: static/index.html not found"
+    exit 1
+fi
+
+echo "✅ All required files found"
 
 # Start dispatcher in background
 echo "🎯 Starting dispatcher..."
 python dispatcher.py --port 7860 > dispatcher.log 2>&1 &
-DISPATCHER_PID=\$!
-
-# Wait for dispatcher to start
-sleep 3
-
-# Start single worker (HF Spaces typically has 1 GPU or CPU)
+DISPATCHER_PID=$!
+echo "📊 Dispatcher PID: $DISPATCHER_PID"
+
+# Wait for dispatcher to start and check if it's running
+echo "⏳ Waiting for dispatcher to initialize..."
+sleep 5
+
+if ! kill -0 $DISPATCHER_PID 2>/dev/null; then
+    echo "❌ Dispatcher failed to start"
+    echo "📋 Dispatcher log:"
+    cat dispatcher.log
+    exit 1
+fi
+
+# Test if dispatcher is responding
+echo "🔍 Testing dispatcher health..."
+curl -f http://localhost:7860/ > /dev/null 2>&1
+if [ $? -eq 0 ]; then
+    echo "✅ Dispatcher is responding to HTTP requests"
+else
+    echo "⚠️ Dispatcher HTTP test failed, but continuing..."
+fi
+
+# Start single worker
 echo "🔧 Starting worker..."
 python worker.py --worker-address localhost:8001 --dispatcher-url http://localhost:7860 > worker.log 2>&1 &
-WORKER_PID=\$!
+WORKER_PID=$!
+echo "📊 Worker PID: $WORKER_PID"
 
 # Wait for worker to initialize
 echo "⏳ Waiting for worker to initialize..."
 sleep 30
 
+# Check if worker is still running
+if ! kill -0 $WORKER_PID 2>/dev/null; then
+    echo "❌ Worker failed to start"
+    echo "📋 Worker log:"
+    cat worker.log
+    echo "📋 Dispatcher log:"
+    cat dispatcher.log
+    exit 1
+fi
+
 echo "✅ System ready!"
 echo "🌍 Web interface: http://localhost:7860"
+echo "📊 Dispatcher PID: $DISPATCHER_PID"
+echo "📊 Worker PID: $WORKER_PID"
 
 # Function to cleanup
 cleanup() {
     echo "🛑 Shutting down..."
-    kill \$DISPATCHER_PID \$WORKER_PID 2>/dev/null || true
+    kill $DISPATCHER_PID $WORKER_PID 2>/dev/null || true
     exit 0
 }
 
 trap cleanup SIGINT SIGTERM
 
+# Keep the script running by following the dispatcher log
+echo "📋 Following dispatcher log (Ctrl+C to stop):"
+tail -f dispatcher.log &
+TAIL_PID=$!
+
 # Wait for dispatcher (main process)
-wait \$DISPATCHER_PID
+wait $DISPATCHER_PID
+
+# Clean up tail process
+kill $TAIL_PID 2>/dev/null || true
 EOF
 
-RUN chmod +x $HOME/app/start_hf_spaces.sh
+RUN chmod +x start_hf_spaces.sh
 
 CMD ["bash", "start_hf_spaces.sh"]