[
  {
    "path": ".github/workflows/CI.yaml",
    "content": "name: CI\n\non:\n  push:\n    branches: [ main ]\n  pull_request:\n    branches: [ main ]\n  workflow_dispatch:\n  \njobs:\n  testing:\n    runs-on: ubuntu-latest\n    steps:\n      - uses: actions/checkout@v3\n      - name: install python 3.9 \n        uses: actions/setup-python@v4\n        with:\n          python-version: '3.9'\n          cache: 'pip' # caching pip dependencies\n      - name: Check formatting (black)\n        run: |\n            pip install black\n            black . --check\n"
  },
  {
    "path": ".gitignore",
    "content": "# Byte-compiled / optimized / DLL files\n__pycache__/\n*.py[cod]\n*$py.class\n\n# C extensions\n*.so\n\n# Distribution / packaging\n.Python\nbuild/\ndevelop-eggs/\ndist/\ndownloads/\neggs/\n.eggs/\nlib/\nlib64/\nparts/\nsdist/\nvar/\nwheels/\npip-wheel-metadata/\nshare/python-wheels/\n*.egg-info/\n.installed.cfg\n*.egg\nMANIFEST\n\n# PyInstaller\n#  Usually these files are written by a python script from a template\n#  before PyInstaller builds the exe, so as to inject date/other infos into it.\n*.manifest\n*.spec\n\n# Installer logs\npip-log.txt\npip-delete-this-directory.txt\n\n# Unit test / coverage reports\nhtmlcov/\n.tox/\n.nox/\n.coverage\n.coverage.*\n.cache\nnosetests.xml\ncoverage.xml\n*.cover\n*.py,cover\n.hypothesis/\n.pytest_cache/\n\n# Translations\n*.mo\n*.pot\n\n# Django stuff:\n*.log\nlocal_settings.py\ndb.sqlite3\ndb.sqlite3-journal\n\n# Flask stuff:\ninstance/\n.webassets-cache\n\n# Scrapy stuff:\n.scrapy\n\n# Sphinx documentation\ndocs/_build/\n\n# PyBuilder\ntarget/\n\n# Jupyter Notebook\n.ipynb_checkpoints\n\n# IPython\nprofile_default/\nipython_config.py\n\n# pyenv\n.python-version\n\n# pipenv\n#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.\n#   However, in case of collaboration, if having platform-specific dependencies or dependencies\n#   having no cross-platform support, pipenv may install dependencies that don't work, or not\n#   install all needed dependencies.\n#Pipfile.lock\n\n# PEP 582; used by e.g. github.com/David-OConnor/pyflow\n__pypackages__/\n\n# Celery stuff\ncelerybeat-schedule\ncelerybeat.pid\n\n# SageMath parsed files\n*.sage.py\n\n# Environments\n.env\n.venv\nenv/\nvenv/\nENV/\nenv.bak/\nvenv.bak/\n\n# Spyder project settings\n.spyderproject\n.spyproject\n\n# Rope project settings\n.ropeproject\n\n# mkdocs documentation\n/site\n\n# mypy\n.mypy_cache/\n.dmypy.json\ndmypy.json\n\n# Pyre type checker\n.pyre/\n.idea/\nxiaogptconfig.json\nxiao_config.json\nxiao_config.json.example\nxiao_config.json.example\n"
  },
  {
    "path": "Dockerfile",
    "content": "FROM python:3.10\nWORKDIR /app\nRUN pip install aiohttp\nCOPY . .\nRUN pip install --no-cache-dir -r requirements.txt\nENV OPENAI_API_KEY=$OPENAI_API_KEY\nENV XDG_CONFIG_HOME=/config\nVOLUME /config\nENTRYPOINT [\"python3\",\"xiaogpt.py\"]"
  },
  {
    "path": "LICENSE",
    "content": "MIT License\n\nCopyright (c) 2023 yihong\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n"
  },
  {
    "path": "README.md",
    "content": "# xiaogpt\nPlay ChatGPT with Xiaomi AI Speaker\n\n![image](https://user-images.githubusercontent.com/15976103/220028375-c193a859-48a1-4270-95b6-ef540e54a621.png)\n\n## Update by zsc\n这个是可行命令\n`python3 xiaogpt.py --hardware LX04 --use_chatgpt_api --use_command`\n\n\n## 一点原理\n\n[不用 root 使用小爱同学和 ChatGPT 交互折腾记](https://github.com/yihong0618/gitblog/issues/258)\n\n\n## 准备\n\n1. ChatGPT id\n2. 小爱音响\n3. 能正常联网的环境或 proxy\n4. python3.8+\n\n## 使用\n\n1. pip install aiohttp # 解决 miserver 依赖\n2. pip install -r requirements.txt\n3. 参考 [MiService](https://github.com/Yonsm/MiService) 项目 README 并在本地 terminal 跑 `micli list` 拿到你音响的 DID 成功 **别忘了设置 export MI_DID=xxx** 这个 MI_DID 用 \n4. 参考 [revChatGPT](https://github.com/acheong08/ChatGPT) 项目 README 配置 chatGPT 的 config\n5. run `python xiaogpt.py --hardware ${your_hardware}` hardware 你看小爱屁股上有型号，输入进来\n6. 跑起来之后就可以问小爱同学问题了，“帮我\"开头的问题，会发送一份给 ChatGPT 然后小爱同学用 tts 回答\n7. 因为现在必须指定 conversation_id 和 parent_id 来持续对话，会自动建一个新的 conversation\n8. 如果上面不可用，可以尝试用手机抓包，https://userprofile.mina.mi.com/device_profile/v2/conversation 找到 cookie 利用 --cookie '${cookie}' cookie 别忘了用单引号包裹\n9. 默认用目前 ubus, 如果你的设备不支持 ubus 可以使用 --use_command 来使用 command 来 tts\n10. 使用 --mute_xiaoai 选项，可以让小爱不回答，但会频繁请求，玩一下可以使用，不建议一直用\n11. 使用 --account ‘${account}’ --password ‘${password}’ 可以不进行步骤 2\n12. 如果有能力可以自行替换唤醒词，也可以去掉唤醒词，源码在 https://github.com/yihong0618/xiaogpt/blob/main/xiaogpt.py#L32\n13. 可以使用 gpt-3 的 api 那样可以更流畅的对话，速度快, 请 google 如何用 openai api, 命令 --use_gpt3\n14. 可以使用 --use_chatgpt_api 的 api 那样可以更流畅的对话，速度特别快，达到了对话的体验, 请 google 如何用 openai api, 命令 --use_chatgpt_api\n\ne.g.\n```shell\npython3 xiaogpt.py --hardware LX06;\n# or\npython3 xiaogpt.py --hardware LX06 --conversation_id=\"xxxxxxxx\";\n# or \npython3 xiaogpt.py --hardware LX06 --cookie ${cookie};\n# 如果你想直接输入账号密码\npython3 xiaogpt.py --hardware LX06 --account ${your_xiaomi_account} --password ${your_password};\n# 如果你想 mute 小米的回答\npython3 xiaogpt.py --hardware LX06  --mute_xiaoai \n# 如果你想使用 gpt3 ai\nexport OPENAI_API_KEY=${your_api_key}\npython3 xiaogpt.py --hardware LX06  --mute_xiaoai --use_gpt3\n# 如果你想用 chatgpt api\nexport OPENAI_API_KEY=${your_api_key}\npython3 xiaogpt.py --hardware LX06 --use_chatgpt_api\n```\n\n## config.json\n如果想通过单一配置文件启动也是可以的, 可以通过 --config 参数指定配置文件, config 文件必须是合法的 JSON 格式\n参数优先级\n- cli args > default > config\n\n```shell\npython3 xiaogpt.py --config xiao_config.json\n```\n或者\n```shell\ncp xiao_config.json.example xiao_config.json\npython3 xiaogpt.py \n```\n\n## 注意\n\n1. 请开启小爱同学的蓝牙\n2. 如果要更改提示词和 PROMPT 在代码最上面自行更改\n3. 目前已知 LX04 和 L05B L05C 可能需要使用 `--use_command`\n\n## QA\n\n1. 用破解么？不用\n2. 连不上 revChatGPT？国情，你得设置 proxy 并且该地区可用的 proxy\n3. 你做这玩意也没用啊？确实。。。但是挺好玩的，有用对你来说没用，对我们来说不一定呀\n4. 想把它变得更好？PR Issue always welcome.\n5. 还有问题？提 Issuse 哈哈\n\n## 视频教程\nhttps://www.youtube.com/watch?v=K4YA8YwzOOA\n\n## Docker\n\n### 常规用法\n\ndocker run -e OPENAI_API_KEY=< your-openapi-key > yihong0618/xiaogpt < 命令行参数 >\n\n如\n\n```shell\ndocker run -e OPENAI_API_KEY=<your-openapi-key> yihong0618/xiaogpt --account=<your-xiaomi-account> --password=<your-xiaomi-password> --hardware=<your-xiaomi-hardware> --use_chatgpt_api\n```\n\n### 使用配置文件\n\n1.xiaogpt的配置文件可通过指定volume /config，以及指定参数--config来处理，如\n\n```shell\ndocker run -e OPENAI_API_KEY=<your-openapi-key> -v <your-config-dir>:/config yihong0618/xiaogpt --account=<your-xiaomi-account> --password=<your-xiaomi-password> --hardware=<your-xiaomi-hardware> --use_chatgpt_api --config=/config/config.json\n```\n\n2.如果使用revChatGPT，则可通过指定volume /config，以及指定环境变量XDG_CONFIG_HOME来处理 ( **revChatGPT配置文件需要放置到<your-config-dir>/revChatGPT/config.json** ) ，如\n\n```shell\ndocker run -e XDG_CONFIG_HOME=/config -v <your-config-dir>:/config yihong0618/xiaogpt --account=<your-xiaomi-account> --password=<your-xiaomi-password> --hardware=<your-xiaomi-hardware> --use_chatgpt_api --config=/config/config.json\n```\n\n# 感谢\n\n- [xiaomi](https://www.mi.com/)\n- @[Yonsm](https://github.com/Yonsm) 的 [MiService](https://github.com/Yonsm/MiService) \n\n## 赞赏\n\n谢谢就够了\n"
  },
  {
    "path": "requirements.txt",
    "content": "rich\ngit+https://github.com/yihong0618/MiService\nrequests\nrevChatGPT\nopenai\n\n"
  },
  {
    "path": "xiaogpt.py",
    "content": "#!/usr/bin/env python3\nimport argparse\nimport asyncio\nimport json\nimport os\nfrom os import environ as env\nimport subprocess\nimport time\nfrom http.cookies import SimpleCookie\nfrom pathlib import Path\n\nimport openai\nfrom aiohttp import ClientSession\nfrom miservice import MiAccount, MiNAService\nfrom requests.utils import cookiejar_from_dict\n#from revChatGPT.V1 import Chatbot, configure\nfrom rich import print\n\nLATEST_ASK_API = \"https://userprofile.mina.mi.com/device_profile/v2/conversation?source=dialogu&hardware={hardware}&timestamp={timestamp}&limit=2\"\nCOOKIE_TEMPLATE = \"deviceId={device_id}; serviceToken={service_token}; userId={user_id}\"\n\nHARDWARE_COMMAND_DICT = {\n    \"LX06\": \"5-1\",\n    \"L05B\": \"5-3\",\n    \"S12A\": \"5-1\",\n    \"LX01\": \"5-1\",\n    \"L06A\": \"5-1\",\n    \"LX04\": \"5-1\",\n    \"L05C\": \"5-3\",\n    \"L17A\": \"7-3\",\n    \"X08E\": \"7-3\",\n    # add more here\n}\nMI_USER = \"\"\nMI_PASS = \"\"\nOPENAI_API_KEY = \"\"\nKEY_WORD = \"帮我\"\nPROMPT = \"请用100字以内回答\"\n\n# simulate the response from xiaoai server by type the input.\nCLI_INTERACTIVE_MODE = False\n\n\n### HELP FUNCTION ###\ndef parse_cookie_string(cookie_string):\n    cookie = SimpleCookie()\n    cookie.load(cookie_string)\n    cookies_dict = {}\n    cookiejar = None\n    for k, m in cookie.items():\n        cookies_dict[k] = m.value\n        cookiejar = cookiejar_from_dict(cookies_dict, cookiejar=None, overwrite=True)\n    return cookiejar\n\n\nclass GPT3Bot:\n    def __init__(self, session):\n        self.api_key = OPENAI_API_KEY\n        self.api_url = \"https://api.openai.com/v1/completions\"\n        self.headers = {\n            \"Content-Type\": \"application/json\",\n            \"Authorization\": f\"Bearer {self.api_key}\",\n        }\n        # TODO support more models here\n        self.data = {\n            \"prompt\": \"\",\n            \"model\": \"text-davinci-003\",\n            \"max_tokens\": 1024,\n            \"temperature\": 1,\n            \"top_p\": 1,\n        }\n        self.session = session\n\n    async def ask(self, query):\n        # TODO Support for continuous dialogue\n        # pass all prompt and answers\n        # PR welcome\n        self.data[\"prompt\"] = query\n        r = await self.session.post(self.api_url, headers=self.headers, json=self.data)\n        return await r.json()\n\n\nclass ChatGPTBot:\n    def __init__(self, session):\n        self.session = session\n        self.history = []\n\n    async def ask(self, query):\n        openai.api_key = OPENAI_API_KEY\n        ms = []\n        for h in self.history:\n            ms.append({\"role\": \"user\", \"content\": h[0]})\n            ms.append({\"role\": \"assistant\", \"content\": h[1]})\n        ms.append({\"role\": \"user\", \"content\": f\"{query}\"})\n        completion = openai.ChatCompletion.create(model=\"gpt-3.5-turbo\", messages=ms)\n        message = (\n            completion[\"choices\"][0]\n            .get(\"message\")\n            .get(\"content\")\n            .encode(\"utf8\")\n            .decode()\n        )\n        self.history.append([f\"{query}\", message])\n        # only keep 5 history\n        self.history = self.history[-5:]\n        return message\n\n\nclass MiGPT:\n    def __init__(\n        self,\n        hardware,\n        cookie=\"\",\n        use_command=False,\n        mute_xiaoai=False,\n        use_gpt3=False,\n        use_chatgpt_api=False,\n        verbose=False,\n    ):\n        self.mi_token_home = Path.home() / \".mi.token\"\n        self.hardware = hardware\n        self.cookie_string = \"\"\n        self.last_timestamp = 0  # timestamp last call mi speaker\n        self.session = None\n        self.chatbot = None  # a little slow to init we move it after xiaomi init\n        self.user_id = \"\"\n        self.device_id = \"\"\n        self.service_token = \"\"\n        self.cookie = cookie\n        self.use_command = use_command\n        self.tts_command = HARDWARE_COMMAND_DICT.get(hardware, \"5-1\")\n        self.conversation_id = None\n        self.parent_id = None\n        self.miboy_account = None\n        self.mina_service = None\n        # try to mute xiaoai config\n        self.mute_xiaoai = mute_xiaoai\n        # mute xiaomi in runtime\n        self.this_mute_xiaoai = mute_xiaoai\n        # if use gpt3 api\n        self.use_gpt3 = use_gpt3\n        self.use_chatgpt_api = use_chatgpt_api\n        self.verbose = verbose\n\n    async def init_all_data(self, session):\n        await self.login_miboy(session)\n        await self._init_data_hardware()\n        with open(self.mi_token_home) as f:\n            user_data = json.loads(f.read())\n        self.user_id = user_data.get(\"userId\")\n        self.service_token = user_data.get(\"micoapi\")[1]\n        self._init_cookie()\n        await self._init_first_data_and_chatbot()\n\n    async def login_miboy(self, session):\n        self.session = session\n        self.account = MiAccount(\n            session,\n            MI_USER,\n            MI_PASS,\n            str(self.mi_token_home),\n        )\n        # Forced login to refresh to refresh token\n        await self.account.login(\"micoapi\")\n        self.mina_service = MiNAService(self.account)\n        print(self.mina_service.__dict__)\n\n    async def _init_data_hardware(self):\n        if self.cookie:\n            # if use cookie do not need init\n            return\n        hardware_data = await self.mina_service.device_list()\n        for h in hardware_data:\n            if h.get(\"hardware\", \"\") == self.hardware:\n                self.device_id = h.get(\"deviceID\")\n                break\n        else:\n            raise Exception(f\"we have no hardware: {self.hardware} please check\")\n\n    def _init_cookie(self):\n        if self.cookie:\n            self.cookie = parse_cookie_string(self.cookie)\n        else:\n            self.cookie_string = COOKIE_TEMPLATE.format(\n                device_id=self.device_id,\n                service_token=self.service_token,\n                user_id=self.user_id,\n            )\n            self.cookie = parse_cookie_string(self.cookie_string)\n\n    async def _init_first_data_and_chatbot(self):\n        data = await self.get_latest_ask_from_xiaoai()\n        self.last_timestamp, self.last_record = self.get_last_timestamp_and_record(data)\n        # TODO refactor this\n        if self.use_gpt3:\n            self.chatbot = GPT3Bot(self.session)\n        elif self.use_chatgpt_api:\n            self.chatbot = ChatGPTBot(self.session)\n        else:\n            self.chatbot = Chatbot(configure())\n\n    async def simulate_xiaoai_question(self):\n        data = {\n            \"code\": 0,\n            \"message\": \"Success\",\n            \"data\": '{\"bitSet\":[0,1,1],\"records\":[{\"bitSet\":[0,1,1,1,1],\"answers\":[{\"bitSet\":[0,1,1,1],\"type\":\"TTS\",\"tts\":{\"bitSet\":[0,1],\"text\":\"Fake Answer\"}}],\"time\":1677851434593,\"query\":\"Fake Question\",\"requestId\":\"fada34f8fa0c3f408ee6761ec7391d85\"}],\"nextEndTime\":1677849207387}',\n        }\n        # Convert the data['data'] value from a string to a dictionary\n        data_dict = json.loads(data[\"data\"])\n        # Get the first item in the records list\n        record = data_dict[\"records\"][0]\n        # Replace the query and time values with user input\n        record[\"query\"] = input(\"Enter the new query: \")\n        record[\"time\"] = int(time.time() * 1000)\n        # Convert the updated data_dict back to a string and update the data['data'] value\n        data[\"data\"] = json.dumps(data_dict)\n        await asyncio.sleep(1)\n\n        return data\n\n    async def get_latest_ask_from_xiaoai(self):\n        if CLI_INTERACTIVE_MODE:\n            r = await self.simulate_xiaoai_question()\n            return r\n\n        r = await self.session.get(\n            LATEST_ASK_API.format(\n                hardware=self.hardware, timestamp=str(int(time.time() * 1000))\n            ),\n            cookies=parse_cookie_string(self.cookie),\n        )\n        return await r.json()\n\n    def get_last_timestamp_and_record(self, data):\n        if d := data.get(\"data\"):\n            records = json.loads(d).get(\"records\")\n            if not records:\n                return 0, None\n            last_record = records[0]\n            timestamp = last_record.get(\"time\")\n            return timestamp, last_record\n\n    async def do_tts(self, value):\n        if CLI_INTERACTIVE_MODE:\n            print(f\"do_tts, CLI_INTERACTIVE_MODE:{value}\")\n            await asyncio.sleep(2)\n            return\n\n        if not self.use_command:\n            try:\n                await self.mina_service.text_to_speech(self.device_id, value)\n            except:\n                # do nothing is ok\n                pass\n        else:\n            subprocess.check_output([\"micli.py\", self.tts_command, value])\n\n    def _normalize(self, message):\n        message = message.replace(\" \", \"--\")\n        message = message.replace(\"\\n\", \"，\")\n        message = message.replace('\"', \"，\")\n        return message\n\n    async def ask_gpt(self, query):\n        if self.use_gpt3:\n            return await self.ask_gpt3(query)\n        elif self.use_chatgpt_api:\n            return await self.ask_chatgpt_api(query)\n        return await self.ask_chatgpt(query)\n\n    async def ask_chatgpt_api(self, query):\n        message = await self.chatbot.ask(query)\n        message = self._normalize(message)\n        return message\n\n    async def ask_gpt3(self, query):\n        data = await self.chatbot.ask(query)\n        choices = data.get(\"choices\")\n        if not choices:\n            print(\"No reply from gpt3\")\n        else:\n            message = choices[0].get(\"text\", \"\")\n            message = self._normalize(message)\n            return message\n\n    async def ask_chatgpt(self, query):\n        # TODO maybe use v2 to async it here\n        if self.conversation_id and self.parent_id:\n            data = list(\n                self.chatbot.ask(\n                    query,\n                    conversation_id=self.conversation_id,\n                    parent_id=self.parent_id,\n                )\n            )[-1]\n        else:\n            data = list(self.chatbot.ask(query))[-1]\n        if message := data.get(\"message\", \"\"):\n            self.conversation_id = data.get(\"conversation_id\")\n            self.parent_id = data.get(\"parent_id\")\n            # xiaoai tts did not support space\n            message = self._normalize(message)\n            return message\n        return \"\"\n\n    async def get_if_xiaoai_is_playing(self):\n        playing_info = await self.mina_service.player_get_status(self.device_id)\n        # WTF xiaomi api\n        is_playing = (\n            json.loads(playing_info.get(\"data\", {}).get(\"info\", \"{}\")).get(\"status\", -1)\n            == 1\n        )\n        return is_playing\n\n    async def stop_if_xiaoai_is_playing(self):\n        is_playing = await self.get_if_xiaoai_is_playing()\n        if is_playing:\n            # stop it\n            await self.mina_service.player_pause(self.device_id)\n\n    async def run_forever(self):\n        print(f\"Running xiaogpt now, 用`{KEY_WORD}`开头来提问\")\n        async with ClientSession() as session:\n            await self.init_all_data(session)\n            while 1:\n                if self.verbose:\n                    print(\n                        f\"Now listening xiaoai new message timestamp: {self.last_timestamp}\"\n                    )\n                try:\n                    r = await self.get_latest_ask_from_xiaoai()\n                except Exception:\n                    # we try to init all again\n                    await self.init_all_data(session)\n                    r = await self.get_latest_ask_from_xiaoai()\n                # spider rule\n                if not self.mute_xiaoai:\n                    await asyncio.sleep(3)\n                else:\n                    await asyncio.sleep(0.3)\n\n                new_timestamp, last_record = self.get_last_timestamp_and_record(r)\n                if new_timestamp > self.last_timestamp:\n                    self.last_timestamp = new_timestamp\n                    query = last_record.get(\"query\", \"\")\n                    if query.find(KEY_WORD) != -1:\n                        # only mute when your clause start's with the keyword\n                        if self.this_mute_xiaoai:\n                            await self.stop_if_xiaoai_is_playing()\n                        self.this_mute_xiaoai = False\n                        # drop 帮我回答\n                        query = query.replace(KEY_WORD, \"\")\n                        query = f\"{query}，{PROMPT}\"\n                        # waiting for xiaoai speaker done\n                        if not self.mute_xiaoai:\n                            await asyncio.sleep(4)\n                        await self.do_tts(\"正在问GPT请耐心等待\")\n                        try:\n                            print(\n                                \"以下是小爱的回答: \",\n                                last_record.get(\"answers\")[0]\n                                .get(\"tts\", {})\n                                .get(\"text\"),\n                            )\n                        except:\n                            print(\"小爱没回\")\n                        message = await self.ask_gpt(query)\n                        # tts to xiaoai with ChatGPT answer\n                        print(\"以下是GPT的回答: \" + message)\n                        await self.do_tts(message)\n                        if self.mute_xiaoai:\n                            while 1:\n                                is_playing = await self.get_if_xiaoai_is_playing()\n                                time.sleep(2)\n                                if not is_playing:\n                                    break\n                            self.this_mute_xiaoai = True\n                else:\n                    if self.verbose:\n                        print(\"No new xiao ai record\")\n\n\nif __name__ == \"__main__\":\n    parser = argparse.ArgumentParser()\n    parser.add_argument(\n        \"--hardware\",\n        dest=\"hardware\",\n        type=str,\n        default=\"\",\n        help=\"小爱 hardware\",\n    )\n    parser.add_argument(\n        \"--account\",\n        dest=\"account\",\n        type=str,\n        default=\"\",\n        help=\"xiaomi account\",\n    )\n    parser.add_argument(\n        \"--password\",\n        dest=\"password\",\n        type=str,\n        default=\"\",\n        help=\"xiaomi password\",\n    )\n    parser.add_argument(\n        \"--openai_key\",\n        dest=\"openai_key\",\n        type=str,\n        default=\"\",\n        help=\"openai api key\",\n    )\n    parser.add_argument(\n        \"--cookie\",\n        dest=\"cookie\",\n        type=str,\n        default=\"\",\n        help=\"xiaomi cookie\",\n    )\n    parser.add_argument(\n        \"--use_command\",\n        dest=\"use_command\",\n        action=\"store_true\",\n        help=\"use command to tts\",\n    )\n    parser.add_argument(\n        \"--mute_xiaoai\",\n        dest=\"mute_xiaoai\",\n        action=\"store_true\",\n        help=\"try to mute xiaoai answer\",\n    )\n    parser.add_argument(\n        \"--verbose\",\n        dest=\"verbose\",\n        action=\"store_true\",\n        help=\"show info\",\n    )\n    parser.add_argument(\n        \"--use_gpt3\",\n        dest=\"use_gpt3\",\n        action=\"store_true\",\n        help=\"if use openai gpt3 api\",\n    )\n    parser.add_argument(\n        \"--use_chatgpt_api\",\n        dest=\"use_chatgpt_api\",\n        action=\"store_true\",\n        help=\"if use openai chatgpt api\",\n    )\n    parser.add_argument(\n        \"--config\",\n        dest=\"config\",\n        type=str,\n        default=\"\",\n        help=\"config file path\",\n    )\n\n    options = parser.parse_args()\n\n    if options.config:\n        config = {}\n        if os.path.exists(options.config):\n            with open(options.config, \"r\") as f:\n                config = json.load(f)\n        else:\n            raise Exception(f\"{options.config} doesn't exist\")\n\n        # update options with config\n        for key, value in config.items():\n            if not getattr(options, key, None):\n                setattr(options, key, value)\n\n    # if set\n    MI_USER = options.account or env.get(\"MI_USER\") or MI_USER\n    MI_PASS = options.password or env.get(\"MI_PASS\") or MI_PASS\n    OPENAI_API_KEY = options.openai_key or env.get(\"OPENAI_API_KEY\")\n    if options.use_gpt3:\n        if not OPENAI_API_KEY:\n            raise Exception(\"Use gpt-3 api need openai API key, please google how to\")\n    if options.use_chatgpt_api:\n        if not OPENAI_API_KEY:\n            raise Exception(\"Use chatgpt api need openai API key, please google how to\")\n\n    miboy = MiGPT(\n        options.hardware,\n        options.cookie,\n        options.use_command,\n        options.mute_xiaoai,\n        options.use_gpt3,\n        options.use_chatgpt_api,\n        options.verbose,\n    )\n    asyncio.run(miboy.run_forever())\n"
  }
]