import time
from bilibili_api import live
from bilibili_api import user
import time as t
from baidu_ai_voice import *
from playsound import playsound
from aip import AipSpeech
import os

//房间号
dm = live.LiveDanmaku(room_display_id="5239322", debug=False, use_wss=True, should_reconnect=True)

//百度智能云控制台的appid api_key...
client = AipSpeech(BAIDUAI_AppID, BAIDUAI_API_Key, BAIDUAI_Secret_Key)


def connect():
    dm.connect()


@dm.on("DANMU_MSG")  # 指定事件名
def on_danmu(msg):
    if ("info" in str(msg)):
        info = msg["data"]["info"]
        message = str(info[2][1]) + ": " + str(info[1])
        print("【弹幕】 from " + message)
        # engine.Speak(f"{str(info[2][1])} 说： {str(info[1])}")
        # t.sleep(1)
        voice = f"{str(info[2][1])} 说： {str(info[1])}".replace(" ", "").replace("：", "")
        if len(message) != 0:
            result = client.synthesis(voice, 'zh', 0, {
                'vol': 5,
            })
            if not isinstance(result, dict):
                with open('./audio.mp3', 'wb') as f:
                    f.write(result)

            playsound("./audio.mp3")
            os.remove("audio.mp3")


if __name__ == '__main__':
    connect()