alihan/browser-use-web-ui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update README with new example

Browse Source
This commit is contained in:
Gregor Žunič
2025-01-07 09:07:50 -08:00
parent 6c29bc1195
commit 5539a838bf
5 changed files with 247 additions and 164 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
.vscode/settings.json vendored Normal file
View File

@@ -0,0 +1,11 @@
{
"python.analysis.typeCheckingMode": "basic",
"[python]": {
"editor.defaultFormatter": "charliermarsh.ruff",
"editor.formatOnSave": true,
"editor.codeActionsOnSave": {
"source.fixAll.ruff": "explicit",
"source.organizeImports.ruff": "explicit"
}
}
}

100
README.md
View File

@@ -1,43 +1,53 @@
# Browser-Use WebUI <img src="./assets/web-ui.png" alt="Browser Use Web UI" width="full"/>
## Background <br/>
This project builds upon the foundation of the [browser-use](https://github.com/browser-use/browser-use), which is designed to make websites accessible for AI agents. We have enhanced the original capabilities by providing: [![GitHub stars](https://img.shields.io/github/stars/browser-use/web-ui?style=social)](https://github.com/browser-use/web-ui/stargazers)
[![Discord](https://img.shields.io/discord/1303749220842340412?color=7289DA&label=Discord&logo=discord&logoColor=white)](https://link.browser-use.com/discord)
[![Documentation](https://img.shields.io/badge/Documentation-📕-blue)](https://docs.browser-use.com)
[![WarmShao](https://img.shields.io/twitter/follow/warmshao?style=social)](https://x.com/warmshao)
1. **A Brand New WebUI:** We offer a comprehensive web interface that supports a wide range of `browser-use` functionalities. This UI is designed to be user-friendly and enables easy interaction with the browser agent. This project builds upon the foundation of the [browser-use](https://github.com/browser-use/browser-use), which is designed to make websites accessible for AI agents.
2. **Expanded LLM Support:** We've integrated support for various Large Language Models (LLMs), including: Gemini, OpenAI, Azure OpenAI, Anthropic, DeepSeek, Ollama etc. And we plan to add support for even more models in the future. We would like to officially thank [WarmShao](https://github.com/warmshao) for his contribution to this project.
3. **Custom Browser Support:** You can use your own browser with our tool, eliminating the need to re-login to sites or deal with other authentication challenges. This feature also supports high-definition screen recording. **WebUI:** is built on Gradio and supports a most of `browser-use` functionalities. This UI is designed to be user-friendly and enables easy interaction with the browser agent.
4. **Customized Agent:** We've implemented a custom agent that enhances `browser-use` with Optimized prompts. **Expanded LLM Support:** We've integrated support for various Large Language Models (LLMs), including: Gemini, OpenAI, Azure OpenAI, Anthropic, DeepSeek, Ollama etc. And we plan to add support for even more models in the future.
<video src="https://github.com/user-attachments/assets/58c0f59e-02b4-4413-aba8-6184616bf181" controls="controls" width="500" height="300" >Your browser does not support playing this video!</video> **Custom Browser Support:** You can use your own browser with our tool, eliminating the need to re-login to sites or deal with other authentication challenges. This feature also supports high-definition screen recording.
**Changelog** <video src="https://github.com/user-attachments/assets/56bc7080-f2e3-4367-af22-6bf2245ff6cb" controls="controls" >Your browser does not support playing this video!</video>
- [x] **2025/01/06:** Thanks to @richard-devbot, a New and Well-Designed WebUI is released. [Video tutorial demo](https://github.com/warmshao/browser-use-webui/issues/1#issuecomment-2573393113).
## Installation Guide
## Environment Installation Read the [quickstart guide](https://docs.browser-use.com/quickstart#prepare-the-environment) or follow the steps below to get started.
1. **Python Version:** Ensure you have Python 3.11 or higher installed. > Python 3.11 or higher is required.
2. **Install `browser-use`:**
```bash First, we recommend using [uv](https://docs.astral.sh/uv/) to setup the Python environment.
pip install browser-use
``` ```bash
3. **Install Playwright:** uv venv --python 3.11
```bash ```
playwright install
``` and activate it with:
4. **Install Dependencies:**
```bash ```bash
pip install -r requirements.txt source .venv/bin/activate
``` ```
5. **Configure Environment Variables:**
- Copy `.env.example` to `.env` and set your environment variables, including API keys for the LLM. Install the dependencies:
- **If using your own browser:**
- Set `CHROME_PATH` to the executable path of your browser (e.g., `C:\Program Files\Google\Chrome\Application\chrome.exe` on Windows). ```bash
- Set `CHROME_USER_DATA` to the user data directory of your browser (e.g.,`C:\Users\<YourUsername>\AppData\Local\Google\Chrome\User Data`). uv pip install -r requirements.txt
```
Then install playwright:
```bash
playwright install
```
## Usage ## Usage
@@ -50,3 +60,35 @@ This project builds upon the foundation of the [browser-use](https://github.com/
- Close all chrome windows - Close all chrome windows
- Open the WebUI in a non-Chrome browser, such as Firefox or Edge. This is important because the persistent browser context will use the Chrome data when running the agent. - Open the WebUI in a non-Chrome browser, such as Firefox or Edge. This is important because the persistent browser context will use the Chrome data when running the agent.
- Check the "Use Own Browser" option within the Browser Settings. - Check the "Use Own Browser" option within the Browser Settings.
## (Optional) Configure Environment Variables
Copy `.env.example` to `.env` and set your environment variables, including API keys for the LLM. With
```bash
cp .env.example .env
```
**If using your own browser:** - Set `CHROME_PATH` to the executable path of your browser and `CHROME_USER_DATA` to the user data directory of your browser.
You can just copy examples down below to your `.env` file.
### Windows
```env
CHROME_PATH="C:\Program Files\Google\Chrome\Application\chrome.exe"
CHROME_USER_DATA="C:\Users\YourUsername\AppData\Local\Google\Chrome\User Data"
```
> Note: Replace `YourUsername` with your actual Windows username for Windows systems.
### Mac
```env
CHROME_PATH="/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
CHROME_USER_DATA="~/Library/Application Support/Google/Chrome/Profile 1"
```
## Changelog
- [x] **2025/01/06:** Thanks to @richard-devbot, a New and Well-Designed WebUI is released. [Video tutorial demo](https://github.com/warmshao/browser-use-webui/issues/1#issuecomment-2573393113).

BIN
assets/web-ui.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 24 KiB

3
requirements.txt
View File

@@ -2,5 +2,4 @@ browser-use
langchain-google-genai langchain-google-genai
pyperclip pyperclip
gradio gradio
langchain-ollama langchain-ollama

297
webui.py
View File

@@ -4,62 +4,56 @@
# @Email : wenshaoguo1026@gmail.com # @Email : wenshaoguo1026@gmail.com
# @Project : browser-use-webui # @Project : browser-use-webui
# @FileName: webui.py # @FileName: webui.py
import pdb
from dotenv import load_dotenv from dotenv import load_dotenv
load_dotenv() load_dotenv()
import argparse import argparse
import os
import asyncio
import gradio as gr import gradio as gr
import asyncio from browser_use.agent.service import Agent
import os
from pprint import pprint
from typing import List, Dict, Any
from playwright.async_api import async_playwright
from browser_use.browser.browser import Browser, BrowserConfig from browser_use.browser.browser import Browser, BrowserConfig
from browser_use.browser.context import ( from browser_use.browser.context import (
BrowserContext,
BrowserContextConfig, BrowserContextConfig,
BrowserContextWindowSize, BrowserContextWindowSize,
) )
from browser_use.agent.service import Agent from playwright.async_api import async_playwright
from src.browser.custom_browser import CustomBrowser, BrowserConfig
from src.browser.custom_context import BrowserContext, BrowserContextConfig
from src.controller.custom_controller import CustomController
from src.agent.custom_agent import CustomAgent from src.agent.custom_agent import CustomAgent
from src.agent.custom_prompts import CustomSystemPrompt from src.agent.custom_prompts import CustomSystemPrompt
from src.browser.custom_browser import BrowserConfig, CustomBrowser
from src.browser.custom_context import BrowserContextConfig
from src.controller.custom_controller import CustomController
from src.utils import utils from src.utils import utils
async def run_browser_agent( async def run_browser_agent(
agent_type, agent_type,
llm_provider, llm_provider,
llm_model_name, llm_model_name,
llm_temperature, llm_temperature,
llm_base_url, llm_base_url,
llm_api_key, llm_api_key,
use_own_browser, use_own_browser,
headless, headless,
disable_security, disable_security,
window_w, window_w,
window_h, window_h,
save_recording_path, save_recording_path,
task, task,
add_infos, add_infos,
max_steps, max_steps,
use_vision use_vision,
): ):
# Ensure the recording directory exists # Ensure the recording directory exists
os.makedirs(save_recording_path, exist_ok=True) os.makedirs(save_recording_path, exist_ok=True)
# Get the list of existing videos before the agent runs # Get the list of existing videos before the agent runs
existing_videos = set(glob.glob(os.path.join(save_recording_path, '*.[mM][pP]4')) + existing_videos = set(
glob.glob(os.path.join(save_recording_path, '*.[wW][eE][bB][mM]'))) glob.glob(os.path.join(save_recording_path, "*.[mM][pP]4"))
+ glob.glob(os.path.join(save_recording_path, "*.[wW][eE][bB][mM]"))
)
# Run the agent # Run the agent
llm = utils.get_llm_model( llm = utils.get_llm_model(
@@ -67,7 +61,7 @@ async def run_browser_agent(
model_name=llm_model_name, model_name=llm_model_name,
temperature=llm_temperature, temperature=llm_temperature,
base_url=llm_base_url, base_url=llm_base_url,
api_key=llm_api_key api_key=llm_api_key,
) )
if agent_type == "org": if agent_type == "org":
final_result, errors, model_actions, model_thoughts = await run_org_agent( final_result, errors, model_actions, model_thoughts = await run_org_agent(
@@ -79,7 +73,7 @@ async def run_browser_agent(
save_recording_path=save_recording_path, save_recording_path=save_recording_path,
task=task, task=task,
max_steps=max_steps, max_steps=max_steps,
use_vision=use_vision use_vision=use_vision,
) )
elif agent_type == "custom": elif agent_type == "custom":
final_result, errors, model_actions, model_thoughts = await run_custom_agent( final_result, errors, model_actions, model_thoughts = await run_custom_agent(
@@ -93,14 +87,16 @@ async def run_browser_agent(
task=task, task=task,
add_infos=add_infos, add_infos=add_infos,
max_steps=max_steps, max_steps=max_steps,
use_vision=use_vision use_vision=use_vision,
) )
else: else:
raise ValueError(f"Invalid agent type: {agent_type}") raise ValueError(f"Invalid agent type: {agent_type}")
# Get the list of videos after the agent runs # Get the list of videos after the agent runs
new_videos = set(glob.glob(os.path.join(save_recording_path, '*.[mM][pP]4')) + new_videos = set(
glob.glob(os.path.join(save_recording_path, '*.[wW][eE][bB][mM]'))) glob.glob(os.path.join(save_recording_path, "*.[mM][pP]4"))
+ glob.glob(os.path.join(save_recording_path, "*.[wW][eE][bB][mM]"))
)
# Find the newly created video # Find the newly created video
latest_video = None latest_video = None
@@ -109,31 +105,34 @@ async def run_browser_agent(
return final_result, errors, model_actions, model_thoughts, latest_video return final_result, errors, model_actions, model_thoughts, latest_video
async def run_org_agent( async def run_org_agent(
llm, llm,
headless, headless,
disable_security, disable_security,
window_w, window_w,
window_h, window_h,
save_recording_path, save_recording_path,
task, task,
max_steps, max_steps,
use_vision use_vision,
): ):
browser = Browser( browser = Browser(
config=BrowserConfig( config=BrowserConfig(
headless=headless, headless=headless,
disable_security=disable_security, disable_security=disable_security,
extra_chromium_args=[f'--window-size={window_w},{window_h}'], extra_chromium_args=[f"--window-size={window_w},{window_h}"],
) )
) )
async with await browser.new_context( async with await browser.new_context(
config=BrowserContextConfig( config=BrowserContextConfig(
trace_path='./tmp/traces', trace_path="./tmp/traces",
save_recording_path=save_recording_path if save_recording_path else None, save_recording_path=save_recording_path if save_recording_path else None,
no_viewport=False, no_viewport=False,
browser_window_size=BrowserContextWindowSize(width=window_w, height=window_h), browser_window_size=BrowserContextWindowSize(
) width=window_w, height=window_h
),
)
) as browser_context: ) as browser_context:
agent = Agent( agent = Agent(
task=task, task=task,
@@ -150,18 +149,19 @@ async def run_org_agent(
await browser.close() await browser.close()
return final_result, errors, model_actions, model_thoughts return final_result, errors, model_actions, model_thoughts
async def run_custom_agent( async def run_custom_agent(
llm, llm,
use_own_browser, use_own_browser,
headless, headless,
disable_security, disable_security,
window_w, window_w,
window_h, window_h,
save_recording_path, save_recording_path,
task, task,
add_infos, add_infos,
max_steps, max_steps,
use_vision use_vision,
): ):
controller = CustomController() controller = CustomController()
playwright = None playwright = None
@@ -177,14 +177,14 @@ async def run_custom_agent(
no_viewport=False, no_viewport=False,
headless=headless, # 保持浏览器窗口可见 headless=headless, # 保持浏览器窗口可见
user_agent=( user_agent=(
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 ' "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
'(KHTML, like Gecko) Chrome/85.0.4183.102 Safari/537.36' "(KHTML, like Gecko) Chrome/85.0.4183.102 Safari/537.36"
), ),
java_script_enabled=True, java_script_enabled=True,
bypass_csp=disable_security, bypass_csp=disable_security,
ignore_https_errors=disable_security, ignore_https_errors=disable_security,
record_video_dir=save_recording_path if save_recording_path else None, record_video_dir=save_recording_path if save_recording_path else None,
record_video_size={'width': window_w, 'height': window_h} record_video_size={"width": window_w, "height": window_h},
) )
else: else:
browser_context_ = None browser_context_ = None
@@ -193,17 +193,21 @@ async def run_custom_agent(
config=BrowserConfig( config=BrowserConfig(
headless=headless, headless=headless,
disable_security=disable_security, disable_security=disable_security,
extra_chromium_args=[f'--window-size={window_w},{window_h}'], extra_chromium_args=[f"--window-size={window_w},{window_h}"],
) )
) )
async with await browser.new_context( async with await browser.new_context(
config=BrowserContextConfig( config=BrowserContextConfig(
trace_path='./tmp/result_processing', trace_path="./tmp/result_processing",
save_recording_path=save_recording_path if save_recording_path else None, save_recording_path=save_recording_path
no_viewport=False, if save_recording_path
browser_window_size=BrowserContextWindowSize(width=window_w, height=window_h), else None,
no_viewport=False,
browser_window_size=BrowserContextWindowSize(
width=window_w, height=window_h
), ),
context=browser_context_ ),
context=browser_context_,
) as browser_context: ) as browser_context:
agent = CustomAgent( agent = CustomAgent(
task=task, task=task,
@@ -212,7 +216,7 @@ async def run_custom_agent(
llm=llm, llm=llm,
browser_context=browser_context, browser_context=browser_context,
controller=controller, controller=controller,
system_prompt_class=CustomSystemPrompt system_prompt_class=CustomSystemPrompt,
) )
history = await agent.run(max_steps=max_steps) history = await agent.run(max_steps=max_steps)
@@ -223,6 +227,7 @@ async def run_custom_agent(
except Exception as e: except Exception as e:
import traceback import traceback
traceback.print_exc() traceback.print_exc()
final_result = "" final_result = ""
errors = str(e) + "\n" + traceback.format_exc() errors = str(e) + "\n" + traceback.format_exc()
@@ -240,10 +245,9 @@ async def run_custom_agent(
return final_result, errors, model_actions, model_thoughts return final_result, errors, model_actions, model_thoughts
import argparse import glob
import gradio as gr
from gradio.themes import Base, Default, Soft, Monochrome, Glass, Origin, Citrus, Ocean from gradio.themes import Citrus, Default, Glass, Monochrome, Ocean, Origin, Soft
import os, glob
# Define the theme map globally # Define the theme map globally
theme_map = { theme_map = {
@@ -253,9 +257,10 @@ theme_map = {
"Glass": Glass(), "Glass": Glass(),
"Origin": Origin(), "Origin": Origin(),
"Citrus": Citrus(), "Citrus": Citrus(),
"Ocean": Ocean() "Ocean": Ocean(),
} }
def create_ui(theme_name="Ocean"): def create_ui(theme_name="Ocean"):
css = """ css = """
.gradio-container { .gradio-container {
@@ -283,25 +288,27 @@ def create_ui(theme_name="Ocean"):
} }
} }
""" """
with gr.Blocks(title="Browser Use WebUI", theme=theme_map[theme_name], css=css, js=js) as demo: with gr.Blocks(
title="Browser Use WebUI", theme=theme_map[theme_name], css=css, js=js
) as demo:
with gr.Row(): with gr.Row():
gr.Markdown( gr.Markdown(
""" """
# 🌐 Browser Use WebUI # 🌐 Browser Use WebUI
### Control your browser with AI assistance ### Control your browser with AI assistance
""", """,
elem_classes=["header-text"] elem_classes=["header-text"],
) )
with gr.Tabs() as tabs: with gr.Tabs() as tabs:
with gr.TabItem("🤖 Agent Settings", id=1): with gr.TabItem("⚙️ Agent Settings", id=1):
with gr.Group(): with gr.Group():
agent_type = gr.Radio( agent_type = gr.Radio(
["org", "custom"], ["org", "custom"],
label="Agent Type", label="Agent Type",
value="custom", value="custom",
info="Select the type of agent to use" info="Select the type of agent to use",
) )
max_steps = gr.Slider( max_steps = gr.Slider(
minimum=1, minimum=1,
@@ -309,26 +316,33 @@ def create_ui(theme_name="Ocean"):
value=100, value=100,
step=1, step=1,
label="Max Run Steps", label="Max Run Steps",
info="Maximum number of steps the agent will take" info="Maximum number of steps the agent will take",
) )
use_vision = gr.Checkbox( use_vision = gr.Checkbox(
label="Use Vision", label="Use Vision",
value=True, value=True,
info="Enable visual processing capabilities" info="Enable visual processing capabilities",
) )
with gr.TabItem("🔧 LLM Configuration", id=2): with gr.TabItem("🔧 LLM Configuration", id=2):
with gr.Group(): with gr.Group():
llm_provider = gr.Dropdown( llm_provider = gr.Dropdown(
["anthropic", "openai", "gemini", "azure_openai", "deepseek", "ollama"], [
"anthropic",
"openai",
"gemini",
"azure_openai",
"deepseek",
"ollama",
],
label="LLM Provider", label="LLM Provider",
value="gemini", value="openai",
info="Select your preferred language model provider" info="Select your preferred language model provider",
) )
llm_model_name = gr.Textbox( llm_model_name = gr.Textbox(
label="Model Name", label="Model Name",
value="gemini-2.0-flash-exp", value="gpt-4o",
info="Specify the model to use" info="Specify the model to use",
) )
llm_temperature = gr.Slider( llm_temperature = gr.Slider(
minimum=0.0, minimum=0.0,
@@ -336,17 +350,14 @@ def create_ui(theme_name="Ocean"):
value=1.0, value=1.0,
step=0.1, step=0.1,
label="Temperature", label="Temperature",
info="Controls randomness in model outputs" info="Controls randomness in model outputs",
) )
with gr.Row(): with gr.Row():
llm_base_url = gr.Textbox( llm_base_url = gr.Textbox(
label="Base URL", label="Base URL", info="API endpoint URL (if required)"
info="API endpoint URL (if required)"
) )
llm_api_key = gr.Textbox( llm_api_key = gr.Textbox(
label="API Key", label="API Key", type="password", info="Your API key"
type="password",
info="Your API key"
) )
with gr.TabItem("🌐 Browser Settings", id=3): with gr.TabItem("🌐 Browser Settings", id=3):
@@ -355,51 +366,51 @@ def create_ui(theme_name="Ocean"):
use_own_browser = gr.Checkbox( use_own_browser = gr.Checkbox(
label="Use Own Browser", label="Use Own Browser",
value=False, value=False,
info="Use your existing browser instance" info="Use your existing browser instance",
) )
headless = gr.Checkbox( headless = gr.Checkbox(
label="Headless Mode", label="Headless Mode",
value=False, value=False,
info="Run browser without GUI" info="Run browser without GUI",
) )
disable_security = gr.Checkbox( disable_security = gr.Checkbox(
label="Disable Security", label="Disable Security",
value=True, value=True,
info="Disable browser security features" info="Disable browser security features",
) )
with gr.Row(): with gr.Row():
window_w = gr.Number( window_w = gr.Number(
label="Window Width", label="Window Width",
value=1920, value=1280,
info="Browser window width" info="Browser window width",
) )
window_h = gr.Number( window_h = gr.Number(
label="Window Height", label="Window Height",
value=1080, value=1100,
info="Browser window height" info="Browser window height",
) )
save_recording_path = gr.Textbox( save_recording_path = gr.Textbox(
label="Recording Path", label="Recording Path",
placeholder="e.g. ./tmp/record_videos", placeholder="e.g. ./tmp/record_videos",
value="./tmp/record_videos", value="./tmp/record_videos",
info="Path to save browser recordings" info="Path to save browser recordings",
) )
with gr.TabItem("📝 Task Settings", id=4): with gr.TabItem("🤖 Run Agent", id=4):
task = gr.Textbox( task = gr.Textbox(
label="Task Description", label="Task Description",
lines=4, lines=4,
placeholder="Enter your task here...", placeholder="Enter your task here...",
value="go to google.com and type 'OpenAI' click search and give me the first url", value="go to google.com and type 'OpenAI' click search and give me the first url",
info="Describe what you want the agent to do" info="Describe what you want the agent to do",
) )
add_infos = gr.Textbox( add_infos = gr.Textbox(
label="Additional Information", label="Additional Information",
lines=3, lines=3,
placeholder="Add any helpful context or instructions...", placeholder="Add any helpful context or instructions...",
info="Optional hints to help the LLM complete the task" info="Optional hints to help the LLM complete the task",
) )
with gr.Row(): with gr.Row():
@@ -414,54 +425,74 @@ def create_ui(theme_name="Ocean"):
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
final_result_output = gr.Textbox( final_result_output = gr.Textbox(
label="Final Result", label="Final Result", lines=3, show_label=True
lines=3,
show_label=True
) )
with gr.Column(): with gr.Column():
errors_output = gr.Textbox( errors_output = gr.Textbox(
label="Errors", label="Errors", lines=3, show_label=True
lines=3,
show_label=True
) )
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
model_actions_output = gr.Textbox( model_actions_output = gr.Textbox(
label="Model Actions", label="Model Actions", lines=3, show_label=True
lines=3,
show_label=True
) )
with gr.Column(): with gr.Column():
model_thoughts_output = gr.Textbox( model_thoughts_output = gr.Textbox(
label="Model Thoughts", label="Model Thoughts", lines=3, show_label=True
lines=3,
show_label=True
) )
# Run button click handler # Run button click handler
run_button.click( run_button.click(
fn=run_browser_agent, fn=run_browser_agent,
inputs=[ inputs=[
agent_type, llm_provider, llm_model_name, llm_temperature, agent_type,
llm_base_url, llm_api_key, use_own_browser, headless, llm_provider,
disable_security, window_w, window_h, save_recording_path, llm_model_name,
task, add_infos, max_steps, use_vision llm_temperature,
llm_base_url,
llm_api_key,
use_own_browser,
headless,
disable_security,
window_w,
window_h,
save_recording_path,
task,
add_infos,
max_steps,
use_vision,
],
outputs=[
final_result_output,
errors_output,
model_actions_output,
model_thoughts_output,
recording_display,
], ],
outputs=[final_result_output, errors_output, model_actions_output, model_thoughts_output, recording_display]
) )
return demo return demo
def main(): def main():
parser = argparse.ArgumentParser(description="Gradio UI for Browser Agent") parser = argparse.ArgumentParser(description="Gradio UI for Browser Agent")
parser.add_argument("--ip", type=str, default="127.0.0.1", help="IP address to bind to") parser.add_argument(
"--ip", type=str, default="127.0.0.1", help="IP address to bind to"
)
parser.add_argument("--port", type=int, default=7788, help="Port to listen on") parser.add_argument("--port", type=int, default=7788, help="Port to listen on")
parser.add_argument("--theme", type=str, default="Ocean", choices=theme_map.keys(), help="Theme to use for the UI") parser.add_argument(
"--theme",
type=str,
default="Ocean",
choices=theme_map.keys(),
help="Theme to use for the UI",
)
parser.add_argument("--dark-mode", action="store_true", help="Enable dark mode") parser.add_argument("--dark-mode", action="store_true", help="Enable dark mode")
args = parser.parse_args() args = parser.parse_args()
demo = create_ui(theme_name=args.theme) demo = create_ui(theme_name=args.theme)
demo.launch(server_name=args.ip, server_port=args.port) demo.launch(server_name=args.ip, server_port=args.port)
if __name__ == '__main__':
if __name__ == "__main__":
main() main()