-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
177 lines (158 loc) · 6.66 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
import json
import os
import uuid
import wave
import discord
import requests
import yaml
from discord import app_commands
from gradio_client import Client
from src import checker, config, guild_config, guild_tts_manager
if __name__ == "__main__":
config = config.Config()
print(f"{config.rvc_default_model}")
rvc_pitch = 0
rvc_client = None
if not config.rvc_disabled:
rvc_client = Client(f"http://{config.rvc_host}:{config.rvc_port}/")
rvc_client.httpx_auth = (
None # hack : 定義されていない値を読み取ろうとするエラー修正
)
result = rvc_client.predict(
f"{config.rvc_default_model}", # 推論ファイル
0,
0,
api_name="/infer_change_voice",
)
guild_tts_manager = guild_tts_manager.guild_tts_manager()
def rvc(filepath):
result = rvc_client.predict(
0,
filepath,
rvc_pitch, # ピッチ
filepath,
"pm", # pm or harvest or crepe
"Howdy!",
"null",
0,
0,
0,
0,
0,
api_name="/infer_convert",
)
print("rvc_pitch:" + str(rvc_pitch))
return result[1]
def generate_wav(text, speaker=1, filepath="./temp/wav/audio.wav"):
host = "localhost"
port = 50021
params = (
("text", text),
("speaker", speaker),
("enable_interrogative_upspeak", True),
)
response1 = requests.post(f"http://{host}:{port}/audio_query", params=params)
print(response1.json())
headers = {
"Content-Type": "application/json",
}
response2 = requests.post(
f"http://{host}:{port}/synthesis",
headers=headers,
params=params,
data=json.dumps(response1.json()),
)
with open(filepath, mode="wb") as f:
f.write(response2.content)
f.close()
def wav_gen_and_get_path(
text,
): # テキストを受取、生成した読み上げ音声のパスを返す。
msg_uuid = str(uuid.uuid4())
generate_wav(text, 2, f"temp/wav/{msg_uuid}.wav")
output_wav_file = os.path.abspath(f"temp/wav/{msg_uuid}.wav")
if not config.rvc_disabled:
rvc_voice_path = rvc(os.path.abspath(f"temp/wav/{msg_uuid}.wav"))
output_wav_file = rvc_voice_path
return output_wav_file
discord_access_token = config.discord_access_token
discord_application_id = config.discord_application_id
client = discord.Client(intents=discord.Intents.all())
tree = app_commands.CommandTree(client)
@client.event
async def on_ready():
print("Bot started!")
print(
f"https://discord.com/api/oauth2/authorize?client_id={discord_application_id}&permissions=3148864&scope=bot%20applications.commands"
)
await tree.sync() # スラッシュコマンドを同期
@tree.command(name="change_rvc", description="change rvc model")
async def change_rvcmodel_command(interaction: discord.Interaction, text: str):
await interaction.response.send_message("処理中...")
rvc_client.predict(text, 0, 0, api_name="/infer_change_voice") # 推論ファイル
await interaction.edit_original_response(
content=f"RVCモデルを '{text}' に変更しました。"
)
@tree.command(name="vspeed", description="読み上げのスピードを変更します。")
async def speed_command(interaction: discord.Interaction, speed: int):
await interaction.response.send_message(
f"この機能は現在実装中です。今後のアップデートで使えるようになります！"
)
@tree.command(name="silent", description="履歴を残さずに読み上げます")
async def silent_command(interaction: discord.Interaction, message: str):
content = message
if len(content) > config.max_text_length:
content = content[0 : config.max_text_length] + "以下省略"
if checker.is_url(content):
content = "URL"
if not checker.ignore_check(content):
wav_path = wav_gen_and_get_path(content)
guild_tts_manager.enqueue(
interaction.guild.voice_client,
interaction.guild,
discord.FFmpegPCMAudio(wav_path),
)
await interaction.response.send_message(
"OK", ephemeral=True, delete_after=3
)
@tree.command(name="vjoin", description="ボイスチャットにボットを追加。")
async def join_command(interaction: discord.Interaction):
if interaction.user.voice is None:
await interaction.response.send_message(
"先に、ボイスチャンネルに接続してください。", ephemeral=True
)
else:
await interaction.user.voice.channel.connect()
await interaction.response.send_message("ボイスチャンネルに接続しました。")
wav_path = wav_gen_and_get_path("接続しました。")
guild_tts_manager.enqueue(
interaction.guild.voice_client,
interaction.guild,
discord.FFmpegPCMAudio(wav_path),
)
@tree.command(name="vleave", description="ボイスチャットから切断します。")
async def join_command(interaction: discord.Interaction):
await interaction.guild.voice_client.disconnect()
await interaction.response.send_message("切断しました。")
@tree.command(name="vpitch", description="ピッチを調節します。")
async def pitch_command(interaction: discord.Interaction, value: int):
global rvc_pitch
rvc_pitch = value
await interaction.response.send_message(f"ピッチを{value}に設定しました。")
@client.event
async def on_message(message):
if not message.author.bot:
if message.guild.voice_client is not None:
content = message.content
if len(content) > config.max_text_length:
content = content[0 : config.max_text_length] + "以下省略"
if checker.is_url(content):
content = "URL"
if not checker.ignore_check(content):
wav_path = wav_gen_and_get_path(content)
guild_tts_manager.enqueue(
message.guild.voice_client,
message.guild,
discord.FFmpegPCMAudio(wav_path),
)
client.run(discord_access_token)