Update app.py

app.py

@@ -3,13 +3,17 @@ import gradio as gr
 from datetime import datetime
 import random
 from selenium import webdriver
-from selenium.common.exceptions import WebDriverException
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+from selenium.common.exceptions import WebDriverException, TimeoutException
 from PIL import Image
 from io import BytesIO
 import base64
+import time
 
 def take_screenshot(url):
-    """웹사이트 스크린샷 촬영 함수"""
+    """웹사이트 스크린샷 촬영 함수 (로딩 대기 시간 추가)"""
     if not url.startswith('http'):
         url = f"https://{url}"
         
@@ -22,14 +26,33 @@ def take_screenshot(url):
     try:
         driver = webdriver.Chrome(options=options)
         driver.get(url)
-        driver.implicitly_wait(10)
+        
+        # 명시적 대기: 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")
         return base64.b64encode(buffered.getvalue()).decode()
+        
     except WebDriverException as e:
-        print(f"스크린샷 촬영 실패: {str(e)}")
+        print(f"스크린샷 촬영 실패: {str(e)} for URL: {url}")
+        return None
+    except Exception as e:
+        print(f"예상치 못한 오류: {str(e)} for URL: {url}")
         return None
     finally:
         if 'driver' in locals():
@@ -297,8 +320,6 @@ def get_vercel_card(deployment, index, is_top_best=False):
         url = f"https://{raw_url}"
     
     name = deployment.get('name', '이름 없는 프로젝트')
-    created = format_timestamp(deployment.get('created'))
-    state = deployment.get('state', 'N/A')
     
     # 카드 ID 생성
     card_id = f"vercel-card-{url.replace('.', '-').replace('/', '-')}"
@@ -307,6 +328,7 @@ def get_vercel_card(deployment, index, is_top_best=False):
     screenshot_html = ""
     if is_top_best:
         try:
+            print(f"스크린샷 캡처 시도: {url}")  # 디버깅용 로그
             screenshot_base64 = take_screenshot(raw_url)
             if screenshot_base64:
                 screenshot_html = f"""
@@ -316,14 +338,16 @@ def get_vercel_card(deployment, index, is_top_best=False):
                              alt="{name} 스크린샷"/>
                     </div>
                 """
+            else:
+                print(f"스크린샷 캡처 실패: {url}")  # 디버깅용 로그
         except Exception as e:
-            print(f"스크린샷 처리 오류: {str(e)}")
+            print(f"스크린샷 처리 오류: {str(e)} for URL: {url}")  # 디버깅용 로그
     
     bg_color = get_pastel_color(index + (20 if not is_top_best else 0))
     tech_emojis = ['⚡', '🚀', '🌟', '✨', '💫', '🔥', '🌈', '🎯', '🎨', '🔮']
     random_emojis = random.sample(tech_emojis, 3)
 
-    # Top Best와 일반 카드의 내용을 다르게 구성
+    # Top Best 카드의 간소화된 정보 섹션
     if is_top_best:
         info_section = f"""
             <div style='margin: 15px 0; color: #444; background: rgba(255,255,255,0.5); 
@@ -338,10 +362,10 @@ def get_vercel_card(deployment, index, is_top_best=False):
             <div style='margin: 15px 0; color: #444; background: rgba(255,255,255,0.5); 
                         padding: 15px; border-radius: 12px;'>
                 <p style='margin: 8px 0;'>
-                    <strong>Status:</strong> ✅ {state}
+                    <strong>Status:</strong> ✅ {deployment.get('state', 'N/A')}
                 </p>
                 <p style='margin: 8px 0;'>
-                    <strong>Created:</strong> 📅 {created}
+                    <strong>Created:</strong> 📅 {format_timestamp(deployment.get('created'))}
                 </p>
                 <p style='margin: 8px 0;'>
                     <strong>URL:</strong> 🔗 {url}