Update state_orchestrator.py

state_orchestrator.py

@@ -339,9 +339,14 @@ class StateOrchestrator:
     async def _handle_audio_playback_completed(self, event: Event):
         """Handle audio playback completion"""
         session_id = event.session_id
-        current_state = self.get_state(session_id)
+        context = self.sessions.get(session_id)
         
-        log_info(f"🎵 Audio playback completed", session_id=session_id, state=current_state)
+        if not context:
+            return
+            
+        current_state = context.state
+        
+        log_info(f"🎵 Audio playback completed", session_id=session_id, state=current_state.value)
         
         if current_state in [ConversationState.PLAYING_WELCOME, ConversationState.PLAYING_RESPONSE]:
             # Transition back to listening
@@ -354,6 +359,13 @@ class StateOrchestrator:
                 data={}
             ))
             
+            # Send STT ready signal to frontend
+            await self.event_bus.publish(Event(
+                type=EventType.STT_READY,
+                session_id=session_id,
+                data={}
+            ))
+            
     async def _handle_stt_error(self, event: Event):
         """Handle STT errors"""
         session_id = event.session_id
@@ -475,14 +487,16 @@ class StateOrchestrator:
             log_info(f"✅ State transition | session_id={session_id}, {old_state.value} → {new_state.value}")
             
             # Emit state transition event
-            await self._event_bus.publish(
-                EventType.STATE_TRANSITION,
-                {
-                    "session_id": session_id,
-                    "old_state": old_state.value,
-                    "new_state": new_state.value,
-                    "timestamp": datetime.utcnow().isoformat()
-                }
+            await self.event_bus.publish(
+                Event(
+                    type=EventType.STATE_TRANSITION,
+                    data={
+                        "old_state": old_state.value,
+                        "new_state": new_state.value,
+                        "timestamp": datetime.utcnow().isoformat()
+                    },
+                    session_id=session_id
+                )
             )
             
             return True