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

optimize code

Browse Source
This commit is contained in:
katiue
2025-01-11 00:00:14 +07:00
parent 808d40cd73
commit 7a86580397
1 changed files with 23 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
webui.py
View File

@@ -86,7 +86,7 @@ async def run_browser_agent(
)
if agent_type == "org":
final_result, errors, model_actions, model_thoughts, recorded_files, trace_file = await run_org_agent(
final_result, errors, model_actions, model_thoughts = await run_org_agent(
llm=llm,
headless=headless,
disable_security=disable_security,
@@ -158,19 +158,21 @@ async def run_org_agent(
trace_path=save_trace_path if save_trace_path else None,
save_recording_path=save_recording_path if save_recording_path else None,
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:
agent = Agent(
task=task,
llm=llm,
use_vision=use_vision,
max_actions_per_step=max_actions_per_step,
tool_call_in_content=tool_call_in_content,
browser_context=browser_context,
max_actions_per_step=max_actions_per_step,
tool_call_in_content=tool_call_in_content
)
history = await agent.run(max_steps=max_steps)
final_result = history.final_result()
errors = history.errors()
model_actions = history.model_actions()
@@ -196,7 +198,7 @@ async def run_custom_agent(
max_steps,
use_vision,
max_actions_per_step,
tool_call_in_content,
tool_call_in_content
):
global _global_browser, _global_browser_context, _global_playwright
@@ -636,22 +638,22 @@ def create_ui(theme_name="Ocean"):
)
with gr.TabItem("🤖 Run Agent", id=4):
task = gr.Textbox(
lines=4,
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",
)
add_infos = gr.Textbox(lines=3, label="Additional Information")
task = gr.Textbox(
lines=4,
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",
)
add_infos = gr.Textbox(lines=3, label="Additional Information")
with gr.Row():
run_button = gr.Button("▶️ Run Agent", variant="primary", scale=2)
stop_button = gr.Button("⏹️ Stop", variant="stop", scale=1)
with gr.Row():
run_button = gr.Button("▶️ Run Agent", variant="primary", scale=2)
stop_button = gr.Button("⏹️ Stop", variant="stop", scale=1)
with gr.Row():
browser_view = gr.HTML(
value="<div>Waiting for browser session...</div>",
label="Live Browser View",
)
with gr.Row():
browser_view = gr.HTML(
value="<div>Waiting for browser session...</div>",
label="Live Browser View",
)
with gr.TabItem("📊 Results", id=5):
with gr.Group():