Speech MCP

tests
"""
Test script for the audio processor module.

This script demonstrates how to use the AudioProcessor class directly.
"""

import sys
import time
import logging
import os

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[logging.StreamHandler()]
)
logger = logging.getLogger(__name__)

# Add the parent directory to sys.path to allow importing speech_mcp
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))

# Import the AudioProcessor
from speech_mcp.audio_processor import AudioProcessor

def on_audio_level(level):
    """Callback function for audio level updates"""
    # Print a simple visualization of the audio level
    bars = int(level * 50)
    print(f"\r[{'#' * bars}{' ' * (50 - bars)}] {level:.2f}", end="")

def main():
    """Main test function"""
    print("Testing AudioProcessor...")
    
    # Create an AudioProcessor with our callback
    processor = AudioProcessor(on_audio_level=on_audio_level)
    
    print("\nAvailable audio devices:")
    devices = processor.get_available_devices()
    for i, device in enumerate(devices):
        print(f"{i}: {device['name']} ({device['channels']} channels, {device['sample_rate']} Hz)")
    
    print("\nRecording audio for 5 seconds or until silence is detected...")
    print("Speak now!")
    
    # Record audio
    audio_file = processor.record_audio()
    
    if audio_file:
        print(f"\nAudio saved to: {audio_file}")
        
        # Play back the recorded audio
        print("Playing back recorded audio...")
        processor.play_audio_file(audio_file)
        
        # Clean up the audio file
        try:
            os.unlink(audio_file)
            print(f"Deleted temporary audio file: {audio_file}")
        except Exception as e:
            print(f"Error deleting audio file: {e}")
    else:
        print("Failed to record audio")
    
    # Clean up resources
    processor.cleanup()
    print("Test complete")

if __name__ == "__main__":
    main()