Update app.py

app.py

@@ -144,7 +144,7 @@ class Config:
     CNN_CHECKPOINT = CHECKPOINT_DIR / HF_CHECKPOINT_FILENAME
     
     # Model configuration
-    WHISPER_MODEL = os.getenv("WHISPER_MODEL", "small")  # Can use 'base' with 16GB RAM!
+    WHISPER_MODEL = os.getenv("WHISPER_MODEL", "small")
     DEVICE = os.getenv("DEVICE", "cpu")
     USE_FP16 = False
     
@@ -328,6 +328,7 @@ async def root():
         "endpoints": {
             "docs": "/docs",
             "health": "/health",
+            "warmup": "/warmup",
             "process": "/process (POST)",
             "enhance": "/enhance (POST)",
             "transcribe": "/transcribe (POST)",
@@ -355,6 +356,27 @@ async def health_check():
         "device": Config.DEVICE,
         "tts_available": tts_available
     }
+PING_COUNT = 0
+
+@app.get("/warmup")
+@app.post("/warmup")
+async def warmup():
+    """
+    Lightweight endpoint to simulate activity for keepalive pings.
+    Accepts both GET (browser) and POST (curl/action).
+    """
+    global PING_COUNT
+    PING_COUNT += 1
+
+    # Minimal CPU work to look like real activity
+    import numpy as np
+    _ = np.zeros((1000,))
+
+    return {
+        "ok": True,
+        "message": "Warmup ping received",
+        "total_pings": PING_COUNT
+    }
 
 
 @app.post("/process", response_model=ProcessResponse)