#!/bin/bash
# Script de validation des screenshots après capture

echo "🔍 Validation des Screenshots"
echo "=============================="
echo ""

# Trouver la dernière session
LATEST_SESSION=$(find /opt/rpa_vision_v3/data/training/sessions -maxdepth 2 -name "sess_*" -type d | sort -r | head -1)

if [ -z "$LATEST_SESSION" ]; then
    echo "❌ Aucune session trouvée"
    exit 1
fi

echo "📁 Session trouvée: $(basename $LATEST_SESSION)"
echo ""

# Vérifier le répertoire shots
SHOTS_DIR="${LATEST_SESSION}/shots"
if [ -d "$SHOTS_DIR" ]; then
    SHOT_COUNT=$(ls -1 "$SHOTS_DIR"/*.png 2>/dev/null | wc -l)
    echo "✅ Répertoire shots/ existe"
    echo "📸 Screenshots trouvés: $SHOT_COUNT fichiers"

    if [ $SHOT_COUNT -gt 0 ]; then
        echo ""
        echo "📋 Liste des screenshots:"
        ls -lh "$SHOTS_DIR"/*.png | awk '{print "  - " $9 " (" $5 ")"}'

        # Calculer la taille totale
        TOTAL_SIZE=$(du -sh "$SHOTS_DIR" | awk '{print $1}')
        echo ""
        echo "💾 Taille totale: $TOTAL_SIZE"
        echo ""
        echo "✅ SUCCÈS: Les screenshots sont bien conservés après traitement"
    else
        echo "❌ ÉCHEC: Le répertoire existe mais est vide"
        exit 1
    fi
else
    echo "❌ ÉCHEC: Le répertoire shots/ n'existe pas"
    echo "   Attendu: $SHOTS_DIR"
    exit 1
fi

# Vérifier le JSON
JSON_FILE=$(find /opt/rpa_vision_v3/data/training/sessions -name "session_$(basename $LATEST_SESSION).json" | head -1)
if [ -f "$JSON_FILE" ]; then
    echo "✅ Fichier JSON trouvé: $(basename $JSON_FILE)"

    # Compter les screenshots dans le JSON
    JSON_SHOT_COUNT=$(grep -o '"screenshot_id"' "$JSON_FILE" | wc -l)
    echo "📊 Screenshots référencés dans JSON: $JSON_SHOT_COUNT"

    if [ $SHOT_COUNT -eq $JSON_SHOT_COUNT ]; then
        echo "✅ Cohérence JSON ↔ Fichiers: OK"
    else
        echo "⚠️  Attention: Différence entre JSON ($JSON_SHOT_COUNT) et fichiers ($SHOT_COUNT)"
    fi
else
    echo "⚠️  Fichier JSON non trouvé"
fi

echo ""
echo "🎯 Résumé:"
echo "  - Session: $(basename $LATEST_SESSION)"
echo "  - Screenshots: $SHOT_COUNT fichiers conservés"
echo "  - Taille: $TOTAL_SIZE"
echo "  - Statut: ✅ Nettoyage prématuré DÉSACTIVÉ avec succès"