fix gradio too many requests exception

utils/tts.py

@@ -20,13 +20,7 @@ try:
 except AttributeError as e:
     # Specifically catch the EspeakWrapper.set_data_path error
     if "EspeakWrapper" in str(e) and "set_data_path" in str(e):
-        logger.warning("Kokoro import failed due to EspeakWrapper.set_data_path issue")
-        logger.info("Try to use Kokoro FastAPI server")
-        client = Client("Remsky/Kokoro-TTS-Zero")
-        result = client.predict(
-                api_name="/lambda"
-        )
-        logger.debug(f"result get back from Kokora FastAPI server: {result}")
+        logger.warning("Kokoro import failed due to EspeakWrapper.set_data_path issue, falling back to Kokoro FastAPI server")
         if result:
             KOKORO_SPACE_AVAILABLE = True
     else:
@@ -36,47 +30,6 @@ except AttributeError as e:
 except ImportError:
     logger.warning("Kokoro TTS engine is not available")
 
-# Try to import Dia as fallback
-# if not KOKORO_AVAILABLE:
-#     try:
-#         logger.info("Attempting to import Dia TTS engine as fallback")
-#         try:
-#             # Check if required dependencies for Dia are available
-#             import torch
-#             logger.info("PyTorch is available for Dia TTS")
-#         except ImportError as torch_err:
-#             logger.error(f"PyTorch dependency for Dia TTS is missing: {str(torch_err)}")
-#             raise ImportError(f"PyTorch dependency required for Dia TTS: {str(torch_err)}") from torch_err
-            
-#         # Try to import the Dia module
-#         try:
-#             from utils.tts_dia import _get_model as get_dia_model
-#             logger.info("Successfully imported Dia TTS module")
-            
-#             # Verify the model can be accessed
-#             logger.info("Verifying Dia model can be accessed")
-#             model_info = get_dia_model.__module__
-#             logger.info(f"Dia model module: {model_info}")
-            
-#             DIA_AVAILABLE = True
-#             logger.info("Dia TTS engine is available as fallback")
-#         except ImportError as module_err:
-#             logger.error(f"Failed to import Dia TTS module: {str(module_err)}")
-#             logger.error(f"Module path: {module_err.__traceback__.tb_frame.f_globals.get('__file__', 'unknown')}")
-#             raise
-#         except AttributeError as attr_err:
-#             logger.error(f"Dia TTS module attribute error: {str(attr_err)}")
-#             logger.error(f"This may indicate the module exists but has incorrect structure")
-#             raise
-#     except ImportError as e:
-#         logger.error(f"Dia TTS engine is not available due to import error: {str(e)}")
-#         logger.error(f"Import path attempted: {e.__traceback__.tb_frame.f_globals.get('__name__', 'unknown')}")
-#         logger.warning("Will use dummy TTS implementation as fallback")
-#     except Exception as e:
-#         logger.error(f"Unexpected error initializing Dia TTS: {str(e)}")
-#         logger.error(f"Error type: {type(e).__name__}")
-#         logger.error("Will use dummy TTS implementation as fallback")
-
 class TTSEngine:
     def __init__(self, lang_code='z'):
         """Initialize TTS Engine with Kokoro or Dia as fallback
@@ -108,7 +61,7 @@ class TTSEngine:
                 self.engine_type = "kokoro_space"
                 logger.info("TTS engine successfully initialized with Kokoro FastAPI server")
             except Exception as kokoro_err:
-                logger.error(f"Failed to initialize Kokoro pipeline: {str(kokoro_err)}")
+                logger.error(f"Failed to initialize Kokoro space: {str(kokoro_err)}")
                 logger.error(f"Error type: {type(kokoro_err).__name__}")
                 logger.info("Will try to fall back to Dia TTS engine")