changes

app.py

@@ -1,8 +1,11 @@
 from flask import Flask, request, jsonify, send_from_directory
 from werkzeug.utils import secure_filename
 from flask_cors import CORS  # Import Flask-CORS
+from apscheduler.schedulers.background import BackgroundScheduler
+import requests
 import os
 import sys
+import atexit
 
 print("Starting Flask app...", file=sys.stderr)
 print("Loading model...", file=sys.stderr)
@@ -57,6 +60,36 @@ def upload_file():
 def uploaded_file(filename):
     return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
 
+# ============ KEEP-ALIVE CRON JOB ============
+# Self-ping to prevent Hugging Face from sleeping
+
+def keep_alive():
+    """Ping the health endpoint to keep the server awake"""
+    try:
+        # Get the URL from environment or use default
+        app_url = os.environ.get('SPACE_HOST', None)
+        if app_url:
+            url = f"https://{app_url}"
+        else:
+            url = os.environ.get('APP_URL', 'http://localhost:5000')
+        
+        response = requests.get(f"{url}/", timeout=30)
+        print(f"[Keep-Alive] Pinged {url} - Status: {response.status_code}", file=sys.stderr)
+    except Exception as e:
+        print(f"[Keep-Alive] Ping failed: {str(e)}", file=sys.stderr)
+
+# Initialize scheduler
+scheduler = BackgroundScheduler(daemon=True)
+# Ping every 14 minutes (Hugging Face sleeps after 15 minutes of inactivity)
+scheduler.add_job(keep_alive, 'interval', minutes=14)
+scheduler.start()
+
+# Shut down the scheduler when exiting the app
+atexit.register(lambda: scheduler.shutdown())
+
+print("Keep-alive cron job started (pinging every 14 minutes)", file=sys.stderr)
+# ============================================
+
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 5000))
     app.run(host='0.0.0.0', port=port, debug=False)

requirements.txt

@@ -18,6 +18,8 @@ Flask==3.0.0
 flask-cors==4.0.0
 Werkzeug==3.0.1
 gunicorn==21.2.0
+APScheduler==3.10.4
+requests==2.31.0
 
 # Image handling
 scikit-image==0.22.0