Update app.py

app.py

@@ -11,14 +11,12 @@ from PIL import Image
 from io import BytesIO
 from datetime import datetime
 import gradio as gr
+from typing import Tuple
 
 USERNAME = "openfree"
-
-# 스크린샷 캐시 저장소
 SCREENSHOT_CACHE = {}
 
 def take_screenshot(url):
-    """웹사이트 스크린샷 촬영"""
     if url in SCREENSHOT_CACHE:
         return SCREENSHOT_CACHE[url]
         
@@ -52,7 +50,6 @@ def take_screenshot(url):
             driver.quit()
 
 def get_pastel_color(index):
-    """인덱스 기반 파스텔 색상 생성"""
     pastel_colors = [
         'rgba(255, 230, 230, 0.8)', 'rgba(255, 230, 255, 0.8)', 
         'rgba(230, 230, 255, 0.8)', 'rgba(230, 255, 255, 0.8)', 
@@ -61,14 +58,13 @@ def get_pastel_color(index):
     return pastel_colors[index % len(pastel_colors)]
 
 def get_space_card(space, index):
-    """스페이스 카드 HTML 생성"""
     space_id = space.get('id', '')
-    space_name = space_id.split('/')[-1]
-    likes = space.get('likes', 0)
+    author, title = space_id.split('/', 1)
+    likes = format(space.get('likes', 0), ',')
     sdk = space.get('sdk', 'N/A')
+    created = space.get('createdAt', '').split('T')[0]
     url = f"https://huggingface.co/spaces/{space_id}"
     
-    # 스크린샷 가져오기
     screenshot = take_screenshot(url)
     bg_style = f"""
         background-image: linear-gradient(
@@ -102,47 +98,83 @@ def get_space_card(space, index):
             padding: 15px;
             border-radius: 10px;
             backdrop-filter: blur(5px);'>
-            <h3 style='margin: 0; color: #333;'>{space_name}</h3>
-            <p style='margin: 10px 0; color: #666;'>SDK: {sdk}</p>
-            <p style='margin: 10px 0; color: #666;'>❤️ {likes}</p>
+            <h3 style='margin: 0; color: #333;'>{title}</h3>
+            <p style='margin: 5px 0; color: #666;'>Author: {author}</p>
+            <p style='margin: 5px 0; color: #666;'>SDK: {sdk}</p>
+            <p style='margin: 5px 0; color: #666;'>❤️ {likes}</p>
+            <p style='margin: 5px 0; color: #666;'>Created: {created}</p>
+            <a href='{url}' target='_blank' style='color: blue; text-decoration: none;'>🔗 View</a>
         </div>
     </div>
     """
 
-def get_user_spaces():
-    """허깅페이스 스페이스 목록 가져오기"""
+def get_user_spaces(progress=gr.Progress()) -> Tuple[str, str]:
     url = f"https://huggingface.co/api/spaces?author={USERNAME}&limit=500"
     try:
+        progress(0, desc="Fetching spaces data...")
         response = requests.get(url)
-        if response.status_code == 200:
-            spaces_data = response.json()
-            html_content = """
+        response.raise_for_status()
+        spaces_data = response.json()
+        
+        progress(0.1, desc="Creating gallery...")
+        html_content = """
+        <div style='
+            padding: 20px;
+            background: #f5f5f5;'>
             <div style='
-                padding: 20px;
-                background: #f5f5f5;'>
-                <div style='
-                    display: grid;
-                    grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
-                    gap: 20px;'>
-            """
-            html_content += "".join(get_space_card(space, idx) for idx, space in enumerate(spaces_data))
-            html_content += "</div></div>"
-            return html_content
-        return "<p>Failed to fetch spaces</p>"
+                display: grid;
+                grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
+                gap: 20px;'>
+        """
+        
+        total = len(spaces_data)
+        for idx, space in enumerate(spaces_data):
+            progress((0.1 + 0.9 * idx/total), desc=f"Loading space {idx+1}/{total}...")
+            html_content += get_space_card(space, idx)
+            
+        html_content += "</div></div>"
+        
+        progress(1.0, desc="Complete!")
+        return html_content, "Gallery refresh complete!"
     except Exception as e:
-        return f"<p>Error: {str(e)}</p>"
+        error_html = f'<div style="color: red; padding: 20px;">Error: {str(e)}</div>'
+        return error_html, f"Error: {str(e)}"
 
-# Gradio 인터페이스 생성
-demo = gr.Interface(
-    fn=lambda: get_user_spaces(),
-    inputs=None,
-    outputs=gr.HTML(),
-    title="Hugging Face Spaces Gallery",
-    css="""
-        .container { max-width: 1200px; margin: 0 auto; }
-        .gradio-container { font-family: 'Arial', sans-serif; }
-    """
-)
+def create_interface():
+    with gr.Blocks(title="Hugging Face Space Gallery") as interface:
+        gr.Markdown("# 🤗 Hugging Face Space Gallery")
+        gr.Markdown(f"Shows spaces by user: {USERNAME}")
+        
+        with gr.Row():
+            refresh_btn = gr.Button("Refresh Gallery", variant="primary")
+        
+        gallery_html = gr.HTML()
+        status = gr.Markdown("Ready")
+        
+        def update():
+            html, status_text = get_user_spaces()
+            return [html, status_text]
+        
+        refresh_btn.click(
+            fn=update,
+            outputs=[gallery_html, status],
+            show_progress=True
+        )
+        
+        interface.load(
+            fn=update,
+            outputs=[gallery_html, status]
+        )
+    
+    return interface
 
 if __name__ == "__main__":
-    demo.launch()
+    try:
+        demo = create_interface()
+        demo.launch(
+            share=True,
+            inbrowser=True,
+            show_api=False
+        )
+    except Exception as e:
+        print(f"Error launching app: {e}")