diff --git a/lib/utils/browser_use/__init__.py b/lib/utils/browser_use/__init__.py index cde80d6..11168e1 100644 --- a/lib/utils/browser_use/__init__.py +++ b/lib/utils/browser_use/__init__.py @@ -5,24 +5,28 @@ proxy_url = setup_proxy() storage_state_path = setup_storage_state() # Create browser profile -profile = BrowserProfile( - # Security settings - disable_security=True, - stealth=True, - - # Display settings - headless=False, - device_scale_factor=1, - window_size={"width": 1600, "height": 900}, - viewport={"width": 1600, "height": 900}, - - # Data persistence - user_data_dir=None, - storage_state=storage_state_path, - - # Network settings - proxy={"server": proxy_url} if proxy_url else None, - - # Additional arguments - args=get_browser_args(), -) \ No newline at end of file +def GetProfile(): + setup_storage_state() + profile = BrowserProfile( + # Security settings + disable_security=True, + stealth=True, + + # Display settings + headless=False, + device_scale_factor=1, + window_size={"width": 1600, "height": 900}, + viewport={"width": 1600, "height": 900}, + + # Data persistence + user_data_dir=None, + storage_state=storage_state_path, + + # Network settings + proxy={"server": proxy_url} if proxy_url else None, + + # Additional arguments + args=get_browser_args(), + ) + + return profile \ No newline at end of file diff --git a/main.py b/main.py index 87a2554..8b150a1 100644 --- a/main.py +++ b/main.py @@ -103,7 +103,7 @@ async def scan_one_url(url: str, skip_html_check: bool = False): # BrowserSession에 profile 전달 session = BrowserSession( playwright=(await async_patchright().start()), - browser_profile=browser_use.profile, + browser_profile=browser_use.GetProfile(), ) # Agent 생성 및 실행 (단일 try-except with 백오프)