Update app.py

app.py

@@ -127,7 +127,10 @@ def generate_audio(text,name_model,speaker_id=None):
     with torch.no_grad():
         for chunk in _inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= speaker_embeddings,chunk_size=64):
             yield  16000,chunk.squeeze().cpu().numpy()#.astype(np.int16).tobytes()    
-
+def generate_audio_ai(text,name_model):
+    text_answer = get_answer_ai(text)
+    text_answer = remove_extra_spaces(text_answer)
+    return yield generate_audio(text_answer,name_model)
 def remove_extra_spaces(text):
   """
   Removes extra spaces between words in a string.
@@ -191,9 +194,9 @@ with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
             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 = generate_audio(text_answer,model_choice)#query(text_answer, API_URL)
+            data_ai = genrate_speech(text_answer,model_choice)#query(text_answer, API_URL)
             if generate_user_audio:  # Generate user audio if needed
-                data_user =generate_audio(text,model_choice)# query(text, API_URL)
+                data_user =genrate_speech(text,model_choice)# query(text, API_URL)
                 return data_user, data_ai, text_answer
             else:
                 return  data_ai  # Return None for user_audio
@@ -207,7 +210,7 @@ with gr.Blocks() as demo:  # Use gr.Blocks to wrap the entire interface
         # Additional button to generate only AI audio
       
         btn_ai_only.click(
-                lambda text, model_choice: process_audio(text, model_choice, False),
+                generate_audio_ai,
                 inputs=[text_input, model_choices],
                 outputs=[ai_audio],
             )