Update app.py

app.py

@@ -158,18 +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.Column():
-                model_choices = gr.Dropdown(
-                    choices=[
-                        "asg2024/vits-ar-sa",
-                        "asg2024/vits-ar-sa-huba",
-                        "asg2024/vits-ar-sa-ms",
-                        "asg2024/vits-ar-sa-magd",
-                        "asg2024/vits-ar-sa-fahd",
-                    ],
-                    label="اختر النموذج",
-                    value="asg2024/vits-ar-sa",
-                )
+    
     with gr.Tab("Chat AI "):
         gr.Markdown("## AI: محادثة صوتية بالذكاء الاصطناعي باللهجة السعودية")
         with gr.Row(): # Arrange input/output components side-by-side
@@ -180,7 +169,18 @@ with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
                     btn = gr.Button("إرسال")
                     btn_ai_only = gr.Button("توليد رد الذكاء الاصطناعي فقط")
                     
-            
+                with gr.Column():
+                        model_choices = gr.Dropdown(
+                            choices=[
+                                "asg2024/vits-ar-sa",
+                                "asg2024/vits-ar-sa-huba",
+                                "asg2024/vits-ar-sa-ms",
+                                "asg2024/vits-ar-sa-magd",
+                                "asg2024/vits-ar-sa-fahd",
+                            ],
+                            label="اختر النموذج",
+                            value="asg2024/vits-ar-sa",
+                        )
                 ai_audio = gr.Audio(label="رد الذكاء الاصطناعي الصوتي")
                 ai_text = gr.Textbox(label="رد الذكاء الاصطناعي النصي")
 
@@ -216,12 +216,24 @@ with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
         with gr.Row():
             text_input = gr.Textbox(label="أدخل النص هنا")
             speaker_id_input = gr.Number(label="معرّف المتحدث (اختياري)", interactive=True)
+            with gr.Column():
+                        model_choices2 = gr.Dropdown(
+                            choices=[
+                                "asg2024/vits-ar-sa",
+                                "asg2024/vits-ar-sa-huba",
+                                "asg2024/vits-ar-sa-ms",
+                                "asg2024/vits-ar-sa-magd",
+                                "asg2024/vits-ar-sa-fahd",
+                            ],
+                            label="اختر النموذج",
+                            value="asg2024/vits-ar-sa",
+                        )
         generate_button = gr.Button("توليد وتشغيل الصوت")
       
         audio_player = gr.Audio(label="أ audio",streaming=True)
     
         # Update the event binding
-    generate_button.click(generate_audio, inputs=[text_input,model_choice], outputs=audio_player)
+        generate_button.click(generate_audio, inputs=[text_input,model_choices2], outputs=audio_player)