Update app.py

app.py

@@ -2,7 +2,7 @@ import requests
 import gradio as gr
 from datetime import datetime
 
-# 테스트를 위해 Gradio 공식 계정으로 변경
+# 테스트를 위해 Gradio 공식 계정으로 설정
 USERNAME = "gradio"
 
 def format_timestamp(timestamp):
@@ -11,8 +11,29 @@ def format_timestamp(timestamp):
         return dt.strftime('%Y-%m-%d %H:%M')
     return 'N/A'
 
+def search_spaces():
+    """Search for spaces using the Hugging Face search API"""
+    url = "https://huggingface.co/api/spaces"
+    headers = {"Accept": "application/json"}
+    
+    try:
+        # First, try to get all spaces
+        response = requests.get(url, headers=headers)
+        if response.status_code == 200:
+            all_spaces = response.json()
+            # Filter spaces by username
+            user_spaces = [space for space in all_spaces 
+                         if space.get('author', '').lower() == USERNAME.lower()]
+            return user_spaces
+    except Exception as e:
+        print(f"Error in search_spaces: {e}")
+    return []
+
 def get_space_card(space):
     """Generate HTML card for a space"""
+    space_id = space.get('id', '')
+    space_name = space_id.split('/')[-1] if space_id else '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);
@@ -20,83 +41,73 @@ 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='{space.get("url", "#")}' target='_blank' 
+            <a href='https://huggingface.co/spaces/{space_id}' target='_blank' 
                style='text-decoration: none; color: #2d2d2d;'>
-                {space.get('title', 'Unnamed Space')}
+                {space_name}
             </a>
         </h3>
-        <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' 
+        <p style='margin: 5px 0;'><strong>Author:</strong> {space.get('author', 'Unknown')}</p>
+        <p style='margin: 5px 0;'><strong>SDK:</strong> {space.get('sdk', 'N/A')}</p>
+        <p style='margin: 5px 0;'><strong>Likes:</strong> {space.get('likes', 0)} ❤️</p>
+        <div style='margin-top: 10px;'>
+            <a href='https://huggingface.co/spaces/{space_id}' target='_blank' 
                style='background-color: #0084ff; color: white; padding: 5px 10px; 
-                      border-radius: 5px; text-decoration: none;'>
+                      border-radius: 5px; text-decoration: none; display: inline-block;'>
                 View Space
             </a>
-            <span style='color: #666;'>
-                ❤️ {space.get('likes', 0)}
-            </span>
         </div>
     </div>
     """
 
 def get_user_spaces():
-    # Hugging Face search API 사용
-    url = "https://huggingface.co/api/spaces"
-    params = {
-        "author": USERNAME,
-        "limit": 100,
-        "full": True
-    }
+    # Get spaces for the user
+    spaces = search_spaces()
     
-    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]}...")  # 디버깅용
+    if not spaces:
+        # Try alternative search method
+        try:
+            response = requests.get(
+                f"https://huggingface.co/search/full-text",
+                params={"q": f"user:{USERNAME} type:space"},
+                headers={"Accept": "application/json"}
+            )
+            if response.status_code == 200:
+                search_results = response.json()
+                if 'hits' in search_results:
+                    spaces = search_results['hits']
+        except Exception as e:
+            print(f"Error in alternative search: {e}")
 
-        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 []
-
-        if not spaces:
-            return "No public Spaces found for this user."
-
-        if not isinstance(spaces, list):
-            spaces = [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>
+    if not spaces:
+        return """
+        <div style='padding: 20px; text-align: center; color: #666;'>
+            <h2>No public Spaces found for user: {USERNAME}</h2>
+            <p>This could be because:</p>
+            <ul style='list-style: none; padding: 0;'>
+                <li>The user has no public spaces</li>
+                <li>The username might be incorrect</li>
+                <li>There might be an issue with the API</li>
+            </ul>
         </div>
         """
 
-        return html_content
+    # 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(space) for space in spaces)}
+        </div>
+    </div>
+    """
 
-    except Exception as e:
-        print(f"Error: {str(e)}")  # 디버깅용
-        return f"Error occurred: {str(e)}"
+    return html_content
 
 # Creating the Gradio interface
 app = gr.Interface(