Spaces:
Sleeping
Sleeping
File size: 1,553 Bytes
69427f5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
"""
Test TTS functionality
"""
import os
from fitness_gradio.ui.tts_utils import GroqTTS, generate_speech_for_text, clean_tts_markup
def test_tts_setup():
"""Test basic TTS setup and functionality."""
print("Testing TTS setup...")
# Check if API key is available
api_key = os.getenv("GROQ_API_KEY")
if not api_key:
print("❌ GROQ_API_KEY not found in environment variables")
print("Please set GROQ_API_KEY to test TTS functionality")
return False
print("✅ GROQ_API_KEY found")
# Test TTS instance creation
try:
tts = GroqTTS()
print("✅ GroqTTS instance created successfully")
except Exception as e:
print(f"❌ Failed to create GroqTTS instance: {e}")
return False
# Test text cleaning
test_text = "**Hello** this is a *test* with `code` and [links](http://example.com)"
cleaned = clean_tts_markup(test_text)
print(f"Text cleaning test:")
print(f" Original: {test_text}")
print(f" Cleaned: {cleaned}")
# Test voice lists
english_voices = GroqTTS.get_available_voices("playai-tts")
arabic_voices = GroqTTS.get_available_voices("playai-tts-arabic")
print(f"✅ Available English voices: {len(english_voices)}")
print(f"✅ Available Arabic voices: {len(arabic_voices)}")
print("🎉 All TTS setup tests passed!")
print("Note: Actual TTS generation will be tested when the UI is used with a valid API key.")
return True
if __name__ == "__main__":
test_tts_setup()
|