🦄 refactor(log): colorful log

core/providers/tts/base.py

@@ -1,12 +1,13 @@
 import asyncio
-import logging
+from config.logger import setup_logging
 import os
 import numpy as np
 import opuslib
 from pydub import AudioSegment
 from abc import ABC, abstractmethod
 
-logger = logging.getLogger(__name__)
+TAG = __name__
+logger = setup_logging()
 
 
 class TTSProviderBase(ABC):
@@ -26,14 +27,14 @@ class TTSProviderBase(ABC):
                 asyncio.run(self.text_to_speak(text, tmp_file))
                 if not os.path.exists(tmp_file):
                     max_repeat_time = max_repeat_time - 1
-                    logger.error(f"语音生成失败: {text}:{tmp_file}，再试{max_repeat_time}次")
+                    logger.bind(tag=TAG).error(f"语音生成失败: {text}:{tmp_file}，再试{max_repeat_time}次")
 
             if max_repeat_time > 0:
-                logger.info(f"语音生成成功: {text}:{tmp_file}，重试{5 - max_repeat_time}次")
+                logger.bind(tag=TAG).info(f"语音生成成功: {text}:{tmp_file}，重试{5 - max_repeat_time}次")
 
             return tmp_file
         except Exception as e:
-            logger.info(f"Failed to generate TTS file: {e}")
+            logger.bind(tag=TAG).info(f"Failed to generate TTS file: {e}")
             return None
 
     @abstractmethod

core/providers/tts/gpt_sovits_v2.py

@@ -3,10 +3,12 @@ import uuid
 import json
 import base64
 import requests
-import logging
+from config.logger import setup_logging
 from datetime import datetime
 from core.providers.tts.base import TTSProviderBase
-logger = logging.getLogger(__name__)
+
+TAG = __name__
+logger = setup_logging()
 
 class TTSProvider(TTSProviderBase):
     def __init__(self, config, delete_audio_file):
@@ -62,4 +64,4 @@ class TTSProvider(TTSProviderBase):
             with open(output_file, "wb") as file:
                 file.write(resp.content)
         else:
-            logger.error(f"GPT_SoVITS_V2 TTS请求失败: {resp.status_code} - {resp.text}")
+            logger.bind(tag=TAG).error(f"GPT_SoVITS_V2 TTS请求失败: {resp.status_code} - {resp.text}")