audio_processor

shared.py

@@ -578,7 +578,7 @@ class RealtimeSpeakerDiarization:
             speaker_id = self.speaker_detector.current_speaker
             similarity = 1.0
             
-            if len(self.audio_processor.audio_buffer) % (SAMPLE_RATE // 2) == 0:  # Every 0.5 seconds
+            if len(self.audio_processor.audio_buffer) >= SAMPLE_RATE and (len(self.audio_processor.audio_buffer) - SAMPLE_RATE) % (SAMPLE_RATE // 2)==0:
                 embedding = self.audio_processor.extract_embedding_from_buffer()
                 if embedding is not None:
                     speaker_id, similarity = self.speaker_detector.add_embedding(embedding)

test_websocket.py

@@ -1,81 +1,26 @@
+import numpy as np
 import asyncio
 import websockets
 import json
-import numpy as np
-import time
 
-# Configuration
-HF_SPACE_URL = "https://androidguy-speaker-diarization.hf.space"
-RENDER_SIGNAL_URL = "wss://render-signal-audio.onrender.com/stream"
-WS_INFERENCE_ENDPOINT = f"wss://{HF_SPACE_URL.replace('https://', '')}/ws_inference"
-WS_RELAY_ENDPOINT = "wss://render-signal-audio.onrender.com/ws_relay"
+async def test_ws():
+    uri = "wss://androidguy-speaker-diarization.hf.space/ws_inference"
+    async with websockets.connect(uri) as websocket:
+        print("Connected")
+
+        print(await websocket.recv())  # connection_established
+
+        for i in range(20):
+            # Generate random "noise" audio instead of silence
+            audio = (np.random.randn(3200) * 3000).astype(np.int16)
+            await websocket.send(audio.tobytes())
+            print(f"Sent audio chunk {i+1}/20")
 
-async def test_websocket_connections():
-    print(f"Testing WebSocket connections...")
-    
-    # Test HF Space /ws_inference connection
-    print(f"Connecting to {WS_INFERENCE_ENDPOINT}...")
-    try:
-        async with websockets.connect(WS_INFERENCE_ENDPOINT, ping_interval=30) as ws:
-            print("Connected to HF Space inference WebSocket!")
-            
-            # Send test data (simulated audio)
-            test_audio = np.zeros(1600, dtype=np.float32).tobytes()
-            await ws.send(test_audio)
-            print(f"Sent test audio data: {len(test_audio)} bytes")
-            
-            # Wait for response
-            for _ in range(5):  # Try to get 5 responses
-                try:
-                    response = await asyncio.wait_for(ws.recv(), timeout=3.0)
-                    if isinstance(response, str):
-                        try:
-                            json_data = json.loads(response)
-                            print(f"Received JSON response: {json.dumps(json_data, indent=2)}")
-                        except json.JSONDecodeError:
-                            print(f"Received non-JSON response: {response[:500]}...")
-                    else:
-                        print(f"Received binary data: {len(response)} bytes")
-                except asyncio.TimeoutError:
-                    print("No response received within timeout")
-                    break
-                
-    except Exception as e:
-        print(f"Error connecting to HF Space WebSocket: {e}")
-        
-    # Test Render Signal WebSocket relay connection
-    print(f"\nConnecting to {WS_RELAY_ENDPOINT}...")
-    try:
-        async with websockets.connect(WS_RELAY_ENDPOINT) as ws:
-            print("Connected to Render Signal WebSocket relay!")
-            
-            # Send test configuration message
-            test_config = {
-                "type": "config",
-                "client_id": f"test_client_{time.time()}",
-                "test": True
-            }
-            await ws.send(json.dumps(test_config))
-            print(f"Sent test config: {test_config}")
-            
-            # Wait for responses
-            for _ in range(5):  # Try to get 5 responses
-                try:
-                    response = await asyncio.wait_for(ws.recv(), timeout=3.0)
-                    if isinstance(response, str):
-                        try:
-                            json_data = json.loads(response)
-                            print(f"Received JSON response: {json.dumps(json_data, indent=2)}")
-                        except json.JSONDecodeError:
-                            print(f"Received non-JSON response: {response[:500]}...")
-                    else:
-                        print(f"Received binary data: {len(response)} bytes")
-                except asyncio.TimeoutError:
-                    print("No response received within timeout")
-                    break
-    except Exception as e:
-        print(f"Error connecting to Render Signal WebSocket: {e}")
+        try:
+            while True:
+                res = await asyncio.wait_for(websocket.recv(), timeout=10)
+                print("Received:", json.dumps(json.loads(res), indent=2))
+        except asyncio.TimeoutError:
+            print("[ERROR] No more responses (timeout)")
 
-# Run tests for direct WebSocket communication
-if __name__ == "__main__":
-    asyncio.run(test_websocket_connections())
+asyncio.run(test_ws())