Update app.py

app.py

@@ -2,6 +2,32 @@ import requests
 import gradio as gr
 from datetime import datetime
 import random
+from selenium import webdriver
+from selenium.common.exceptions import WebDriverException
+from PIL import Image
+from io import BytesIO
+
+def take_screenshot(url):
+    """웹사이트 스크린샷 촬영 함수"""
+    options = webdriver.ChromeOptions()
+    options.add_argument('--headless')
+    options.add_argument('--no-sandbox')
+    options.add_argument('--disable-dev-shm-usage')
+    
+    try:
+        driver = webdriver.Chrome(options=options)
+        driver.set_window_size(1080, 720)  # 스크린샷 크기 설정
+        driver.get(url)
+        driver.implicitly_wait(10)
+        screenshot = driver.get_screenshot_as_png()
+        return Image.open(BytesIO(screenshot))
+    except WebDriverException as e:
+        print(f"스크린샷 촬영 실패: {str(e)}")
+        return Image.new('RGB', (1, 1))  # 오류 시 빈 이미지 반환
+    finally:
+        if driver:
+            driver.quit()
+
 
 USERNAME = "openfree"
 
@@ -254,15 +280,55 @@ def get_vercel_deployments():
         return []
 
 def get_vercel_card(deployment, index):
-    """Generate HTML card for a Vercel deployment with like button"""
+    """Vercel 배포 카드 HTML 생성 함수"""
     raw_url = deployment.get('url', '')
     
-    # URL 처리 로직 수정
+    # URL 처리
     if raw_url.startswith('http'):
-        url = raw_url  # 전체 URL이 제공된 경우 그대로 사용
+        url = raw_url
     else:
-        project_name = raw_url[:6] if len(raw_url) >= 6 else raw_url
-        url = f"{project_name}.vercel.app"
+        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('/', '-')}"
+
+    
+    # 스크린샷 이미지 가져오기
+    try:
+        screenshot = take_screenshot(url)
+        screenshot_html = f"""
+            <div style="width: 100%; height: 200px; overflow: hidden; border-radius: 10px; margin-bottom: 15px;">
+                <img src="data:image/png;base64,{screenshot}" 
+                     style="width: 100%; height: 100%; object-fit: cover;"
+                     alt="{name} 스크린샷"/>
+            </div>
+        """
+    except Exception as e:
+        print(f"스크린샷 처리 오류: {str(e)}")
+        screenshot_html = ""  # 오류 시 스크린샷 영역 생략
+
+    # 나머지 카드 스타일링 코드는 기존과 동일...
+    return f"""
+    <div id="{card_id}" class="vercel-card"
+         style='border: none; 
+                padding: 25px; 
+                margin: 15px; 
+                border-radius: 20px; 
+                background-color: {get_pastel_color(index)};
+                box-shadow: 0 4px 15px rgba(0,0,0,0.1);'>
+        {screenshot_html}
+        <h3>{name}</h3>
+        <div style='margin: 15px 0;'>
+            <p>상태: {state}</p>
+            <p>생성일: {created}</p>
+            <p>URL: <a href="{url}" target="_blank">{url}</a></p>
+        </div>
+    </div>
+    """    
     
     # Hugging Face 스페이스 URL인 경우 직접 사용
     if 'huggingface.co' in url:
@@ -685,4 +751,7 @@ with demo:
     html_output = gr.HTML(value=get_user_spaces())  # 초기 로드 시 직접 함수 호출
 
 if __name__ == "__main__":
+    demo = gr.Blocks()
+    with demo:
+        gr.HTML(value=get_user_spaces())
     demo.launch()