Add an informative error when extension requirements are missing

extensions/coqui_tts/script.py

@@ -6,27 +6,14 @@ import time
 from pathlib import Path
 
 import gradio as gr
+from TTS.api import TTS
+from TTS.utils.synthesizer import Synthesizer
 
 from modules import chat, shared, ui_chat
 from modules.logging_colors import logger
 from modules.ui import create_refresh_button
 from modules.utils import gradio
 
-try:
-    from TTS.api import TTS
-    from TTS.utils.synthesizer import Synthesizer
-except ModuleNotFoundError:
-    logger.error(
-        "Could not find the TTS module. Make sure to install the requirements for the coqui_tts extension."
-        "\n"
-        "\nLinux / Mac:\npip install -r extensions/coqui_tts/requirements.txt\n"
-        "\nWindows:\npip install -r extensions\\coqui_tts\\requirements.txt\n"
-        "\n"
-        "If you used the one-click installer, paste the command above in the terminal window launched after running the \"cmd_\" script. On Windows, that's \"cmd_windows.bat\"."
-    )
-
-    raise
-
 os.environ["COQUI_TOS_AGREED"] = "1"
 
 params = {

modules/extensions.py

@@ -32,8 +32,13 @@ def load_extensions():
         if name in available_extensions:
             if name != 'api':
                 logger.info(f'Loading the extension "{name}"...')
+            try:
                 try:
                     exec(f"import extensions.{name}.script")
+                except ModuleNotFoundError:
+                    logger.error(f"Could not import the requirements for '{name}'. Make sure to install the requirements for the extension.\n\nLinux / Mac:\n\npip install -r extensions/{name}/requirements.txt --upgrade\n\nWindows:\n\npip install -r extensions\\{name}\\requirements.txt --upgrade\n\nIf you used the one-click installer, paste the command above in the terminal window opened after launching the cmd script for your OS.")
+                    raise
+
                 extension = getattr(extensions, name).script
                 apply_settings(extension, name)
                 if extension not in setup_called and hasattr(extension, "setup"):