Update webui_with_vnc.py

webui_with_vnc.py

@@ -6,14 +6,25 @@ import gradio as gr
 from src.webui.interface import theme_map, create_ui
 
 html="""
-<div id="vnc-container" style="width:100%;height:800px;border:1px solid #ccc"></div>
+<div id="vnc-container" style="width:100%;height:800px;border:2px solid #444;border-radius:8px;"></div>
 <script type="module">
   import RFB from "https://unpkg.com/@novnc/novnc/core/rfb.js";
 
-  const rfb = new RFB(document.getElementById('vnc-container'),
-                      "ws://localhost:6080/websockify");
-  rfb.scaleViewport = true;
-  rfb.background = 'black';
+  // Connect directly to your WebSocket proxy
+  const url = "ws://localhost:6080/websockify";
+  const target = document.getElementById('vnc-container');
+
+  try {
+    const rfb = new RFB(target, url);
+    rfb.viewOnly = false;   // true = disable keyboard/mouse input
+    rfb.scaleViewport = true;
+    rfb.resizeSession = true;
+    rfb.background = "black";
+
+    console.log("✅ Connected to VNC via WebSocket:", url);
+  } catch (err) {
+    target.innerHTML = "<p style='color:red'> ❌ Failed to connect to VNC server.<br>" + err + "</p>";
+  }
 </script>
 """