#!/usr/bin/env python3
"""
Test rapide de Qwen3-VL:8b avec Ollama
"""

import sys
from pathlib import Path

# Ajouter le répertoire geniusia2 au path
sys.path.insert(0, str(Path(__file__).parent / "geniusia2"))

from core.llm_manager import LLMManager
from core.logger import Logger
import numpy as np

def test_qwen3_vl():
    """Test de connexion à Qwen3-VL."""
    print("=" * 60)
    print("Test de Qwen3-VL:8b")
    print("=" * 60)
    print()
    
    # Créer un logger
    logger = Logger()
    
    # Initialiser le LLM Manager avec Qwen3-VL
    print("1. Initialisation du LLM Manager...")
    try:
        llm = LLMManager(
            model_name="qwen3-vl:8b",
            logger=logger
        )
        print("   ✓ LLM Manager initialisé")
    except Exception as e:
        print(f"   ✗ Erreur: {e}")
        return False
    
    # Créer une image de test simple
    print("\n2. Création d'une image de test...")
    test_image = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
    print("   ✓ Image de test créée (100x100)")
    
    # Tester le raisonnement visuel
    print("\n3. Test du raisonnement visuel...")
    try:
        result = llm.reason_about_action(
            frame=test_image,
            detections=[],
            context={"window_title": "Test Window"}
        )
        print(f"   ✓ Raisonnement effectué")
        print(f"   Résultat: {result}")
    except Exception as e:
        print(f"   ⚠ Erreur (normal si pas de détections): {e}")
    
    print("\n" + "=" * 60)
    print("✓ Test terminé")
    print("=" * 60)
    print()
    print("Qwen3-VL:8b est opérationnel !")
    return True

if __name__ == "__main__":
    success = test_qwen3_vl()
    sys.exit(0 if success else 1)
