test_functionality.py

#!/usr/bin/env python3
"""
Test script to verify TTS service core functionality is preserved after logging fixes.
This can be run in the Hugging Face environment to verify everything works.
"""

def test_tts_functionality():
    """Test core TTS functionality without actually running synthesis"""
    print("🧪 Testing TTS Service Core Functionality")
    print("=" * 60)
    
    try:
        # Test imports and basic setup
        print("1️⃣ Testing imports...")
        import torch
        import numpy as np
        print("   ✅ Core dependencies imported successfully")
        
        # Test device detection
        print("2️⃣ Testing device detection...")
        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
        print(f"   ✅ Device detected: {device}")
        
        # Test logging functions
        print("3️⃣ Testing logging functions...")
        from app import safe_log, logging_ok
        print(f"   ✅ Logging status: {'OK' if logging_ok else 'FALLBACK'}")
        safe_log("info", "Test message from functionality test")
        print("   ✅ Safe logging works")
        
        # Test MCP availability
        print("4️⃣ Testing MCP integration...")
        from app import MCP_AVAILABLE
        print(f"   ✅ MCP Available: {'Yes' if MCP_AVAILABLE else 'No'}")
        
        # Test voice presets
        print("5️⃣ Testing voice presets...")
        from app import VOICE_PRESETS
        print(f"   ✅ Voice presets loaded: {len(VOICE_PRESETS)} options")
        for i, (code, desc) in enumerate(VOICE_PRESETS[:3]):  # Show first 3
            print(f"      {i+1}. {code}: {desc}")
        if len(VOICE_PRESETS) > 3:
            print(f"      ... and {len(VOICE_PRESETS)-3} more")
        
        print("\n🎉 All functionality tests passed!")
        return True
        
    except Exception as e:
        print(f"\n❌ Functionality test failed: {e}")
        import traceback
        traceback.print_exc()
        return False

def test_synthesis_stub():
    """Test synthesis function structure without actual model loading"""
    print("\n🎤 Testing Synthesis Function Structure")
    print("=" * 60)
    
    try:
        from app import synthesize_speech
        print("✅ synthesize_speech function imported successfully")
        
        # Test with empty input (should return error gracefully)
        result = synthesize_speech("", "v2/en_speaker_6")
        if result[0] is None and "Please enter some text" in result[1]:
            print("✅ Empty input handling works correctly")
        else:
            print("⚠️ Empty input handling might need review")
        
        print("✅ Synthesis function structure is correct")
        return True
        
    except Exception as e:
        print(f"❌ Synthesis function test failed: {e}")
        return False

def test_gradio_interface():
    """Test that Gradio interface can be created"""
    print("\n🖥️ Testing Gradio Interface Creation")
    print("=" * 60)
    
    try:
        # This would normally create the interface
        # We'll just verify the imports work
        import gradio as gr
        print("✅ Gradio imported successfully")
        
        # Test that our app structure is compatible
        from app import VOICE_PRESETS, get_system_info
        info = get_system_info()
        print("✅ System info function works")
        print(f"   System info preview: {info[:100]}...")
        
        print("✅ Gradio interface structure is compatible")
        return True
        
    except ImportError:
        print("⚠️ Gradio not available (expected in local environment)")
        return True  # This is expected locally
    except Exception as e:
        print(f"❌ Gradio interface test failed: {e}")
        return False

if __name__ == "__main__":
    print("🔧 TTS Service Functionality Verification")
    print("🎯 Testing that logging fixes preserve core functionality")
    print("=" * 80)
    
    success = True
    
    # Run all tests
    if not test_tts_functionality():
        success = False
    
    if not test_synthesis_stub():
        success = False
        
    if not test_gradio_interface():
        success = False
    
    print("\n" + "=" * 80)
    if success:
        print("🎉 ALL FUNCTIONALITY TESTS PASSED!")
        print("✅ TTS service core functionality is preserved")
        print("✅ Logging fixes don't break existing features")
        print("✅ Ready for deployment with improved logging")
    else:
        print("❌ SOME FUNCTIONALITY TESTS FAILED!")
        print("❌ Review the errors above before deployment")
        
    print(f"\n📋 Test Summary:")
    print(f"   • Core imports and device detection: {'✅' if success else '❌'}")
    print(f"   • Logging system integration: {'✅' if success else '❌'}")
    print(f"   • MCP compatibility: {'✅' if success else '❌'}")
    print(f"   • Voice presets and synthesis: {'✅' if success else '❌'}")
    print(f"   • Interface compatibility: {'✅' if success else '❌'}")