Upload folder using huggingface_hub

keep_alive.py

@@ -2,12 +2,12 @@
 HuggingFace Spaces Keep-Alive Pinger
 
 HuggingFace Spaces는 일정 시간 요청이 없으면 Sleep 모드로 전환됩니다.
-이 모듈은 5분마다 자신의 /health + OSRM Space를 호출하여 Cold Start를 방지합니다.
+이 모듈은 Backend(5분) + OSRM Space(30분) 주기로 ping하여 Cold Start를 방지합니다.
 
 @module keep_alive
 @description
-- 5분(300초) 주기로 자기 자신의 /health 엔드포인트 호출
-- OSRM Space도 함께 ping하여 sleep 방지
+- Backend: 5분(300초) 주기로 자기 자신의 /health 엔드포인트 호출
+- OSRM: 30분(1800초) 주기로 경량 경로 요청 (HF sleep 타임아웃 48시간 대비 충분)
 - asyncio 백그라운드 태스크로 실행
 - 에러 발생 시에도 계속 실행 (무한 루프)
 - 로깅을 통한 상태 모니터링
@@ -43,12 +43,15 @@ _OSRM_SELF_HOSTED = "router.project-osrm.org" not in _OSRM_BASE_URL
 # 경량 OSRM ping: 제주 애월 근처 짧은 경로 요청
 _OSRM_PING_URL = f"{_OSRM_BASE_URL}/route/v1/foot/126.3397,33.4627;126.3450,33.4630" if _OSRM_SELF_HOSTED else None
 
-# Keep-alive 설정
-PING_INTERVAL_SECONDS = 300  # 5분
+# Keep-alive 설정 - Backend
+PING_INTERVAL_SECONDS = 300  # 5분 (사용자 응답 속도 유지)
 PING_TIMEOUT_SECONDS = 10    # 요청 타임아웃
-OSRM_PING_TIMEOUT_SECONDS = 30  # OSRM은 cold start 시 더 오래 걸림
 MAX_CONSECUTIVE_FAILURES = 10  # 최대 연속 실패 허용 (로깅용)
 
+# Keep-alive 설정 - OSRM (HF sleep 타임아웃 48시간이므로 30분이면 충분)
+OSRM_PING_INTERVAL_SECONDS = 1800  # 30분
+OSRM_PING_TIMEOUT_SECONDS = 30  # OSRM은 cold start 시 더 오래 걸림
+
 # 상태 추적 - Backend
 _ping_task: Optional[asyncio.Task] = None
 _consecutive_failures: int = 0
@@ -177,18 +180,13 @@ async def ping_osrm() -> bool:
 
 async def keep_alive_loop():
     """
-    5분마다 자기 자신 + OSRM Space를 호출하여 HuggingFace Spaces sleep 방지.
-
-    - 무한 루프로 실행
-    - 에러가 발생해도 계속 실행
-    - 연속 실패 시 경고 로그
+    5분마다 Backend self-ping. HuggingFace Spaces sleep 방지.
     """
     global _consecutive_failures
 
-    osrm_status = f", osrm={_OSRM_PING_URL}" if _OSRM_PING_URL else ""
     logger.info(
-        f"[keep-alive] Starting keep-alive pinger. "
-        f"url={HF_SPACE_URL}, interval={PING_INTERVAL_SECONDS}s{osrm_status}"
+        f"[keep-alive] Starting backend pinger. "
+        f"url={HF_SPACE_URL}, interval={PING_INTERVAL_SECONDS}s"
     )
 
     # 첫 ping은 서버 시작 후 1분 뒤에 실행 (완전히 준비될 때까지 대기)
@@ -196,18 +194,47 @@ async def keep_alive_loop():
 
     while True:
         try:
-            # Backend self-ping과 OSRM ping 동시 실행
-            tasks = [ping_self()]
-            if _OSRM_PING_URL:
-                tasks.append(ping_osrm())
-            await asyncio.gather(*tasks)
+            await ping_self()
 
-            # 연속 실패 경고
             if _consecutive_failures >= MAX_CONSECUTIVE_FAILURES:
                 logger.error(
                     f"[keep-alive] {_consecutive_failures} consecutive failures. "
                     f"Service may be unhealthy."
                 )
+
+        except asyncio.CancelledError:
+            logger.info("[keep-alive] Backend keep-alive task cancelled")
+            raise
+
+        except Exception as e:
+            logger.error(
+                f"[keep-alive] Loop error (will continue): "
+                f"{type(e).__name__}: {str(e)[:100]}"
+            )
+
+        await asyncio.sleep(PING_INTERVAL_SECONDS)
+
+
+async def osrm_keep_alive_loop():
+    """
+    30분마다 OSRM Space ping. HF sleep 타임아웃(48시간) 대비 충분한 간격.
+    자체 호스팅 OSRM이 아니면 즉시 종료.
+    """
+    if not _OSRM_PING_URL:
+        return
+
+    logger.info(
+        f"[keep-alive:osrm] Starting OSRM pinger. "
+        f"url={_OSRM_BASE_URL}, interval={OSRM_PING_INTERVAL_SECONDS}s (30min)"
+    )
+
+    # OSRM warmup은 server.py에서 처리하므로 여기선 30분 뒤 첫 ping
+    await asyncio.sleep(OSRM_PING_INTERVAL_SECONDS)
+
+    while True:
+        try:
+            await ping_osrm()
+
             if _osrm_consecutive_failures >= MAX_CONSECUTIVE_FAILURES:
                 logger.error(
                     f"[keep-alive:osrm] {_osrm_consecutive_failures} consecutive failures. "
@@ -215,53 +242,62 @@ async def keep_alive_loop():
                 )
 
         except asyncio.CancelledError:
-            logger.info("[keep-alive] Keep-alive task cancelled")
+            logger.info("[keep-alive:osrm] OSRM keep-alive task cancelled")
             raise
 
         except Exception as e:
-            # 예상치 못한 에러도 무시하고 계속 실행
             logger.error(
-                f"[keep-alive] Loop error (will continue): "
+                f"[keep-alive:osrm] Loop error (will continue): "
                 f"{type(e).__name__}: {str(e)[:100]}"
             )
 
-        # 다음 ping까지 대기
-        await asyncio.sleep(PING_INTERVAL_SECONDS)
+        await asyncio.sleep(OSRM_PING_INTERVAL_SECONDS)
+
+
+_osrm_ping_task: Optional[asyncio.Task] = None
 
 
 def start_keep_alive_task() -> asyncio.Task:
     """
     Keep-alive 백그라운드 태스크를 시작합니다.
+    Backend(5분) + OSRM(30분) 별도 루프로 실행.
 
     Returns:
-        asyncio.Task: 생성된 태스크 객체
+        asyncio.Task: Backend 태스크 객체
     """
-    global _ping_task
+    global _ping_task, _osrm_ping_task
 
     if _ping_task is not None and not _ping_task.done():
         logger.warning("[keep-alive] Task already running")
         return _ping_task
 
     _ping_task = asyncio.create_task(keep_alive_loop())
-    logger.info("[keep-alive] Background task started")
+    logger.info("[keep-alive] Backend background task started (5min interval)")
+
+    if _OSRM_PING_URL and (_osrm_ping_task is None or _osrm_ping_task.done()):
+        _osrm_ping_task = asyncio.create_task(osrm_keep_alive_loop())
+        logger.info("[keep-alive:osrm] OSRM background task started (30min interval)")
+
     return _ping_task
 
 
 async def stop_keep_alive_task():
     """
-    Keep-alive 태스크를 중지합니다.
+    Keep-alive 태스크를 모두 중지합니다.
     """
-    global _ping_task
+    global _ping_task, _osrm_ping_task
 
-    if _ping_task is not None and not _ping_task.done():
-        _ping_task.cancel()
-        try:
-            await _ping_task
-        except asyncio.CancelledError:
-            pass
-        logger.info("[keep-alive] Background task stopped")
+    for name, task in [("backend", _ping_task), ("osrm", _osrm_ping_task)]:
+        if task is not None and not task.done():
+            task.cancel()
+            try:
+                await task
+            except asyncio.CancelledError:
+                pass
+            logger.info(f"[keep-alive] {name} task stopped")
 
     _ping_task = None
+    _osrm_ping_task = None
 
 
 def get_keep_alive_stats() -> dict:
@@ -274,7 +310,7 @@ def get_keep_alive_stats() -> dict:
     stats = {
         "enabled": _ping_task is not None and not _ping_task.done(),
         "target_url": HF_SPACE_URL,
-        "interval_seconds": PING_INTERVAL_SECONDS,
+        "interval_seconds": PING_INTERVAL_SECONDS,  # 5분
         "total_pings": _total_pings,
         "successful_pings": _successful_pings,
         "success_rate": round(_successful_pings / _total_pings * 100, 2) if _total_pings > 0 else 0,
@@ -283,8 +319,9 @@ def get_keep_alive_stats() -> dict:
 
     if _OSRM_SELF_HOSTED:
         stats["osrm"] = {
-            "enabled": True,
+            "enabled": _osrm_ping_task is not None and not _osrm_ping_task.done(),
             "target_url": _OSRM_BASE_URL,
+            "interval_seconds": OSRM_PING_INTERVAL_SECONDS,  # 30분
             "total_pings": _osrm_total_pings,
             "successful_pings": _osrm_successful_pings,
             "success_rate": round(_osrm_successful_pings / _osrm_total_pings * 100, 2) if _osrm_total_pings > 0 else 0,