temporarily disable dia

requirements.txt

@@ -1,4 +1,4 @@
-streamlit>=1.21
+streamlit>=1.31
 nltk>=3.8
 librosa>=0.10
 ffmpeg-python>=0.2
@@ -12,4 +12,4 @@ soundfile>=0.13.0
 kokoro>=0.7.9
 ordered-set>=4.1.0
 phonemizer-fork>=3.3.2
-descript-audio-codec
+# descript-audio-codec

utils/tts.py

@@ -26,45 +26,45 @@ 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
+# 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")
+#         # 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}")
+#             # 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")
+#             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'):
@@ -92,13 +92,13 @@ class TTSEngine:
                 # Fall through to try Dia
         
         # Try Dia if Kokoro is not available or failed to initialize
-        if self.engine_type is None and DIA_AVAILABLE:
-            logger.info("Using Dia as fallback TTS engine")
-            # For Dia, we don't need to initialize anything here
-            # The model will be lazy-loaded when needed
-            self.pipeline = None
-            self.engine_type = "dia"
-            logger.info("TTS engine initialized with Dia (lazy loading)")
+        # if self.engine_type is None and DIA_AVAILABLE:
+        #     logger.info("Using Dia as fallback TTS engine")
+        #     # For Dia, we don't need to initialize anything here
+        #     # The model will be lazy-loaded when needed
+        #     self.pipeline = None
+        #     self.engine_type = "dia"
+        #     logger.info("TTS engine initialized with Dia (lazy loading)")
         
         # Use dummy if no TTS engines are available
         if self.engine_type is None: