Update Dockerfile

Dockerfile

@@ -25,6 +25,9 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
     build-essential \
     libsndfile-dev \
     software-properties-common \
+    xfce4 xfce4-goodies \
+    x11vnc \
+    novnc websockify \
  && rm -rf /var/lib/apt/lists/*
 
 # Clone repository
@@ -67,5 +70,9 @@ EXPOSE 7860
 RUN chown -R user:user /web-ui && chmod -R 755 /web-ui
 
 # Use Conda environment Python explicitly
-CMD ["/home/user/miniconda/envs/webui/bin/python", "webui.py", "--ip", "0.0.0.0", "--port", "7860"]
-
+# CMD ["/home/user/miniconda/envs/webui/bin/python", "webui.py", "--ip", "0.0.0.0", "--port", "7860"]
+CMD bash -c "\
+    x11vnc -forever -nopw -display :0 & \
+    websockify --web=/usr/share/novnc/ 7860 localhost:5900 & \
+    /home/user/miniconda/envs/webui/bin/python webui.py --ip 0.0.0.0 --port 7860 \
+"