Spaces:
Runtime error
Runtime error
| import os | |
| from distutils.util import strtobool | |
| import gradio as gr | |
| import logging | |
| from gradio.components import Component | |
| from src.webui.webui_manager import WebuiManager | |
| from src.utils import config | |
| logger = logging.getLogger(__name__) | |
| async def close_browser(webui_manager: WebuiManager): | |
| """ | |
| Close browser | |
| """ | |
| if webui_manager.bu_current_task and not webui_manager.bu_current_task.done(): | |
| webui_manager.bu_current_task.cancel() | |
| webui_manager.bu_current_task = None | |
| if webui_manager.bu_browser_context: | |
| logger.info("⚠️ Closing browser context when changing browser config.") | |
| await webui_manager.bu_browser_context.close() | |
| webui_manager.bu_browser_context = None | |
| if webui_manager.bu_browser: | |
| logger.info("⚠️ Closing browser when changing browser config.") | |
| await webui_manager.bu_browser.close() | |
| webui_manager.bu_browser = None | |
| def create_browser_settings_tab(webui_manager: WebuiManager): | |
| """ | |
| Creates a browser settings tab. | |
| """ | |
| # Create components and store them in a dictionary | |
| # Explicitly define component IDs for clarity and consistency | |
| components_to_register = { | |
| "browser_binary_path": gr.Textbox( | |
| label="Browser Binary Path", | |
| lines=1, | |
| interactive=True, | |
| placeholder="e.g. \'/Applications/Google\\ Chrome.app/Contents/MacOS/Google\\ Chrome\\\''" | |
| ), | |
| "browser_user_data_dir": gr.Textbox( | |
| label="Browser User Data Dir", | |
| lines=1, | |
| interactive=True, | |
| placeholder="Leave it empty if you use your default user data", | |
| ), | |
| "use_own_browser": gr.Checkbox( | |
| label="Use Own Browser", | |
| value=bool(strtobool(os.getenv("USE_OWN_BROWSER", "false"))), | |
| info="Use your existing browser instance", | |
| interactive=True | |
| ), | |
| "keep_browser_open": gr.Checkbox( | |
| label="Keep Browser Open", | |
| value=bool(strtobool(os.getenv("KEEP_BROWSER_OPEN", "true"))), | |
| info="Keep Browser Open between Tasks", | |
| interactive=True | |
| ), | |
| "browser_headless": gr.Checkbox( | |
| label="Headless Mode", | |
| value=False, | |
| info="Run browser without GUI", | |
| interactive=True | |
| ), | |
| "disable_security": gr.Checkbox( | |
| label="Disable Security", | |
| value=False, | |
| info="Disable browser security", | |
| interactive=True | |
| ), | |
| "browser_viewport_width": gr.Number( | |
| label="Window Width", | |
| value=1280, | |
| info="Browser window width", | |
| interactive=True | |
| ), | |
| "browser_viewport_height": gr.Number( | |
| label="Window Height", | |
| value=1100, | |
| info="Browser window height", | |
| interactive=True | |
| ), | |
| "cdp_url": gr.Textbox( | |
| label="CDP URL", | |
| value=os.getenv("BROWSER_CDP", None), | |
| info="CDP URL for browser remote debugging", | |
| interactive=True, | |
| ), | |
| "wss_url": gr.Textbox( | |
| label="WSS URL", | |
| info="WSS URL for browser remote debugging", | |
| interactive=True, | |
| ), | |
| "save_recording_path": gr.Textbox( | |
| label="Recording Path", | |
| placeholder="e.g. ./tmp/record_videos", | |
| info="Path to save browser recordings", | |
| interactive=True, | |
| ), | |
| "save_trace_path": gr.Textbox( | |
| label="Trace Path", | |
| placeholder="e.g. ./tmp/traces", | |
| info="Path to save Agent traces", | |
| interactive=True, | |
| ), | |
| "save_agent_history_path": gr.Textbox( | |
| label="Agent History Save Path", | |
| value="./tmp/agent_history", | |
| info="Specify the directory where agent history should be saved.", | |
| interactive=True, | |
| ), | |
| "save_download_path": gr.Textbox( | |
| label="Save Directory for browser downloads", | |
| value="./tmp/downloads", | |
| info="Specify the directory where downloaded files should be saved.", | |
| interactive=True, | |
| ), | |
| } | |
| # Register components with the WebuiManager using their full IDs | |
| webui_manager.add_components("browser_settings", components_to_register) | |
| # Group and render components within the Gradio UI | |
| with gr.Group(): | |
| with gr.Row(): | |
| components_to_register["browser_binary_path"] | |
| components_to_register["browser_user_data_dir"] | |
| with gr.Group(): | |
| with gr.Row(): | |
| components_to_register["use_own_browser"] | |
| components_to_register["keep_browser_open"] | |
| components_to_register["browser_headless"] | |
| components_to_register["disable_security"] | |
| with gr.Group(): | |
| with gr.Row(): | |
| components_to_register["browser_viewport_width"] | |
| components_to_register["browser_viewport_height"] | |
| with gr.Group(): | |
| with gr.Row(): | |
| components_to_register["cdp_url"] | |
| components_to_register["wss_url"] | |
| with gr.Group(): | |
| with gr.Row(): | |
| components_to_register["save_recording_path"] | |
| components_to_register["save_trace_path"] | |
| with gr.Row(): | |
| components_to_register["save_agent_history_path"] | |
| components_to_register["save_download_path"] | |
| async def close_wrapper(): | |
| """Wrapper for handle_clear.""" | |
| await close_browser(webui_manager) | |
| # Attach change events to relevant components to trigger browser close | |
| components_to_register["browser_headless"].change(close_wrapper) | |
| components_to_register["keep_browser_open"].change(close_wrapper) | |
| components_to_register["disable_security"].change(close_wrapper) | |
| components_to_register["use_own_browser"].change(close_wrapper) | |
| components_to_register["browser_viewport_width"].change(close_wrapper) | |
| components_to_register["browser_viewport_height"].change(close_wrapper) | |
| components_to_register["cdp_url"].change(close_wrapper) | |
| components_to_register["wss_url"].change(close_wrapper) | |
| # Return a dictionary of components for interface.py to use for rendering | |
| return components_to_register | |