From 2d2aa0d6893f77a5c99d5eab223b4646c94d71d1 Mon Sep 17 00:00:00 2001
From: meshkatshb <meshkat97@yahoo.com>
Date: Wed, 22 Jan 2025 11:58:49 +0330
Subject: [PATCH] feat: add default config to load from hard-code instead of a
 file

---
 default_config.pkl                   | Bin 554 -> 0 bytes
 src/utils/default_config_settings.py |  27 +++++++++++++++++++++++++++
 webui.py                             |   4 ++--
 3 files changed, 29 insertions(+), 2 deletions(-)
 delete mode 100644 default_config.pkl

diff --git a/default_config.pkl b/default_config.pkl
deleted file mode 100644
index 7d7b0c60df90e56470bd5d6e3e785ec025354be3..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 554
zcmY*Wv2GMG5Ivy-g(HxlL`sRf&~Rz!5mgi=Q9i-)*~6}Nw#Tw(Z@D5;N|0O*|H&s{
z?~*8S<$3RE-kX_+<)2Tl7Sp@u&t;`L2qbAAfy)~`WJ1s7tye!~CWx7@O*w6q(qu7Y
z86iy!U$4q)$RKw%SAu_)Q$lo7E9ayQQe&Xnx#EY;c^MP#tbtTbFhWq4%ZG-0G~i@V
zzS(z?zI;V4M~xxy)l1QkfG@xOog>f9Hmdp#$#Ge5dDp}$TDgTi_wUP53x#8ln}lPo
zj=#&tEkJy%W%IA#K@Z9}$hrQHjm>J~KxWWGvb3+Uk88>XCD53R4PC_!MjH%{a^%~^
z>!+&c+nZ%M$!hmZEu*5I%gOqJe7u;O)Lj|CTnC)Al$3`5y2DvFK~-=z3Fh1MwjL$z
zUp9O}aaoeeTP{}}3PRB#b`I7Wy->l3c3g3OSI7Q-buP5CdMh%hr2C2P><)wnK|P2c
LHf0iRwZr-!oCel5

diff --git a/src/utils/default_config_settings.py b/src/utils/default_config_settings.py
index 3af4dc2..4853ebf 100644
--- a/src/utils/default_config_settings.py
+++ b/src/utils/default_config_settings.py
@@ -3,6 +3,33 @@ import pickle
 import uuid
 
 
+def default_config():
+    """Prepare the default configuration"""
+    return {
+        "agent_type": "custom",
+        "max_steps": 100,
+        "max_actions_per_step": 10,
+        "use_vision": True,
+        "tool_call_in_content": True,
+        "llm_provider": "openai",
+        "llm_model_name": "gpt-4o",
+        "llm_temperature": 1.0,
+        "llm_base_url": "",
+        "llm_api_key": "",
+        "use_own_browser": False,
+        "keep_browser_open": False,
+        "headless": False,
+        "disable_security": True,
+        "enable_recording": True,
+        "window_w": 1280,
+        "window_h": 1100,
+        "save_recording_path": "./tmp/record_videos",
+        "save_trace_path": "./tmp/traces",
+        "save_agent_history_path": "./tmp/agent_history",
+        "task": "go to google.com and type 'OpenAI' click search and give me the first url",
+    }
+
+
 def load_config_from_file(config_file):
     """Load settings from a UUID.pkl file."""
     try:
diff --git a/webui.py b/webui.py
index 74fef9e..6e0d14a 100644
--- a/webui.py
+++ b/webui.py
@@ -39,7 +39,7 @@ from src.browser.config import BrowserPersistenceConfig
 from src.browser.custom_context import BrowserContextConfig, CustomBrowserContext
 from src.controller.custom_controller import CustomController
 from gradio.themes import Citrus, Default, Glass, Monochrome, Ocean, Origin, Soft, Base
-from src.utils.default_config_settings import load_config_from_file, save_config_to_file
+from src.utils.default_config_settings import default_config, load_config_from_file, save_config_to_file
 from src.utils.utils import update_model_dropdown, get_latest_files, capture_screenshot
 
 from dotenv import load_dotenv
@@ -1024,7 +1024,7 @@ def main():
     parser.add_argument("--dark-mode", action="store_true", help="Enable dark mode")
     args = parser.parse_args()
 
-    config_dict = load_config_from_file("./default_config.pkl") or {}
+    config_dict = default_config()
 
     demo = create_ui(config_dict, theme_name=args.theme)
     demo.launch(server_name=args.ip, server_port=args.port)