Update app.py

app.py

@@ -51,15 +51,25 @@ def log_memory_usage():
     threading.Timer(60, log_memory_usage).start()
 
 # ----------------------------------------------------------------------
-# Keep‑alive to prevent idle timeout (runs every 10 minutes)
+# Keep‑alive to prevent idle timeout (external pings)
 # ----------------------------------------------------------------------
 def keep_alive():
-    """Periodically ping the app to avoid idle timeout."""
-    url = "http://127.0.0.1:7860/"
+    """Periodically ping the public Space URL to prevent idle timeout."""
+    # Determine the public URL of the Space
+    space_id = os.environ.get('SPACE_ID')
+    if space_id:
+        # Convert "username/space-name" to "username-space-name.hf.space"
+        url = f"https://{space_id.replace('/', '-')}.hf.space/"
+        logging.info(f"Using external URL for keep‑alive: {url}")
+    else:
+        # Fallback to localhost when running locally
+        url = "http://127.0.0.1:7860/"
+        logging.warning("No SPACE_ID found, using localhost for keep‑alive – will not prevent sleep!")
+
     while True:
-        time.sleep(600)   # 10 minutes
+        time.sleep(300)   # 5 minutes
         try:
-            with urllib.request.urlopen(url, timeout=5) as response:
+            with urllib.request.urlopen(url, timeout=10) as response:
                 status = response.getcode()
                 logging.info(f"Keep‑alive ping: {status}")
         except Exception as e: