Update main.py

main.py

@@ -63,7 +63,7 @@ def create_plot():
     
     return fig
 
-def run_main(audio_file,shift):
+def run_main(audio_file,shift, speaker_id):
     # 固定の引数を設定
     class Args:
         pass
@@ -71,7 +71,8 @@ def run_main(audio_file,shift):
     args = Args()
     args.config = "configs/base.yaml"
     args.model = "./vits_pretrain/sovits5.0.pretrain.pth"
-    args.spk = "./configs/singers/singer0001.npy"
+    speaker_str = f"{speaker_id:04d}"
+    args.spk = f"./configs/singers/singer{speaker_str}.npy"
     args.wave = audio_file
     args.shift = shift
     
@@ -103,7 +104,16 @@ with gr.Blocks() as demo:
         type="filepath"  # ファイルパスとして取得
     )
 
-    plot = gr.Plot(value=create_plot())
+    with gr.Row():
+        plot = gr.Plot(value=create_plot())
+        # Speaker ID の選択
+        speaker_id = gr.Number(
+            label="Speaker ID (1-56)",
+            value=1,
+            minimum=1,
+            maximum=56,
+            step=1
+        )
     
     # Pitch シフトのスライダー
     shift = gr.Slider(
@@ -123,10 +133,10 @@ with gr.Blocks() as demo:
     
     run_btn.click(
         fn=run_main,
-        inputs=[input_audio, shift],
+        inputs=[input_audio, shift, speaker_id],
         outputs=[output_audio]
     )
 
 # アプリケーションの起動
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()