Update app.py

app.py

@@ -108,7 +108,7 @@ def create_demo():
         gr.Markdown(
             """
             # 🗣️ AI Voice Assistant
-            Welcome to your personal voice assistant! Simply record your voice, and I will respond with both text and speech. Powered by advanced AI models.
+            Welcome to your personal voice assistant! Simply record your voice, and I will respond with both text and speech. The assistant will automatically start listening after playing its response. Powered by advanced AI models.
             """
         )
 
@@ -139,13 +139,32 @@ def create_demo():
         audio_input.change(process_audio, inputs=[audio_input, voice_volume], outputs=[chat_output, audio_output])
         clear_button.click(lambda: (None, None), None, [chat_output, audio_output])
 
-        # JavaScript to handle autoplay and automatic submission
+        # JavaScript to handle autoplay, automatic submission, and auto-listen
         demo.load(None, js="""
             function() {
-                document.querySelector("audio").addEventListener("stop", function() {
-                    setTimeout(function() {
-                        document.querySelector('button[title="Submit"]').click();
-                    }, 500);
+                var recordButton;
+                
+                function findRecordButton() {
+                    var buttons = document.querySelectorAll('button');
+                    for (var i = 0; i < buttons.length; i++) {
+                        if (buttons[i].textContent.includes('Record from microphone')) {
+                            return buttons[i];
+                        }
+                    }
+                    return null;
+                }
+
+                function startListening() {
+                    if (!recordButton) {
+                        recordButton = findRecordButton();
+                    }
+                    if (recordButton) {
+                        recordButton.click();
+                    }
+                }
+
+                document.querySelector("audio").addEventListener("ended", function() {
+                    setTimeout(startListening, 500);
                 });
                 
                 function playAssistantAudio() {