process_audio_chunk result is not sent back to the client

inference.py

@@ -245,12 +245,27 @@ async def ws_inference(websocket: WebSocket):
                     # Process audio chunk
                     result = diart.process_audio_chunk(data)
                     
+                    # Send processing result back to client
+                    if result:
+                        result_message = json.dumps({
+                            "type": "processing_result",
+                            "timestamp": time.time(),
+                            "data": result
+                        })
+                        await websocket.send_text(result_message)
+                    
                     # Log processing result (optional)
                     if connection_stats["total_audio_chunks"] % 100 == 0:  # Log every 100 chunks
                         logger.debug(f"Processed {connection_stats['total_audio_chunks']} audio chunks")
                 
                 elif not diart:
                     logger.warning("Received audio data but diarization system not initialized")
+                    error_message = json.dumps({
+                        "type": "error",
+                        "message": "Diarization system not initialized",
+                        "timestamp": time.time()
+                    })
+                    await websocket.send_text(error_message)
                     
             except Exception as e:
                 logger.error(f"Error processing audio chunk: {e}")
@@ -258,6 +273,7 @@ async def ws_inference(websocket: WebSocket):
                 error_message = json.dumps({
                     "type": "error",
                     "message": "Error processing audio",
+                    "details": str(e),
                     "timestamp": time.time()
                 })
                 await websocket.send_text(error_message)

ui.py

@@ -187,11 +187,41 @@ def build_ui():
                                         }
                                         break;
                                         
+                                    case 'processing_result':
+                                        // Handle individual audio chunk processing result
+                                        console.log('Processing result:', message.data);
+                                        
+                                        // Update status info if needed
+                                        if (message.data && message.data.status === "processed") {
+                                            const statusElem = document.getElementById('status-text');
+                                            if (statusElem) {
+                                                const speakerId = message.data.speaker_id !== undefined ? 
+                                                    `Speaker ${message.data.speaker_id + 1}` : '';
+                                                
+                                                if (speakerId) {
+                                                    statusElem.textContent = `Connected - ${speakerId} active`;
+                                                }
+                                            }
+                                        } else if (message.data && message.data.status === "error") {
+                                            updateStatus('error', message.data.message || 'Processing error');
+                                        }
+                                        break;
+                                        
                                     case 'connection':
                                         console.log('Connection status:', message.status);
                                         updateStatus(message.status === 'connected' ? 'connected' : 'warning');
                                         break;
                                         
+                                    case 'connection_established':
+                                        console.log('Connection established:', message);
+                                        updateStatus('connected');
+                                        
+                                        // If initial conversation is provided, display it
+                                        if (message.conversation) {
+                                            document.getElementById("conversation").innerHTML = message.conversation;
+                                        }
+                                        break;
+                                        
                                     case 'conversation_update':
                                         if (message.conversation_html) {
                                             document.getElementById("conversation").innerHTML = message.conversation_html;