Add self-ping background task to prevent HF Space from sleeping

api/main.py

@@ -24,6 +24,21 @@ from api.routers import profile as profile_router
 # ── Rate limiter ──────────────────────────────────────────────────────────────
 limiter = Limiter(key_func=get_remote_address)
 
+import asyncio
+import httpx
+
+async def keep_awake():
+    """Background task to ping the server and prevent HF Space from sleeping."""
+    while True:
+        try:
+            await asyncio.sleep(600)  # Ping every 10 minutes
+            async with httpx.AsyncClient() as client:
+                await client.get("http://127.0.0.1:7860/health", timeout=10.0)
+                print("[Keep-Awake] Pinged health endpoint to prevent sleep.")
+        except asyncio.CancelledError:
+            break
+        except Exception as e:
+            print(f"[Keep-Awake] Ping failed: {e}")
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -34,8 +49,14 @@ async def lifespan(app: FastAPI):
     print("[Lifespan] Loading embedding model (first run may download ~420MB)...")
     get_embedder()
     print("[Lifespan] Embedding model loaded successfully.")
+    
+    # Start the keep-awake background task
+    ping_task = asyncio.create_task(keep_awake())
+    
     yield
-    # Shutdown: close DB pool
+    
+    # Shutdown: cancel task and close DB pool
+    ping_task.cancel()
     try:
         from db.pool import get_pool
         get_pool().closeall()