Spaces:
Running
Running
shiny-rufflet
commited on
Commit
·
7d653ef
1
Parent(s):
966a19e
- app.py +6 -8
- gradio_tabs/sample.py +16 -34
app.py
CHANGED
@@ -8,7 +8,7 @@ start_time = time.time()
|
|
8 |
# Hugging Face Spacesでは推論のみ行うのが一般的なので、
|
9 |
# only_infer=True を指定して不要なモデルのダウンロードをスキップします。
|
10 |
# これにより起動が高速化します。
|
11 |
-
initialize.main(only_infer=True)
|
12 |
end_time = time.time()
|
13 |
print(f"--- [2/3] Model Setup Complete. Elapsed time: {end_time - start_time:.2f} seconds ---")
|
14 |
# ===================================================================
|
@@ -26,8 +26,8 @@ import torch
|
|
26 |
# 拡張機能タブから必要なものだけをインポート
|
27 |
from gradio_tabs.single import create_synthesis_app
|
28 |
from gradio_tabs.merge import create_merge_app
|
29 |
-
#
|
30 |
-
from gradio_tabs.sample import
|
31 |
|
32 |
from config import get_path_config
|
33 |
from style_bert_vits2.constants import GRADIO_THEME, VERSION
|
@@ -44,7 +44,7 @@ update_dict()
|
|
44 |
|
45 |
|
46 |
parser = argparse.ArgumentParser()
|
47 |
-
parser.add_argument("--device", type=
|
48 |
parser.add_argument("--no_autolaunch", action="store_true")
|
49 |
|
50 |
args = parser.parse_args()
|
@@ -58,8 +58,6 @@ else:
|
|
58 |
|
59 |
path_config = get_path_config()
|
60 |
|
61 |
-
# この時点で、initialize.pyが作成したモデルへのシンボリックリンクが
|
62 |
-
# Path(path_config.assets_root) 以下に存在しているはずです。
|
63 |
model_holder = TTSModelHolder(Path(path_config.assets_root), device)
|
64 |
|
65 |
# 起動時にトップへスクロールさせるjsを追加
|
@@ -71,9 +69,9 @@ with gr.Blocks(theme=GRADIO_THEME, js="() => { window.scrollTo(0, 0); }") as app
|
|
71 |
create_synthesis_app(model_holder=model_holder)
|
72 |
with gr.Tab("融☆合"):
|
73 |
create_merge_app(model_holder=model_holder)
|
74 |
-
#
|
75 |
with gr.Tab("サンプル"):
|
76 |
-
|
77 |
|
78 |
# Hugging Face Spacesでは、share=Trueは不要(自動で公開URLが生成される)
|
79 |
# また、サーバー名とポートも指定しない方が安定します。
|
|
|
8 |
# Hugging Face Spacesでは推論のみ行うのが一般的なので、
|
9 |
# only_infer=True を指定して不要なモデルのダウンロードをスキップします。
|
10 |
# これにより起動が高速化します。
|
11 |
+
initialize.main(only_infer=True)
|
12 |
end_time = time.time()
|
13 |
print(f"--- [2/3] Model Setup Complete. Elapsed time: {end_time - start_time:.2f} seconds ---")
|
14 |
# ===================================================================
|
|
|
26 |
# 拡張機能タブから必要なものだけをインポート
|
27 |
from gradio_tabs.single import create_synthesis_app
|
28 |
from gradio_tabs.merge import create_merge_app
|
29 |
+
# ▼▼▼ ここを追加 ▼▼▼
|
30 |
+
from gradio_tabs.sample import create_player_tab # sample.pyから関数をインポート
|
31 |
|
32 |
from config import get_path_config
|
33 |
from style_bert_vits2.constants import GRADIO_THEME, VERSION
|
|
|
44 |
|
45 |
|
46 |
parser = argparse.ArgumentParser()
|
47 |
+
parser.add_argument("--device", type=str, default="cuda")
|
48 |
parser.add_argument("--no_autolaunch", action="store_true")
|
49 |
|
50 |
args = parser.parse_args()
|
|
|
58 |
|
59 |
path_config = get_path_config()
|
60 |
|
|
|
|
|
61 |
model_holder = TTSModelHolder(Path(path_config.assets_root), device)
|
62 |
|
63 |
# 起動時にトップへスクロールさせるjsを追加
|
|
|
69 |
create_synthesis_app(model_holder=model_holder)
|
70 |
with gr.Tab("融☆合"):
|
71 |
create_merge_app(model_holder=model_holder)
|
72 |
+
# ▼▼▼ ここを追加 ▼▼▼
|
73 |
with gr.Tab("サンプル"):
|
74 |
+
create_player_tab() # インポートした関数を呼び出してUIを構築
|
75 |
|
76 |
# Hugging Face Spacesでは、share=Trueは不要(自動で公開URLが生成される)
|
77 |
# また、サーバー名とポートも指定しない方が安定します。
|
gradio_tabs/sample.py
CHANGED
@@ -4,7 +4,7 @@ import glob
|
|
4 |
import re
|
5 |
|
6 |
# --- 定数定義 ---
|
7 |
-
# Style-Bert-VITS2
|
8 |
OUTPUT_DIR = "sample_audio"
|
9 |
|
10 |
# --- 1. ヘルパー関数 ---
|
@@ -14,54 +14,41 @@ def natural_sort_key(s):
|
|
14 |
"""
|
15 |
return [int(text) if text.isdigit() else text.lower() for text in re.split(r'(\d+)', s)]
|
16 |
|
17 |
-
# --- 2. Gradioアプリのロジックとなる関数 ---
|
18 |
def get_folders():
|
19 |
-
"""
|
20 |
if not os.path.exists(OUTPUT_DIR) or not os.path.isdir(OUTPUT_DIR):
|
21 |
-
# outputフォルダが存在しない場合、空のリストと警告を返す
|
22 |
print(f"警告: '{OUTPUT_DIR}' フォルダが見つかりません。")
|
23 |
return []
|
24 |
|
25 |
-
# is_dir() でディレクトリのみをフィルタリング
|
26 |
folders = [f.name for f in os.scandir(OUTPUT_DIR) if f.is_dir()]
|
27 |
-
|
28 |
-
# フォルダ名を自然順でソート
|
29 |
folders.sort(key=natural_sort_key)
|
30 |
return folders
|
31 |
|
32 |
def update_file_list(folder_name):
|
33 |
"""
|
34 |
選択されたフォルダに応じて、ファイルリストの選択肢を更新します。
|
35 |
-
ファイルパス全体ではなく、内容がわかるテキスト部分のみを表示します。
|
36 |
"""
|
37 |
if not folder_name:
|
38 |
return gr.update(choices=[], value=None, label="ファイルを選択")
|
39 |
|
40 |
-
# [XXX]/[XXX]_[YYY].wav のパターンに一致するファイルを検索
|
41 |
search_pattern = os.path.join(OUTPUT_DIR, folder_name, f"{folder_name}_*.wav")
|
42 |
file_paths = sorted(glob.glob(search_pattern))
|
43 |
|
44 |
-
# 表示用の選択肢リストを作成 (表示名, 実際の値) のタプルのリスト
|
45 |
choices_list = []
|
46 |
for path in file_paths:
|
47 |
-
# ファイル名部分を取得 (例: FN1_こんにちは.wav)
|
48 |
basename = os.path.basename(path)
|
49 |
-
# プレフィックスと拡張子を除去して表示名を作成 (例: こんにちは)
|
50 |
prefix_to_remove = f"{folder_name}_"
|
51 |
if basename.startswith(prefix_to_remove):
|
52 |
display_name = basename[len(prefix_to_remove):]
|
53 |
else:
|
54 |
display_name = basename
|
55 |
|
56 |
-
# .wav拡張子を削除
|
57 |
display_name = display_name.removesuffix(".wav")
|
58 |
-
|
59 |
choices_list.append((display_name, path))
|
60 |
|
61 |
if not choices_list:
|
62 |
return gr.update(choices=[], value=None, label=f"{folder_name} にはファイルがありません")
|
63 |
|
64 |
-
# Gradioコンポーネントを更新して返す
|
65 |
return gr.update(choices=choices_list, value=None, label=f"「{folder_name}」のファイルリスト")
|
66 |
|
67 |
def play_audio(filepath):
|
@@ -71,8 +58,11 @@ def play_audio(filepath):
|
|
71 |
return gr.update(value=filepath, visible=bool(filepath))
|
72 |
|
73 |
|
74 |
-
# ---
|
75 |
-
|
|
|
|
|
|
|
76 |
gr.Markdown(
|
77 |
f"""
|
78 |
# Style-Bert-VITS2 音声再生プレイヤー
|
@@ -81,7 +71,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
81 |
"""
|
82 |
)
|
83 |
|
84 |
-
# 初期のフォルダリストを取得
|
85 |
initial_folders = get_folders()
|
86 |
|
87 |
if not initial_folders:
|
@@ -93,36 +82,29 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
93 |
label="1. フォルダ(話者)を選択",
|
94 |
interactive=True
|
95 |
)
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
|
|
|
|
101 |
|
102 |
-
# 最初は非表示にしておき、ファイルが選択されたら表示する
|
103 |
audio_player = gr.Audio(label="再生プレイヤー", type="filepath", visible=False)
|
104 |
|
105 |
-
# ---
|
106 |
-
|
107 |
-
# フォルダセレクタの値が変更されたら、ファイルリストを更新する
|
108 |
folder_selector.change(
|
109 |
fn=update_file_list,
|
110 |
inputs=folder_selector,
|
111 |
outputs=file_selector
|
112 |
).then(
|
113 |
-
# フォルダが切り替わったらプレイヤーを非表示にする
|
114 |
fn=lambda: gr.update(value=None, visible=False),
|
115 |
inputs=None,
|
116 |
outputs=audio_player
|
117 |
)
|
118 |
|
119 |
-
# ファイルセレクタの値が変更されたら、オーディオプレイヤーを更新する
|
120 |
file_selector.change(
|
121 |
fn=play_audio,
|
122 |
inputs=file_selector,
|
123 |
outputs=audio_player
|
124 |
-
)
|
125 |
-
|
126 |
-
# --- 5. アプリの実行 ---
|
127 |
-
if __name__ == "__main__":
|
128 |
-
demo.launch()
|
|
|
4 |
import re
|
5 |
|
6 |
# --- 定数定義 ---
|
7 |
+
# Style-Bert-VITS2の実際の出力フォルダ名を指定
|
8 |
OUTPUT_DIR = "sample_audio"
|
9 |
|
10 |
# --- 1. ヘルパー関数 ---
|
|
|
14 |
"""
|
15 |
return [int(text) if text.isdigit() else text.lower() for text in re.split(r'(\d+)', s)]
|
16 |
|
|
|
17 |
def get_folders():
|
18 |
+
"""outputsディレクトリ内のサブフォルダのリストを自然順ソートで取得します。"""
|
19 |
if not os.path.exists(OUTPUT_DIR) or not os.path.isdir(OUTPUT_DIR):
|
|
|
20 |
print(f"警告: '{OUTPUT_DIR}' フォルダが見つかりません。")
|
21 |
return []
|
22 |
|
|
|
23 |
folders = [f.name for f in os.scandir(OUTPUT_DIR) if f.is_dir()]
|
|
|
|
|
24 |
folders.sort(key=natural_sort_key)
|
25 |
return folders
|
26 |
|
27 |
def update_file_list(folder_name):
|
28 |
"""
|
29 |
選択されたフォルダに応じて、ファイルリストの選択肢を更新します。
|
|
|
30 |
"""
|
31 |
if not folder_name:
|
32 |
return gr.update(choices=[], value=None, label="ファイルを選択")
|
33 |
|
|
|
34 |
search_pattern = os.path.join(OUTPUT_DIR, folder_name, f"{folder_name}_*.wav")
|
35 |
file_paths = sorted(glob.glob(search_pattern))
|
36 |
|
|
|
37 |
choices_list = []
|
38 |
for path in file_paths:
|
|
|
39 |
basename = os.path.basename(path)
|
|
|
40 |
prefix_to_remove = f"{folder_name}_"
|
41 |
if basename.startswith(prefix_to_remove):
|
42 |
display_name = basename[len(prefix_to_remove):]
|
43 |
else:
|
44 |
display_name = basename
|
45 |
|
|
|
46 |
display_name = display_name.removesuffix(".wav")
|
|
|
47 |
choices_list.append((display_name, path))
|
48 |
|
49 |
if not choices_list:
|
50 |
return gr.update(choices=[], value=None, label=f"{folder_name} にはファイルがありません")
|
51 |
|
|
|
52 |
return gr.update(choices=choices_list, value=None, label=f"「{folder_name}」のファイルリスト")
|
53 |
|
54 |
def play_audio(filepath):
|
|
|
58 |
return gr.update(value=filepath, visible=bool(filepath))
|
59 |
|
60 |
|
61 |
+
# --- 2. Gradio UIの構築 ---
|
62 |
+
def create_player_tab():
|
63 |
+
"""
|
64 |
+
音声再生プレイヤーのUIコンポーネントを作成し、タブ内に配置できるようにする関数。
|
65 |
+
"""
|
66 |
gr.Markdown(
|
67 |
f"""
|
68 |
# Style-Bert-VITS2 音声再生プレイヤー
|
|
|
71 |
"""
|
72 |
)
|
73 |
|
|
|
74 |
initial_folders = get_folders()
|
75 |
|
76 |
if not initial_folders:
|
|
|
82 |
label="1. フォルダ(話者)を選択",
|
83 |
interactive=True
|
84 |
)
|
85 |
+
# フォルダリストが長い場合に見やすくなるようにコンポーネントを分ける
|
86 |
+
with gr.Column(scale=2): # 横幅の比率を調整
|
87 |
+
file_selector = gr.Radio(
|
88 |
+
label="2. ファイルを選択",
|
89 |
+
choices=[],
|
90 |
+
interactive=True
|
91 |
+
)
|
92 |
|
|
|
93 |
audio_player = gr.Audio(label="再生プレイヤー", type="filepath", visible=False)
|
94 |
|
95 |
+
# --- 3. コンポーネント間の連携を設定 ---
|
|
|
|
|
96 |
folder_selector.change(
|
97 |
fn=update_file_list,
|
98 |
inputs=folder_selector,
|
99 |
outputs=file_selector
|
100 |
).then(
|
|
|
101 |
fn=lambda: gr.update(value=None, visible=False),
|
102 |
inputs=None,
|
103 |
outputs=audio_player
|
104 |
)
|
105 |
|
|
|
106 |
file_selector.change(
|
107 |
fn=play_audio,
|
108 |
inputs=file_selector,
|
109 |
outputs=audio_player
|
110 |
+
)
|
|
|
|
|
|
|
|