Update app.py

app.py

@@ -346,6 +346,9 @@ def clear_all():
     # For chunk duration (reset to 10)
     outputs.append(10)
     
+    # For example selector (reset to None)
+    outputs.append(None)
+    
     return outputs
 
 # Function to load example audio
@@ -483,8 +486,14 @@ with gr.Blocks(title="Affective Virtual Environments - Chunked Processing") as i
     # Function to handle example selection
     def load_example(example_name):
         if not example_name:
-            return None
-        return load_example_audio(example_name)
+            return None, None  # Return None for both audio and example selector
+        
+        # Get the path to the example audio file
+        example_path = load_example_audio(example_name)
+        
+        # Return the example path to update the audio component
+        # The waveform will be automatically displayed by Gradio's Audio component
+        return example_path, example_name
     
     # Set up the button click
     process_btn.click(
@@ -509,14 +518,14 @@ with gr.Blocks(title="Affective Virtual Environments - Chunked Processing") as i
             container['sentiment'],
             container['image'],
             container['music']
-        ]] + [loading_indicator] + [chunk_duration_input]
+        ]] + [loading_indicator] + [chunk_duration_input] + [example_selector]
     )
     
     # Set up the example loading button
     load_example_btn.click(
         fn=load_example,
         inputs=[example_selector],
-        outputs=[audio_input]
+        outputs=[audio_input, example_selector]
     )
 
 interface.launch()