Update loader.py

loader.py

@@ -7,6 +7,10 @@ import httpx
 app = FastAPI()
 
 # ========= CONFIG =========
+PRIVATE_STATIC_URL = os.getenv(
+    "PRIVATE_STATIC_URL",
+    "https://cheekeong2025-iip-grading.static.hf.space"  # ✅ fallback URL
+)
 PRIVATE_SPACE_ID = os.getenv("PRIVATE_SPACE_ID")
 HF_TOKEN = os.getenv("HF_TOKEN")
 
@@ -17,8 +21,15 @@ if not PRIVATE_SPACE_ID:
         "and add PRIVATE_SPACE_ID=cheekeong2025/iip-grading"
     )
 
+if not PRIVATE_STATIC_URL:
+    raise RuntimeError(
+        "❌ PRIVATE_STATIC_URL is not set.\n"
+        "Go to your Hugging Face Space → Settings → Repository secrets\n"
+        "and add PRIVATE_STATIC_URL=https://<your-private-static-space>.hf.space"
+    )
+
 if not HF_TOKEN:
-    print("[loader] ⚠️  Warning: HF_TOKEN not set — attempting anonymous access.")
+    print("[loader] ⚠️ Warning: HF_TOKEN not set — attempting anonymous access.")
 
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 # ===========================
@@ -27,9 +38,10 @@ HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 @app.get("/")
 async def root():
     """Serve the index.html from the private static Space."""
+    print(f"[loader] Fetching root from {PRIVATE_STATIC_URL}/index.html")
     async with httpx.AsyncClient(timeout=None) as client:
         try:
-            r = await client.get(PRIVATE_STATIC_URL, headers=HEADERS)
+            r = await client.get(f"{PRIVATE_STATIC_URL}/index.html", headers=HEADERS)
             r.raise_for_status()
         except Exception as e:
             return PlainTextResponse(f"Error fetching root: {e}", status_code=500)
@@ -40,6 +52,7 @@ async def root():
 async def proxy(path: str, request: Request):
     """Proxy all static assets (CSS, JS, images, etc.) from the private Space."""
     url = f"{PRIVATE_STATIC_URL.rstrip('/')}/{path}"
+    print(f"[loader] Fetching asset: {url}")
     try:
         async with httpx.AsyncClient(timeout=None) as client:
             r = await client.get(url, headers=HEADERS)
@@ -53,4 +66,9 @@ async def proxy(path: str, request: Request):
 
 @app.get("/health")
 async def health():
-    return {"status": "ok", "source": PRIVATE_STATIC_URL}
+    """Simple health check."""
+    return {
+        "status": "ok",
+        "source": PRIVATE_STATIC_URL,
+        "space_id": PRIVATE_SPACE_ID
+    }