Update Dockerfile

Dockerfile

@@ -69,17 +69,31 @@ RUN chown -R user:user /home/user
 EXPOSE 7080
 
 RUN echo '#!/bin/bash\n\
+set -e\n\
 \n\
+echo "Starting Ubuntu Desktop..."\n\
+\n\
+mkdir -p /home/user/.vnc\n\
 echo "password" | vncpasswd -f > /home/user/.vnc/passwd\n\
 chmod 600 /home/user/.vnc/passwd\n\
 \n\
+echo "Starting VNC server..."\n\
 vncserver :1 -geometry 1280x720 -depth 24\n\
 \n\
-sleep 2\n\
+echo "Waiting for VNC server to start..."\n\
+sleep 5\n\
 \n\
+echo "Starting noVNC..."\n\
 websockify --web=/usr/share/novnc/ 7080 localhost:5901 &\n\
 \n\
-wait' > /startup.sh && chmod +x /startup.sh
+echo "Desktop ready! Access at http://localhost:7080"\n\
+echo "Press Ctrl+C to stop"\n\
+\n\
+trap "echo Shutting down...; vncserver -kill :1; exit 0" SIGTERM SIGINT\n\
+\n\
+while true; do\n\
+    sleep 1\n\
+done' > /startup.sh && chmod +x /startup.sh
 
 USER user
 WORKDIR /home/user