#!/usr/bin/env python3
"""
Script de test pour vérifier le fonctionnement du bot de sous-titres
"""

import os
import sys
import time

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

from fonction.first_class import Subtitle_translation

def test_subtitle_bot():
    """Test du bot de sous-titres"""
    print("🧪 Test du bot de 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")
    
    # Créer une instance du bot de sous-titres
    try:
        subtitle_bot = Subtitle_translation(config_file)
        print("✅ Bot de sous-titres créé avec succès")
        
        # Tester la méthode verif_file_transcribe
        print("🔍 Test de la vérification des fichiers...")
        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")
            for time_key, text in subtitle_bot.subtitle.items():
                print(f"   {time_key}: {text}")
        else:
            print("⚠️  Aucun sous-titre créé")
        
        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_subtitle_bot()
    if success:
        print("\n✅ Test réussi !")
    else:
        print("\n❌ Test échoué !")