Update app.py

app.py

@@ -25,31 +25,53 @@ def init_TTS(config):
     tts = TTS(config['inference']['model']).to(device)
     return tts
 
+                dataset_choice = gr.Radio(label="Choose Dataset", 
+                                          choices=["Hugging Face Hub Dataset", "Upload Your Own"], 
+                                          value="Hugging Face Hub Dataset")
+                dataset_predefined = gr.Textbox(label="Hugging Face Hub Training Dataset", 
+                                                value='yahma/alpaca-cleaned', 
+                                                visible=True)
+
 
 @spaces.GPU
-def generate_speech(microphone, text):
+def generate_speech(voice_choice, microphone, text):
     # Generate speech using the provided text, speaker voice, and language
+
+    if voice_choice=="Record":
+        speaker = microphone
+    
+    else:
+        speaker = config['inference']['speaker_wav']
     
     tts.tts_to_file(text=text,
                     file_path=config['inference']['file_path'],
-                    speaker_wav=microphone, # config['inference']['speaker_wav']
+                    speaker_wav=microphone,
                     language=config['inference']['language'])
     return config['inference']['file_path']
 
 
+
+
+
 ###################################### main ######################################
 def main(config):
+
+    voice_choice = gr.Radio(label="Record or use predefined voice.", 
+                            choices=["Record", "Predefined (Nancy)"], 
+                            value="Record")
     
     # Create the Gradio interface
     demo = gr.Interface(
         fn=generate_speech,
         inputs=[
+            voice_choice,
+            
             gr.Audio(label="Audio", sources="microphone", type="filepath", elem_id='audio'),
             gr.Textbox(label="Enter your text")
         ],
         outputs="audio",
-        title="Voice Synthesis with Coqui-XTTS",
-        description="Synthesize speech using predefined target voice and language."
+        title="Voice cloning and Synthesis with Coqui-XTTS",
+        description="Clone your voice and Synthesize speech using predefined target voice and language."
     )
 
     # Launch the interface