Update app.py

app.py

@@ -110,21 +110,17 @@ def process_audio(audio, state):
     
     return gr.update(), state
 
-def cleanup(state):
-    """Cleanup when the interface is closed"""
-    if state and state.get('is_running', False):
-        audio_processor.stop()
-        state['is_running'] = False
-    return state
+# Define event handler for cleanup
+def on_close():
+    audio_processor.stop()
 
 with gr.Blocks() as demo:
     gr.Markdown("# Real-time Audio Emotion Detection")
     gr.Markdown("Speak into your microphone. Emotions are detected in 3-second chunks.")
     
-    state = gr.State(None)
+    state = gr.State({'is_running': False})
     output = gr.Textbox(label="Detected Emotions (Last 5 chunks)", lines=2)
     
-    # Updated Audio component syntax for Gradio 4.39.0
     audio_input = gr.Audio(
         sources=["microphone"],
         type="numpy",
@@ -139,9 +135,10 @@ with gr.Blocks() as demo:
         outputs=[output, state],
         show_progress=False
     )
-    
-    demo.load(lambda: None, None, state)
-    demo.close(cleanup, state, state)
 
-# Launch with a smaller queue size for faster updates
-demo.queue(max_size=10).launch(share=True)
+# Launch with cleanup handling
+demo.queue(max_size=10).launch(share=True, prevent_thread_lock=True)
+
+# Register cleanup
+import atexit
+atexit.register(on_close)