#!/bin/bash
# Déploiement Option B - FAISS + GraphBuilder
# Corrections pour workflows automatiques

set -e  # Stop on error

echo "=== Déploiement Option B - Workflows Automatiques ==="
echo ""

# Couleurs
GREEN='\033[0;32m'
BLUE='\033[0;34m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo -e "${BLUE}1. Sauvegarde des fichiers actuels${NC}"
sudo cp /opt/rpa_vision_v3/server/processing_pipeline.py \
        /opt/rpa_vision_v3/server/processing_pipeline.py.backup_$(date +%Y%m%d_%H%M%S)
echo "  ✓ processing_pipeline.py sauvegardé"

sudo cp /opt/rpa_vision_v3/core/graph/graph_builder.py \
        /opt/rpa_vision_v3/core/graph/graph_builder.py.backup_$(date +%Y%m%d_%H%M%S)
echo "  ✓ graph_builder.py sauvegardé"

echo ""
echo -e "${BLUE}2. Déploiement des corrections${NC}"

# Correction FAISS (typo dimension → dimensions)
sudo cp /home/dom/ai/rpa_vision_v3/processing_pipeline.py \
        /opt/rpa_vision_v3/server/processing_pipeline.py
sudo chown rpa:rpa /opt/rpa_vision_v3/server/processing_pipeline.py
echo "  ✓ processing_pipeline.py déployé (fix FAISS)"

# Correction GraphBuilder (nouvelle API ScreenTemplate)
sudo cp /home/dom/ai/rpa_vision_v3/graph_builder.py \
        /opt/rpa_vision_v3/core/graph/graph_builder.py
sudo chown rpa:rpa /opt/rpa_vision_v3/core/graph/graph_builder.py
echo "  ✓ graph_builder.py déployé (Option B progressive)"

echo ""
echo -e "${BLUE}3. Création du dossier prototypes${NC}"
sudo mkdir -p /opt/rpa_vision_v3/data/training/prototypes
sudo chown -R rpa:rpa /opt/rpa_vision_v3/data/training/prototypes
echo "  ✓ /opt/rpa_vision_v3/data/training/prototypes créé"

echo ""
echo -e "${BLUE}4. Redémarrage du worker${NC}"
sudo systemctl restart rpa-vision-v3-worker.service
sleep 2
echo "  ✓ Worker redémarré"

echo ""
echo -e "${BLUE}5. Vérification du statut${NC}"
if systemctl is-active --quiet rpa-vision-v3-worker.service; then
    echo -e "  ${GREEN}✓ Worker actif${NC}"
else
    echo -e "  ${YELLOW}⚠ Worker non actif, vérifier logs${NC}"
    sudo journalctl -u rpa-vision-v3-worker -n 20 --no-pager
    exit 1
fi

echo ""
echo -e "${GREEN}=== Déploiement terminé avec succès ===${NC}"
echo ""
echo "Prochaines étapes:"
echo "  1. Tester sur une nouvelle session : cd /home/dom/ai/rpa_vision_v3/agent_v0 && ./run.sh"
echo "  2. Ou reprocesser sessions existantes : bash /home/dom/ai/rpa_vision_v3/reprocess_sessions.sh"
echo ""
echo "Logs à surveiller:"
echo "  journalctl -u rpa-vision-v3-worker -f | grep -E '(Embeddings|Workflow|FAISS)'"