Update webui_with_vnc.py

webui_with_vnc.py

@@ -1,15 +1,33 @@
-import gradio as gr
+from dotenv import load_dotenv
 import os
+import argparse
+import gradio as gr
+from src.webui.interface import theme_map, create_ui
+
+load_dotenv()
+
+def main():
+    parser = argparse.ArgumentParser(description="Gradio WebUI for Browser Agent")
+    parser.add_argument("--ip", type=str, default=os.getenv("HOST", "0.0.0.0"), help="IP address to bind to")
+    parser.add_argument("--port", type=int, default=int(os.getenv("PORT", "7860")), help="Port to listen on")
+    parser.add_argument("--theme", type=str, default=os.getenv("THEME", "Ocean"), choices=theme_map.keys(), help="Theme to use for the UI")
+    args = parser.parse_args()
+
+    # Existing UI
+    base_ui = create_ui(theme_name=args.theme)
+
+    # Combine into single Gradio app
+    with gr.Blocks(theme=args.theme) as demo:
+        gr.Markdown("# 🌐 Browser Agent Web-UI")
 
-def greet(name):
-    return f"Hello {name}!"
+        with gr.Tab("Main App"):
+            base_ui.render()
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+        with gr.Tab("Web VNC"):
+            gr.HTML('<iframe src="http://localhost:6080/vnc.html" width="100%" height="800"></iframe>')
 
-# Serve noVNC files inside /web-vnc
-novnc_path = "/web-ui/static/novnc"
+    demo.queue().launch(server_name=args.ip, server_port=args.port)
 
-if os.path.exists(novnc_path):
-    demo.add_static_route("/web-vnc", novnc_path)
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    main()