Update app.py

app.py

@@ -1376,14 +1376,30 @@ async def restart_space(request: Request):
 # ---- Status endpoint (no auth needed — for frontend health checks) ----
 @app.get("/status")
 async def get_status():
-    import psutil
+    # Read container-level memory from /sys/fs/cgroup (accurate for HF Spaces)
+    # psutil reads the host machine memory which is misleading on shared containers
     try:
-        mem = psutil.virtual_memory()
-        mem_used_mb = round((mem.total - mem.available) / 1024 / 1024)
-        mem_total_mb = round(mem.total / 1024 / 1024)
-        mem_pct = round(mem.percent)
+        # cgroup v2 (HF Spaces uses this)
+        with open('/sys/fs/cgroup/memory.current') as f:
+            mem_used_bytes = int(f.read().strip())
+        with open('/sys/fs/cgroup/memory.max') as f:
+            val = f.read().strip()
+            mem_total_bytes = int(val) if val != 'max' else 16 * 1024**3  # 16GB default
+        mem_used_mb = round(mem_used_bytes / 1024 / 1024)
+        mem_total_mb = round(mem_total_bytes / 1024 / 1024)
+        mem_pct = round(mem_used_bytes / mem_total_bytes * 100)
     except:
-        mem_used_mb = mem_total_mb = mem_pct = 0
+        # Fallback: try cgroup v1
+        try:
+            with open('/sys/fs/cgroup/memory/memory.usage_in_bytes') as f:
+                mem_used_bytes = int(f.read().strip())
+            with open('/sys/fs/cgroup/memory/memory.limit_in_bytes') as f:
+                mem_total_bytes = int(f.read().strip())
+            mem_used_mb = round(mem_used_bytes / 1024 / 1024)
+            mem_total_mb = round(mem_total_bytes / 1024 / 1024)
+            mem_pct = round(mem_used_bytes / mem_total_bytes * 100)
+        except:
+            mem_used_mb = mem_total_mb = mem_pct = 0
 
     kb_files = glob.glob(os.path.join(KNOWLEDGE_DIR, "*.txt"))
     chunks = vectorstore.index.ntotal if vectorstore else 0