Update app.py

app.py

@@ -1,4 +1,4 @@
-from TTS.api import TTS
+# from TTS.api import TTS
 import gradio as gr
 from gradio import Dropdown
 from scipy.io.wavfile import write
@@ -30,13 +30,14 @@ script_choices = {
         "Random": "Grandma, I can’t find your email address. I need to send you something important."
     }
 }
-tts = TTS("tts_models/multilingual/multi-dataset/bark", gpu=True)
+# tts = TTS("tts_models/multilingual/multi-dataset/bark", gpu=True)
 
 
-def infer(prompt, input_wav_file, script_type):
+def infer(prompt, input_wav_file, script_type,selected_theme):
     print("Prompt:", prompt)
     print("Input WAV File:", input_wav_file)
     print("Script Type:", script_type)
+    print(selected_theme)
     print("""
 —————
 NEW INFERENCE:
@@ -70,7 +71,7 @@ NEW INFERENCE:
     else:
         prompt = prompt
 
-    theme_dict = script_choices.get(theme_dropdown.value, {})
+    theme_dict = script_choices.get(selected_theme, {})
     chosen_script = theme_dict.get(script_type, "")
     
     gr.Info("Generating audio from prompt")
@@ -208,7 +209,7 @@ with gr.Blocks(css=css) as demo:
 
         submit_btn.click(
             fn=infer,
-            inputs=[script_text, micro_in, script_type_dropdown],
+            inputs=[script_text, micro_in, script_type_dropdown,theme_dropdown],
             outputs=[cloned_out, video_out, npz_file, folder_path]
         )
 demo.queue(api_open=False, max_size=10).launch()