Update app.py

app.py

@@ -5,9 +5,10 @@ from huggingface_hub import HfApi
 HF_TOKEN = os.getenv("HF_TOKEN")
 api = HfApi(token=HF_TOKEN)
 
-def scan_spaces_zero_a10g(username: str):
+def scan_spaces_detailed(username: str):
     """
-    Scan all Spaces for a user and suggest ZeroGPU if SpaceRuntime.hardware == 'zero-a10g'
+    Scan all Spaces for a user and log all metadata.
+    Mark Hardware: ZeroGPU if CPU only.
     """
     output = ""
     try:
@@ -19,18 +20,27 @@ def scan_spaces_zero_a10g(username: str):
             try:
                 info = api.space_info(space.id, token=HF_TOKEN)
                 status = getattr(info, "status", "unknown")
+                tags = getattr(info, "tags", [])
+                hardware_field = getattr(info, "hardware", None)
 
-                # Check if SpaceRuntime object exists and has hardware attribute
-                space_runtime = getattr(info, "SpaceRuntime", None)
-                if space_runtime and getattr(space_runtime, "hardware", "").lower() == "zero-a10g":
-                    hardware = "ZeroGPU"
+                # Determine ZeroGPU
+                if hardware_field:
+                    hardware = "ZeroGPU" if all("cpu" in h.lower() for h in hardware_field) else "GPU"
                 else:
-                    hardware = "None"
+                    # fallback to tags
+                    hardware = "ZeroGPU" if "cpu" in tags else "GPU"
 
-                output += f"{space.id} | Status: {status} | Hardware: {hardware}\n"
+                # Log all metadata
+                output += f"Space: {space.id}\n"
+                output += f"  Status: {status}\n"
+                output += f"  Hardware: {hardware}\n"
+                output += f"  Tags: {tags}\n"
+                output += f"  Raw metadata: {info}\n"
+                output += "-"*50 + "\n"
 
             except Exception as e:
                 output += f"{space.id} | Error retrieving info: {e}\n"
+                output += "-"*50 + "\n"
 
     except Exception as e:
         return f"Error: {e}"
@@ -39,11 +49,11 @@ def scan_spaces_zero_a10g(username: str):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Hugging Face Spaces Zero-A10G (ZeroGPU) Scanner")
+    gr.Markdown("## Hugging Face Spaces Detailed ZeroGPU Scanner")
     username_input = gr.Textbox(label="Hugging Face Username", value="rahul7star")
     scan_button = gr.Button("Scan Spaces")
-    output_box = gr.Textbox(label="ZeroGPU Status", lines=25)
+    output_box = gr.Textbox(label="Detailed Space Metadata", lines=30)
     
-    scan_button.click(scan_spaces_zero_a10g, inputs=username_input, outputs=output_box)
+    scan_button.click(scan_spaces_detailed, inputs=username_input, outputs=output_box)
 
 demo.launch()