Update main.py

main.py

@@ -7,6 +7,7 @@ import threading
 import json
 import re
 import psutil
+import httpx
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse, HTMLResponse
 
@@ -366,6 +367,28 @@ pool = BrowserPool()
 pool.start()
 
 
+# ====================================================================
+# Keep-Alive (يمنع HuggingFace من إيقاف Space بسبب عدم النشاط)
+# ====================================================================
+
+_keep_alive_count = 0
+
+async def _keep_alive():
+    global _keep_alive_count
+    # انتظر حتى يكون السيرفر جاهزاً
+    await asyncio.sleep(30)
+    print("[KEEP-ALIVE] Started — ping every 4 minutes ✓")
+    while True:
+        await asyncio.sleep(240)  # كل 4 دقائق
+        try:
+            async with httpx.AsyncClient(timeout=10) as client:
+                r = await client.get("http://localhost:7860/")
+            _keep_alive_count += 1
+            print(f"[KEEP-ALIVE] Ping #{_keep_alive_count} → {r.status_code} ✓")
+        except Exception as e:
+            print(f"[KEEP-ALIVE] Ping failed (non-fatal): {e}")
+
+
 # ====================================================================
 # Helpers
 # ====================================================================
@@ -440,7 +463,6 @@ def _auth(request: Request) -> bool:
 
 
 def _auth_token(request: Request, token: str = "") -> bool:
-    """يقبل المصادقة من Header أو من ?token= في الـ URL"""
     header_key = (
         request.headers.get("authorization", "")
         .replace("Bearer ", "").strip()
@@ -481,6 +503,12 @@ def _make_completion(start_time, model, text, messages, tools=None):
 app = FastAPI(title="Duck.ai API Server")
 
 
+@app.on_event("startup")
+async def startup_event():
+    asyncio.create_task(_keep_alive())
+    print("[APP] Keep-alive task registered ✓")
+
+
 @app.post("/v1/chat/completions")
 async def chat_completions(request: Request):
     try:
@@ -584,7 +612,6 @@ async def list_models(request: Request):
 
 @app.get("/health")
 async def health(request: Request, token: str = ""):
-    # ── /health يقبل Header أو ?token= ──────────────────────
     if not _auth_token(request, token):
         return JSONResponse(status_code=401, content={"error": {"message": "Invalid API Key"}})
 
@@ -609,6 +636,7 @@ async def health(request: Request, token: str = ""):
         "total_requests":    pool._total_requests,
         "rejected_requests": pool._rejected,
         "queue_timeout_sec": QUEUE_TIMEOUT,
+        "keep_alive_pings":  _keep_alive_count,
         "workers":           stats,
         "ram": {
             "used_gb":  round(mem.used  / 1024**3, 2),
@@ -622,16 +650,16 @@ async def health(request: Request, token: str = ""):
 @app.get("/")
 async def root():
     return {
-        "status":  "running",
-        "message": "Duck.ai API Pool Server is active!",
-        "docs":    "/docs",
-        "health":  "/health",
+        "status":           "running",
+        "message":          "Duck.ai API Pool Server is active!",
+        "docs":             "/docs",
+        "health":           "/health",
+        "keep_alive_pings": _keep_alive_count,
     }
 
 
 @app.get("/dashboard", response_class=HTMLResponse)
 async def dashboard(request: Request, token: str = ""):
-    # ── يقبل Header أو ?token= في الـ URL ───────────────────
     if not _auth_token(request, token):
         return HTMLResponse(
             "<h1 style='font-family:sans-serif;text-align:center;margin-top:20%;"