Add files via upload

tts_server.py

@@ -650,9 +650,19 @@ async def generate_audio(text, voice, language, temperature, repetition_penalty,
     response = model_engine.generate_tts(text, voice, language, temperature, repetition_penalty, speed, pitch, output_file, streaming)
     # If streaming, then return the generator as-is, otherwise just exhaust it and return
     if streaming:
-        return response
+        # Return an async generator that yields from the response
+        async def stream_response():
+            async for chunk in response:
+                yield chunk
+        return stream_response()
+    else:
+        # Exhaust the generator and handle any errors
+        try:
             async for _ in response:
                 pass
+        except Exception as e:
+            print(f"[GEN] Error during audio generation: {str(e)}")
+            raise
 
 ###########################
 #### PREVIEW VOICE API ####