Update app-backup.py

app-backup.py

@@ -2,6 +2,38 @@ 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
+import base64
+
+def take_screenshot(url):
+    """웹사이트 스크린샷 촬영 함수"""
+    if not url.startswith('http'):
+        url = f"https://{url}"
+        
+    options = webdriver.ChromeOptions()
+    options.add_argument('--headless')
+    options.add_argument('--no-sandbox')
+    options.add_argument('--disable-dev-shm-usage')
+    options.add_argument('--window-size=1080,720')
+    
+    try:
+        driver = webdriver.Chrome(options=options)
+        driver.get(url)
+        driver.implicitly_wait(10)
+        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)}")
+        return None
+    finally:
+        if 'driver' in locals():
+            driver.quit()
 
 USERNAME = "openfree"
 
@@ -193,9 +225,13 @@ def get_space_card(space, index):
     """
 
 def get_vercel_deployments():
-    """Vercel API를 통해 배포된 서비스 정보 가져오기"""
+    """Vercel API를 통해 모든 배포된 서비스 정보 가져오기 (페이지네이션 적용)"""
     token = "A8IFZmgW2cqA4yUNlLPnci0N"
-    url = "https://api.vercel.com/v6/deployments?limit=100"  # limit 파라미터 추가
+    base_url = "https://api.vercel.com/v6/deployments"
+    all_deployments = []
+    has_next = True
+    page = 1
+    until = None  # 첫 요청에서는 until 파라미터 없음
     
     headers = {
         "Authorization": f"Bearer {token}",
@@ -203,57 +239,115 @@ def get_vercel_deployments():
     }
     
     try:
-        response = requests.get(url, headers=headers)
-        if response.status_code != 200:
-            print(f"Vercel API Error: {response.text}")
-            return []
+        while has_next:
+            # URL 구성 (페이지네이션 파라미터 포함)
+            url = f"{base_url}?limit=100"
+            if until:
+                url += f"&until={until}"
+            
+            print(f"Fetching page {page}... URL: {url}")  # 디버깅용
+            
+            response = requests.get(url, headers=headers)
+            if response.status_code != 200:
+                print(f"Vercel API Error: {response.text}")
+                break
+            
+            data = response.json()
+            current_deployments = data.get('deployments', [])
+            
+            if not current_deployments:  # 더 이상 데이터가 없으면 종료
+                break
+                
+            all_deployments.extend(current_deployments)
+            
+            # 다음 페이지를 위한 until 값 설정
+            pagination = data.get('pagination', {})
+            until = pagination.get('next')
+            has_next = bool(until)  # until 값이 있으면 다음 페이지 존재
             
-        deployments = response.json().get('deployments', [])
+            print(f"Page {page} fetched. Got {len(current_deployments)} deployments")  # 디버깅용
+            page += 1
+        
+        print(f"Total deployments fetched: {len(all_deployments)}")  # 디버깅용
         
         # 상태가 'READY'이고 'url'이 있는 배포만 필터링하고 'javis1' 제외
         active_deployments = [
-            dep for dep in deployments 
+            dep for dep in all_deployments 
             if dep.get('state') == 'READY' and 
                dep.get('url') and 
-               'javis1' not in dep.get('name', '').lower()  # javis1 제외
+               'javis1' not in dep.get('name', '').lower()
         ]
         
+        print(f"Active deployments after filtering: {len(active_deployments)}")  # 디버깅용
         return active_deployments
+        
     except Exception as e:
         print(f"Error fetching Vercel deployments: {str(e)}")
         return []
 
 
-def get_vercel_card(deployment, index):
-    """Generate HTML card for a Vercel deployment with like button"""
+def get_vercel_card(deployment, index, is_top_best=False):
+    """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"
-    
-    # Hugging Face 스페이스 URL인 경우 직접 사용
-    if 'huggingface.co' in url:
-        final_url = url
-    else:
-        final_url = f"https://{url}" if not url.startswith('http') else url
+        url = f"https://{raw_url}"
     
+    name = deployment.get('name', '이름 없는 프로젝트')
     created = format_timestamp(deployment.get('created'))
-    name = deployment.get('name', 'Unnamed Project')
     state = deployment.get('state', 'N/A')
     
-    # 고유 ID 생성 (카드 식별용)
+    # 카드 ID 생성
     card_id = f"vercel-card-{url.replace('.', '-').replace('/', '-')}"
     
-
+    # Top Best 항목일 경우의 스크린샷 처리
+    screenshot_html = ""
+    if is_top_best:
+        try:
+            screenshot_base64 = take_screenshot(raw_url)
+            if screenshot_base64:
+                screenshot_html = f"""
+                    <div style="width: 100%; height: 200px; overflow: hidden; border-radius: 10px; margin-bottom: 15px;">
+                        <img src="data:image/png;base64,{screenshot_base64}" 
+                             style="width: 100%; height: 100%; object-fit: cover;"
+                             alt="{name} 스크린샷"/>
+                    </div>
+                """
+        except Exception as e:
+            print(f"스크린샷 처리 오류: {str(e)}")
     
-    bg_color = get_pastel_color(index + 20)
+    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와 일반 카드의 내용을 다르게 구성
+    if is_top_best:
+        info_section = f"""
+            <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>URL:</strong> 🔗 {url}
+                </p>
+            </div>
+        """
+    else:
+        info_section = f"""
+            <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}
+                </p>
+                <p style='margin: 8px 0;'>
+                    <strong>Created:</strong> 📅 {created}
+                </p>
+                <p style='margin: 8px 0;'>
+                    <strong>URL:</strong> 🔗 {url}
+                </p>
+            </div>
+        """
 
     return f"""
     <div id="{card_id}" class="vercel-card"
@@ -269,7 +363,7 @@ def get_vercel_card(deployment, index):
                 overflow: hidden;'
          onmouseover='this.style.transform="translateY(-5px) scale(1.02)"; this.style.boxShadow="0 8px 25px rgba(0,0,0,0.15)"'
          onmouseout='this.style.transform="translateY(0) scale(1)"; this.style.boxShadow="0 4px 15px rgba(0,0,0,0.1)"'>
-         <!-- ... (이전 코드와 동일) ... -->
+         {screenshot_html}
          <h3 style='color: #2d2d2d; 
                     margin: 0 0 20px 0; 
                     font-size: 1.4em;
@@ -277,25 +371,13 @@ def get_vercel_card(deployment, index):
                     align-items: center;
                     gap: 10px;'>
              <span style='font-size: 1.3em'>{random_emojis[0]}</span>
-             <a href='{final_url}' target='_blank' 
+             <a href='{url}' target='_blank' 
                 style='text-decoration: none; color: #2d2d2d;'>
                  {name}
              </a>
              <span style='font-size: 1.3em'>{random_emojis[1]}</span>
          </h3>    
-
-        <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}
-            </p>
-            <p style='margin: 8px 0;'>
-                <strong>Created:</strong> 📅 {created}
-            </p>
-            <p style='margin: 8px 0;'>
-                <strong>URL:</strong> 🔗 https://{url}
-            </p>
-        </div>
+         {info_section}
          <div style='margin-top: 20px; display: flex; justify-content: space-between; align-items: center;'>
              <div class="like-section" style="display: flex; align-items: center; gap: 10px;">
                  <button onclick="toggleLike('{card_id}')" class="like-button" 
@@ -304,7 +386,7 @@ def get_vercel_card(deployment, index):
                  </button>
                  <span class="like-count" style="font-size: 1.2em; color: #666;">0</span>
              </div>
-             <a href='{final_url}' target='_blank' 
+             <a href='{url}' target='_blank' 
                 style='background: linear-gradient(45deg, #0084ff, #00a3ff);
                        color: white; 
                        padding: 10px 20px; 
@@ -517,7 +599,6 @@ TOP_BEST_URLS = [
 ]
 
 
-
 def get_user_spaces():
     # 기존 Hugging Face 스페이스 가져오기
     url = f"https://huggingface.co/api/spaces?author={USERNAME}&limit=500"
@@ -537,11 +618,12 @@ def get_user_spaces():
             if not should_exclude_space(space.get('id', '').split('/')[-1])
         ]
 
-        # Vercel 배포 가져오기
-        vercel_deployments = get_vercel_deployments()
+        # TOP_BEST_URLS 항목 수
+        top_best_count = len(TOP_BEST_URLS)
         
-        # 순수 Vercel 배포 수만 카운트
-        vercel_count = len(vercel_deployments) if vercel_deployments else 0
+        # Vercel API를 통한 실제 배포 수
+        vercel_deployments = get_vercel_deployments()
+        actual_vercel_count = len(vercel_deployments) if vercel_deployments else 0
 
         html_content = f"""
         <div style='padding: 20px; background-color: #f5f5f5;'>
@@ -555,23 +637,28 @@ def get_user_spaces():
                     </a>
                 </p>
                 <p style='color: #666; margin: 0;'>
-                    Found {vercel_count} Vercel deployments and {len(user_spaces)} Hugging Face spaces
+                    Found {actual_vercel_count} Vercel deployments and {len(user_spaces)} Hugging Face spaces<br>
+                    (Plus {top_best_count} featured items in Top Best section)
                 </p>
             </div>
 
-            
             <!-- Top Best -->
             <h3 style='color: #333; margin: 20px 0;'>🏆 Top Best</h3>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
-                {"".join(get_vercel_card({"url": url["url"], "created": url["created"], "name": url["name"], "state": url["state"]}, idx) 
-                        for idx, url in enumerate(TOP_BEST_URLS))}
+                {"".join(get_vercel_card(
+                    {"url": url["url"], "created": url["created"], "name": url["name"], "state": url["state"]}, 
+                    idx,
+                    is_top_best=True
+                ) for idx, url in enumerate(TOP_BEST_URLS))}
             </div>
             
             <!-- Vercel Deployments -->
+            {f'''
             <h3 style='color: #333; margin: 20px 0;'>⚡ Vercel Deployments</h3>
             <div id="vercel-container" style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
                 {"".join(get_vercel_card(dep, idx) for idx, dep in enumerate(vercel_deployments))}
             </div>
+            ''' if vercel_deployments else ''}
             
             <!-- Hugging Face Spaces -->
             <h3 style='color: #333; margin: 20px 0;'>🤗 Hugging Face Spaces</h3>
@@ -633,7 +720,6 @@ def get_user_spaces():
         </script>
         """
 
-        
         return html_content
 
     except Exception as e:
@@ -645,12 +731,14 @@ def get_user_spaces():
             <p>Please try again later.</p>
         </div>
         """
+
         
 # Creating the Gradio interface
 demo = gr.Blocks()
 
 with demo:
-    html_output = gr.HTML(value=get_user_spaces())  # 초기 로드 시 직접 함수 호출
+    html_output = gr.HTML(value=get_user_spaces())
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+