Fix: Add local file persistence to JobRegistry to prevent 404s after space restart

backend/core/job_registry.py

@@ -1,10 +1,32 @@
 import threading
+import json
+import os
+import logging
 from typing import Dict, Any
 
+logger = logging.getLogger(__name__)
+
 class JobRegistry:
-    def __init__(self):
+    def __init__(self, storage_file: str = "/tmp/job_registry.json"):
         self._jobs: Dict[str, Dict[str, Any]] = {}
         self._lock = threading.Lock()
+        self.storage_file = storage_file
+        self._load()
+
+    def _save(self):
+        try:
+            with open(self.storage_file, "w") as f:
+                json.dump(self._jobs, f)
+        except Exception as e:
+            logger.error(f"Failed to save job registry to disk: {e}")
+
+    def _load(self):
+        if os.path.exists(self.storage_file):
+            try:
+                with open(self.storage_file, "r") as f:
+                    self._jobs = json.load(f)
+            except Exception as e:
+                logger.error(f"Failed to load job registry from disk: {e}")
 
     def create_job(self, job_id: str, initial_data: Dict[str, Any] = None):
         with self._lock:
@@ -14,11 +36,13 @@ class JobRegistry:
                 "results": None,
                 **(initial_data or {})
             }
+            self._save()
 
     def update_job(self, job_id: str, **kwargs):
         with self._lock:
             if job_id in self._jobs:
                 self._jobs[job_id].update(kwargs)
+                self._save()
 
     def get_job(self, job_id: str) -> Dict[str, Any]:
         with self._lock:
@@ -28,6 +52,7 @@ class JobRegistry:
         with self._lock:
             if job_id in self._jobs:
                 del self._jobs[job_id]
+                self._save()
 
     def list_jobs(self) -> Dict[str, Dict[str, Any]]:
         with self._lock: