Update app.py

app.py

@@ -12,6 +12,29 @@ TOKEN = os.environ.get("DEBUG")  # keep your existing env var
 
 API = HfApi(token=TOKEN)
 
+def preflight_status():
+    # 1) Check token presence
+    if not TOKEN:
+        return ("❌ No HF token found in env var 'DEBUG'. "
+                "Add a secret named DEBUG in the Space settings (a token with 'write' scope).")
+
+    # 2) Check identity
+    try:
+        me = API.whoami(token=TOKEN)
+        user_str = me.get("name") or me.get("username") or "unknown-user"
+    except Exception as e:
+        return f"❌ Token error: cannot authenticate ({e})."
+
+    # 3) Check dataset access
+    repo_id = "AIEnergyScore/tested_proprietary_models"
+    try:
+        info = API.repo_info(repo_id=repo_id, repo_type="dataset", token=TOKEN)
+        # If this succeeds, you at least have read access; write failure will surface during upload.
+        return f"✅ Connected as **{user_str}**. Dataset **{repo_id}** reachable."
+    except Exception as e:
+        return (f"⚠️ Auth OK as **{user_str}**, but cannot access dataset "
+                f"**{repo_id}** ({e}). Make sure the token user has write access.")
+
 # ---------- Upload to HF dataset (kept from your original) ----------
 def add_docker_eval(zip_file):
     new_fid = os.path.basename(zip_file)
@@ -61,6 +84,7 @@ with gr.Blocks(title="AI Energy Score") as demo:
     """)
 
     gr.Markdown("<div style='text-align:center;'><h2>Submission Portal</h2></div>")
+    preflight_box = gr.Markdown(preflight_status())
 
     with gr.Row():
         # -------- Open Models ----------
@@ -108,13 +132,24 @@ Run the benchmark **in your own environment** and upload the logs here.
                 if not temp_path:
                     gr.Warning("No file selected.")
                     return "❌ No file uploaded."
-                # Run your original uploader (pushes to HF dataset + toast)
+            
+                if not TOKEN:
+                    gr.Warning("Missing HF token in env var 'DEBUG'.")
+                    return "❌ Upload blocked: missing token (DEBUG)."
+            
+                # Enforce .zip
+                if not str(temp_path).lower().endswith(".zip"):
+                    gr.Warning("Only .zip files are accepted.")
+                    return "❌ Please upload a .zip file."
+            
                 try:
-                    add_docker_eval(temp_path)
-                    return f"✅ Received and submitted: **{os.path.basename(temp_path)}**"
+                    # Your existing uploader: pushes to AIEnergyScore/tested_proprietary_models/submitted_models/
+                    add_docker_eval(temp_path)  # shows a toast on success/failure internally
+                    basename = os.path.basename(temp_path)
+                    return f"✅ Received and submitted: **{basename}**"
                 except Exception as e:
-                    gr.Warning(str(e))
-                    return f"❌ Upload failed: {e}"
+                    gr.Warning(f"Upload error: {e}")
+                    return f"❌ Upload failed — {e}"
 
             # IMPORTANT: bind inside Blocks context
             upload_button.upload(