fix: restart Hindsight after pg_restore to reload data

pg_restore modifies PG underneath but the running API
has cached state. Restarting forces reconnection to
the restored database.

restore.py now returns exit codes:
0 = data restored (restart needed)
2 = no backup found (skip restart)
1 = error

scripts/entrypoint.sh

@@ -53,12 +53,44 @@ done
 
 # ============================================================
 # STEP 4: Restore from backup (PG is now running)
+# After pg_restore, we must restart Hindsight so the API
+# reconnects to PG and sees the restored data.
+# Exit codes from restore.py:
+#   0 = data restored successfully
+#   2 = no backup found (skip restart)
+#   1 = error
 # ============================================================
 if [ -n "${HF_TOKEN:-}" ]; then
     log "Attempting restore from HF Dataset..."
-    python3 /opt/backup/restore.py || {
-        log "Restore failed or no backup found — continuing with fresh database"
-    }
+    python3 /opt/backup/restore.py
+    restore_exit=$?
+
+    if [ "$restore_exit" -eq 0 ]; then
+        log "Restore succeeded — restarting Hindsight to load restored data..."
+        kill "$HINDSIGHT_PID" 2>/dev/null || true
+        wait "$HINDSIGHT_PID" 2>/dev/null || true
+        sleep 2
+
+        /app/start-all.sh &
+        HINDSIGHT_PID=$!
+        log "Hindsight restarted (PID $HINDSIGHT_PID)"
+
+        for i in $(seq 1 60); do
+            if curl -sf http://localhost:${HINDSIGHT_API_PORT:-7860}/health > /dev/null 2>&1; then
+                log "Hindsight is healthy after restore"
+                break
+            fi
+            if ! kill -0 "$HINDSIGHT_PID" 2>/dev/null; then
+                log "Hindsight died after restore restart"
+                exit 1
+            fi
+            sleep 5
+        done
+    elif [ "$restore_exit" -eq 2 ]; then
+        log "No backup found — continuing with fresh database"
+    else
+        log "Restore failed — continuing with fresh database"
+    fi
 else
     log "HF_TOKEN not set — skipping restore"
 fi

scripts/restore.py

@@ -41,16 +41,22 @@ def find_pg_bin(name: str) -> str:
     raise FileNotFoundError(f"{name} not found in ~/.pg0/installation/")
 
 
-def main() -> None:
+EXIT_RESTORED = 0   # Data was restored — caller should restart Hindsight
+EXIT_ERROR = 1       # Restore failed
+EXIT_NO_BACKUP = 2   # No backup found — skip restart
+
+
+def main() -> int:
+    """Returns exit code: 0=restored, 1=error, 2=no backup."""
     if not HF_TOKEN:
         log("HF_TOKEN not set — skipping restore")
-        return
+        return EXIT_NO_BACKUP
 
     try:
         from huggingface_hub import HfApi, hf_hub_download
     except ImportError:
         log("huggingface_hub not installed — skipping restore")
-        return
+        return EXIT_NO_BACKUP
 
     api = HfApi(token=HF_TOKEN)
 
@@ -59,11 +65,11 @@ def main() -> None:
         files = list(api.list_repo_files(repo_id=HF_REPO, repo_type="dataset"))
     except Exception as e:
         log(f"Cannot access repo {HF_REPO}: {e}")
-        return
+        return EXIT_ERROR
 
     if "snapshots/latest.pgdump" not in files:
         log("No pg_dump backup found in HF Dataset — starting fresh")
-        return
+        return EXIT_NO_BACKUP
 
     log(f"Downloading latest backup from {HF_REPO}...")
 
@@ -114,17 +120,20 @@ def main() -> None:
         ]
         if real_errors:
             log(f"pg_restore had errors: {'; '.join(real_errors[:5])}")
+            return EXIT_ERROR
         else:
             log("pg_restore completed (minor warnings only)")
     else:
         log("pg_restore completed successfully")
 
-    log("Restore complete")
+    log("Restore complete — Hindsight should be restarted to load restored data")
+    return EXIT_RESTORED
 
 
 if __name__ == "__main__":
     try:
-        main()
+        code = main()
+        sys.exit(code)
     except Exception as e:
         log(f"FAILED: {e}")
         sys.exit(1)