#!/usr/bin/env python3
"""
Script de test pour le traitement manuel des sous-titres
"""

import os
import sys
import json

# Ajouter le chemin vers les modules
sys.path.append('.')

def test_manual_subtitle_processing():
    """Test du traitement manuel des sous-titres"""
    print("🧪 Test du traitement manuel des sous-titres")
    print("=" * 50)
    
    # Vérifier que le fichier de configuration existe
    config_file = "config/config.json"
    if not os.path.exists(config_file):
        print(f"❌ Fichier de configuration non trouvé: {config_file}")
        return False
    
    # Vérifier que le dossier record existe et contient des fichiers
    record_dir = "record"
    if not os.path.exists(record_dir):
        print(f"❌ Dossier record non trouvé: {record_dir}")
        return False
    
    audio_files = [f for f in os.listdir(record_dir) if f.endswith('.mp3')]
    if not audio_files:
        print(f"❌ Aucun fichier audio trouvé dans {record_dir}")
        return False
    
    print(f"✅ {len(audio_files)} fichiers audio trouvés")
    print("📁 Fichiers audio disponibles:")
    for file in audio_files[:5]:  # Afficher les 5 premiers fichiers
        print(f"   - {file}")
    
    # Créer une instance du bot de sous-titres
    try:
        from fonction.first_class import Subtitle_translation
        
        subtitle_bot = Subtitle_translation(config_file)
        print("✅ Bot de sous-titres créé avec succès")
        
        # Lancer le traitement manuel
        print("🔍 Lancement du traitement manuel...")
        subtitle_bot.verif_file_transcribe()
        
        # Vérifier si des sous-titres ont été créés
        if subtitle_bot.subtitle:
            print("✅ Sous-titres créés avec succès")
            print("📝 Sous-titres générés:")
            for time_key, text in subtitle_bot.subtitle.items():
                print(f"   {time_key}: {text}")
        else:
            print("⚠️  Aucun sous-titre créé")
        
        # Vérifier le stockage
        from fonction.first_class import storage
        subtitle_data = storage.read("subtitle_data")
        if subtitle_data:
            print(f"💾 {len(subtitle_data)} sous-titre(s) sauvegardé(s) dans le stockage")
        
        return True
        
    except Exception as e:
        print(f"❌ Erreur lors du test: {e}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == '__main__':
    success = test_manual_subtitle_processing()
    if success:
        print("\n✅ Test réussi !")
    else:
        print("\n❌ Test échoué !")