Update app.py

app.py

@@ -2,8 +2,8 @@ import requests
 import gradio as gr
 from datetime import datetime
 
-# 테스트를 위해 알려진 사용자로 변경
-USERNAME = "multimodalart"  # 많은 public spaces를 가진 사용자
+# 테스트를 위해 Gradio 공식 계정으로 변경
+USERNAME = "gradio"
 
 def format_timestamp(timestamp):
     if timestamp:
@@ -13,7 +13,6 @@ def format_timestamp(timestamp):
 
 def get_space_card(space):
     """Generate HTML card for a space"""
-    space_name = space["id"].split('/')[-1] if 'id' in space else space.get('name', 'Unknown')
     return f"""
     <div style='border: 1px solid #ddd; padding: 15px; margin: 10px; border-radius: 8px; 
                 background-color: white; box-shadow: 2px 2px 5px rgba(0,0,0,0.1);
@@ -21,68 +20,83 @@ def get_space_card(space):
          onmouseover='this.style.transform="scale(1.02)"'
          onmouseout='this.style.transform="scale(1)"'>
         <h3 style='color: #2d2d2d; margin: 0 0 10px 0;'>
-            <a href='https://huggingface.co/spaces/{USERNAME}/{space_name}' target='_blank' 
+            <a href='{space.get("url", "#")}' target='_blank' 
                style='text-decoration: none; color: #2d2d2d;'>
-                {space_name}
+                {space.get('title', 'Unnamed Space')}
             </a>
         </h3>
-        <p style='margin: 5px 0;'><strong>Status:</strong> 
-           <span style='color: {"green" if space.get("status") == "running" else "orange"}'>
-           {space.get("status", "unknown")}</span>
-        </p>
-        <p style='margin: 5px 0;'><strong>SDK:</strong> {space.get("sdk", "N/A")}</p>
-        <div style='margin-top: 10px;'>
-            <a href='https://huggingface.co/spaces/{USERNAME}/{space_name}' target='_blank' 
+        <p style='margin: 5px 0; color: #666;'>{space.get('description', 'No description available')}</p>
+        <div style='margin-top: 10px; display: flex; justify-content: space-between; align-items: center;'>
+            <a href='{space.get("url", "#")}' target='_blank' 
                style='background-color: #0084ff; color: white; padding: 5px 10px; 
-                      border-radius: 5px; text-decoration: none; display: inline-block;'>
+                      border-radius: 5px; text-decoration: none;'>
                 View Space
             </a>
+            <span style='color: #666;'>
+                ❤️ {space.get('likes', 0)}
+            </span>
         </div>
     </div>
     """
 
 def get_user_spaces():
-    # spaces 검색 API 사용
-    response = requests.get(
-        "https://huggingface.co/api/spaces",
-        params={"search": USERNAME},
-        headers={"Accept": "application/json"}
-    )
+    # Hugging Face search API 사용
+    url = "https://huggingface.co/api/spaces"
+    params = {
+        "author": USERNAME,
+        "limit": 100,
+        "full": True
+    }
+    
+    try:
+        response = requests.get(url, params=params)
+        print(f"Status Code: {response.status_code}")  # 디버깅용
+        print(f"URL: {response.url}")  # 디버깅용
+        print(f"Response: {response.text[:500]}...")  # 디버깅용
 
-    # 디버깅을 위한 출력
-    print(f"Status Code: {response.status_code}")
-    print(f"Response: {response.text[:500]}...")  # 응답의 처음 500자만 출력
+        if response.status_code != 200:
+            return f"Error: Failed to fetch spaces (Status Code: {response.status_code})"
 
-    if response.status_code != 200:
-        return f"Error: Failed to fetch spaces (Status Code: {response.status_code})"
+        spaces = response.json()
+        
+        if not spaces:
+            # 다른 API 엔드포인트 시도
+            alternate_url = f"https://huggingface.co/api/spaces/{USERNAME}"
+            response = requests.get(alternate_url)
+            spaces = response.json() if response.status_code == 200 else []
 
-    try:
-        all_spaces = response.json()
-        # 해당 사용자의 spaces만 필터링
-        user_spaces = [space for space in all_spaces if space.get('author', '') == USERNAME]
-    except Exception as e:
-        return f"Error parsing response: {str(e)}"
+        if not spaces:
+            return "No public Spaces found for this user."
 
-    if not user_spaces:
-        return "No public Spaces found for this user."
+        if not isinstance(spaces, list):
+            spaces = [spaces]
 
-    # Create HTML grid layout with title and count
-    html_content = f"""
-    <div style='padding: 20px; background-color: #f5f5f5;'>
-        <div style='margin-bottom: 20px;'>
-            <p style='color: #666; margin: 0;'>Found {len(user_spaces)} public spaces for {USERNAME}</p>
-        </div>
-        <div style='
-            display: grid;
-            grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
-            gap: 20px;
-        '>
-            {"".join(get_space_card(space) for space in user_spaces)}
+        # Create HTML grid layout
+        html_content = f"""
+        <div style='padding: 20px; background-color: #f5f5f5;'>
+            <div style='margin-bottom: 20px;'>
+                <p style='color: #666; margin: 0;'>Found {len(spaces)} public spaces for {USERNAME}</p>
+            </div>
+            <div style='
+                display: grid;
+                grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
+                gap: 20px;
+            '>
+                {"".join(get_space_card({
+                    'title': space.get('id', '').split('/')[-1],
+                    'description': space.get('description', ''),
+                    'url': f"https://huggingface.co/spaces/{space.get('id', '')}",
+                    'likes': space.get('likes', 0)
+                }) for space in spaces)}
+            </div>
         </div>
-    </div>
-    """
+        """
 
-    return html_content
+        return html_content
+
+    except Exception as e:
+        print(f"Error: {str(e)}")  # 디버깅용
+        return f"Error occurred: {str(e)}"
 
 # Creating the Gradio interface
 app = gr.Interface(