aroniscunt/Browser_Web_UI_Automation

1from dotenv import load_dotenv
2load_dotenv()
3import argparse
4from src.webui.interface import theme_map, create_ui
5
6
7def main():
8    parser = argparse.ArgumentParser(description="Gradio WebUI for Browser Agent")
9    parser.add_argument("--ip", type=str, default="127.0.0.1", help="IP address to bind to")
10    parser.add_argument("--port", type=int, default=7788, help="Port to listen on")
11    parser.add_argument("--theme", type=str, default="Ocean", choices=theme_map.keys(), help="Theme to use for the UI")
12    args = parser.parse_args()
13
14    demo = create_ui(theme_name=args.theme)
15    demo.queue().launch(server_name=args.ip, server_port=args.port, share=True)
16
17
18if __name__ == '__main__':
19    main()
20