Update app.py

app.py

@@ -20,6 +20,10 @@ from selenium.common.exceptions import TimeoutException, WebDriverException
 from webdriver_manager.chrome import ChromeDriverManager
 from PIL import Image
 import requests
+from dotenv import load_dotenv
+
+# 환경변수 로드
+load_dotenv()
 
 # 전역 변수
 driver = None
@@ -31,7 +35,19 @@ def get_claude_client():
     api_key = os.getenv("ANTHROPIC_API_KEY")
     if not api_key:
         raise ValueError("ANTHROPIC_API_KEY 환경변수가 설정되지 않았습니다.")
-    return anthropic.Anthropic(api_key=api_key)
+    
+    try:
+        # 더 안전한 클라이언트 초기화
+        client = anthropic.Anthropic(
+            api_key=api_key,
+            max_retries=2,
+            timeout=60.0
+        )
+        return client
+    except Exception as e:
+        print(f"Claude 클라이언트 초기화 오류: {e}")
+        # 기본 초기화 방식으로 재시도
+        return anthropic.Anthropic(api_key=api_key)
 
 def init_browser():
     """브라우저를 초기화하고 Google Trends 페이지로 이동"""
@@ -52,7 +68,12 @@ def init_browser():
         chrome_options.add_argument("--disable-blink-features=AutomationControlled")
         chrome_options.add_argument("--disable-features=VizDisplayCompositor")
         chrome_options.add_argument("--window-size=1920,1080")
-        chrome_options.add_argument("--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
+        chrome_options.add_argument("--disable-background-timer-throttling")
+        chrome_options.add_argument("--disable-backgrounding-occluded-windows")
+        chrome_options.add_argument("--disable-renderer-backgrounding")
+        chrome_options.add_argument("--disable-features=TranslateUI")
+        chrome_options.add_argument("--disable-ipc-flooding-protection")
+        chrome_options.add_argument("--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36")
         
         # ChromeDriver 서비스 설정
         service = Service(ChromeDriverManager().install())
@@ -71,7 +92,7 @@ def init_browser():
         )
         
         # 추가 로딩 시간
-        time.sleep(3)
+        time.sleep(5)
         
         is_browser_ready = True
         print("브라우저 초기화 완료!")
@@ -182,6 +203,7 @@ def analyze_with_claude(image: Image.Image, user_prompt: str = "") -> str:
         1. 현재 가장 인기 있는 검색어들과 그 순위
         2. 각 검색어의 증가 추세나 특이사항
         3. 검색어들에서 발견되는 주요 트렌드나 패턴
+        4. 시사적/문화적 맥락에서의 해석
         
         한국어로 상세하고 구체적으로 분석해주세요.
         """
@@ -189,8 +211,8 @@ def analyze_with_claude(image: Image.Image, user_prompt: str = "") -> str:
         prompt = user_prompt if user_prompt.strip() else default_prompt
         
         response = client.messages.create(
-            model="claude-3-sonnet-20240229",
-            max_tokens=1000,
+            model="claude-3-5-sonnet-20241022",
+            max_tokens=1500,
             messages=[
                 {
                     "role": "user",
@@ -219,6 +241,24 @@ def analyze_with_claude(image: Image.Image, user_prompt: str = "") -> str:
         print(error_msg)
         return error_msg
 
+def reinit_browser():
+    """브라우저 재초기화"""
+    global driver, is_browser_ready
+    
+    print("브라우저 재초기화 시작...")
+    
+    # 기존 브라우저 정리
+    cleanup_browser()
+    is_browser_ready = False
+    
+    # 새로 초기화
+    init_browser()
+    
+    if is_browser_ready:
+        return "브라우저 재초기화 완료!"
+    else:
+        return "브라우저 재초기화 실패. 다시 시도해주세요."
+
 def refresh_trends_page():
     """Google Trends 페이지 새로고침"""
     global driver
@@ -319,6 +359,7 @@ def create_interface():
                 
                 analyze_btn = gr.Button("분석 시작", variant="primary")
                 refresh_btn = gr.Button("페이지 새로고침", variant="secondary")
+                reinit_btn = gr.Button("브라우저 재초기화", variant="secondary")
                 status_btn = gr.Button("브라우저 상태 확인", variant="secondary")
             
             with gr.Column(scale=2):
@@ -342,6 +383,11 @@ def create_interface():
             outputs=status_output
         )
         
+        reinit_btn.click(
+            fn=reinit_browser,
+            outputs=status_output
+        )
+        
         status_btn.click(
             fn=check_browser_status,
             outputs=status_output
@@ -355,13 +401,13 @@ def main():
     print("Google Trends 실시간 분석기 시작!")
     print("=" * 50)
     
-    # API 키 확인
-    try:
-        get_claude_client()
-        print("✓ Claude API 키 확인됨")
-    except ValueError as e:
-        print(f"✗ {e}")
+    # API 키 존재 여부만 확인 (실제 클라��언트 초기화는 나중에)
+    api_key = os.getenv("ANTHROPIC_API_KEY")
+    if not api_key:
+        print("✗ ANTHROPIC_API_KEY 환경변수가 설정되지 않았습니다.")
         return
+    else:
+        print("✓ Claude API 키 확인됨")
     
     print("✓ Gradio 웹 인터페이스 준비 중...")