Skip to content

Commit dce50eb

Browse files
MeshkatShBMeshkatShB
committed
refactor: moved functions to utils for more readability
1 parent 0e152d5 commit dce50eb

File tree

2 files changed

+74
-71
lines changed

2 files changed

+74
-71
lines changed

‎src/utils/default_config_settings.py‎

Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os
22
import pickle
33
import uuid
4+
import gradio as gr
45

56

67
def default_config():
@@ -47,3 +48,75 @@ def save_config_to_file(settings, save_dir="./tmp/webui_settings"):
4748
with open(config_file, 'wb') as f:
4849
pickle.dump(settings, f)
4950
return f"Configuration saved to {config_file}"
51+
52+
53+
def save_current_config(*args):
54+
current_config = {
55+
"agent_type": args[0],
56+
"max_steps": args[1],
57+
"max_actions_per_step": args[2],
58+
"use_vision": args[3],
59+
"tool_call_in_content": args[4],
60+
"llm_provider": args[5],
61+
"llm_model_name": args[6],
62+
"llm_temperature": args[7],
63+
"llm_base_url": args[8],
64+
"llm_api_key": args[9],
65+
"use_own_browser": args[10],
66+
"keep_browser_open": args[11],
67+
"headless": args[12],
68+
"disable_security": args[13],
69+
"enable_recording": args[14],
70+
"window_w": args[15],
71+
"window_h": args[16],
72+
"save_recording_path": args[17],
73+
"save_trace_path": args[18],
74+
"save_agent_history_path": args[19],
75+
"task": args[20],
76+
}
77+
return save_config_to_file(current_config)
78+
79+
80+
def update_ui_from_config(config_file):
81+
if config_file is not None:
82+
loaded_config = load_config_from_file(config_file.name)
83+
if isinstance(loaded_config, dict):
84+
return (
85+
gr.update(value=loaded_config.get("agent_type", "custom")),
86+
gr.update(value=loaded_config.get("max_steps", 100)),
87+
gr.update(value=loaded_config.get("max_actions_per_step", 10)),
88+
gr.update(value=loaded_config.get("use_vision", True)),
89+
gr.update(value=loaded_config.get("tool_call_in_content", True)),
90+
gr.update(value=loaded_config.get("llm_provider", "openai")),
91+
gr.update(value=loaded_config.get("llm_model_name", "gpt-4o")),
92+
gr.update(value=loaded_config.get("llm_temperature", 1.0)),
93+
gr.update(value=loaded_config.get("llm_base_url", "")),
94+
gr.update(value=loaded_config.get("llm_api_key", "")),
95+
gr.update(value=loaded_config.get("use_own_browser", False)),
96+
gr.update(value=loaded_config.get("keep_browser_open", False)),
97+
gr.update(value=loaded_config.get("headless", False)),
98+
gr.update(value=loaded_config.get("disable_security", True)),
99+
gr.update(value=loaded_config.get("enable_recording", True)),
100+
gr.update(value=loaded_config.get("window_w", 1280)),
101+
gr.update(value=loaded_config.get("window_h", 1100)),
102+
gr.update(value=loaded_config.get("save_recording_path", "./tmp/record_videos")),
103+
gr.update(value=loaded_config.get("save_trace_path", "./tmp/traces")),
104+
gr.update(value=loaded_config.get("save_agent_history_path", "./tmp/agent_history")),
105+
gr.update(value=loaded_config.get("task", "")),
106+
"Configuration loaded successfully."
107+
)
108+
else:
109+
return (
110+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
111+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
112+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
113+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
114+
gr.update(), "Error: Invalid configuration file."
115+
)
116+
return (
117+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
118+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
119+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
120+
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
121+
gr.update(), "No file selected."
122+
)

‎webui.py‎

Lines changed: 1 addition & 71 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
from src.browser.custom_context import BrowserContextConfig, CustomBrowserContext
4040
from src.controller.custom_controller import CustomController
4141
from gradio.themes import Citrus, Default, Glass, Monochrome, Ocean, Origin, Soft, Base
42-
from src.utils.default_config_settings import default_config, load_config_from_file, save_config_to_file
42+
from src.utils.default_config_settings import default_config, load_config_from_file, save_config_to_file, save_current_config, update_ui_from_config
4343
from src.utils.utils import update_model_dropdown, get_latest_files, capture_screenshot
4444

4545
from dotenv import load_dotenv
@@ -809,50 +809,6 @@ def create_ui(config, theme_name="Ocean"):
809809
interactive=False
810810
)
811811

812-
def update_ui_from_config(config_file):
813-
if config_file is not None:
814-
loaded_config = load_config_from_file(config_file.name)
815-
if isinstance(loaded_config, dict):
816-
return (
817-
gr.update(value=loaded_config.get("agent_type", "custom")),
818-
gr.update(value=loaded_config.get("max_steps", 100)),
819-
gr.update(value=loaded_config.get("max_actions_per_step", 10)),
820-
gr.update(value=loaded_config.get("use_vision", True)),
821-
gr.update(value=loaded_config.get("tool_call_in_content", True)),
822-
gr.update(value=loaded_config.get("llm_provider", "openai")),
823-
gr.update(value=loaded_config.get("llm_model_name", "gpt-4o")),
824-
gr.update(value=loaded_config.get("llm_temperature", 1.0)),
825-
gr.update(value=loaded_config.get("llm_base_url", "")),
826-
gr.update(value=loaded_config.get("llm_api_key", "")),
827-
gr.update(value=loaded_config.get("use_own_browser", False)),
828-
gr.update(value=loaded_config.get("keep_browser_open", False)),
829-
gr.update(value=loaded_config.get("headless", False)),
830-
gr.update(value=loaded_config.get("disable_security", True)),
831-
gr.update(value=loaded_config.get("enable_recording", True)),
832-
gr.update(value=loaded_config.get("window_w", 1280)),
833-
gr.update(value=loaded_config.get("window_h", 1100)),
834-
gr.update(value=loaded_config.get("save_recording_path", "./tmp/record_videos")),
835-
gr.update(value=loaded_config.get("save_trace_path", "./tmp/traces")),
836-
gr.update(value=loaded_config.get("save_agent_history_path", "./tmp/agent_history")),
837-
gr.update(value=loaded_config.get("task", "")),
838-
"Configuration loaded successfully."
839-
)
840-
else:
841-
return (
842-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
843-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
844-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
845-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
846-
gr.update(), "Error: Invalid configuration file."
847-
)
848-
return (
849-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
850-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
851-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
852-
gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
853-
gr.update(), "No file selected."
854-
)
855-
856812
load_config_button.click(
857813
fn=update_ui_from_config,
858814
inputs=[config_file_input],
@@ -865,32 +821,6 @@ def update_ui_from_config(config_file):
865821
]
866822
)
867823

868-
def save_current_config(*args):
869-
current_config = {
870-
"agent_type": args[0],
871-
"max_steps": args[1],
872-
"max_actions_per_step": args[2],
873-
"use_vision": args[3],
874-
"tool_call_in_content": args[4],
875-
"llm_provider": args[5],
876-
"llm_model_name": args[6],
877-
"llm_temperature": args[7],
878-
"llm_base_url": args[8],
879-
"llm_api_key": args[9],
880-
"use_own_browser": args[10],
881-
"keep_browser_open": args[11],
882-
"headless": args[12],
883-
"disable_security": args[13],
884-
"enable_recording": args[14],
885-
"window_w": args[15],
886-
"window_h": args[16],
887-
"save_recording_path": args[17],
888-
"save_trace_path": args[18],
889-
"save_agent_history_path": args[19],
890-
"task": args[20],
891-
}
892-
return save_config_to_file(current_config)
893-
894824
save_config_button.click(
895825
fn=save_current_config,
896826
inputs=[

0 commit comments

Comments
 (0)