import os
from dotenv import load_dotenv

load_dotenv()

class Config:   
    def __init__(self):
        self.openai_api_key = os.getenv("OPENAI_API_KEY")
        self.openai_api_base = os.getenv("OPENAI_API_BASE")
        
        
        ### minio
        self.minio_bucket_name = os.getenv("MINIO_BUCKET_NAME")
        self.minio_access_key = os.getenv("MINIO_ACCESS_KEY")
        self.minio_secret_key = os.getenv("MINIO_SECRET_KEY")
        self.minio_endpoint_url = os.getenv("MINIO_ENDPOINT_URL")
        self.minio_cdn_url = os.getenv("MINIO_CDN_URL")
        
        
        #### google cloud tts
        self.google_project_id = os.getenv("GOOGLE_PROJECT_ID")
        self.google_private_key_id = os.getenv("GOOGLE_PRIVATE_KEY_ID")
        self.google_private_key = os.getenv("GOOGLE_PRIVATE_KEY")
        self.google_client_email = os.getenv("GOOGLE_CLIENT_EMAIL")
        self.google_client_id = os.getenv("GOOGLE_CLIENT_ID")
        self.google_auth_uri = os.getenv("GOOGLE_AUTH_URI")
        self.google_token_uri = os.getenv("GOOGLE_TOKEN_URI")
        self.google_auth_provider_x509_cert_url = os.getenv("GOOGLE_AUTH_PROVIDER_X509_CERT_URL")
        self.google_client_x509_cert_url = os.getenv("GOOGLE_CLIENT_X509_CERT_URL")
        self.google_universe_domain = os.getenv("GOOGLE_UNIVERSE_DOMAIN")
        
        # 添加一个方法来获取完整的Google服务账号配置
        self.google_service_account_info = {
            "type": "service_account",
            "project_id": self.google_project_id,
            "private_key_id": self.google_private_key_id,
            "private_key": self.google_private_key,
            "client_email": self.google_client_email,
            "client_id": self.google_client_id,
            "auth_uri": self.google_auth_uri,
            "token_uri": self.google_token_uri,
            "auth_provider_x509_cert_url": self.google_auth_provider_x509_cert_url,
            "client_x509_cert_url": self.google_client_x509_cert_url,
            "universe_domain": self.google_universe_domain
        }

global_config = Config()

import asyncio
import aiofiles
import json
from google.cloud import texttospeech
from google.oauth2 import service_account
from config import global_config

class GoogleTTS:
    def __init__(self):
        # 使用配置文件中的服务账号信息初始化客户端
        credentials = service_account.Credentials.from_service_account_info(
            global_config.google_service_account_info
        )
        self.client = texttospeech.TextToSpeechClient(credentials=credentials)

        
    async def synthesize(self,
                        text: str,
                        output_path: str,
                        language_code: str = "zh-CN",
                        ssml_gender: str = "NEUTRAL",
                        audio_encoding: str = "MP3") -> None:
        """
        异步合成指定文本到 MP3 文件。

        :param text:           要合成的文本
        :param output_path:    输出文件路径
        :param language_code:  语言代码，如 zh-CN
        :param ssml_gender:    声音性别：MALE/FEMALE/NEUTRAL
        :param audio_encoding: 音频编码：MP3/LINEAR16/OGG_OPUS
        """
        # 枚举映射
        gender = getattr(texttospeech.SsmlVoiceGender, ssml_gender.upper())
        encoding = getattr(texttospeech.AudioEncoding, audio_encoding.upper())

        def _run_tts() -> bytes:
            synthesis_input = texttospeech.SynthesisInput(text=text)
            voice = texttospeech.VoiceSelectionParams(
                language_code=language_code,
                ssml_gender=gender
            )
            audio_config = texttospeech.AudioConfig(audio_encoding=encoding)
            resp = self.client.synthesize_speech(
                input=synthesis_input,
                voice=voice,
                audio_config=audio_config
            )
            return resp.audio_content

        audio_bytes = await asyncio.to_thread(_run_tts)
        # 写入文件
        async with aiofiles.open(output_path, "wb") as f:
            await f.write(audio_bytes)
        print(f"✅ 已合成到 {output_path}")

google_tts = GoogleTTS()

if __name__ == "__main__":
    async def test_synthesize():
        out_path = "output.mp3"
        await google_tts.synthesize(
            text="TTS（语音合成）技术负责将文字转化为富含情感、个性化的“声音”；形象驱动技术让语音与表情、口型、肢体动作精准同步，塑造逼真视觉形象；音视频工程解决实时渲染、低延迟传输与高质量画面输出的技术挑战",
            output_path=out_path
        )
        from cos import minio_client
        url = await minio_client.minio_upload_file_streaming(out_path, "test.mp3", "tmp", "audio/mpeg")
        print(url)

    asyncio.run(test_synthesize())