From c654bff974b1bd0f93487a7a58d6590a73b858b3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= <aleix@daily.co>
Date: Fri, 30 Jan 2026 16:03:58 -0800
Subject: [PATCH] move loguru setup from __init__ to __main__

This allows other applications that also use Kokoro to setup loguru properly.
---
 kokoro/__init__.py | 19 -------------------
 kokoro/__main__.py | 17 +++++++++++++++++
 2 files changed, 17 insertions(+), 19 deletions(-)

diff --git a/kokoro/__init__.py b/kokoro/__init__.py
index 9156e5c5..bf2b0341 100644
--- a/kokoro/__init__.py
+++ b/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
diff --git a/kokoro/__main__.py b/kokoro/__main__.py
index 34ee21a3..42f45e7a 100644
--- a/kokoro/__main__.py
+++ b/kokoro/__main__.py
@@ -13,6 +13,7 @@
 """
 
 import argparse
+import sys
 import wave
 from pathlib import Path
 from typing import Generator, TYPE_CHECKING
@@ -20,6 +21,22 @@
 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