Update webui_with_vnc.py

webui_with_vnc.py

@@ -5,7 +5,17 @@ import argparse
 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>
+<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';
+</script>
+"""
     
 # # Example theme map for the argument parser (add more themes if you have them)
 # theme_map = {
@@ -82,8 +92,8 @@ def main():
 
         with gr.Tab("Web VNC"):
             # The iframe must point to the actual noVNC server (default: localhost:6080/vnc.html)
-            gr.HTML(f'<p>{response}</p><iframe src="http://localhost:6080/vnc.html" width="100%" height="800" style="border:1px solid red;"></iframe>')
-
+            # gr.HTML(f'<iframe src="http://localhost:6080/vnc.html" width="100%" height="800" style="border:1px solid red;"></iframe>')
+            gr.HTML(html)
     demo.queue().launch(server_name=args.ip, server_port=args.port)
 
 if __name__ == "__main__":