Adding cleanup logic to startup routine

app.py

@@ -1,9 +1,8 @@
-
-
 import os
 import uuid
 import traceback
 import sys
+import time
 import io
 import zipfile
 import cv2
@@ -58,22 +57,6 @@ def index():
 
 # Load model once at startup, use CUDA if available
 MODEL_DEVICE = 'cuda' if cuda.is_available() else 'cpu'
-_model = None
-def get_model():
-    global _model
-    if _model is None:
-        _model = YOLO(WEIGHTS_FILE)
-        if MODEL_DEVICE == 'cuda':
-            _model.to('cuda')
-    return _model
-
-def cleanup_session(session_id):
-    # Remove files for this session
-    upload_dir = Path(app.config['UPLOAD_FOLDER']) / session_id
-    results_dir = Path(app.config['RESULTS_FOLDER']) / session_id
-    for d in [upload_dir, results_dir]:
-        if d.exists():
-            shutil.rmtree(d)
 
 # save the uploaded files
 @app.route('/uploads', methods=['POST'])
@@ -165,7 +148,6 @@ def start_processing():
         "sessionId": session_id
     }
     session['job_state'] = job_state
-    filename_map = session.get('filename_map', {})
     upload_dir = Path(app.config['UPLOAD_FOLDER']) / session_id
     results_dir = Path(app.config['RESULTS_FOLDER']) / session_id
     # clean out old results if needed
@@ -446,18 +428,20 @@ def print_startup_info():
             print(f"Could not get NemaQuant script details: {e}")
 
 @app.before_request
-def ensure_session_id():
+def startup_routine():
+    # Ensure session id
     if 'id' not in session:
         session['id'] = str(uuid.uuid4())
-
-# Explicit endpoint for safe session cleanup
-@app.route('/cleanup', methods=['POST'])
-def cleanup_endpoint():
-    if 'id' in session:
-        cleanup_session(session['id'])
-        session.clear()
-        return jsonify({'status': 'cleaned up'})
-    return jsonify({'error': 'No session to clean up'}), 400
+    print('Running periodic cleanup of old sessions...')
+    # Cleanup old session folders
+    max_age_hours = 24
+    now = time.time()
+    for base_dir in [UPLOAD_FOLDER, RESULTS_FOLDER, ANNOT_FOLDER]:
+        for session_dir in Path(base_dir).iterdir():
+            if session_dir.is_dir():
+                mtime = session_dir.stat().st_mtime
+                if now - mtime > max_age_hours * 3600:
+                    shutil.rmtree(session_dir)
 
 if __name__ == '__main__':
     print_startup_info()