Update app.py

app.py

@@ -120,7 +120,7 @@ def create_demo():
 
             with gr.Column(scale=1):
                 chat_output = gr.Textbox(label="💬 AI Response", elem_id="chat-output", lines=5, interactive=False)
-                audio_output = gr.Audio(label="🔊 AI Voice Response", autoplay=True, elem_id="audio-output")
+                audio_output = gr.Audio(label="🔊 AI Voice Response", elem_id="audio-output")
 
         # Add some spacing and a divider
         gr.Markdown("---")
@@ -129,35 +129,42 @@ def create_demo():
         def process_audio(audio, volume):
             logging.info(f"Received audio: {audio}")
             if audio is None:
-                return "No audio detected. Please try recording again.", None, None
+                return "No audio detected. Please try recording again.", None
             response, audio_path = transcribe_and_chat(audio)
             # Adjust volume for the response audio
             adjusted_audio_path = asyncio.run(text_to_speech_stream(response, volume))
             logging.info(f"Response: {response}, Audio path: {adjusted_audio_path}")
-            return response, adjusted_audio_path, None  # Return None to clear the audio input
+            return response, adjusted_audio_path
 
-        audio_input.change(process_audio, inputs=[audio_input, voice_volume], outputs=[chat_output, audio_output, audio_input])
+        audio_input.change(process_audio, inputs=[audio_input, voice_volume], outputs=[chat_output, audio_output])
         clear_button.click(lambda: (None, None, None), None, [chat_output, audio_output, audio_input])
 
-        # JavaScript to handle autoplay and automatic submission
+        # JavaScript to handle autoplay and automatic clearing
         demo.load(None, js="""
             function() {
-                document.querySelector("audio").addEventListener("stop", function() {
-                    setTimeout(function() {
-                        document.querySelector('button[title="Submit"]').click();
-                    }, 500);
-                });
-                
+                let isPlaying = false;
+
                 function playAssistantAudio() {
                     var audioElements = document.querySelectorAll('audio');
                     if (audioElements.length > 1) {
                         var assistantAudio = audioElements[1];
                         if (assistantAudio) {
+                            isPlaying = true;
                             assistantAudio.play();
+                            assistantAudio.onended = function() {
+                                isPlaying = false;
+                                clearAllInputs();
+                            };
                         }
                     }
                 }
 
+                function clearAllInputs() {
+                    if (!isPlaying) {
+                        document.querySelector('#clear-button').click();
+                    }
+                }
+
                 document.addEventListener('gradioAudioLoaded', function(event) {
                     playAssistantAudio();
                 });
@@ -170,7 +177,9 @@ def create_demo():
                 document.addEventListener("visibilitychange", function() {
                     var audioElements = document.querySelectorAll('audio');
                     audioElements.forEach(function(audio) {
-                        audio.play();
+                        if (isPlaying) {
+                            audio.play();
+                        }
                     });
                 });
             }