Upload 2 files

scripts/init_pg.sh

@@ -0,0 +1,49 @@
+#!/bin/bash
+# init_pg.sh — Inicia PostgreSQL + restaura backup + sobe app com 2 workers
+
+set -e
+
+echo "🔧 [INIT] Iniciando PostgreSQL..."
+
+# Inicia PostgreSQL como usuário postgres
+pg_ctlcluster 14 main start 2>/dev/null || pg_ctlcluster 15 main start 2>/dev/null || \
+  su - postgres -c "pg_ctl -D $PGDATA start" 2>/dev/null || \
+  service postgresql start
+
+# Aguarda PostgreSQL estar pronto
+for i in $(seq 1 30); do
+    if pg_isready -q 2>/dev/null; then
+        echo "✅ [INIT] PostgreSQL pronto"
+        break
+    fi
+    sleep 1
+done
+
+# Cria usuário e banco se não existirem
+su - postgres -c "psql -tc \"SELECT 1 FROM pg_roles WHERE rolname='akira'\" | grep -q 1" || \
+    su - postgres -c "psql -c \"CREATE USER akira WITH PASSWORD 'akira' SUPERUSER;\""
+
+su - postgres -c "psql -tc \"SELECT 1 FROM pg_database WHERE datname='akira'\" | grep -q 1" || \
+    su - postgres -c "psql -c \"CREATE DATABASE akira OWNER akira;\""
+
+# Tenta restaurar backup se existir
+echo "🔄 [INIT] Verificando backup..."
+BACKUP_FILE="/akira/data/cloud_sync/akira_dump.sql"
+if [ -f "$BACKUP_FILE" ]; then
+    echo "📥 [INIT] Restaurando backup do PostgreSQL..."
+    PGPASSWORD=akira psql -h localhost -U akira -d akira -f "$BACKUP_FILE" 2>/dev/null || \
+        echo "⚠️ [INIT] Restauração falhou (banco pode já ter dados)"
+fi
+
+echo "🚀 [INIT] Iniciando Akira com 2 workers..."
+
+# Inicia aplicação com 2 workers + 8 threads cada
+exec gunicorn \
+    --bind 0.0.0.0:7860 \
+    --workers 2 \
+    --threads 8 \
+    --timeout 180 \
+    --preload \
+    --graceful-timeout 30 \
+    --keep-alive 5 \
+    main:app

scripts/pg_backup.sh

@@ -0,0 +1,80 @@
+#!/bin/bash
+# pg_backup.sh — Backup periódico do PostgreSQL → HF Dataset
+# Executa a cada 2 horas via cron ou background loop
+
+set -e
+
+BACKUP_DIR="/akira/data/cloud_sync"
+BACKUP_FILE="$BACKUP_DIR/akira_dump.sql"
+DATASET_NAME="${HF_BACKUP_DATASET:-akira-db-backup}"
+SYNC_INTERVAL=${BACKUP_SYNC_INTERVAL:-7200}
+
+mkdir -p "$BACKUP_DIR"
+
+backup_pg() {
+    echo "🔄 [BACKUP] Iniciando backup do PostgreSQL..."
+    
+    PGPASSWORD=akira pg_dump -h localhost -U akira -d akira \
+        --format=custom \
+        --compress=9 \
+        --file="$BACKUP_DIR/akira_dump.custom" 2>/dev/null
+    
+    PGPASSWORD=akira pg_dump -h localhost -U akira -d akira \
+        > "$BACKUP_FILE" 2>/dev/null
+    
+    if [ $? -eq 0 ]; then
+        echo "✅ [BACKUP] Dump concluído: $(du -h $BACKUP_FILE | cut -f1)"
+        
+        # Upload para HF Dataset se HF_TOKEN estiver configurado
+        if [ -n "$HF_TOKEN" ]; then
+            echo "📤 [BACKUP] Enviando para HF Dataset: $DATASET_NAME"
+            python3 -c "
+import os, sys
+from huggingface_hub import HfApi
+try:
+    api = HfApi(token='$HF_TOKEN')
+    repo_id = os.environ.get('HF_AUTHOR', 'akra35567') + '/' + '$DATASET_NAME'
+    api.upload_folder(
+        folder_path='$BACKUP_DIR',
+        repo_id=repo_id,
+        repo_type='dataset',
+        allow_patterns=['akira_dump.*'],
+    )
+    print('✅ [BACKUP] Upload concluído')
+except Exception as e:
+    print(f'⚠️ [BACKUP] Upload falhou: {e}')
+" 2>/dev/null || echo "⚠️ [BACKUP] Upload pulado"
+        fi
+    else
+        echo "❌ [BACKUP] Dump falhou"
+    fi
+}
+
+restore_pg() {
+    echo "📥 [RESTORE] Verificando backup para restaurar..."
+    
+    if [ -f "$BACKUP_FILE" ]; then
+        echo "📥 [RESTORE] Restaurando backup..."
+        PGPASSWORD=akira psql -h localhost -U akira -d akira -f "$BACKUP_FILE" 2>/dev/null
+        echo "✅ [RESTORE] Concluído"
+    elif [ -f "$BACKUP_DIR/akira_dump.custom" ]; then
+        echo "📥 [RESTORE] Restaurando backup custom..."
+        PGPASSWORD=akira pg_restore -h localhost -U akira -d akira "$BACKUP_DIR/akira_dump.custom" 2>/dev/null
+        echo "✅ [RESTORE] Concluído"
+    else
+        echo "ℹ️ [RESTORE] Nenhum backup encontrado"
+    fi
+}
+
+# Se chamado com argumento "restore"
+if [ "$1" = "restore" ]; then
+    restore_pg
+    exit 0
+fi
+
+# Loop de backup periódico
+echo "⏱️ [BACKUP] Loop iniciado (intervalo: ${SYNC_INTERVAL}s)"
+while true; do
+    backup_pg
+    sleep $SYNC_INTERVAL
+done