#!/usr/bin/env python3
"""
Script de teste para verificar se todas as dependências estão funcionando
"""

def test_imports():
    """Testa se todas as importações funcionam"""
    try:
        import gradio as gr
        print("✓ Gradio importado com sucesso")
        print(f"  Versão do Gradio: {gr.__version__}")
    except ImportError as e:
        print(f"✗ Erro ao importar Gradio: {e}")
        return False
    
    try:
        from transformers import TrOCRProcessor, VisionEncoderDecoderModel
        print("✓ Transformers importado com sucesso")
    except ImportError as e:
        print(f"✗ Erro ao importar Transformers: {e}")
        return False
    
    try:
        from PIL import Image
        print("✓ PIL/Pillow importado com sucesso")
    except ImportError as e:
        print(f"✗ Erro ao importar PIL: {e}")
        return False
    
    try:
        import torch
        print("✓ PyTorch importado com sucesso")
        print(f"  Versão do PyTorch: {torch.__version__}")
    except ImportError as e:
        print(f"✗ Erro ao importar PyTorch: {e}")
        return False
    
    return True

def test_model_loading():
    """Testa se o modelo pode ser carregado"""
    try:
        from transformers import TrOCRProcessor, VisionEncoderDecoderModel
        
        print("Testando carregamento do modelo...")
        processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
        model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
        
        print("✓ Modelo carregado com sucesso")
        return True
    except Exception as e:
        print(f"✗ Erro ao carregar modelo: {e}")
        return False

if __name__ == "__main__":
    print("🧪 Executando testes do Professor IA")
    print("=" * 50)
    
    if test_imports():
        print("\n📦 Todas as importações estão funcionando!")
        
        if test_model_loading():
            print("\n🤖 Modelo carregado com sucesso!")
            print("\n✅ Todos os testes passaram! O aplicativo deve funcionar no Hugging Face.")
        else:
            print("\n⚠️ Problema com o carregamento do modelo.")
    else:
        print("\n❌ Problemas com as importações. Verifique o requirements.txt")