Update app.py

app.py

@@ -18,14 +18,16 @@ from pathlib import Path
 CACHE_DIR = Path("screenshot_cache")
 CACHE_DIR.mkdir(exist_ok=True)
 
-def get_cached_screenshot(url: str) -> str:
-    """캐시된 스크린샷 가져오기 또는 새로 생성"""
-    cache_file = CACHE_DIR / f"{base64.b64encode(url.encode()).decode()}.png"
-    
-    if cache_file.exists():
-        with open(cache_file, "rb") as f:
-            return base64.b64encode(f.read()).decode()
-            
+import time  # time.sleep() 사용을 위해 추가
+
+def take_screenshot(url):
+    """웹사이트 스크린샷 촬영 함수 (로딩 대기 시간 추가)"""
+    if url in SCREENSHOT_CACHE:
+        return SCREENSHOT_CACHE[url]
+        
+    if not url.startswith('http'):
+        url = f"https://{url}"
+        
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
@@ -35,18 +37,37 @@ def get_cached_screenshot(url: str) -> str:
     try:
         driver = webdriver.Chrome(options=options)
         driver.get(url)
-        WebDriverWait(driver, 10).until(
-            EC.presence_of_element_located((By.TAG_NAME, "body"))
-        )
+        
+        # 명시적 대기: body 요소가 로드될 때까지 대기 (최대 10초)
+        try:
+            WebDriverWait(driver, 10).until(
+                EC.presence_of_element_located((By.TAG_NAME, "body"))
+            )
+        except TimeoutException:
+            print(f"페이지 로딩 타임아웃: {url}")
+        
+        # 추가 대기 시간 (1초)
+        time.sleep(1)
+        
+        # JavaScript 실행 완료 대기
+        driver.execute_script("return document.readyState") == "complete"
+        
+        # 스크린샷 촬영
         screenshot = driver.get_screenshot_as_png()
+        img = Image.open(BytesIO(screenshot))
+        buffered = BytesIO()
+        img.save(buffered, format="PNG")
+        base64_image = base64.b64encode(buffered.getvalue()).decode()
         
-        # 캐시 파일 저장
-        with open(cache_file, "wb") as f:
-            f.write(screenshot)
-            
-        return base64.b64encode(screenshot).decode()
+        # 캐시에 저장
+        SCREENSHOT_CACHE[url] = base64_image
+        return base64_image
+        
+    except WebDriverException as e:
+        print(f"스크린샷 촬영 실패: {str(e)} for URL: {url}")
+        return None
     except Exception as e:
-        print(f"Screenshot error for {url}: {str(e)}")
+        print(f"예상치 못한 오류: {str(e)} for URL: {url}")
         return None
     finally:
         if 'driver' in locals():
@@ -109,8 +130,7 @@ def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
         response.raise_for_status()
         spaces = response.json()
         
-        # 좋아요 수로 정렬하고 상위 10개만 선택
-        spaces.sort(key=lambda x: x.get('likes', 0), reverse=True)
+        # 상위 10개만 선택 (원본 순서 유지)
         top_spaces = spaces[:10]
         
         progress(0.1, desc="Creating gallery...")
@@ -125,7 +145,7 @@ def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
             likes = format(space.get('likes', 0), ',')
             created = space.get('createdAt', '').split('T')[0]
             
-            screenshot = get_cached_screenshot(space_url)
+            screenshot = take_screenshot(space_url)
             bg_color = f"rgba({random.randint(230,255)}, {random.randint(230,255)}, {random.randint(230,255)}, 0.8)"
             
             html_content += f"""