Update app.py

app.py

@@ -37,51 +37,55 @@ def   get_answer_ai(text):
           response = AI.send_message(text)
           return response.text
 
+with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
+    with gr.Tab("محادثة صوتية بالذكاء الاصطناعي باللهجة السعودية"):
+        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.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="رد الذكاء الاصطناعي النصي")
 
-text_input = gr.Textbox(label="ُEnter any text")
-user_audio = gr.Audio(label="Your Speech")
-ai_audio = gr.Audio(label="AI Response Speech")
-ai_text = gr.Textbox(label="AI Response Text")
+        btn = gr.Button("إرسال")
 
-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"],  # Replace with your model names
-    label="Choose Model",
-    value="asg2024/vits-ar-sa"  # Default selection
-)
-   
-def reverse_audio(text,model_choice):
-    API_URL = f"https://api-inference.huggingface.co/models/{model_choice}" 
-    text_answer=get_answer_ai(text)
-    dataq = query(text,API_URL)
-    text_answer=remove_extra_spaces(text_answer)
-    dataa = query(text_answer,API_URL)
-    
-    # data_enhanc = remove_noise_enhance(data)
+        # Use a single button to trigger both functionalities
+        def process_audio(text, model_choice, generate_user_audio=True):
+            API_URL = f"https://api-inference.huggingface.co/models/{model_choice}"
+            text_answer = get_answer_ai(text)
+            text_answer = remove_extra_spaces(text_answer)
+            data_ai = query(text_answer, API_URL)
+            if generate_user_audio:  # Generate user audio if needed
+                data_user = query(text, API_URL)
+                return data_user, data_ai, text_answer
+            else:
+                return None, data_ai, text_answer  # Return None for user_audio
 
-    
-    return dataq,dataa,text_answer#(16000, np.flipud(data))
+        btn.click(
+            process_audio,  # Call the combined function
+            inputs=[text_input, model_choices],
+            outputs=[user_audio, ai_audio, ai_text],
+        )
 
-def reverse_audio_2(text,model_choice):
-    API_URL = f"https://api-inference.huggingface.co/models/{model_choice}" 
-    text_answer=get_answer_ai(text)
-    text_answer=remove_extra_spaces(text_answer)
-    dataa = query(text_answer,API_URL)
-    return dataa
-    
-demo = gr.Interface(
-    fn=reverse_audio,
-    inputs=[text_input,model_choices],
-    outputs=[user_audio,ai_audio,ai_text],
-    title=" محادثة صوتية بالذكاء الاصطناعي باللهجة السعودية "
-)
+        # Additional button to generate only AI audio
+        with gr.Row():
+            btn_ai_only = gr.Button("توليد رد الذكاء الاصطناعي فقط")
+            btn_ai_only.click(
+                lambda text, model_choice: process_audio(text, model_choice, False),
+                inputs=[text_input, model_choices],
+                outputs=[ai_audio],
+            )
 
-# panal_2 = gr.Interface(
-#     fn=reverse_audio_2,
-#     inputs=[text_input,model_choices],
-#     outputs=[ai_audio],
-#     title=" محادثة صوتية بالذكاء الاصطناعي باللهجة السعودية "
-# )
-
-# demo = gr.TabbedInterface([panal_1, panal_2],['panal_1', 'panal_2'])
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()