diff --git a/src/f5_tts/infer/infer_gradio.py b/src/f5_tts/infer/infer_gradio.py
index 87d1276..edcaca5 100644
--- a/src/f5_tts/infer/infer_gradio.py
+++ b/src/f5_tts/infer/infer_gradio.py
@@ -567,7 +567,7 @@ Have a conversation with an AI using your reference voice!
 
             last_user_message, last_ai_response = history[-1]
             if not last_ai_response:
-                return None
+                return None, gr.update(interactive=True)
 
             audio_result, _ = infer(
                 ref_audio,
@@ -579,7 +579,7 @@ Have a conversation with an AI using your reference voice!
                 speed=1.0,
                 show_info=print,  # show_info=print no pull to top when generating
             )
-            return audio_result
+            return audio_result, gr.update(interactive=True)
 
         def clear_conversation():
             """Reset the conversation"""
@@ -603,7 +603,7 @@ Have a conversation with an AI using your reference voice!
         ).then(
             generate_audio_response,
             inputs=[chatbot_interface, ref_audio_chat, ref_text_chat, model_choice_chat, remove_silence_chat],
-            outputs=audio_output_chat,
+            outputs=[audio_output_chat, audio_input_chat],
         )
 
         # Handle clear button