simple_voice_test.py

#!/usr/bin/env python3
"""
Simple Interactive Voice Chat Test for Autumn AI Assistant
"""

import asyncio
import sys
import os
from pathlib import Path

# Add project root to path
sys.path.append(os.getcwd())

from core.kokoro_tts import KokoroTTSEngine

async def simple_voice_chat():
    """Simple interactive voice chat test."""
    print("🍂 Autumn's Voice Chat Test")
    print("=" * 30)
    print("🎵 Initializing voice system...")
    
    # Initialize TTS engine
    engine = KokoroTTSEngine(device="auto")
    success = await engine.initialize()
    
    if not success:
        print("❌ Failed to initialize voice system")
        return
    
    print(f"✅ Voice system ready!")
    print(f"🔥 Using ONNX: {engine.use_onnx}")
    print(f"🎭 Device: {engine.device}")
    print()
    print("Type messages for Autumn to speak!")
    print("Commands: 'quit' to exit, emotions: neutral, friendly, happy, calm, serious")
    print()
    
    current_emotion = "friendly"
    
    while True:
        try:
            # Get user input
            user_input = input(f"[{current_emotion}] Enter text: ").strip()
            
            if user_input.lower() in ['quit', 'exit', 'bye']:
                print("🍂 Goodbye!")
                await engine.speak("Goodbye! Take care!", emotion="friendly")
                break
            
            # Check for emotion change
            emotions = ['neutral', 'friendly', 'happy', 'calm', 'serious']
            if user_input.lower() in emotions:
                current_emotion = user_input.lower()
                print(f"🎭 Voice emotion changed to: {current_emotion}")
                await engine.speak(f"Voice emotion changed to {current_emotion}", emotion=current_emotion)
                continue
            
            if not user_input:
                continue
            
            print(f"🎵 Speaking with emotion: {current_emotion}")
            
            # Generate and play speech
            success = await engine.speak(user_input, emotion=current_emotion)
            
            if success:
                print("✅ Speech completed!")
            else:
                print("❌ Speech failed!")
                
        except KeyboardInterrupt:
            print("\n👋 Goodbye!")
            break
        except Exception as e:
            print(f"❌ Error: {e}")
    
    await engine.shutdown()
    print("🔊 Voice system shut down.")

if __name__ == "__main__":
    try:
        asyncio.run(simple_voice_chat())
    except KeyboardInterrupt:
        print("\n👋 Test interrupted. Goodbye!")
    except Exception as e:
        print(f"❌ Test failed: {e}")