From 94ca383b82a6dc22f686ee36d376b66b8202e88d Mon Sep 17 00:00:00 2001
From: "tv0924@icloud.com" <j93es@naver.com>
Date: Mon, 9 Jun 2025 10:33:08 +0900
Subject: [PATCH] =?UTF-8?q?[Fix]=20=EC=97=90=EB=9F=AC=20=EB=B0=9C=EC=83=9D?=
 =?UTF-8?q?=20=EC=8B=9C,=20=EC=9E=AC=EC=8B=9C=EB=8F=84=20=ED=9B=84=20retur?=
 =?UTF-8?q?n=20=EB=90=98=EC=A7=80=20=EC=95=8A=EB=8A=94=20=EB=AC=B8?=
 =?UTF-8?q?=EC=A0=9C=20=ED=95=B4=EA=B2=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lib/prompt.py | 4 ++--
 main.py       | 9 +++++----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/lib/prompt.py b/lib/prompt.py
index 26ab2b4..9bb58e7 100644
--- a/lib/prompt.py
+++ b/lib/prompt.py
@@ -2,7 +2,7 @@
 extend_planner_system_message = """
 🎯 Mission: Collect Initial SSO Redirect URLs (For Browser Automation)
 
-※ **절대로 구글 검색, Bing 검색 등 어떤 외부 검색 기능도 사용하지 말고, 주어진 로그인 페이지 URL을 직접 방문하여 탐색하세요.**
+※ **모든 STEP에서 구글 검색, Bing 검색 등 어떤 외부 검색 기능도 절대 사용하지 않고, 초기에 주어진 URL에서 탐색하세요.**
 
 0. **초기 블록(Block) 체크**
    - 브라우저가 로그인 페이지에 접근하려 할 때, **페이지가 차단(blocked)** 되거나 **방화벽, CAPTCHA, 접근 제한** 등으로 인해 정상적으로 로드되지 않으면 즉시 프로세스를 종료하고 아래 JSON만 반환해야 합니다.  
@@ -17,7 +17,7 @@ extend_planner_system_message = """
    - 이후 단계로 절대 넘어가지 않도록 합니다.
 
 1. **로그인 페이지 탐색**
-   - **클라이언트(비엔터프라이즈) 로그인 페이지**로 직접 이동합니다. (검색 엔진을 사용하여 찾아서는 안 됩니다.)
+   - **클라이언트(비엔터프라이즈) 로그인 페이지**로 직접 이동합니다. **검색 엔진을 사용하여 찾아서는 안 됩니다.**
    - 접근 후 **개인정보/쿠키/동의 팝업**이 뜨면, 이를 반드시 **닫거나(Dismiss)** 처리하고 계속 진행합니다.
    - (이미 0단계에서 블록 여부를 확인했으므로, 이 단계에서는 페이지가 정상 로드되었다고 가정합니다.)
 
diff --git a/main.py b/main.py
index 92c0c00..bb22a49 100644
--- a/main.py
+++ b/main.py
@@ -151,17 +151,18 @@ async def scan_one_url(url: str, skip_html_check: bool = False):
             break
         
         except Exception as e:
-            if try_cnt >= 3:
+            await clean_resources(agent, context, browser)
+            
+            if try_cnt >= 1:
                 print(f"❌ {url} 스캔에 실패했습니다. 에러: {e}")
                 logger(f"❌ {url} 스캔에 실패했습니다. 에러: {e}")
-                break
+                return
             try_cnt += 1
             print(f"⚠️ 에러 발생: {e}. {try_cnt}번째 재시도 중...")
             
-            await clean_resources(agent, context, browser)
 
             # 1분 대기
-            await asyncio.sleep(60)
+            await asyncio.sleep(5)
             # 반복문을 통해 재시도
             continue