Spaces:
Running
Running
| import gradio as gr | |
| from gradio import themes | |
| from src.webui.webui_manager import WebuiManager | |
| from src.webui.components.agent_settings_tab import create_agent_settings_tab | |
| from src.webui.components.browser_settings_tab import create_browser_settings_tab | |
| from src.webui.components.browser_use_agent_tab import create_browser_use_agent_tab | |
| from src.webui.components.deep_research_agent_tab import create_deep_research_agent_tab | |
| from src.webui.components.load_save_config_tab import create_load_save_config_tab | |
| from src.webui.components.documentation_tab import create_documentation_tab | |
| from src.webui.components.vayner_client_research_tab import create_vayner_client_research_tab | |
| theme_map = { | |
| "Default": themes.Default(), | |
| "Soft": themes.Soft(), | |
| "Monochrome": themes.Monochrome(), | |
| "Glass": themes.Glass(), | |
| "Origin": themes.Origin(), | |
| "Citrus": themes.Citrus(), | |
| "Ocean": themes.Ocean(), | |
| "Base": themes.Base() | |
| } | |
| def create_ui(theme_name="Ocean"): | |
| css = """ | |
| .gradio-container { | |
| width: 100vw !important; | |
| max-width: 100% !important; | |
| margin-left: auto !important; | |
| margin-right: auto !important; | |
| padding-top: 10px !important; | |
| } | |
| .header-text { | |
| text-align: center; | |
| margin-bottom: 20px; | |
| } | |
| .tab-header-text { | |
| text-align: center; | |
| } | |
| .theme-section { | |
| margin-bottom: 10px; | |
| padding: 15px; | |
| border-radius: 10px; | |
| } | |
| """ | |
| # dark mode in default | |
| js_func = """ | |
| function refresh() { | |
| const url = new URL(window.location); | |
| if (url.searchParams.get('__theme') !== 'dark') { | |
| url.searchParams.set('__theme', 'dark'); | |
| window.location.href = url.href; | |
| } | |
| } | |
| """ | |
| ui_manager = WebuiManager() | |
| with gr.Blocks( | |
| title="Browser Use WebUI", theme=theme_map[theme_name], css=css, js=js_func, | |
| ) as demo: | |
| with gr.Row(): | |
| gr.Markdown( | |
| """ | |
| # 🌐 Browser Use WebUI | |
| ### Control your browser with AI assistance | |
| """, | |
| elem_classes=["header-text"], | |
| ) | |
| with gr.Tabs() as tabs: | |
| with gr.TabItem("Vayner Client Research"): | |
| create_vayner_client_research_tab(ui_manager) | |
| with gr.TabItem("📚 Documentation"): | |
| create_documentation_tab(ui_manager) | |
| return demo | |