Update app.py

app.py

@@ -11,29 +11,8 @@ 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);
@@ -41,18 +20,16 @@ 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/{space_id}' target='_blank' 
+            <a href='{space["url"]}' target='_blank' 
                style='text-decoration: none; color: #2d2d2d;'>
-                {space_name}
+                {space["title"]}
             </a>
         </h3>
-        <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' 
+        <p style='margin: 5px 0;'>{space.get("description", "No description available")}</p>
+        <div style='margin-top: 10px; display: flex; justify-content: space-between; align-items: center;'>
+            <a href='{space["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>
         </div>
@@ -60,54 +37,70 @@ def get_space_card(space):
     """
 
 def get_user_spaces():
-    # Get spaces for the user
-    spaces = search_spaces()
-    
-    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}")
+    # Hugging Face API v2 사용
+    url = f"https://huggingface.co/api/spaces?search={USERNAME}"
+    headers = {
+        "Accept": "application/json",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+    }
 
-    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>
+    try:
+        response = requests.get(url, headers=headers)
+        print(f"Status Code: {response.status_code}")  # 디버깅용
+        print(f"Response: {response.text[:500]}...")   # 디버깅용
+
+        if response.status_code != 200:
+            return f"Error: Failed to fetch spaces (Status Code: {response.status_code})"
+
+        spaces_data = response.json()
+        
+        # 사용자의 spaces만 필터링
+        user_spaces = []
+        for space in spaces_data:
+            if space.get('author', {}).get('name', '').lower() == USERNAME.lower():
+                space_info = {
+                    "title": space.get('id', '').split('/')[-1],
+                    "description": space.get('description', ''),
+                    "url": f"https://huggingface.co/spaces/{space.get('id', '')}",
+                }
+                user_spaces.append(space_info)
+
+        if not user_spaces:
+            return f"""
+            <div style='padding: 20px; text-align: center; color: #666;'>
+                <h2>No public Spaces found for user: {USERNAME}</h2>
+                <p>Try visiting: <a href='https://huggingface.co/{USERNAME}' target='_blank'>
+                    https://huggingface.co/{USERNAME}</a></p>
+            </div>
+            """
+
+        # 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(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)}
+            </div>
         </div>
         """
 
-    # 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>
-    """
+        return html_content
 
-    return html_content
+    except Exception as e:
+        print(f"Error: {str(e)}")  # 디버깅용
+        return f"""
+        <div style='padding: 20px; text-align: center; color: #666;'>
+            <h2>Error occurred while fetching spaces</h2>
+            <p>Error details: {str(e)}</p>
+            <p>Please try again later or check if the username is correct.</p>
+        </div>
+        """
 
 # Creating the Gradio interface
 app = gr.Interface(
@@ -124,6 +117,6 @@ app = gr.Interface(
     """
 )
 
-# Launch the Gradio app
+# Launch the Gradio app with sharing enabled
 if __name__ == "__main__":
-    app.launch()
+    app.launch(share=True)  # share=True를 추가하여 공개 링크 생성