diff --git a/src/plugins/built_in/maizone_refactored/services/image_service.py b/src/plugins/built_in/maizone_refactored/services/image_service.py index 64b431d56..69ecb1687 100644 --- a/src/plugins/built_in/maizone_refactored/services/image_service.py +++ b/src/plugins/built_in/maizone_refactored/services/image_service.py @@ -4,13 +4,17 @@ """ import base64 +import random from collections.abc import Callable from pathlib import Path +from io import BytesIO +from PIL import Image import aiofiles import aiohttp from src.common.logger import get_logger +from src.plugin_system.apis import llm_api, config_api logger = get_logger("MaiZone.ImageService") @@ -38,7 +42,7 @@ class ImageService: try: enable_ai_image = bool(self.get_config("send.enable_ai_image", False)) api_key = str(self.get_config("models.siliconflow_apikey", "")) - image_dir = str(self.get_config("send.image_directory", "./data/plugins/maizone_refactored/images")) + image_dir = str(self.get_config("send.image_directory", "./data/plugins/maizon_refactored/images")) image_num_raw = self.get_config("send.ai_image_number", 1) image_num = int(image_num_raw if image_num_raw is not None else 1) @@ -52,49 +56,165 @@ class ImageService: # 确保图片目录存在 Path(image_dir).mkdir(parents=True, exist_ok=True) + # 生成图片提示词 + image_prompt = await self._generate_image_prompt(story) + if not image_prompt: + logger.error("生成图片提示词失败") + return False + logger.info(f"正在为说说生成 {image_num} 张AI配图...") - return await self._call_siliconflow_api(api_key, story, image_dir, image_num) + return await self._call_siliconflow_api(api_key, image_prompt, image_dir, image_num) except Exception as e: logger.error(f"处理AI配图时发生异常: {e}") return False - async def _call_siliconflow_api(self, api_key: str, story: str, image_dir: str, batch_size: int) -> bool: + async def _generate_image_prompt(self, story_content: str) -> str: + """ + 使用LLM生成图片提示词，基于说说内容。 + + :param story_content: 说说内容 + :return: 生成的图片提示词，失败时返回空字符串 + """ + try: + # 获取配置 + identity = config_api.get_global_config("identity", "年龄为19岁,是女孩子,身高为160cm,黑色短发") + enable_ref = bool(self.get_config("models.image_ref", True)) + + # 构建提示词 + prompt = f""" + 请根据以下QQ空间说说内容配图，并构建生成配图的风格和prompt。 + 说说主人信息：'{identity}'。 + 说说内容:'{story_content}'。 + 请注意：仅回复用于生成图片的prompt，不要输出多余内容(包括前后缀，冒号和引号，括号()，表情包，at或 @等 )。 + """ + if enable_ref: + prompt += "说说主人的人设参考图片将随同提示词一起发送给生图AI，可使用'in the style of'或'根据图中人物'等描述引导生成风格" + + # 获取模型配置 + models = llm_api.get_available_models() + prompt_model = self.get_config("models.text_model", "replyer") + model_config = models.get(prompt_model) + + if not model_config: + logger.error(f"找不到模型配置: {prompt_model}") + return "" + + # 调用LLM生成提示词 + logger.info("正在生成图片提示词...") + success, image_prompt, reasoning, model_name = await llm_api.generate_with_model( + prompt=prompt, + model_config=model_config, + request_type="story.generate", + temperature=0.3, + max_tokens=1000 + ) + + if success: + logger.info(f'成功生成图片提示词: {image_prompt}') + return image_prompt + else: + logger.error('生成图片提示词失败') + return "" + + except Exception as e: + logger.error(f"生成图片提示词时发生异常: {e}") + return "" + + async def _call_siliconflow_api(self, api_key: str, image_prompt: str, image_dir: str, batch_size: int) -> bool: """ 调用硅基流动（SiliconFlow）的API来生成图片。 :param api_key: SiliconFlow API密钥。 - :param story: 用于生成图片的文本内容（说说）。 + :param image_prompt: 用于生成图片的提示词。 :param image_dir: 图片保存目录。 :param batch_size: 生成图片的数量。 :return: API调用是否成功。 """ url = "https://api.siliconflow.cn/v1/images/generations" headers = { - "accept": "application/json", - "authorization": f"Bearer {api_key}", - "content-type": "application/json", + "Authorization": f"Bearer {api_key}", + "Content-Type": "application/json" } - payload = {"prompt": story, "n": batch_size, "response_format": "b64_json", "style": "cinematic-default"} + data = { + "model": "Kwai-Kolors/Kolors", + "prompt": image_prompt, + "negative_prompt": "lowres, bad anatomy, bad hands, text, error, cropped, worst quality, low quality, " + "normal quality, jpeg artifacts, signature, watermark, username, blurry", + "seed": random.randint(1, 9999999999), + "batch_size": batch_size, + } + + # 检查是否启用参考图片 + enable_ref = bool(self.get_config("models.image_ref", True)) + if enable_ref: + # 修复：使用Path对象正确获取父目录 + parent_dir = Path(image_dir).parent + ref_images = list(parent_dir.glob("done_ref.*")) + if ref_images: + try: + image = Image.open(ref_images[0]) + encoded_image = self._encode_image_to_base64(image) + data["image"] = encoded_image + logger.info("已添加参考图片到生成参数") + except Exception as e: + logger.warning(f"加载参考图片失败: {e}") try: async with aiohttp.ClientSession() as session: - async with session.post(url, json=payload, headers=headers) as response: - if response.status == 200: - data = await response.json() - for i, img_data in enumerate(data.get("data", [])): - b64_json = img_data.get("b64_json") - if b64_json: - image_bytes = base64.b64decode(b64_json) - file_path = Path(image_dir) / f"image_{i + 1}.png" - async with aiofiles.open(file_path, "wb") as f: - await f.write(image_bytes) - logger.info(f"成功保存AI图片到: {file_path}") - return True - else: + # 发送生成请求 + async with session.post(url, json=data, headers=headers) as response: + if response.status != 200: error_text = await response.text() - logger.error(f"AI生图API请求失败，状态码: {response.status}, 错误信息: {error_text}") + logger.error(f'生成图片出错，错误码[{response.status}]') + logger.error(f'错误响应: {error_text}') return False + + json_data = await response.json() + image_urls = [img["url"] for img in json_data["images"]] + + # 下载并保存图片 + for i, img_url in enumerate(image_urls): + try: + # 下载图片 + async with session.get(img_url) as img_response: + img_response.raise_for_status() + img_data = await img_response.read() + + # 处理图片 + image = Image.open(BytesIO(img_data)) + + # 保存图片 + filename = f"image_{i}.png" + save_path = Path(image_dir) / filename + image.save(save_path) + logger.info(f"图片已保存至: {save_path}") + + except Exception as e: + logger.error(f"下载图片失败: {str(e)}") + return False + + return True + except Exception as e: logger.error(f"调用AI生图API时发生异常: {e}") return False + + def _encode_image_to_base64(self, img: Image.Image) -> str: + """ + 将PIL.Image对象编码为base64 data URL + + :param img: PIL图片对象 + :return: base64 data URL字符串 + """ + try: + img_format = (img.format or "PNG").upper() + buffer = BytesIO() + img.save(buffer, format=img_format) + byte_data = buffer.getvalue() + mime_type = f"image/{img_format.lower()}" + encoded_string = base64.b64encode(byte_data).decode("utf-8") + return f"data:{mime_type};base64,{encoded_string}" + except Exception as e: + logger.error(f"编码图片为base64失败: {e}") + return "" \ No newline at end of file diff --git a/src/plugins/built_in/tts_voice_plugin/LICENSE b/src/plugins/built_in/tts_voice_plugin/LICENSE deleted file mode 100644 index 0ad25db4b..000000000 --- a/src/plugins/built_in/tts_voice_plugin/LICENSE +++ /dev/null @@ -1,661 +0,0 @@ - GNU AFFERO GENERAL PUBLIC LICENSE - Version 3, 19 November 2007 - - Copyright (C) 2007 Free Software Foundation, Inc. - Everyone is permitted to copy and distribute verbatim copies - of this license document, but changing it is not allowed. - - Preamble - - The GNU Affero General Public License is a free, copyleft license for -software and other kinds of works, specifically designed to ensure -cooperation with the community in the case of network server software. - - The licenses for most software and other practical works are designed -to take away your freedom to share and change the works. By contrast, -our General Public Licenses are intended to guarantee your freedom to -share and change all versions of a program--to make sure it remains free -software for all its users. - - When we speak of free software, we are referring to freedom, not -price. Our General Public Licenses are designed to make sure that you -have the freedom to distribute copies of free software (and charge for -them if you wish), that you receive source code or can get it if you -want it, that you can change the software or use pieces of it in new -free programs, and that you know you can do these things. - - Developers that use our General Public Licenses protect your rights -with two steps: (1) assert copyright on the software, and (2) offer -you this License which gives you legal permission to copy, distribute -and/or modify the software. - - A secondary benefit of defending all users' freedom is that -improvements made in alternate versions of the program, if they -receive widespread use, become available for other developers to -incorporate. Many developers of free software are heartened and -encouraged by the resulting cooperation. However, in the case of -software used on network servers, this result may fail to come about. -The GNU General Public License permits making a modified version and -letting the public access it on a server without ever releasing its -source code to the public. - - The GNU Affero General Public License is designed specifically to -ensure that, in such cases, the modified source code becomes available -to the community. It requires the operator of a network server to -provide the source code of the modified version running there to the -users of that server. Therefore, public use of a modified version, on -a publicly accessible server, gives the public access to the source -code of the modified version. - - An older license, called the Affero General Public License and -published by Affero, was designed to accomplish similar goals. This is -a different license, not a version of the Affero GPL, but Affero has -released a new version of the Affero GPL which permits relicensing under -this license. - - The precise terms and conditions for copying, distribution and -modification follow. - - TERMS AND CONDITIONS - - 0. Definitions. - - "This License" refers to version 3 of the GNU Affero General Public License. - - "Copyright" also means copyright-like laws that apply to other kinds of -works, such as semiconductor masks. - - "The Program" refers to any copyrightable work licensed under this -License. Each licensee is addressed as "you". "Licensees" and -"recipients" may be individuals or organizations. - - To "modify" a work means to copy from or adapt all or part of the work -in a fashion requiring copyright permission, other than the making of an -exact copy. The resulting work is called a "modified version" of the -earlier work or a work "based on" the earlier work. - - A "covered work" means either the unmodified Program or a work based -on the Program. - - To "propagate" a work means to do anything with it that, without -permission, would make you directly or secondarily liable for -infringement under applicable copyright law, except executing it on a -computer or modifying a private copy. Propagation includes copying, -distribution (with or without modification), making available to the -public, and in some countries other activities as well. - - To "convey" a work means any kind of propagation that enables other -parties to make or receive copies. Mere interaction with a user through -a computer network, with no transfer of a copy, is not conveying. - - An interactive user interface displays "Appropriate Legal Notices" -to the extent that it includes a convenient and prominently visible -feature that (1) displays an appropriate copyright notice, and (2) -tells the user that there is no warranty for the work (except to the -extent that warranties are provided), that licensees may convey the -work under this License, and how to view a copy of this License. If -the interface presents a list of user commands or options, such as a -menu, a prominent item in the list meets this criterion. - - 1. Source Code. - - The "source code" for a work means the preferred form of the work -for making modifications to it. "Object code" means any non-source -form of a work. - - A "Standard Interface" means an interface that either is an official -standard defined by a recognized standards body, or, in the case of -interfaces specified for a particular programming language, one that -is widely used among developers working in that language. - - The "System Libraries" of an executable work include anything, other -than the work as a whole, that (a) is included in the normal form of -packaging a Major Component, but which is not part of that Major -Component, and (b) serves only to enable use of the work with that -Major Component, or to implement a Standard Interface for which an -implementation is available to the public in source code form. A -"Major Component", in this context, means a major essential component -(kernel, window system, and so on) of the specific operating system -(if any) on which the executable work runs, or a compiler used to -produce the work, or an object code interpreter used to run it. - - The "Corresponding Source" for a work in object code form means all -the source code needed to generate, install, and (for an executable -work) run the object code and to modify the work, including scripts to -control those activities. However, it does not include the work's -System Libraries, or general-purpose tools or generally available free -programs which are used unmodified in performing those activities but -which are not part of the work. For example, Corresponding Source -includes interface definition files associated with source files for -the work, and the source code for shared libraries and dynamically -linked subprograms that the work is specifically designed to require, -such as by intimate data communication or control flow between those -subprograms and other parts of the work. - - The Corresponding Source need not include anything that users -can regenerate automatically from other parts of the Corresponding -Source. - - The Corresponding Source for a work in source code form is that -same work. - - 2. Basic Permissions. - - All rights granted under this License are granted for the term of -copyright on the Program, and are irrevocable provided the stated -conditions are met. This License explicitly affirms your unlimited -permission to run the unmodified Program. The output from running a -covered work is covered by this License only if the output, given its -content, constitutes a covered work. This License acknowledges your -rights of fair use or other equivalent, as provided by copyright law. - - You may make, run and propagate covered works that you do not -convey, without conditions so long as your license otherwise remains -in force. You may convey covered works to others for the sole purpose -of having them make modifications exclusively for you, or provide you -with facilities for running those works, provided that you comply with -the terms of this License in conveying all material for which you do -not control copyright. Those thus making or running the covered works -for you must do so exclusively on your behalf, under your direction -and control, on terms that prohibit them from making any copies of -your copyrighted material outside their relationship with you. - - Conveying under any other circumstances is permitted solely under -the conditions stated below. Sublicensing is not allowed; section 10 -makes it unnecessary. - - 3. Protecting Users' Legal Rights From Anti-Circumvention Law. - - No covered work shall be deemed part of an effective technological -measure under any applicable law fulfilling obligations under article -11 of the WIPO copyright treaty adopted on 20 December 1996, or -similar laws prohibiting or restricting circumvention of such -measures. - - When you convey a covered work, you waive any legal power to forbid -circumvention of technological measures to the extent such circumvention -is effected by exercising rights under this License with respect to -the covered work, and you disclaim any intention to limit operation or -modification of the work as a means of enforcing, against the work's -users, your or third parties' legal rights to forbid circumvention of -technological measures. - - 4. Conveying Verbatim Copies. - - You may convey verbatim copies of the Program's source code as you -receive it, in any medium, provided that you conspicuously and -appropriately publish on each copy an appropriate copyright notice; -keep intact all notices stating that this License and any -non-permissive terms added in accord with section 7 apply to the code; -keep intact all notices of the absence of any warranty; and give all -recipients a copy of this License along with the Program. - - You may charge any price or no price for each copy that you convey, -and you may offer support or warranty protection for a fee. - - 5. Conveying Modified Source Versions. - - You may convey a work based on the Program, or the modifications to -produce it from the Program, in the form of source code under the -terms of section 4, provided that you also meet all of these conditions: - - a) The work must carry prominent notices stating that you modified - it, and giving a relevant date. - - b) The work must carry prominent notices stating that it is - released under this License and any conditions added under section - 7. This requirement modifies the requirement in section 4 to - "keep intact all notices". - - c) You must license the entire work, as a whole, under this - License to anyone who comes into possession of a copy. This - License will therefore apply, along with any applicable section 7 - additional terms, to the whole of the work, and all its parts, - regardless of how they are packaged. This License gives no - permission to license the work in any other way, but it does not - invalidate such permission if you have separately received it. - - d) If the work has interactive user interfaces, each must display - Appropriate Legal Notices; however, if the Program has interactive - interfaces that do not display Appropriate Legal Notices, your - work need not make them do so. - - A compilation of a covered work with other separate and independent -works, which are not by their nature extensions of the covered work, -and which are not combined with it such as to form a larger program, -in or on a volume of a storage or distribution medium, is called an -"aggregate" if the compilation and its resulting copyright are not -used to limit the access or legal rights of the compilation's users -beyond what the individual works permit. Inclusion of a covered work -in an aggregate does not cause this License to apply to the other -parts of the aggregate. - - 6. Conveying Non-Source Forms. - - You may convey a covered work in object code form under the terms -of sections 4 and 5, provided that you also convey the -machine-readable Corresponding Source under the terms of this License, -in one of these ways: - - a) Convey the object code in, or embodied in, a physical product - (including a physical distribution medium), accompanied by the - Corresponding Source fixed on a durable physical medium - customarily used for software interchange. - - b) Convey the object code in, or embodied in, a physical product - (including a physical distribution medium), accompanied by a - written offer, valid for at least three years and valid for as - long as you offer spare parts or customer support for that product - model, to give anyone who possesses the object code either (1) a - copy of the Corresponding Source for all the software in the - product that is covered by this License, on a durable physical - medium customarily used for software interchange, for a price no - more than your reasonable cost of physically performing this - conveying of source, or (2) access to copy the - Corresponding Source from a network server at no charge. - - c) Convey individual copies of the object code with a copy of the - written offer to provide the Corresponding Source. This - alternative is allowed only occasionally and noncommercially, and - only if you received the object code with such an offer, in accord - with subsection 6b. - - d) Convey the object code by offering access from a designated - place (gratis or for a charge), and offer equivalent access to the - Corresponding Source in the same way through the same place at no - further charge. You need not require recipients to copy the - Corresponding Source along with the object code. If the place to - copy the object code is a network server, the Corresponding Source - may be on a different server (operated by you or a third party) - that supports equivalent copying facilities, provided you maintain - clear directions next to the object code saying where to find the - Corresponding Source. Regardless of what server hosts the - Corresponding Source, you remain obligated to ensure that it is - available for as long as needed to satisfy these requirements. - - e) Convey the object code using peer-to-peer transmission, provided - you inform other peers where the object code and Corresponding - Source of the work are being offered to the general public at no - charge under subsection 6d. - - A separable portion of the object code, whose source code is excluded -from the Corresponding Source as a System Library, need not be -included in conveying the object code work. - - A "User Product" is either (1) a "consumer product", which means any -tangible personal property which is normally used for personal, family, -or household purposes, or (2) anything designed or sold for incorporation -into a dwelling. In determining whether a product is a consumer product, -doubtful cases shall be resolved in favor of coverage. For a particular -product received by a particular user, "normally used" refers to a -typical or common use of that class of product, regardless of the status -of the particular user or of the way in which the particular user -actually uses, or expects or is expected to use, the product. A product -is a consumer product regardless of whether the product has substantial -commercial, industrial or non-consumer uses, unless such uses represent -the only significant mode of use of the product. - - "Installation Information" for a User Product means any methods, -procedures, authorization keys, or other information required to install -and execute modified versions of a covered work in that User Product from -a modified version of its Corresponding Source. The information must -suffice to ensure that the continued functioning of the modified object -code is in no case prevented or interfered with solely because -modification has been made. - - If you convey an object code work under this section in, or with, or -specifically for use in, a User Product, and the conveying occurs as -part of a transaction in which the right of possession and use of the -User Product is transferred to the recipient in perpetuity or for a -fixed term (regardless of how the transaction is characterized), the -Corresponding Source conveyed under this section must be accompanied -by the Installation Information. But this requirement does not apply -if neither you nor any third party retains the ability to install -modified object code on the User Product (for example, the work has -been installed in ROM). - - The requirement to provide Installation Information does not include a -requirement to continue to provide support service, warranty, or updates -for a work that has been modified or installed by the recipient, or for -the User Product in which it has been modified or installed. Access to a -network may be denied when the modification itself materially and -adversely affects the operation of the network or violates the rules and -protocols for communication across the network. - - Corresponding Source conveyed, and Installation Information provided, -in accord with this section must be in a format that is publicly -documented (and with an implementation available to the public in -source code form), and must require no special password or key for -unpacking, reading or copying. - - 7. Additional Terms. - - "Additional permissions" are terms that supplement the terms of this -License by making exceptions from one or more of its conditions. -Additional permissions that are applicable to the entire Program shall -be treated as though they were included in this License, to the extent -that they are valid under applicable law. If additional permissions -apply only to part of the Program, that part may be used separately -under those permissions, but the entire Program remains governed by -this License without regard to the additional permissions. - - When you convey a copy of a covered work, you may at your option -remove any additional permissions from that copy, or from any part of -it. (Additional permissions may be written to require their own -removal in certain cases when you modify the work.) You may place -additional permissions on material, added by you to a covered work, -for which you have or can give appropriate copyright permission. - - Notwithstanding any other provision of this License, for material you -add to a covered work, you may (if authorized by the copyright holders of -that material) supplement the terms of this License with terms: - - a) Disclaiming warranty or limiting liability differently from the - terms of sections 15 and 16 of this License; or - - b) Requiring preservation of specified reasonable legal notices or - author attributions in that material or in the Appropriate Legal - Notices displayed by works containing it; or - - c) Prohibiting misrepresentation of the origin of that material, or - requiring that modified versions of such material be marked in - reasonable ways as different from the original version; or - - d) Limiting the use for publicity purposes of names of licensors or - authors of the material; or - - e) Declining to grant rights under trademark law for use of some - trade names, trademarks, or service marks; or - - f) Requiring indemnification of licensors and authors of that - material by anyone who conveys the material (or modified versions of - it) with contractual assumptions of liability to the recipient, for - any liability that these contractual assumptions directly impose on - those licensors and authors. - - All other non-permissive additional terms are considered "further -restrictions" within the meaning of section 10. If the Program as you -received it, or any part of it, contains a notice stating that it is -governed by this License along with a term that is a further -restriction, you may remove that term. If a license document contains -a further restriction but permits relicensing or conveying under this -License, you may add to a covered work material governed by the terms -of that license document, provided that the further restriction does -not survive such relicensing or conveying. - - If you add terms to a covered work in accord with this section, you -must place, in the relevant source files, a statement of the -additional terms that apply to those files, or a notice indicating -where to find the applicable terms. - - Additional terms, permissive or non-permissive, may be stated in the -form of a separately written license, or stated as exceptions; -the above requirements apply either way. - - 8. Termination. - - You may not propagate or modify a covered work except as expressly -provided under this License. Any attempt otherwise to propagate or -modify it is void, and will automatically terminate your rights under -this License (including any patent licenses granted under the third -paragraph of section 11). - - However, if you cease all violation of this License, then your -license from a particular copyright holder is reinstated (a) -provisionally, unless and until the copyright holder explicitly and -finally terminates your license, and (b) permanently, if the copyright -holder fails to notify you of the violation by some reasonable means -prior to 60 days after the cessation. - - Moreover, your license from a particular copyright holder is -reinstated permanently if the copyright holder notifies you of the -violation by some reasonable means, this is the first time you have -received notice of violation of this License (for any work) from that -copyright holder, and you cure the violation prior to 30 days after -your receipt of the notice. - - Termination of your rights under this section does not terminate the -licenses of parties who have received copies or rights from you under -this License. If your rights have been terminated and not permanently -reinstated, you do not qualify to receive new licenses for the same -material under section 10. - - 9. Acceptance Not Required for Having Copies. - - You are not required to accept this License in order to receive or -run a copy of the Program. Ancillary propagation of a covered work -occurring solely as a consequence of using peer-to-peer transmission -to receive a copy likewise does not require acceptance. However, -nothing other than this License grants you permission to propagate or -modify any covered work. These actions infringe copyright if you do -not accept this License. Therefore, by modifying or propagating a -covered work, you indicate your acceptance of this License to do so. - - 10. Automatic Licensing of Downstream Recipients. - - Each time you convey a covered work, the recipient automatically -receives a license from the original licensors, to run, modify and -propagate that work, subject to this License. You are not responsible -for enforcing compliance by third parties with this License. - - An "entity transaction" is a transaction transferring control of an -organization, or substantially all assets of one, or subdividing an -organization, or merging organizations. If propagation of a covered -work results from an entity transaction, each party to that -transaction who receives a copy of the work also receives whatever -licenses to the work the party's predecessor in interest had or could -give under the previous paragraph, plus a right to possession of the -Corresponding Source of the work from the predecessor in interest, if -the predecessor has it or can get it with reasonable efforts. - - You may not impose any further restrictions on the exercise of the -rights granted or affirmed under this License. For example, you may -not impose a license fee, royalty, or other charge for exercise of -rights granted under this License, and you may not initiate litigation -(including a cross-claim or counterclaim in a lawsuit) alleging that -any patent claim is infringed by making, using, selling, offering for -sale, or importing the Program or any portion of it. - - 11. Patents. - - A "contributor" is a copyright holder who authorizes use under this -License of the Program or a work on which the Program is based. The -work thus licensed is called the contributor's "contributor version". - - A contributor's "essential patent claims" are all patent claims -owned or controlled by the contributor, whether already acquired or -hereafter acquired, that would be infringed by some manner, permitted -by this License, of making, using, or selling its contributor version, -but do not include claims that would be infringed only as a -consequence of further modification of the contributor version. For -purposes of this definition, "control" includes the right to grant -patent sublicenses in a manner consistent with the requirements of -this License. - - Each contributor grants you a non-exclusive, worldwide, royalty-free -patent license under the contributor's essential patent claims, to -make, use, sell, offer for sale, import and otherwise run, modify and -propagate the contents of its contributor version. - - In the following three paragraphs, a "patent license" is any express -agreement or commitment, however denominated, not to enforce a patent -(such as an express permission to practice a patent or covenant not to -sue for patent infringement). To "grant" such a patent license to a -party means to make such an agreement or commitment not to enforce a -patent against the party. - - If you convey a covered work, knowingly relying on a patent license, -and the Corresponding Source of the work is not available for anyone -to copy, free of charge and under the terms of this License, through a -publicly available network server or other readily accessible means, -then you must either (1) cause the Corresponding Source to be so -available, or (2) arrange to deprive yourself of the benefit of the -patent license for this particular work, or (3) arrange, in a manner -consistent with the requirements of this License, to extend the patent -license to downstream recipients. "Knowingly relying" means you have -actual knowledge that, but for the patent license, your conveying the -covered work in a country, or your recipient's use of the covered work -in a country, would infringe one or more identifiable patents in that -country that you have reason to believe are valid. - - If, pursuant to or in connection with a single transaction or -arrangement, you convey, or propagate by procuring conveyance of, a -covered work, and grant a patent license to some of the parties -receiving the covered work authorizing them to use, propagate, modify -or convey a specific copy of the covered work, then the patent license -you grant is automatically extended to all recipients of the covered -work and works based on it. - - A patent license is "discriminatory" if it does not include within -the scope of its coverage, prohibits the exercise of, or is -conditioned on the non-exercise of one or more of the rights that are -specifically granted under this License. You may not convey a covered -work if you are a party to an arrangement with a third party that is -in the business of distributing software, under which you make payment -to the third party based on the extent of your activity of conveying -the work, and under which the third party grants, to any of the -parties who would receive the covered work from you, a discriminatory -patent license (a) in connection with copies of the covered work -conveyed by you (or copies made from those copies), or (b) primarily -for and in connection with specific products or compilations that -contain the covered work, unless you entered into that arrangement, -or that patent license was granted, prior to 28 March 2007. - - Nothing in this License shall be construed as excluding or limiting -any implied license or other defenses to infringement that may -otherwise be available to you under applicable patent law. - - 12. No Surrender of Others' Freedom. - - If conditions are imposed on you (whether by court order, agreement or -otherwise) that contradict the conditions of this License, they do not -excuse you from the conditions of this License. If you cannot convey a -covered work so as to satisfy simultaneously your obligations under this -License and any other pertinent obligations, then as a consequence you may -not convey it at all. For example, if you agree to terms that obligate you -to collect a royalty for further conveying from those to whom you convey -the Program, the only way you could satisfy both those terms and this -License would be to refrain entirely from conveying the Program. - - 13. Remote Network Interaction; Use with the GNU General Public License. - - Notwithstanding any other provision of this License, if you modify the -Program, your modified version must prominently offer all users -interacting with it remotely through a computer network (if your version -supports such interaction) an opportunity to receive the Corresponding -Source of your version by providing access to the Corresponding Source -from a network server at no charge, through some standard or customary -means of facilitating copying of software. This Corresponding Source -shall include the Corresponding Source for any work covered by version 3 -of the GNU General Public License that is incorporated pursuant to the -following paragraph. - - Notwithstanding any other provision of this License, you have -permission to link or combine any covered work with a work licensed -under version 3 of the GNU General Public License into a single -combined work, and to convey the resulting work. The terms of this -License will continue to apply to the part which is the covered work, -but the work with which it is combined will remain governed by version -3 of the GNU General Public License. - - 14. Revised Versions of this License. - - The Free Software Foundation may publish revised and/or new versions of -the GNU Affero General Public License from time to time. Such new versions -will be similar in spirit to the present version, but may differ in detail to -address new problems or concerns. - - Each version is given a distinguishing version number. If the -Program specifies that a certain numbered version of the GNU Affero General -Public License "or any later version" applies to it, you have the -option of following the terms and conditions either of that numbered -version or of any later version published by the Free Software -Foundation. If the Program does not specify a version number of the -GNU Affero General Public License, you may choose any version ever published -by the Free Software Foundation. - - If the Program specifies that a proxy can decide which future -versions of the GNU Affero General Public License can be used, that proxy's -public statement of acceptance of a version permanently authorizes you -to choose that version for the Program. - - Later license versions may give you additional or different -permissions. However, no additional obligations are imposed on any -author or copyright holder as a result of your choosing to follow a -later version. - - 15. Disclaimer of Warranty. - - THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY -APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT -HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY -OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, -THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR -PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM -IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF -ALL NECESSARY SERVICING, REPAIR OR CORRECTION. - - 16. Limitation of Liability. - - IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING -WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS -THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY -GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE -USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF -DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD -PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS), -EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF -SUCH DAMAGES. - - 17. Interpretation of Sections 15 and 16. - - If the disclaimer of warranty and limitation of liability provided -above cannot be given local legal effect according to their terms, -reviewing courts shall apply local law that most closely approximates -an absolute waiver of all civil liability in connection with the -Program, unless a warranty or assumption of liability accompanies a -copy of the Program in return for a fee. - - END OF TERMS AND CONDITIONS - - How to Apply These Terms to Your New Programs - - If you develop a new program, and you want it to be of the greatest -possible use to the public, the best way to achieve this is to make it -free software which everyone can redistribute and change under these terms. - - To do so, attach the following notices to the program. It is safest -to attach them to the start of each source file to most effectively -state the exclusion of warranty; and each file should have at least -the "copyright" line and a pointer to where the full notice is found. - - - Copyright (C) - - This program is free software: you can redistribute it and/or modify - it under the terms of the GNU Affero General Public License as published - by the Free Software Foundation, either version 3 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU Affero General Public License for more details. - - You should have received a copy of the GNU Affero General Public License - along with this program. If not, see . - -Also add information on how to contact you by electronic and paper mail. - - If your software can interact with users remotely through a computer -network, you should also make sure that it provides a way for users to -get its source. For example, if your program is a web application, its -interface could display a "Source" link that leads users to an archive -of the code. There are many ways you could offer source, and different -solutions will be better for different programs; see section 13 for the -specific requirements. - - You should also get your employer (if you work as a programmer) or school, -if any, to sign a "copyright disclaimer" for the program, if necessary. -For more information on this, and how to apply and follow the GNU AGPL, see -. diff --git a/src/plugins/built_in/tts_voice_plugin/README.md b/src/plugins/built_in/tts_voice_plugin/README.md deleted file mode 100644 index b172bd39c..000000000 --- a/src/plugins/built_in/tts_voice_plugin/README.md +++ /dev/null @@ -1,82 +0,0 @@ -# GPT-SoVITS 语音合成插件（多风格版） - -## 简介 -本插件基于 GPT-SoVITS，实现文本转语音（TTS）功能，支持多种语音风格和多语言（中文/英文/日文）。适用于需要将文本内容以语音形式发送的场景，如语音回复、朗读长文本、增强表达效果等。 -(此版本适用于麦麦09x版) -## 功能特性 -- 支持多种语音风格（如默认、温柔等，可自定义扩展） -- 支持多语言自动检测与指定（zh/en/ja） -- 可配置参考音频、参考文本、模型权重等参数 -- 兼容多种聊天平台（如 QQ、Telegram 等） -- 关键词激活，灵活触发 - -## 安装与配置 -1. **拷贝插件文件夹**到你的插件目录下（如 `plugins/tts_voice_plugin/`）。 -2. **配置 `config.toml`**，参考下方配置示例： - -```toml -[plugin] -enabled = true -config_version = "1.0.0" - -[tts] -timeout = 30 -max_text_length = 500 -server = "http://127.0.0.1:9880" - -[tts_styles.default] -refer_wav = "" -prompt_text = "" -prompt_language = "zh" -gpt_weights = "" -sovits_weights = "" - -[tts_styles.gentle] -refer_wav = "" -prompt_text = "" -prompt_language = "zh" -gpt_weights = "" -sovits_weights = "" -``` - -- `server`：TTS 服务后端地址（需部署 GPT-SoVITS 服务） -- `tts_styles`：可自定义多种风格，每种风格可配置不同参考音频、文本、模型权重等 - -## 使用方法 -- 在聊天中输入关键词（如“语音”、“说话”、“朗读”、“voice”、“tts”等）即可触发语音合成功能 -- 可通过参数指定风格、语言、参考音频等 -- 支持自动检测文本语言 - -## 主要参数说明 -| 参数名 | 说明 | -|------------------|----------------------------| -| text | 要转换为语音的文本内容 | -| text_language | 文本语言（zh/en/ja） | -| refer_wav_path | 参考音频路径（可选） | -| prompt_text | 参考音频文本（可选） | -| prompt_language | 参考音频语言（可选） | -| voice_style | 语音风格（如 default/gentle）| - -## 扩展风格 -如需添加新风格，在 `config.toml` 的 `[tts_styles]` 下增加分组即可。例如： - -```toml -[tts_styles."活泼"] -refer_wav = "path/to/lively.wav" -prompt_text = "活泼的语气" -prompt_language = "zh" -gpt_weights = "" -sovits_weights = "" -``` - -## 依赖 -- Python 3.7+ -- aiohttp -- GPT-SoVITS 服务端 - -## 作者 -- 插件作者：靓仔 -- 插件版本：1.0.0 - -## License -AGPL-3.0 diff --git a/src/plugins/built_in/tts_voice_plugin/__init__.py b/src/plugins/built_in/tts_voice_plugin/__init__.py deleted file mode 100644 index 463dcc244..000000000 --- a/src/plugins/built_in/tts_voice_plugin/__init__.py +++ /dev/null @@ -1,18 +0,0 @@ -from src.plugin_system.base.plugin_metadata import PluginMetadata - -__plugin_meta__ = PluginMetadata( - name="GPT-SoVITS 语音合成插件", - description="基于 GPT-SoVITS 的文本转语音插件，支持多种语言和多风格语音合成。", - usage=" ", - version="2.0.0", - author="靓仔", - license="AGPL-v3.0", - repository_url="https://github.com/xuqian13/tts_voice_plugin", - keywords=["tts", "语音合成", "文本转语音", "gpt-sovits", "语音", "朗读", "多风格", "语音播报"], - categories=["Utility", "Communication", "Accessibility"], - extra={ - "is_built_in": False, - "plugin_type": "tools", - }, - python_dependencies = ["aiohttp", "soundfile", "pedalboard"] -) diff --git a/src/plugins/built_in/tts_voice_plugin/actions/tts_action.py b/src/plugins/built_in/tts_voice_plugin/actions/tts_action.py deleted file mode 100644 index 409b72767..000000000 --- a/src/plugins/built_in/tts_voice_plugin/actions/tts_action.py +++ /dev/null @@ -1,205 +0,0 @@ -""" -TTS 语音合成 Action -""" - -from pathlib import Path -from typing import ClassVar - -import toml - -from src.chat.utils.self_voice_cache import register_self_voice -from src.common.logger import get_logger -from src.plugin_system.base.base_action import BaseAction, ChatMode - -from ..services.manager import get_service - -logger = get_logger("tts_voice_plugin.action") - - -def _get_available_styles() -> list[str]: - """动态读取配置文件，获取所有可用的TTS风格名称""" - try: - # 这个路径构建逻辑是为了确保无论从哪里启动，都能准确定位到配置文件 - plugin_file = Path(__file__).resolve() - # Bot/src/plugins/built_in/tts_voice_plugin/actions -> Bot - bot_root = plugin_file.parent.parent.parent.parent.parent.parent - config_file = bot_root / "config" / "plugins" / "tts_voice_plugin" / "config.toml" - - if not config_file.is_file(): - logger.warning("在 tts_action 中未找到 tts_voice_plugin 的配置文件，无法动态加载风格列表。") - return ["default"] - - config = toml.loads(config_file.read_text(encoding="utf-8")) - - styles_config = config.get("tts_styles", []) - if not isinstance(styles_config, list): - - return ["default"] - - # 使用显式循环和类型检查来提取 style_name，以确保 Pylance 类型检查通过 - style_names: list[str] = [] - for style in styles_config: - if isinstance(style, dict): - name = style.get("style_name") - # 确保 name 是一个非空字符串 - if isinstance(name, str) and name: - style_names.append(name) - - return style_names if style_names else ["default"] - except Exception as e: - logger.error(f"动态加载TTS风格列表时出错: {e}") - return ["default"] # 出现任何错误都回退 - - -# 在类定义之前执行函数，获取风格列表 -AVAILABLE_STYLES = _get_available_styles() -STYLE_OPTIONS_DESC = ", ".join(f"'{s}'" for s in AVAILABLE_STYLES) - - -class TTSVoiceAction(BaseAction): - """ - 通过关键词或规划器自动触发 TTS 语音合成 - """ - - action_name = "tts_voice_action" - action_description = "将你生成好的文本转换为语音并发送。你必须提供要转换的文本。" - - mode_enable = ChatMode.ALL - parallel_action = False - - action_parameters: ClassVar[dict] = { - "tts_voice_text": { - "type": "string", - "description": "需要转换为语音并发送的完整、自然、适合口语的文本内容。", - "required": True - }, - "voice_style": { - "type": "string", - "description": f"语音的风格。可用选项: [{STYLE_OPTIONS_DESC}]。请根据对话的情感和上下文选择一个最合适的风格。如果未提供，将使用默认风格。", - "required": False - }, - "text_language": { - "type": "string", - "description": ( - "指定用于合成的语言模式，请务必根据文本内容选择最精确、范围最小的选项以获得最佳效果。" - "可用选项说明：\n" - "- 'zh': 中文与英文混合 (最优选)\n" - "- 'ja': 日文与英文混合 (最优选)\n" - "- 'yue': 粤语与英文混合 (最优选)\n" - "- 'ko': 韩文与英文混合 (最优选)\n" - "- 'en': 纯英文\n" - "- 'all_zh': 纯中文\n" - "- 'all_ja': 纯日文\n" - "- 'all_yue': 纯粤语\n" - "- 'all_ko': 纯韩文\n" - "- 'auto': 多语种混合自动识别 (备用选项，当前两种语言时优先使用上面的精确选项)\n" - "- 'auto_yue': 多语种混合自动识别（包含粤语）(备用选项)" - ), - "required": False - } - } - - action_require: ClassVar[list] = [ - "在调用此动作时，你必须在 'text' 参数中提供要合成语音的完整回复内容。这是强制性的。", - "当用户明确请求使用语音进行回复时，例如‘发个语音听听’、‘用语音说’等。", - "当对话内容适合用语音表达，例如讲故事、念诗、撒嬌或进行角色扮演时。", - "在表达特殊情感（如安慰、鼓励、庆祝）的场景下，可以主动使用语音来增强感染力。", - "不要在日常的、简短的问答或闲聊中频繁使用语音，避免打扰用户。", - "提供的 'text' 内容必须是纯粹的对话，不能包含任何括号或方括号括起来的动作、表情、或场景描述（例如，不要出现 '(笑)' 或 '[歪头]'）", - "**重要**：此动作专为语音合成设计，因此 'text' 参数的内容必须是纯净、标准的口语文本。请务必抑制你通常的、富有表现力的文本风格，不要使用任何辅助聊天或增强视觉效果的特殊符号（例如 '♪', '～', '∽', '☆' 等），因为它们无法被正确合成为语音。", - "【**最终规则**】'text' 参数中，所有句子和停顿【必须】使用且只能使用以下四个标准标点符号：'，' (逗号)、'。' (句号)、'？' (问号)、'！' (叹号)。任何其他符号，特别是 '...'、'～' 以及任何表情符号或装饰性符号，都【严禁】出现，否则将导致语音合成严重失败。" - ] - - def __init__(self, *args, **kwargs): - super().__init__(*args, **kwargs) - # 关键配置项现在由 TTSService 管理 - self.tts_service = get_service("tts") - - async def go_activate(self, llm_judge_model=None) -> bool: - """ - 判断此 Action 是否应该被激活。 - 满足以下任一条件即可激活： - 1. 55% 的随机概率 - 2. 匹配到预设的关键词 - 3. LLM 判断当前场景适合发送语音 - """ - # 条件1: 随机激活 - if await self._random_activation(0.25): - logger.info(f"{self.log_prefix} 随机激活成功 (25%)") - return True - - # 条件2: 关键词激活 - keywords = [ - "发语音", "语音", "说句话", "用语音说", "听你", "听声音", "想你", "想听声音", - "讲个话", "说段话", "念一下", "读一下", "用嘴说", "说", "能发语音吗", "亲口" - ] - if await self._keyword_match(keywords): - logger.info(f"{self.log_prefix} 关键词激活成功") - return True - - # 条件3: LLM 判断激活 - # 注意：这里我们复用 action_require 里的描述，让 LLM 的判断更精准 - if await self._llm_judge_activation( - llm_judge_model=llm_judge_model - ): - logger.info(f"{self.log_prefix} LLM 判断激活成功") - return True - - logger.debug(f"{self.log_prefix} 所有激活条件均未满足，不激活") - return False - - async def execute(self) -> tuple[bool, str]: - """ - 执行 Action 的核心逻辑 - """ - try: - if not self.tts_service: - logger.error(f"{self.log_prefix} TTSService 未注册或初始化失败，静默处理。") - return False, "TTSService 未注册或初始化失败" - - initial_text = self.action_data.get("tts_voice_text", "").strip() - voice_style = self.action_data.get("voice_style", "default") - # 新增：从决策模型获取指定的语言模式 - text_language = self.action_data.get("text_language") # 如果模型没给，就是 None - logger.info(f"{self.log_prefix} 接收到规划器初步文本: '{initial_text[:70]}...', 指定风格: {voice_style}, 指定语言: {text_language}") - - # 1. 使用规划器提供的文本 - text = initial_text - if not text: - logger.warning(f"{self.log_prefix} 规划器提供的文本为空，静默处理。") - return False, "规划器提供的文本为空" - - # 2. 调用 TTSService 生成语音 - logger.info(f"{self.log_prefix} 使用最终文本进行语音合成: '{text[:70]}...'") - audio_b64 = await self.tts_service.generate_voice( - text=text, - style_hint=voice_style, - language_hint=text_language # 新增：将决策模型指定的语言传递给服务 - ) - - if audio_b64: - # 在发送语音前，将文本注册到缓存中 - register_self_voice(audio_b64, text) - await self.send_custom(message_type="voice", content=audio_b64) - logger.info(f"{self.log_prefix} GPT-SoVITS语音发送成功") - await self.store_action_info( - action_prompt_display=f"将文本转换为语音并发送 (风格:{voice_style})", - action_done=True - ) - return True, f"成功生成并发送语音，文本长度: {len(text)}字符" - else: - logger.error(f"{self.log_prefix} TTS服务未能返回音频数据，静默处理。") - await self.store_action_info( - action_prompt_display="语音合成失败: TTS服务未能返回音频数据", - action_done=False - ) - return False, "语音合成失败" - - except Exception as e: - logger.error(f"{self.log_prefix} 语音合成过程中发生未知错误: {e!s}") - await self.store_action_info( - action_prompt_display=f"语音合成失败: {e!s}", - action_done=False - ) - return False, f"语音合成出错: {e!s}" - diff --git a/src/plugins/built_in/tts_voice_plugin/commands/tts_command.py b/src/plugins/built_in/tts_voice_plugin/commands/tts_command.py deleted file mode 100644 index 6b9cceeb2..000000000 --- a/src/plugins/built_in/tts_voice_plugin/commands/tts_command.py +++ /dev/null @@ -1,76 +0,0 @@ -""" -TTS 语音合成命令 -""" -from typing import ClassVar - -from src.common.logger import get_logger -from src.plugin_system.base.command_args import CommandArgs -from src.plugin_system.base.plus_command import PlusCommand -from src.plugin_system.utils.permission_decorators import require_permission - -from ..services.manager import get_service - -logger = get_logger("tts_voice_plugin.command") - - -class TTSVoiceCommand(PlusCommand): - """ - 通过命令手动触发 TTS 语音合成 - """ - - command_name: str = "tts" - command_description: str = "使用GPT-SoVITS将文本转换为语音并发送" - command_aliases: ClassVar[list[str]] = ["语音合成", "说"] - command_usage = "/tts <要说的文本> [风格]" - - def __init__(self, *args, **kwargs): - super().__init__(*args, **kwargs) - - @require_permission("plugin.tts_voice_plugin.command.use") - async def execute(self, args: CommandArgs) -> tuple[bool, str, bool]: - """ - 执行命令的核心逻辑 - """ - all_args = args.get_args() - if not all_args: - await self.send_text("请提供要转换为语音的文本内容哦！") - return False, "缺少文本参数", True - - try: - tts_service = get_service("tts") - if not tts_service: - raise RuntimeError("TTSService 未注册或初始化失败") - - # 获取可用风格列表 - available_styles = tts_service.tts_styles.keys() - - text_to_speak = "" - style_hint = "default" - - # 检查最后一个参数是否是有效的风格 - if len(all_args) > 1 and all_args[-1] in available_styles: - style_hint = all_args[-1] - text_to_speak = " ".join(all_args[:-1]) - else: - # 如果最后一个参数不是风格，则全部都是文本 - text_to_speak = " ".join(all_args) - # 保持默认风格，让 service 层决定是否需要情感分析 - style_hint = "default" - - if not text_to_speak: - await self.send_text("请提供要转换为语音的文本内容哦！") - return False, "文本内容为空", True - - audio_b64 = await tts_service.generate_voice(text_to_speak, style_hint) - - if audio_b64: - await self.send_type(message_type="voice", content=audio_b64) - return True, "语音发送成功", True - else: - await self.send_text("❌ 语音合成失败，请检查服务状态或配置。") - return False, "语音合成失败", True - - except Exception as e: - logger.error(f"执行 /tts 命令时出错: {e}") - await self.send_text("❌ 语音合成时发生了意想不到的错误，请查看日志。") - return False, "命令执行异常", True diff --git a/src/plugins/built_in/tts_voice_plugin/plugin.py b/src/plugins/built_in/tts_voice_plugin/plugin.py deleted file mode 100644 index 305b8b96b..000000000 --- a/src/plugins/built_in/tts_voice_plugin/plugin.py +++ /dev/null @@ -1,112 +0,0 @@ -""" -TTS Voice 插件 - 重构版 -""" -from pathlib import Path -from typing import Any, ClassVar - -import toml - -from src.common.logger import get_logger -from src.plugin_system import BasePlugin, ComponentInfo, register_plugin -from src.plugin_system.base.component_types import PermissionNodeField - -from .actions.tts_action import TTSVoiceAction -from .commands.tts_command import TTSVoiceCommand -from .services.manager import register_service -from .services.tts_service import TTSService - -logger = get_logger("tts_voice_plugin") - - -@register_plugin -class TTSVoicePlugin(BasePlugin): - """ - GPT-SoVITS 语音合成插件 - 重构版 - """ - - plugin_name = "tts_voice_plugin" - plugin_description = "基于GPT-SoVITS的文本转语音插件（重构版）" - plugin_version = "3.1.2" - plugin_author = "Kilo Code & 靚仔" - enable_plugin = True - config_file_name = "config.toml" - dependencies: ClassVar[list[str]] = [] - - permission_nodes: ClassVar[list[PermissionNodeField]] = [ - PermissionNodeField(node_name="command.use", description="是否可以使用 /tts 命令"), - ] - - config_schema: ClassVar[dict] = {} - - config_section_descriptions: ClassVar[dict] = { - "plugin": "插件基本配置", - "components": "组件启用控制", - "tts": "TTS语音合成基础配置", - "tts_advanced": "TTS高级参数配置（语速、采样、批处理等）", - "tts_styles": "TTS风格参数配置（每个分组为一种风格）" - } - - def __init__(self, *args, **kwargs): - super().__init__(*args, **kwargs) - self.tts_service = None - - def _get_config_wrapper(self, key: str, default: Any = None) -> Any: - """ - 配置获取的包装器，用于解决 get_config 无法直接获取动态表（如 tts_styles）和未在 schema 中定义的节的问题。 - 由于插件系统的 schema 为空时不会加载未定义的键，这里手动读取配置文件以获取所需配置。 - """ - # 需要手动加载的顶级配置节 - manual_load_keys = ["tts_styles", "spatial_effects", "tts_advanced", "tts"] - top_key = key.split(".")[0] - - if top_key in manual_load_keys: - try: - plugin_file = Path(__file__).resolve() - bot_root = plugin_file.parent.parent.parent.parent.parent - config_file = bot_root / "config" / "plugins" / self.plugin_name / self.config_file_name - - if not config_file.is_file(): - logger.error(f"TTS config file not found at robustly constructed path: {config_file}") - return default - - full_config = toml.loads(config_file.read_text(encoding="utf-8")) - - # 支持点状路径访问 - value = full_config - for k in key.split("."): - if isinstance(value, dict): - value = value.get(k) - else: - return default - - return value if value is not None else default - - except Exception as e: - logger.error(f"Failed to manually load '{key}' from config: {e}") - return default - - return self.get_config(key, default) - - async def on_plugin_loaded(self): - """ - 插件加载完成后的回调，初始化并注册服务。 - """ - logger.info("初始化 TTSVoicePlugin...") - - # 实例化 TTSService，并传入 get_config 方法 - self.tts_service = TTSService(self._get_config_wrapper) - - # 注册服务 - register_service("tts", self.tts_service) - logger.info("TTSService 已成功初始化并注册。") - - def get_plugin_components(self) -> list[tuple[ComponentInfo, type]]: - """ - 返回插件包含的组件列表。 - """ - components = [] - if self.get_config("components.action_enabled", True): - components.append((TTSVoiceAction.get_action_info(), TTSVoiceAction)) - if self.get_config("components.command_enabled", True): - components.append((TTSVoiceCommand.get_plus_command_info(), TTSVoiceCommand)) - return components diff --git a/src/plugins/built_in/tts_voice_plugin/services/manager.py b/src/plugins/built_in/tts_voice_plugin/services/manager.py deleted file mode 100644 index 661a1c6bf..000000000 --- a/src/plugins/built_in/tts_voice_plugin/services/manager.py +++ /dev/null @@ -1,19 +0,0 @@ -""" -服务管理器 - -用于注册和获取插件内部使用的服务实例。 -""" - -from typing import Any - -_services: dict[str, Any] = {} - - -def register_service(name: str, instance: Any) -> None: - """注册一个服务实例""" - _services[name] = instance - - -def get_service(name: str) -> Any: - """获取一个已注册的服务实例""" - return _services.get(name) diff --git a/src/plugins/built_in/tts_voice_plugin/services/tts_service.py b/src/plugins/built_in/tts_voice_plugin/services/tts_service.py deleted file mode 100644 index eed12d8dd..000000000 --- a/src/plugins/built_in/tts_voice_plugin/services/tts_service.py +++ /dev/null @@ -1,344 +0,0 @@ -""" -TTS 核心服务 -""" -import asyncio -import base64 -import io -import os -import re -from collections.abc import Callable -from typing import Any - -import aiohttp -import soundfile as sf -from pedalboard import Convolution, Pedalboard, Reverb -from pedalboard.io import AudioFile - -from src.common.logger import get_logger - -logger = get_logger("tts_voice_plugin.service") - - -class TTSService: - """封装了TTS合成的核心逻辑""" - - def __init__(self, get_config_func: Callable[[str, Any], Any]): - self.get_config = get_config_func - self.tts_styles: dict[str, Any] = {} - self.timeout: int = 60 - self.max_text_length: int = 500 - self._load_config() - - def _load_config(self) -> None: - """加载插件配置""" - try: - self.timeout = self.get_config("tts.timeout", 60) - self.max_text_length = self.get_config("tts.max_text_length", 500) - self.tts_styles = self._load_tts_styles() - - if self.tts_styles: - logger.info(f"TTS服务已成功加载风格: {list(self.tts_styles.keys())}") - else: - logger.warning("TTS风格配置为空，请检查配置文件") - except Exception as e: - logger.error(f"TTS服务配置加载失败: {e}") - - def _load_tts_styles(self) -> dict[str, dict[str, Any]]: - """加载 TTS 风格配置""" - styles = {} - global_server = self.get_config("tts.server", "http://127.0.0.1:9880") - tts_styles_config = self.get_config("tts_styles", []) - - if not isinstance(tts_styles_config, list): - logger.error(f"tts_styles 配置不是一个列表, 而是 {type(tts_styles_config)}") - return styles - - default_cfg = next((s for s in tts_styles_config if s.get("style_name") == "default"), None) - if not default_cfg: - logger.error("在 tts_styles 配置中未找到 'default' 风格，这是必需的。") - return styles - - default_refer_wav = default_cfg.get("refer_wav_path", "") - default_prompt_text = default_cfg.get("prompt_text", "") - default_gpt_weights = default_cfg.get("gpt_weights", "") - default_sovits_weights = default_cfg.get("sovits_weights", "") - - if not default_refer_wav: - logger.warning("TTS 'default' style is missing 'refer_wav_path'.") - - for style_cfg in tts_styles_config: - if not isinstance(style_cfg, dict): - - continue - - style_name = style_cfg.get("style_name") - if not style_name: - - continue - - styles[style_name] = { - "url": global_server, - "name": style_cfg.get("name", style_name), - "refer_wav_path": style_cfg.get("refer_wav_path", default_refer_wav), - "prompt_text": style_cfg.get("prompt_text", default_prompt_text), - "prompt_language": style_cfg.get("prompt_language", "zh"), - "gpt_weights": style_cfg.get("gpt_weights", default_gpt_weights), - "sovits_weights": style_cfg.get("sovits_weights", default_sovits_weights), - "speed_factor": style_cfg.get("speed_factor"), - "text_language": style_cfg.get("text_language", "auto"), # 新增：读取文本语言模式 - } - return styles - - def _determine_final_language(self, text: str, mode: str) -> str: - """根据配置的语言策略和文本内容，决定最终发送给API的语言代码""" - # 如果策略是具体的语言（如 all_zh, ja），直接使用 - if mode not in ["auto", "auto_yue"]: - return mode - - # 对于 auto 和 auto_yue 策略，进行内容检测 - # 优先检测粤语 - if mode == "auto_yue": - cantonese_keywords = ["嘅", "喺", "咗", "唔", "係", "啲", "咩", "乜", "喂"] - if any(keyword in text for keyword in cantonese_keywords): - logger.info("在 auto_yue 模式下检测到粤语关键词，最终语言: yue") - return "yue" - - # 检测日语（简单启发式规则） - japanese_chars = len(re.findall(r"[\u3040-\u309f\u30a0-\u30ff]", text)) - if japanese_chars > 5 and japanese_chars > len(re.findall(r"[\u4e00-\u9fff]", text)) * 0.5: - logger.info("检测到日语字符，最终语言: ja") - return "ja" - - # 默认回退到中文 - logger.info(f"在 {mode} 模式下未检测到特定语言，默认回退到: zh") - return "zh" - - def _clean_text_for_tts(self, text: str) -> str: - # 1. 基本清理 - text = re.sub(r"[\(（\[【].*?[\)）\]】]", "", text) - text = re.sub(r"([，。！？、；：,.!?;:~\-`])\1+", r"\1", text) - text = re.sub(r"~{2,}|～{2,}", "，", text) - text = re.sub(r"\.{3,}|…{1,}", "。", text) - - # 2. 词语替换 - replacements = {"www": "哈哈哈", "hhh": "哈哈", "233": "哈哈", "666": "厉害", "88": "拜拜"} - for old, new in replacements.items(): - text = text.replace(old, new) - - # 3. 移除不必要的字符 (恢复使用更安全的原版正则，避免误删) - text = re.sub(r"[^\u4e00-\u9fff\u3040-\u309f\u30a0-\u30ffa-zA-Z0-9\s，。！？、；：,.!?;:~～]", "", text) - - # 4. 确保结尾有标点 - if text and not text.endswith(tuple("，。！？、；：,.!?;:")): - text += "。" - - # 5. 智能截断 (保留改进的截断逻辑) - if len(text) > self.max_text_length: - cut_text = text[:self.max_text_length] - punctuation = "。！？.…" - last_punc_pos = max(cut_text.rfind(p) for p in punctuation) - - if last_punc_pos != -1: - text = cut_text[:last_punc_pos + 1] - else: - last_comma_pos = max(cut_text.rfind(p) for p in "，、；,;") - if last_comma_pos != -1: - text = cut_text[:last_comma_pos + 1] - else: - text = cut_text - - return text.strip() - - async def _call_tts_api(self, server_config: dict, text: str, text_language: str, **kwargs) -> bytes | None: - """ - 最终修复版：先切换模型，然后仅通过路径发送合成请求。 - """ - ref_wav_path = kwargs.get("refer_wav_path") - if not ref_wav_path: - logger.error(f"API 调用失败：缺少 refer_wav_path。当前风格配置: {server_config}") - return None - try: - base_url = server_config["url"].rstrip("/") - - # --- 步骤一：像稳定版一样，先切换模型 --- - async def switch_model_weights(weights_path: str | None, weight_type: str): - if not weights_path: - - return - api_endpoint = f"/set_{weight_type}_weights" - switch_url = f"{base_url}{api_endpoint}" - try: - async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=self.timeout)) as session: - async with session.get(switch_url, params={"weights_path": weights_path}) as resp: - if resp.status != 200: - error_text = await resp.text() - logger.error(f"切换 {weight_type} 模型失败: {resp.status} - {error_text}") - else: - logger.info(f"成功切换 {weight_type} 模型为: {weights_path}") - except Exception as e: - logger.error(f"请求切换 {weight_type} 模型时发生网络异常: {e}") - - await switch_model_weights(kwargs.get("gpt_weights"), "gpt") - await switch_model_weights(kwargs.get("sovits_weights"), "sovits") - - # --- 步骤二：构建纯净的、不含Base64的请求数据 --- - data = { - "text": text, - "text_lang": text_language, - "ref_audio_path": ref_wav_path, - "prompt_text": kwargs.get("prompt_text", ""), - "prompt_lang": kwargs.get("prompt_language", "zh"), - # 在稳定版中，这两个参数是通过API切换的，而不是直接放在请求体里 - # "gpt_model_path": kwargs.get("gpt_weights"), - # "sovits_model_path": kwargs.get("sovits_weights"), - } - - # 合并高级配置 - advanced_config = self.get_config("tts_advanced", {}) - if isinstance(advanced_config, dict): - data.update({k: v for k, v in advanced_config.items() if v is not None}) - - # 优先使用风格特定的语速 - if server_config.get("speed_factor") is not None: - data["speed_factor"] = server_config["speed_factor"] - - # --- 步骤三：发送最终的合成请求 --- - tts_url = base_url if base_url.endswith("/tts") else f"{base_url}/tts" - logger.info(f"发送到 TTS API 的数据: {data}") - - async with aiohttp.ClientSession() as session: - async with session.post(tts_url, json=data, timeout=aiohttp.ClientTimeout(total=self.timeout)) as response: - if response.status == 200: - return await response.read() - else: - error_info = await response.text() - logger.error(f"TTS API调用失败: {response.status} - {error_info}") - return None - except asyncio.TimeoutError: - logger.error("TTS服务请求超时") - return None - except Exception as e: - logger.error(f"TTS API调用异常: {e}") - return None - - async def _apply_spatial_audio_effect(self, audio_data: bytes) -> bytes | None: - """根据配置应用空间效果（混响和卷积）""" - try: - effects_config = self.get_config("spatial_effects", {}) - if not effects_config.get("enabled", False): - - return audio_data - - # 获取插件目录和IR文件路径 - # 基于 __file__ 构建稳健的、独立于当前工作目录的路径 - plugin_file = os.path.abspath(__file__) - # services -> tts_voice_plugin -> plugins -> Bot - bot_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(plugin_file)))) - ir_path = os.path.join(bot_root, "assets", "small_room_ir.wav") - - effects = [] - - # 根据配置添加Reverb效果 - if effects_config.get("reverb_enabled", False): - effects.append(Reverb( - room_size=effects_config.get("room_size", 0.15), - damping=effects_config.get("damping", 0.5), - wet_level=effects_config.get("wet_level", 0.33), - dry_level=effects_config.get("dry_level", 0.4), - width=effects_config.get("width", 1.0) - )) - - # 根据配置添加Convolution效果 - if effects_config.get("convolution_enabled", False) and os.path.exists(ir_path): - effects.append(Convolution( - impulse_response_filename=ir_path, - mix=effects_config.get("convolution_mix", 0.5) - )) - elif effects_config.get("convolution_enabled"): - logger.warning(f"卷积混响已启用，但IR文件不存在 ({ir_path})，跳过该效果。") - - if not effects: - - - return audio_data - - # 将原始音频数据加载到内存中的 AudioFile 对象 - with io.BytesIO(audio_data) as audio_stream: - with AudioFile(audio_stream, "r") as f: - board = Pedalboard(effects) - effected = board(f.read(f.frames), f.samplerate) - - # 将处理后的音频数据写回内存中的字节流 - with io.BytesIO() as output_stream: - # 使用 soundfile 写入，因为它更稳定 - sf.write(output_stream, effected.T, f.samplerate, format="WAV") - processed_audio_data = output_stream.getvalue() - - logger.info("成功应用空间效果。") - return processed_audio_data - - except Exception as e: - logger.error(f"应用空间效果时出错: {e}") - return audio_data # 如果出错，返回原始音频 - - async def generate_voice(self, text: str, style_hint: str = "default", language_hint: str | None = None) -> str | None: - self._load_config() - - if not self.tts_styles: - logger.error("TTS风格配置为空，无法生成语音。") - return None - - style = style_hint if style_hint in self.tts_styles else "default" - if style not in self.tts_styles: - if "default" in self.tts_styles: - style = "default" - logger.warning(f"指定风格 '{style_hint}' 不存在，自动回退到: 'default'") - elif self.tts_styles: - style = next(iter(self.tts_styles)) - logger.warning(f"指定风格 '{style_hint}' 和 'default' 均不存在，自动回退到第一个可用风格: {style}") - else: - logger.error("没有任何可用的TTS风格配置") - return None - - server_config = self.tts_styles[style] - clean_text = self._clean_text_for_tts(text) - if not clean_text: - - return None - - # 语言决策流程： - # 1. 优先使用决策模型直接指定的 language_hint (最高优先级) - if language_hint: - final_language = language_hint - logger.info(f"使用决策模型指定的语言: {final_language}") - else: - # 2. 如果模型未指定，则使用风格配置的 language_policy - language_policy = server_config.get("text_language", "auto") - final_language = self._determine_final_language(clean_text, language_policy) - logger.info(f"决策模型未指定语言，使用策略 '{language_policy}' -> 最终语言: {final_language}") - - logger.info(f"开始TTS语音合成，文本：{clean_text[:50]}..., 风格：{style}, 最终语言: {final_language}") - - audio_data = await self._call_tts_api( - server_config=server_config, text=clean_text, text_language=final_language, - refer_wav_path=server_config.get("refer_wav_path"), - prompt_text=server_config.get("prompt_text"), - prompt_language=server_config.get("prompt_language"), - gpt_weights=server_config.get("gpt_weights"), - sovits_weights=server_config.get("sovits_weights"), - ) - - if audio_data: - # 检查是否启用空间音频效果 - spatial_config = self.get_config("spatial_effects", {}) - if spatial_config.get("enabled", False): - logger.info("检测到已启用空间音频效果，开始处理...") - processed_audio = await self._apply_spatial_audio_effect(audio_data) - if processed_audio: - logger.info("空间音频效果应用成功！") - audio_data = processed_audio - else: - logger.warning("空间音频效果应用失败，将使用原始音频。") - - return base64.b64encode(audio_data).decode("utf-8") - return None