Update startup.sh for persistence

startup.sh

@@ -1,32 +1,26 @@
 #!/bin/bash
 echo "===== Startup $(date) ====="
 
-# Security: Set restrictive permissions
 umask 077
 
-# Install huggingface_hub
 pip3 install --no-cache-dir -q huggingface_hub==0.23.0 --break-system-packages 2>/dev/null
 
-# Install Node.js
 if ! command -v node &> /dev/null; then
   echo "[Install] Node.js..."
   curl -fsSL https://deb.nodesource.com/setup_20.x | bash - 2>/dev/null
   apt-get install -y nodejs 2>/dev/null
 fi
 
-# Install Claude CLI
 echo "[Install] Claude CLI..."
 npm install -g @anthropic-ai/claude-code 2>/dev/null
 
-# Install Codex CLI
 echo "[Install] Codex CLI..."
 npm install -g @openai/codex 2>/dev/null
 
-# Create secure storage
-mkdir -p /persistent/.secrets
+mkdir -p /persistent/.secrets /persistent/sessions
 chmod 700 /persistent/.secrets
 
-# Restore from dataset (encrypted)
+# Restore from dataset
 if [ -n "$HF_TOKEN" ] && [ -n "$HF_USERNAME" ]; then
   echo "[Storage] Restoring from dataset..."
   python3 -c "
@@ -36,37 +30,76 @@ token = os.environ.get('HF_TOKEN')
 username = os.environ.get('HF_USERNAME')
 repo_id = f'{username}/vps-storage'
 try:
-    snapshot_download(repo_id=repo_id, repo_type='dataset', local_dir='/persistent', token=token, ignore_patterns=['.gitattributes','README.md'])
+    snapshot_download(repo_id=repo_id, repo_type='dataset', local_dir='/persistent', token=token, ignore_patterns=['.gitattributes','README.md','.git/*'])
     print('[Storage] Restored!')
 except: print('[Storage] Fresh start')
 " 2>/dev/null
 fi
 
-# Secure backup (only backs up code, not secrets)
+# Reinstall packages
+if [ -s /persistent/installed_packages.txt ]; then
+  echo "[Packages] Reinstalling tracked packages..."
+  while IFS= read -r pkg; do
+    [ -z "$pkg" ] && continue
+    if [[ "$pkg" == apt:* ]]; then
+      apt_pkg="${pkg#apt:}"
+      apt-get install -y -qq "$apt_pkg" 2>/dev/null
+    else
+      pip3 install -q "$pkg" --break-system-packages 2>/dev/null
+    fi
+  done < /persistent/installed_packages.txt
+  echo "[Packages] Done!"
+fi
+
+# Auto-save every 3 minutes
 python3 -c "
-import os, time, threading, hashlib
-def secure_backup():
+import os, time, threading
+def auto_save():
     while True:
-        time.sleep(300)
+        time.sleep(180)
         token = os.environ.get('HF_TOKEN')
         username = os.environ.get('HF_USERNAME')
         if token and username:
             try:
                 from huggingface_hub import HfApi
                 api = HfApi(token=token)
-                # Only upload code files, skip secrets
                 api.upload_folder(
                     folder_path='/persistent',
                     repo_id=f'{username}/vps-storage',
                     repo_type='dataset',
-                    ignore_patterns=['.secrets/', '*.key', '*.pem', '.env'],
-                    commit_message='secure-backup'
+                    token=token,
+                    commit_message='auto-save',
+                    ignore_patterns=['.secrets/', '*.key', '*.pem', '.env', '.git/*']
                 )
             except: pass
-threading.Thread(target=secure_backup, daemon=True).start()
+threading.Thread(target=auto_save, daemon=True).start()
 " &
 
-# Keep-alive ping (human-like)
+# Package tracker
+python3 -c "
+import os, time, json, threading
+def watch_pkgs():
+    seen = set()
+    if os.path.exists('/persistent/installed_packages.txt'):
+        with open('/persistent/installed_packages.txt') as f:
+            seen = {l.strip() for l in f if l.strip()}
+    while True:
+        time.sleep(15)
+        try:
+            result = os.popen('pip3 list --format=json 2>/dev/null').read()
+            if result:
+                pkgs = json.loads(result)
+                for pkg in pkgs:
+                    name = pkg.get('name','')
+                    if name and name not in seen:
+                        with open('/persistent/installed_packages.txt', 'a') as f:
+                            f.write(name + '\n')
+                        seen.add(name)
+        except: pass
+threading.Thread(target=watch_pkgs, daemon=True).start()
+" &
+
+# Keep-alive ping
 python3 -c "
 import os, time, random, threading, requests
 def ping():
@@ -74,14 +107,35 @@ def ping():
         try:
             host = os.environ.get('SPACE_HOST','')
             if host:
-                headers = {
-                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
-                    'Accept': 'text/html,application/xhtml+xml',
-                }
-                requests.get(f'https://{host}/', headers=headers, timeout=10)
+                requests.get(f'https://{host}/', headers={'User-Agent': 'Mozilla/5.0'}, timeout=10)
         except: pass
-        time.sleep(random.randint(900, 3600))
+        time.sleep(random.randint(600, 1800))
 threading.Thread(target=ping, daemon=True).start()
 " &
 
+# Bashrc for persistent env
+cat > /root/.bashrc << 'EOF'
+export HISTFILE=/persistent/.bash_history
+export HISTSIZE=10000
+export HISTFILESIZE=20000
+export HISTCONTROL=ignoredups:erasedups
+shopt -s histappend
+alias pip='pip3'
+alias install='track_pkg'
+track_pkg() {
+    pip3 install "$@"
+    pkg_name=$(echo "$@" | awk '{print $1}' | sed 's/[^a-zA-Z0-9_-]//g')
+    if [ -n "$pkg_name" ] && ! grep -q "^${pkg_name}$" /persistent/installed_packages.txt 2>/dev/null; then
+        echo "$pkg_name" >> /persistent/installed_packages.txt
+        echo "[Tracking] Added: $pkg_name"
+    fi
+}
+EOF
+
+cat > /root/.profile << 'EOF'
+if [ -f /root/.bashrc ]; then . /root/.bashrc; fi
+EOF
+
+echo "[Startup] Ready!"
+
 exec uvicorn app:app --host 0.0.0.0 --port 7860 --ws websockets --proxy-headers --forwarded-allow-ips="*"