Update gradio_tabs/single.py

gradio_tabs/single.py

@@ -27,9 +27,8 @@ from typing import Dict, Any, List, Tuple, Optional, Set
 # --- タイムゾーン定義 ---
 # グローバルな定数としてJSTを定義
 JST = timezone(timedelta(hours=9), 'JST')
-# ▼▼▼ 変更点: FNモデル用のキャッシュパスを定義 ▼▼▼
+# FNモデル用のキャッシュパスを定義
 FN_MODEL_CACHE_PATH = Path("/tmp/sbv2_merger_cache")
-# ▲▲▲ 変更点 ▲▲▲
 
 
 # --- モック（本来はライブラリからインポート） ---
@@ -182,14 +181,12 @@ def format_and_sort_model_names(dir_list: List[str]) -> List[Tuple[str, str]]:
     result_list.extend(sorted(unparsed_models))
     return result_list
 
-# ▼▼▼ 変更点: FNモデルソート用のヘルパー関数を追加 ▼▼▼
 def get_fn_model_sort_key(name: str) -> int:
     """FNモデル名からソート用の数値を抽出する。例: 'FN10' -> 10"""
     match = re.search(r'FN(\d+)', name, re.IGNORECASE)
     if match:
         return int(match.group(1))
     return float('inf') # マッチしないものは後ろに配置
-# ▲▲▲ 変更点 ▲▲▲
 
 
 def set_random_seed(seed: int):
@@ -209,15 +206,13 @@ def get_directory_size(directory_path: Path) -> int:
         for dirpath, _, filenames in os.walk(directory_path):
             for f in filenames:
                 fp = os.path.join(dirpath, f)
-                # シンボリックリンクが壊れている場合などを考慮
                 if not os.path.islink(fp):
                     try:
                         total_size += os.path.getsize(fp)
                     except OSError:
-                        # ファイルがスキャン中に削除された場合など
                         pass
     except FileNotFoundError:
-        return 0 # ディレクトリが存在しない場合は0
+        return 0
     return total_size
 
 def format_bytes(size_bytes: int) -> str:
@@ -518,9 +513,7 @@ def create_synthesis_app(model_holder: TTSModelHolder) -> gr.Blocks:
 
                     with gr.Column(scale=1):
                         with gr.Row():
-                            # ▼▼▼ 変更点: 「融☆合モデルを使う」を「FNモデルモード」に変更し、最初は非表示に ▼▼▼
                             use_fn_model_mode_checkbox = gr.Checkbox(label="FNモデルモード", value=False, interactive=True, scale=3, visible=False)
-                            # ▲▲▲ 変更点 ▲▲▲
                             refresh_model_list_button = gr.Button("再読込", scale=1)
                         selected_model_dropdown = gr.Dropdown(label="話者", choices=[], value=None, interactive=True)
                         current_styles_dropdown = gr.Dropdown(label="スタイル", choices=[], type="value", interactive=True)
@@ -619,10 +612,8 @@ def create_synthesis_app(model_holder: TTSModelHolder) -> gr.Blocks:
                 default_weight = styles_map[first_key].get("weight", DEFAULT_STYLE_WEIGHT)
             return gr.update(choices=display_names, value=default_display_name), gr.update(value=default_weight), styles_map
 
-        # ▼▼▼ 変更点: FNモデルモードに対応したモデルリスト更新関数 ▼▼▼
         def action_refresh_model_list(use_fn_model_mode: bool):
             """モデルリストを再読み込みし、UIとバックエンドの状態を同期させる。"""
-            # 既存のシンボリックリンクをクリア
             if assets_root_path.exists():
                 for item in assets_root_path.iterdir():
                     if item.is_symlink():
@@ -631,7 +622,6 @@ def create_synthesis_app(model_holder: TTSModelHolder) -> gr.Blocks:
                         except OSError as e:
                             print(f"Failed to remove symlink {item}: {e}")
 
-            # FNモデルキャッシュパスからFNモデルを探し、シンボリックリンクを作成
             fn_models_found = []
             if FN_MODEL_CACHE_PATH.exists() and FN_MODEL_CACHE_PATH.is_dir():
                 for item in FN_MODEL_CACHE_PATH.iterdir():
@@ -646,28 +636,25 @@ def create_synthesis_app(model_holder: TTSModelHolder) -> gr.Blocks:
             
             fn_mode_checkbox_update = gr.update(visible=bool(fn_models_found))
 
-            # バックエンドのモデルリストを更新
             model_holder.refresh()
             
             model_dropdown_update = gr.update(choices=[], value=None)
             if use_fn_model_mode and fn_models_found:
-                # FNモデルモードの場合: FNモデルを数値でソートして表示
                 sorted_fn_models = sorted(fn_models_found, key=get_fn_model_sort_key)
                 value = sorted_fn_models[0] if sorted_fn_models else None
                 model_dropdown_update = gr.update(choices=sorted_fn_models, value=value)
             else:
-                # 通常モードの場合: 通常モデルとマージモデルを表示
                 ui_model_list = [p.name for p in assets_root_path.iterdir() if p.is_dir() and not p.is_symlink()]
                 formatted_choices = format_and_sort_model_names(ui_model_list)
                 value = formatted_choices[0][1] if formatted_choices else None
                 model_dropdown_update = gr.update(choices=formatted_choices, value=value)
 
-            # 選択されたモデルに基づいてスタイルUIを更新
-            selected_model_for_style = model_dropdown_update.value
+            # ▼▼▼ 変更点: .value を ['value'] に修正 ▼▼▼
+            selected_model_for_style = model_dropdown_update['value']
+            # ▲▲▲ 変更点 ▲▲▲
             style_dropdown_update, style_weight_update, styles_data_state_update = load_styles_for_ui(selected_model_for_style)
             
             return fn_mode_checkbox_update, model_dropdown_update, style_dropdown_update, style_weight_update, styles_data_state_update
-        # ▲▲▲ 変更点 ▲▲▲
 
 
         def on_model_select_change(selected_model_name: Optional[str]):
@@ -1080,7 +1067,6 @@ def create_synthesis_app(model_holder: TTSModelHolder) -> gr.Blocks:
 
 
         # --- イベントリスナー接続 ---
-        # ▼▼▼ 変更点: FNモデルモードに対応したイベントリスナーの定義 ▼▼▼
         refresh_triggers = [refresh_model_list_button.click, use_fn_model_mode_checkbox.change]
         outputs_for_refresh = [use_fn_model_mode_checkbox, selected_model_dropdown, current_styles_dropdown, style_weight_for_synth_slider, all_styles_data_state]
         
@@ -1166,14 +1152,12 @@ def create_synthesis_app(model_holder: TTSModelHolder) -> gr.Blocks:
         app.load(action_refresh_model_list, 
                  inputs=[use_fn_model_mode_checkbox], 
                  outputs=outputs_for_refresh)
-        # ▲▲▲ 変更点 ▲▲▲
     return app
 
 # --- アプリケーションの起動 ---
 if __name__ == "__main__":
     if Path("model_assets").exists(): shutil.rmtree("model_assets")
 
-    # ▼▼▼ 変更点: FNモデルキャッシュパスの作成とGradioへの許可 ▼▼▲
     # モックFNモデルを作成してテスト
     FN_MODEL_CACHE_PATH.mkdir(exist_ok=True, parents=True)
     (FN_MODEL_CACHE_PATH / "FN1").mkdir(exist_ok=True)
@@ -1188,7 +1172,6 @@ if __name__ == "__main__":
     assets_dir_path.mkdir(exist_ok=True)
     allowed_paths = [str(assets_dir_path)]
     
-    # FNモデルのキャッシュパスへのアクセスを許可
     if FN_MODEL_CACHE_PATH.exists() and FN_MODEL_CACHE_PATH.is_dir():
         allowed_paths.append(str(FN_MODEL_CACHE_PATH.resolve()))
     else:
@@ -1209,5 +1192,4 @@ if __name__ == "__main__":
 
     print(f"Gradioに次のパスへのアクセスを許可します: {', '.join(allowed_paths)}")
 
-    app.launch(allowed_paths=allowed_paths)
-    # ▲▲▲ 変更点 ▲▲▲
+    app.launch(allowed_paths=allowed_paths)