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")

        with gr.Tab("Main App"):
            base_ui.render()

        with gr.Tab("Web VNC"):
            gr.HTML('<iframe src="http://localhost:6080/vnc.html" width="100%" height="800"></iframe>')

    demo.queue().launch(server_name=args.ip, server_port=args.port)


if __name__ == "__main__":
    main()