from modules.scraper import VideoScraper
from modules.llm_processor import LLMProcessor
import logging

# Configuration du logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def test_pipeline(url):
    # Vérifier qu'on a une URL
    if not url:
        print("❌ Erreur: Veuillez fournir une URL à tester")
        return
    
    try:
        # 1. Scraping
        scraper = VideoScraper()
        print("🔍 Récupération des données de la vidéo...")
        video_data = scraper.scrape_video(url)
        print(f"✅ Données récupérées: {video_data}\n")
        
        # 2. Génération du formulaire
        llm = LLMProcessor()
        print("🤖 Génération du formulaire SCAM...")
        result = llm.process_video(video_data)
        
        if result['success']:
            print("✅ Formulaire généré:")
            print(result['form_data'])
        else:
            print(f"❌ Erreur: {result['error']}")
        
    except Exception as e:
        print(f"❌ Erreur: {str(e)}")

if __name__ == "__main__":
    import sys
    if len(sys.argv) < 2:
        print("❌ Usage: python test_pipeline.py <URL>")
        sys.exit(1)
    test_pipeline(sys.argv[1])