Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 from typing import List, Dict, Union
 import concurrent.futures
 import traceback
+import base64
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -40,6 +41,16 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     except ValueError as e:
         return f"JSON decoding error: {str(e)}"
 
+def capture_thumbnail(space_id: str) -> str:
+    screenshot_url = f"https://huggingface.co/spaces/{space_id}/thumbnail.png"
+    try:
+        response = requests.get(screenshot_url, headers=get_headers())
+        if response.status_code == 200:
+            return response.content
+    except requests.RequestException:
+        pass
+    return None
+
 def format_space(space: Dict) -> Dict:
     space_id = space.get('id', 'Unknown')
     space_name = space_id.split('/')[-1] if '/' in space_id else space_id
@@ -51,12 +62,15 @@ def format_space(space: Dict) -> Dict:
     space_likes = space.get('likes', 'N/A')
     space_url = f"https://huggingface.co/spaces/{space_id}"
     
+    thumbnail = capture_thumbnail(space_id)
+    
     return {
         "id": space_id,
         "name": space_name,
         "author": space_author,
         "likes": space_likes,
-        "url": space_url
+        "url": space_url,
+        "thumbnail": thumbnail
     }
 
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
@@ -105,6 +119,11 @@ def create_ui():
     .space-row {
         margin-bottom: 5px !important;
     }
+    .thumbnail {
+        max-width: 100px;
+        max-height: 100px;
+        object-fit: contain;
+    }
     """
 
     with gr.Blocks(css=css) as demo:
@@ -114,13 +133,18 @@ def create_ui():
             with gr.Column(scale=1):
                 for space in formatted_spaces:
                     with gr.Row(elem_classes="space-row"):
-                        gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
-                        button = gr.Button("클릭", elem_classes="minimal-button")
-                        button.click(
-                            lambda s=space: on_select(s),
-                            inputs=[],
-                            outputs=[info_output, app_py_content]
-                        )
+                        if space['thumbnail']:
+                            gr.Image(value=space['thumbnail'], show_label=False, elem_classes="thumbnail")
+                        else:
+                            gr.Markdown("No thumbnail")
+                        with gr.Column():
+                            gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
+                            button = gr.Button("클릭", elem_classes="minimal-button")
+                            button.click(
+                                lambda s=space: on_select(s),
+                                inputs=[],
+                                outputs=[info_output, app_py_content]
+                            )
             
             with gr.Column(scale=1):
                 info_output = gr.Textbox(label="Space 정보 및 요약", lines=10)