Update webui_with_vnc.py

webui_with_vnc.py

@@ -27,6 +27,18 @@ html="""
   }
 </script>
 """
+def get_vnc_iframe():
+    # get hostname + IP
+    hostname = socket.gethostname()
+    ip = socket.gethostbyname(hostname)
+
+    # build iframe URL
+    url = f"http://{ip}:6080/vnc.html?host={ip}&port=6080"
+    print("############## URL :",url)
+    # return HTML iframe
+    return f"""
+    <iframe src="{url}" width="100%" height="800" style="border:1px solid red;"></iframe>
+    """
     
 # # Example theme map for the argument parser (add more themes if you have them)
 # theme_map = {
@@ -104,7 +116,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'<iframe src="http://localhost:6080/vnc.html" width="100%" height="800" style="border:1px solid red;"></iframe>')
-            gr.HTML(html)
+            # gr.HTML(html)
+            gr.HTML(get_vnc_iframe())
     demo.queue().launch(server_name=args.ip, server_port=args.port)
 
 if __name__ == "__main__":