Update app.py

app.py

@@ -15,20 +15,21 @@ from selenium.common.exceptions import TimeoutException, NoSuchElementException
 from concurrent.futures import ThreadPoolExecutor
 import threading
 
-# Hugging Face Space 환경 확인
-IS_HUGGINGFACE = os.getenv("SPACE_ID") is not None
-print(f'IS_HUGGINGFACE = {IS_HUGGINGFACE}')
+# .env 파일 로드 (있는 경우)
+try:
+    from dotenv import load_dotenv
+    load_dotenv()
+    print(".env 파일 로드 완료")
+except ImportError:
+    print("python-dotenv가 설치되지 않았습니다. 환경변수를 수동으로 설정해주세요.")
+except Exception as e:
+    print(f".env 파일 로드 중 오류: {e}")
 
-# Claude API 설정 (Hugging Face Secrets에서 가져오기)
+# Claude API 설정
 ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
 
-if not ANTHROPIC_API_KEY:
-    print("ANTHROPIC_API_KEY가 설정되지 않음, Secret 확인 필요")
-else:
-    print("Claude API 키 확인 완료")
-
-# Claude 클라이언트를 생성
 def create_claude_client():
+    """Claude 클라이언트를 생성하는 함수 (빠른 버전)"""
     try:
         client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY)
         print("Claude API 클라이언트 생성 성공")
@@ -37,36 +38,25 @@ def create_claude_client():
         print(f"Claude API 클라이언트 생성 실패: {e}")
         return None
 
-# Google Trends 자동화 클래스
 class GoogleTrendsAutomator:
+    """최적화된 Google Trends 자동화 클래스"""
+    
     def __init__(self):
         self.driver = None
-    
-    # Chrome 드라이버 설정
+        
     def setup_driver(self):
+        """초고속 Chrome 드라이버 설정"""
         options = Options()
         
-        # Hugging Face Space 환경 설정
-        if IS_HUGGINGFACE:
-            print("Hugging Face Space 환경, 헤드리스 모드로 실행")
-            options.add_argument("--headless=new")  # 새로운 헤드리스 모드
+        # 환경별 설정
+        if os.getenv("SPACE_ID"):  # Hugging Face Space
+            options.add_argument("--headless")
             options.add_argument("--no-sandbox")
             options.add_argument("--disable-dev-shm-usage")
             options.add_argument("--disable-gpu")
-            options.add_argument("--disable-software-rasterizer")
-            options.add_argument("--remote-debugging-port=9222")
-        else:
+        else:  # 로컬 환경
             options.add_argument("--headless")
         
-        # # 환경별 설정
-        # if os.getenv("SPACE_ID"):  # Hugging Face Space
-        #     options.add_argument("--headless")
-        #     options.add_argument("--no-sandbox")
-        #     options.add_argument("--disable-dev-shm-usage")
-        #     options.add_argument("--disable-gpu")
-        # else:  # 로컬 환경
-        #     options.add_argument("--headless")
-        
         # 속도 최적화 설정
         options.add_argument("--window-size=1280,720")  # 해상도 줄임
         options.add_argument("--disable-blink-features=AutomationControlled")
@@ -85,8 +75,7 @@ class GoogleTrendsAutomator:
         
         # 메모리 최적화
         options.add_argument("--memory-pressure-off")
-        # options.add_argument("--max_old_space_size=4096")
-        options.add_argument("--max_old_space_size=2048")# 메모리 제한
+        options.add_argument("--max_old_space_size=4096")
         
         # 이미지/CSS 비활성화로 로딩 속도 향상
         prefs = {
@@ -105,66 +94,34 @@ class GoogleTrendsAutomator:
         options.add_argument("--accept-lang=ko-KR,ko;q=0.9,en;q=0.8")
         
         try:
-            # Hugging Face 환경에서는 webdriver-manager 우선 사용
-            if IS_HUGGINGFACE:
-                try:
-                    from webdriver_manager.chrome import ChromeDriverManager
-                    from selenium.webdriver.chrome.service import Service
-                    
-                    print("ChromeDriver 자동 설치 중...")
-                    service = Service(ChromeDriverManager().install())
-                    self.driver = webdriver.Chrome(service=service, options=options)
-                    print("ChromeDriver 설치 및 초기화 성공")
-                    
-                except Exception as e:
-                    print(f"webdriver-manager 설치 실패: {e}")
-                    # 시스템 ChromeDriver 시도
-                    try:
-                        self.driver = webdriver.Chrome(options=options)
-                        print("시스템 ChromeDriver 사용 성공")
-                    except Exception as sys_e:
-                        print(f"시스템 ChromeDriver도 실패: {sys_e}")
-                        return False
-            else:
-                # 로컬 환경
-                try:
-                    from webdriver_manager.chrome import ChromeDriverManager
-                    from selenium.webdriver.chrome.service import Service
-                    
-                    service = Service(ChromeDriverManager().install())
-                    self.driver = webdriver.Chrome(service=service, options=options)
-                    print("로컬 ChromeDriver 설치 성공")
-                except:
-                    self.driver = webdriver.Chrome(options=options)
-                    print("로컬 시스템 ChromeDriver 사용")
-            # # 여러 방법으로 ChromeDriver 시도
-            # driver_created = False
-            
-            # # 방법 1: webdriver-manager 사용
-            # try:
-            #     from webdriver_manager.chrome import ChromeDriverManager
-            #     from selenium.webdriver.chrome.service import Service
+            # 여러 방법으로 ChromeDriver 시도
+            driver_created = False
+            
+            # 방법 1: webdriver-manager 사용
+            try:
+                from webdriver_manager.chrome import ChromeDriverManager
+                from selenium.webdriver.chrome.service import Service
                 
-            #     service = Service(ChromeDriverManager().install())
-            #     self.driver = webdriver.Chrome(service=service, options=options)
-            #     print("webdriver-manager로 ChromeDriver 자동 설치 성공")
-            #     driver_created = True
+                service = Service(ChromeDriverManager().install())
+                self.driver = webdriver.Chrome(service=service, options=options)
+                print("webdriver-manager로 ChromeDriver 자동 설치 성공")
+                driver_created = True
                 
-            # except Exception as wm_error:
-            #     print(f"webdriver-manager 실패: {wm_error}")
-            
-            # # 방법 2: 시스템 ChromeDriver 사용
-            # if not driver_created:
-            #     try:
-            #         self.driver = webdriver.Chrome(options=options)
-            #         print("시스템 ChromeDriver 사용 성공")
-            #         driver_created = True
-            #     except Exception as sys_error:
-            #         print(f"시스템 ChromeDriver 실패: {sys_error}")
-            
-            # if not driver_created:
-            #     print("모든 ChromeDriver 초기화 방법 실패")
-            #     return False
+            except Exception as wm_error:
+                print(f"webdriver-manager 실패: {wm_error}")
+            
+            # 방법 2: 시스템 ChromeDriver 사용
+            if not driver_created:
+                try:
+                    self.driver = webdriver.Chrome(options=options)
+                    print("시스템 ChromeDriver 사용 성공")
+                    driver_created = True
+                except Exception as sys_error:
+                    print(f"시스템 ChromeDriver 실패: {sys_error}")
+            
+            if not driver_created:
+                print("모든 ChromeDriver 초기화 방법 실패")
+                return False
             
             # 타임아웃 설정 (속도 최적화)
             self.driver.set_page_load_timeout(15)  # 페이지 로딩 타임아웃 15초
@@ -186,8 +143,8 @@ class GoogleTrendsAutomator:
             print(f"드라이버 설정 최종 실패: {e}")
             return False
     
-    # 빠르게 Element 찾기, Timeout 단축 고려
     def safe_find_element(self, selectors: list, timeout: int = 3):
+        """초고속 요소 찾기 (타임아웃 단축)"""
         for selector_type, selector in selectors:
             try:
                 if selector_type == "xpath":
@@ -209,8 +166,8 @@ class GoogleTrendsAutomator:
                 continue
         return None
     
-    # 즉시 클릭
     def safe_click(self, selectors: list, timeout: int = 3) -> bool:
+        """즉시 클릭 (대기 시간 최소화)"""
         element = self.safe_find_element(selectors, timeout)
         if element:
             try:
@@ -225,8 +182,8 @@ class GoogleTrendsAutomator:
                     return False
         return False
     
-    # 병렬로 설정 변경 시도
     def parallel_change_settings(self, region: str, period: str, category: str, progress_callback=None) -> dict:
+        """병렬로 설정 변경 시도 (실험적)"""
         results = {
             'region': False,
             'period': False, 
@@ -265,8 +222,8 @@ class GoogleTrendsAutomator:
         
         return results
     
-    # 지역 변경
     def change_region(self, target_region: str) -> bool:
+        """초고속 지역 변경"""
         try:
             region_selectors = [
                 ("xpath", "//button[@aria-label='대한민국, 위치 선택']"),
@@ -302,8 +259,8 @@ class GoogleTrendsAutomator:
             print(f"지역 변경 실패: {e}")
             return False
     
-    # 기간 변경
     def change_time_period(self, target_period: str) -> bool:
+        """초고속 기간 변경"""
         try:
             period_selectors = [
                 ("xpath", "//button[contains(@aria-label, '기간 선택')]"),
@@ -338,8 +295,8 @@ class GoogleTrendsAutomator:
             print(f"기간 변경 실패: {e}")
             return False
     
-    # 카테고리 ��경
     def change_category(self, target_category: str) -> bool:
+        """초고속 카테고리 변경"""
         try:
             category_selectors = [
                 ("xpath", "//button[contains(@aria-label, '카테고리 선택')]"),
@@ -377,8 +334,8 @@ class GoogleTrendsAutomator:
             print(f"카테고리 변경 실패: {e}")
             return False
     
-    # Google Trends 캡처
     def capture_trends(self, region: str, period: str, category: str, progress_callback=None) -> Tuple[Optional[str], bool, str]:
+        """Google Trends 캡처 (최적화된 버전)"""
         error_msg = ""
         
         try:
@@ -446,8 +403,8 @@ class GoogleTrendsAutomator:
             if self.driver:
                 self.driver.quit()
 
-# Claude API로 스크린샷 분석
 def analyze_with_claude(screenshot_b64: str, region: str, period: str, category: str) -> str:
+    """Claude API로 스크린샷 분석 (최적화된 버전)"""
     try:
         # API 키 체크
         if not ANTHROPIC_API_KEY or ANTHROPIC_API_KEY == "your_api_key_here":