move loguru setup from __init__ to __main__

kokoro/__init__.py

@@ -1,23 +1,4 @@
 __version__ = '0.9.4'
 
-from loguru import logger
-import sys
-
-# Remove default handler
-logger.remove()
-
-# Add custom handler with clean format including module and line number
-logger.add(
-    sys.stderr,
-    format="<green>{time:HH:mm:ss}</green> | <cyan>{module:>16}:{line}</cyan> | <level>{level: >8}</level> | <level>{message}</level>",
-    colorize=True,
-    level="INFO" # "DEBUG" to enable logger.debug("message") and up prints 
-                 # "ERROR" to enable only logger.error("message") prints
-                 # etc
-)
-
-# Disable before release or as needed
-logger.disable("kokoro")
-
 from .model import KModel
 from .pipeline import KPipeline

kokoro/__main__.py

@@ -13,13 +13,30 @@
 """
 
 import argparse
+import sys
 import wave
 from pathlib import Path
 from typing import Generator, TYPE_CHECKING
 
 import numpy as np
 from loguru import logger
 
+# Remove default handler
+logger.remove()
+
+# Add custom handler with clean format including module and line number
+logger.add(
+    sys.stderr,
+    format="<green>{time:HH:mm:ss}</green> | <cyan>{module:>16}:{line}</cyan> | <level>{level: >8}</level> | <level>{message}</level>",
+    colorize=True,
+    level="INFO" # "DEBUG" to enable logger.debug("message") and up prints
+                 # "ERROR" to enable only logger.error("message") prints
+                 # etc
+)
+
+# Disable before release or as needed
+logger.disable("kokoro")
+
 languages = [
     "a",  # American English
     "b",  # British English