changes in cleaning of tempfiles

app.py

@@ -6,8 +6,15 @@ import imageio.v2 as iio
 import tifffile
 import tempfile
 import urllib.request
+import os
 
-from core.utils import *
+from core.utils import (
+    get_sr_mode, normalize_stack, upscale, load_stack,
+    _is_demo_url, _demo_path_for_url,
+    _start_cleaner, citation_markdown, documentation_markdown
+)
+
+_start_cleaner()
 
 # Globals
 original_frames, aligned_frames = [], []
@@ -90,8 +97,13 @@ def load_from_url(request: gr.Request):
     if "file_url" in params:
         try:
             url = params["file_url"]
-            tmp_path = tempfile.mktemp(suffix=".tif")
-            urllib.request.urlretrieve(url, tmp_path)
+            if _is_demo_url(url):
+                tmp_path = _demo_path_for_url(url)
+                if not os.path.exists(tmp_path):
+                    urllib.request.urlretrieve(url, tmp_path)
+            else:
+                tmp_path = tempfile.mktemp(suffix=".tif")  # goes to WORK_DIR
+                urllib.request.urlretrieve(url, tmp_path)
             return [gr.update(value=tmp_path)]
         except Exception as e:
             print(f"[URL load failed] {e}")
@@ -276,8 +288,13 @@ with gr.Blocks() as demo:
         if "file_url" in params:
             try:
                 url = params["file_url"]
-                tmp_path = tempfile.mktemp(suffix=".tif")
-                urllib.request.urlretrieve(url, tmp_path)
+                if _is_demo_url(url):
+                    tmp_path = _demo_path_for_url(url)
+                    if not os.path.exists(tmp_path):
+                        urllib.request.urlretrieve(url, tmp_path)
+                else:
+                    tmp_path = tempfile.mktemp(suffix=".tif")  # goes to WORK_DIR
+                    urllib.request.urlretrieve(url, tmp_path)
                 stack = iio.mimread(tmp_path)
                 max_frame = len(stack) - 1
 
@@ -293,10 +310,24 @@ with gr.Blocks() as demo:
         # Two-stack file case (for stack-based alignment)
         if "file_url_1" in params and "file_url_2" in params:
             try:
-                tmp_path_1 = tempfile.mktemp(suffix=".tif")
-                tmp_path_2 = tempfile.mktemp(suffix=".tif")
-                urllib.request.urlretrieve(params["file_url_1"], tmp_path_1)
-                urllib.request.urlretrieve(params["file_url_2"], tmp_path_2)
+                u1, u2 = params["file_url_1"], params["file_url_2"]
+
+                if _is_demo_url(u1):
+                    tmp_path_1 = _demo_path_for_url(u1)
+                    if not os.path.exists(tmp_path_1):
+                        urllib.request.urlretrieve(u1, tmp_path_1)
+                else:
+                    tmp_path_1 = tempfile.mktemp(suffix=".tif")  # goes to WORK_DIR
+                    urllib.request.urlretrieve(u1, tmp_path_1)
+
+                if _is_demo_url(u2):
+                    tmp_path_2 = _demo_path_for_url(u2)
+                    if not os.path.exists(tmp_path_2):
+                        urllib.request.urlretrieve(u2, tmp_path_2)
+                else:
+                    tmp_path_2 = tempfile.mktemp(suffix=".tif")  # goes to WORK_DIR
+                    urllib.request.urlretrieve(u2, tmp_path_2)
+
                 results[5] = tmp_path_1  # ref_input
                 results[6] = tmp_path_2  # mov_input
             except Exception as e:

core/utils.py

@@ -2,6 +2,11 @@ from pystackreg import StackReg
 import numpy as np
 import imageio.v2 as iio
 from PIL import Image
+import time
+import threading
+import hashlib
+import tempfile
+import os
 
 def get_sr_mode(mode_str): return {
     "TRANSLATION": StackReg.TRANSLATION,
@@ -31,6 +36,48 @@ def load_stack(file):
     return normalize_stack(stack)
 
 
+##### CACHE #####
+
+# App-scoped cache dirs under the OS temp
+APP_TMP_ROOT = os.path.join(tempfile.gettempdir(), "psr_cache")
+WORK_DIR = os.path.join(APP_TMP_ROOT, "work")   # cleaned periodically
+DEMO_DIR = os.path.join(APP_TMP_ROOT, "demo")   # persistent demo cache
+os.makedirs(WORK_DIR, exist_ok=True)
+os.makedirs(DEMO_DIR, exist_ok=True)
+
+# Direct all tempfile.* calls to WORK_DIR so cleanup is safe
+tempfile.tempdir = WORK_DIR
+
+TTL_SECONDS = 30 * 60  # 30 minutes
+
+def _cleanup_old_files(folder, older_than_seconds):
+    now = time.time()
+    for root, _, files in os.walk(folder):
+        for name in files:
+            p = os.path.join(root, name)
+            try:
+                if now - os.path.getmtime(p) > older_than_seconds:
+                    os.remove(p)
+            except Exception as e:
+                print(f"[cleanup] {e}")
+
+def _start_cleaner():
+    def loop():
+        while True:
+            _cleanup_old_files(WORK_DIR, TTL_SECONDS)  # only clean WORK_DIR
+            time.sleep(TTL_SECONDS)
+    threading.Thread(target=loop, daemon=True).start()
+
+
+def _is_demo_url(url: str) -> bool:
+    return ("github.com/glichtner/pystackreg" in url) or \
+           ("raw.githubusercontent.com/glichtner/pystackreg" in url)
+
+def _demo_path_for_url(url: str, suffix=".tif"):
+    h = hashlib.sha256(url.encode("utf-8")).hexdigest()[:16]
+    return os.path.join(DEMO_DIR, f"psr-demo-{h}{suffix}")
+
+
 ##### Markdowns #####
 
 citation_markdown = """