Update app.py

app.py

@@ -5,14 +5,15 @@ from huggingface_hub import HfApi
 HF_TOKEN = os.getenv("HF_TOKEN")
 api = HfApi(token=HF_TOKEN)
 
-def scan_spaces_detailed(username: str, filter_hw: str):
+
+def scan_spaces_detailed(username: str):
     """
-    Scan all Spaces for a user, mark ZeroGPU, and filter by hardware.
+    Scan all Spaces for a user and return full data in a table.
     """
     try:
         spaces = api.list_spaces(author=username, token=HF_TOKEN)
         if not spaces:
-            return []
+            return [], []
 
         results = []
 
@@ -22,7 +23,7 @@ def scan_spaces_detailed(username: str, filter_hw: str):
                 status = getattr(info, "status", "unknown")
                 tags = getattr(info, "tags", [])
 
-                # Check runtime requested_hardware
+                # Detect hardware
                 runtime = getattr(info, "runtime", None)
                 if runtime:
                     requested_hw = getattr(runtime, "requested_hardware", None)
@@ -37,44 +38,69 @@ def scan_spaces_detailed(username: str, filter_hw: str):
                     else:
                         hardware = "ZeroGPU" if "cpu" in tags else "GPU"
 
-                results.append({
-                    "Space ID": space.id,
-                    "Status": status,
-                    "Hardware": hardware,
-                    "Tags": ", ".join(tags)
-                })
+                results.append([
+                    space.id,
+                    status,
+                    hardware,
+                    ", ".join(tags)
+                ])
 
             except Exception as e:
-                results.append({
-                    "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.append([
+                    space.id,
+                    f"Error retrieving info: {e}",
+                    "Unknown",
+                    ""
+                ])
 
-        # Apply hardware filter
-        if filter_hw != "All":
-            results = [r for r in results if r["Hardware"] == filter_hw]
+        # Sort ZeroGPU first
+        results.sort(key=lambda x: 0 if x[2] == "ZeroGPU" else 1)
 
-        return results
+        # Return table and unique hardware values for filter
+        hardware_values = sorted(list({r[2] for r in results}))
+        return results, ["All"] + hardware_values
 
     except Exception as e:
-        return [{"Space ID": "Error", "Status": str(e), "Hardware": "", "Tags": ""}]
+        return [[f"Error: {e}", "", "", ""]], ["All"]
+
+
+def filter_table(rows, selected_hw):
+    """Filter the table by hardware locally (no API call)."""
+    if not rows or selected_hw == "All":
+        return rows
+    return [r for r in rows if r[2] == selected_hw]
+
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Hugging Face Spaces Scanner with Hardware Filter")
-    
+    gr.Markdown("## 🤖 Hugging Face Spaces ZeroGPU Scanner")
+
     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_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, filter_hw], outputs=output_table)
+        scan_button = gr.Button("Scan Spaces")
+
+    with gr.Row():
+        filter_hw = gr.Dropdown(label="Filter by Hardware", choices=["All"], value="All", interactive=True)
+
+    output_table = gr.Dataframe(
+        headers=["Space ID", "Status", "Hardware", "Tags"],
+        datatype=["str", "str", "str", "str"],
+        interactive=False,
+        wrap=True
+    )
+
+    # On scan: fetch all data and populate table + dropdown choices
+    scan_event = scan_button.click(
+        fn=scan_spaces_detailed,
+        inputs=username_input,
+        outputs=[output_table, filter_hw]
+    )
+
+    # When filter changes: re-filter table without re-calling API
+    filter_hw.change(
+        fn=filter_table,
+        inputs=[output_table, filter_hw],
+        outputs=output_table
+    )
 
 demo.launch()