Update app.py

app.py

@@ -5,14 +5,14 @@ from huggingface_hub import HfApi
 HF_TOKEN = os.getenv("HF_TOKEN")
 api = HfApi(token=HF_TOKEN)
 
-def scan_spaces_detailed(username: str):
+def scan_spaces_detailed(username: str, filter_hw: str):
     """
-    Scan all Spaces for a user, mark ZeroGPU, and sort ZeroGPU spaces first.
+    Scan all Spaces for a user, mark ZeroGPU, and filter by hardware.
     """
     try:
         spaces = api.list_spaces(author=username, token=HF_TOKEN)
         if not spaces:
-            return f"No Spaces found for user '{username}'"
+            return []
 
         results = []
 
@@ -38,45 +38,43 @@ def scan_spaces_detailed(username: str):
                         hardware = "ZeroGPU" if "cpu" in tags else "GPU"
 
                 results.append({
-                    "id": space.id,
-                    "status": status,
-                    "hardware": hardware,
-                    "tags": tags
+                    "Space ID": space.id,
+                    "Status": status,
+                    "Hardware": hardware,
+                    "Tags": ", ".join(tags)
                 })
 
             except Exception as e:
                 results.append({
-                    "id": space.id,
-                    "status": f"Error retrieving info: {e}",
-                    "hardware": "Unknown",
-                    "tags": []
+                    "Space ID": space.id,
+                    "Status": f"Error retrieving info: {e}",
+                    "Hardware": "Unknown",
+                    "Tags": ""
                 })
 
         # Sort ZeroGPU spaces first
-        results.sort(key=lambda x: 0 if x["hardware"] == "ZeroGPU" else 1)
+        results.sort(key=lambda x: 0 if x["Hardware"] == "ZeroGPU" else 1)
 
-        # Format output for Gradio (hide raw metadata)
-        output = ""
-        for r in results:
-            line = f"Space: {r['id']}\n"
-            line += f"  Status: {r['status']}\n"
-            line += f"  Hardware: {r['hardware']}\n"
-            line += f"  Tags: {r['tags']}\n"
-            line += "-"*50 + "\n"
-            output += line
+        # Apply hardware filter
+        if filter_hw != "All":
+            results = [r for r in results if r["Hardware"] == filter_hw]
 
-        return output
+        return results
 
     except Exception as e:
-        return f"Error: {e}"
+        return [{"Space ID": "Error", "Status": str(e), "Hardware": "", "Tags": ""}]
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Hugging Face Spaces ZeroGPU Scanner (Sorted)")
-    username_input = gr.Textbox(label="Hugging Face Username", value="rahul7star")
+    gr.Markdown("## Hugging Face Spaces Scanner with Hardware Filter")
+    
+    with gr.Row():
+        username_input = gr.Textbox(label="Hugging Face Username", value="rahul7star")
+        filter_hw = gr.Dropdown(choices=["All", "ZeroGPU", "GPU"], value="All", label="Filter by Hardware")
+    
     scan_button = gr.Button("Scan Spaces")
-    output_box = gr.Textbox(label="Spaces Summary", lines=30)
+    output_table = gr.Dataframe(headers=["Space ID", "Status", "Hardware", "Tags"], datatype=["str","str","str","str"], interactive=False)
     
-    scan_button.click(scan_spaces_detailed, inputs=username_input, outputs=output_box)
+    scan_button.click(scan_spaces_detailed, inputs=[username_input, filter_hw], outputs=output_table)
 
 demo.launch()