Update Dockerfile

Dockerfile

@@ -14,6 +14,8 @@ RUN apt-get update && apt-get install -y \
     apt-transport-https \
     ca-certificates \
     git \
+    xvfb \
+    x11-utils \
     && rm -rf /var/lib/apt/lists/*
 
 # Add Antigravity repository
@@ -32,8 +34,21 @@ RUN apt-get update && apt-get install -y \
 # Create data directory for Antigravity
 RUN mkdir -p /tmp/antigravity-data
 
+# Create startup script
+RUN cat > /start.sh << 'EOF'
+#!/bin/bash
+set -x
+echo "Starting Antigravity..."
+export DISPLAY=:99
+Xvfb :99 -screen 0 1024x768x24 > /dev/null 2>&1 &
+sleep 2
+antigravity --no-sandbox --user-data-dir=/tmp/antigravity-data
+EOF
+
+RUN chmod +x /start.sh
+
 # Expose Hugging Face Spaces default port
 EXPOSE 7860
 
-# Default command with required flags for running as root
-CMD ["sh", "-c", "antigravity --no-sandbox --user-data-dir=/tmp/antigravity-data 2>&1"]
+# Default command with startup script
+CMD ["/start.sh"]