Files
twitchBot-intelligent/test_manual_subtitle.py
T

79 lines
2.6 KiB
Python

#!/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é !")