Update vc.py

vc.py

@@ -51,7 +51,7 @@ def bark_ez(text, n='1', land='en'):
     return (f'{uid}-output.wav')
 
 
-def custom_bark(inp, tog, in_aud=None, trim_aud=None, in_aud_mic=None):
+def custom_bark(inp, tog, speaker,in_aud=None, trim_aud=None, in_aud_mic=None):
     if tog=="Custom":
         if in_aud_mic != None:
             speaker_wav=in_aud_mic
@@ -64,7 +64,7 @@ def custom_bark(inp, tog, in_aud=None, trim_aud=None, in_aud_mic=None):
         tts.tts_to_file(inp, speaker_wav=speaker_wav, language="en", file_path=f"{uid}-output.wav")
         return (f"{uid}-output.wav")
     if tog=="Preset":
-        return (bark_ez(inp))
+        return (bark_ez(inp,speaker))
 def load_video_yt(vid):
     yt = YouTube(vid)
     vid = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename=f"{uid}-tmp.mp4")
@@ -93,6 +93,13 @@ def pre_aud(inp):
     song.export(f"{uid}-tmp_aud.mp4", format="mp4")
     print(f'pre_aud:: {f"{uid}-tmp_aud.mp4"}')
     return inp
+def tog_in(tog):
+    if tog=="Preset":
+        return (gr.update(visible=True),gr.update(visible=False))
+    if tog=="Custom":
+        return (gr.update(visible=False),gr.update(visible=True))   
+
+
 with gr.Blocks() as app:
     with gr.Group():
         with gr.Row():
@@ -100,11 +107,14 @@ with gr.Blocks() as app:
             with gr.Column():
                 alt_go_btn = gr.Button()
                 out_audio = gr.Audio(interactive=False)
-    with gr.Group():      
+
         with gr.Row():
             gr.Markdown('''<H1> Audio Source:''')
         with gr.Row():
-            tog = gr.Radio(label="Input Type", choices=["Preset","Custom"], value="Preset")
+            tog = gr.Radio(label="Input Type", choices=["Preset","Custom"], value="Preset")    
+    with gr.Group(visible=True) as group_1:
+        speaker_num = gr.Dropdown(label="Speaker Voice", choices=num_list,value="1")
+    with gr.Group(visible=False) as group_2:      
         with gr.Row():
             with gr.Column():
 
@@ -121,11 +131,11 @@ with gr.Blocks() as app:
                 load_yt_btn = gr.Button("Load URL")    
                 yt_vid = gr.Video(interactive=False)
                 
-
-
+         
+    tog.change(tog_in,tog,[group_1,group_2])
     #in_aud_file.change(pre_aud,in_aud_file,aud_file)
     load_yt_btn.click(load_video_yt, in_aud_yt, [yt_vid,in_aud_file,aud_file])
     trim_clip_btn.click(trim_clip,[in_aud_file, start_time, end_time],trim_aud)
-    alt_go_btn.click(custom_bark, [in_text,tog,in_aud_file,trim_aud], out_audio)
+    alt_go_btn.click(custom_bark, [in_text,tog,speaker_num,in_aud_file,trim_aud], out_audio)
 
 app.launch()