Michael Hu commited on
Commit
feebf14
·
1 Parent(s): e3843d2

handle voice generation failure exception

Browse files
Files changed (1) hide show
  1. utils/tts.py +12 -7
utils/tts.py CHANGED
@@ -164,13 +164,18 @@ class TTSEngine:
164
  # Use Kokoro FastAPI server for TTS generation
165
  logger.info("Generating speech using Kokoro FastAPI server")
166
  logger.info(f"text to generate speech on is: {text}")
167
- result = self.client.predict(
168
- text=text,
169
- voice_names=voice,
170
- speed=speed,
171
- api_name="/generate_speech_from_ui"
172
- )
173
- logger.info(f"Received audio from Kokoro FastAPI server: {result}")
 
 
 
 
 
174
  elif self.engine_type == "dia":
175
  # Use Dia for TTS generation
176
  try:
 
164
  # Use Kokoro FastAPI server for TTS generation
165
  logger.info("Generating speech using Kokoro FastAPI server")
166
  logger.info(f"text to generate speech on is: {text}")
167
+ try:
168
+ result = self.client.predict(
169
+ text=text,
170
+ voice_names='af_nova',
171
+ speed=speed,
172
+ api_name="/generate_speech_from_ui"
173
+ )
174
+ logger.info(f"Received audio from Kokoro FastAPI server: {result}")
175
+ except Exception as e:
176
+ logger.error(f"Failed to generate speech from Kokoro FastAPI server: {str(e)}")
177
+ logger.error(f"Error type: {type(e).__name__}")
178
+ logger.info("Falling back to dummy audio generation")
179
  elif self.engine_type == "dia":
180
  # Use Dia for TTS generation
181
  try: