Update app.py

app.py

@@ -1,34 +1,11 @@
 import os
 import gradio as gr
-import requests
 from huggingface_hub import HfApi
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 api = HfApi(token=HF_TOKEN)
 
-def check_space_live(space_id: str):
-    """
-    Query the live HF Space to see if it's running on CPU/ZeroGPU.
-    Returns: 'ZeroGPU' or 'GPU' or 'Offline'
-    """
-    try:
-        # Most HF Spaces expose this endpoint
-        url = f"https://hf.space/embed/{space_id}/+/api/status"
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
-        resp = requests.get(url, headers=headers, timeout=5)
-        data = resp.json()
-        # 'cpu' or 'gpu' in hardware field
-        if "gpu" in str(data.get("hardware", "")).lower():
-            return "GPU"
-        else:
-            return "ZeroGPU"
-    except Exception:
-        return "Offline"
-
-def scan_spaces_live(username: str):
-    """
-    Scan all Spaces of a user and return live ZeroGPU status.
-    """
+def scan_spaces(username: str):
     output = ""
     try:
         spaces = api.list_spaces(author=username, token=HF_TOKEN)
@@ -36,21 +13,25 @@ def scan_spaces_live(username: str):
             return f"No Spaces found for user '{username}'"
 
         for space in spaces:
-            live_status = check_space_live(space.id)
-            output += f"{space.id} | Live Status: {live_status}\n"
-
+            try:
+                info = api.space_info(space_id=space.id, token=HF_TOKEN)
+                status = getattr(info, "status", "unknown")
+                hardware = getattr(info, "hardware", [])
+                zero_gpu = "Yes" if (not hardware or all("cpu" in h.lower() for h in hardware)) else "No"
+                output += f"{space.id} | Status: {status} | ZeroGPU: {zero_gpu}\n"
+            except Exception as e:
+                output += f"{space.id} | Error retrieving info: {e}\n"
     except Exception as e:
         return f"Error: {e}"
-
     return output
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Hugging Face Spaces Live ZeroGPU Scanner")
+    gr.Markdown("## Hugging Face Spaces ZeroGPU Status Scanner")
     username_input = gr.Textbox(label="Hugging Face Username", value="rahul7star")
-    scan_button = gr.Button("Scan Live Spaces")
-    output_box = gr.Textbox(label="Live ZeroGPU Status", lines=25)
+    scan_button = gr.Button("Scan Spaces")
+    output_box = gr.Textbox(label="ZeroGPU Status", lines=25)
     
-    scan_button.click(scan_spaces_live, inputs=username_input, outputs=output_box)
+    scan_button.click(scan_spaces, inputs=username_input, outputs=output_box)
 
 demo.launch()