Update app.py

app.py

@@ -118,7 +118,7 @@ def  genrate_speech(text,name_model):
              ).waveform.cpu().numpy().reshape(-1)
     return model.config.sampling_rate,wav
 
-def generate_audio(text, speaker_id=None):
+def generate_audio(text,name_model,speaker_id=None):
     inputs = tokenizer(text, return_tensors="pt")#.input_ids
 
     speaker_embeddings = None
@@ -158,17 +158,7 @@ def   get_answer_ai(text):
           return response.text
 
 with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
-    with gr.Tab("Chat AI "):
-        gr.Markdown("## AI: محادثة صوتية بالذكاء الاصطناعي باللهجة السعودية")
-        with gr.Row(): # Arrange input/output components side-by-side
-            with gr.Column():
-                text_input = gr.Textbox(label="أدخل أي نص")
-                user_audio = gr.Audio(label="صوتك")
-                with gr.Row():
-                    btn = gr.Button("إرسال")
-                    btn_ai_only = gr.Button("توليد رد الذكاء الاصطناعي فقط")
-                    
-            with gr.Column():
+    with gr.Column():
                 model_choices = gr.Dropdown(
                     choices=[
                         "asg2024/vits-ar-sa",
@@ -180,6 +170,17 @@ with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
                     label="اختر النموذج",
                     value="asg2024/vits-ar-sa",
                 )
+    with gr.Tab("Chat AI "):
+        gr.Markdown("## AI: محادثة صوتية بالذكاء الاصطناعي باللهجة السعودية")
+        with gr.Row(): # Arrange input/output components side-by-side
+            with gr.Column():
+                text_input = gr.Textbox(label="أدخل أي نص")
+                user_audio = gr.Audio(label="صوتك")
+                with gr.Row():
+                    btn = gr.Button("إرسال")
+                    btn_ai_only = gr.Button("توليد رد الذكاء الاصطناعي فقط")
+                    
+            
                 ai_audio = gr.Audio(label="رد الذكاء الاصطناعي الصوتي")
                 ai_text = gr.Textbox(label="رد الذكاء الاصطناعي النصي")
 
@@ -220,7 +221,7 @@ with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
         audio_player = gr.Audio(label="أ audio",streaming=True)
     
         # Update the event binding
-        generate_button.click(generate_audio, inputs=[text_input], outputs=audio_player)
+        generate_button.click(generate_audio, inputs=[text_input,model_choice], outputs=audio_player)