mirror of
https://github.com/justLV/onju-v2
synced 2026-04-21 15:47:55 +00:00
f0f6e38e7c
- git mv test_client, test_mic, test_speaker into tests/ - Add tests/test_stall.py (benchmarks the Gemini stall classifier against conversational/fetch/capture/act/follow-up queries) - Add tests/test_stream.py (raw SSE chunk inspection against the agentic gateway) - Update config path resolution in the new tests to climb one level - Update README Testing section with new tests/ paths
117 lines
3.7 KiB
Python
117 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Receive and record audio from ESP32 via UDP.
|
|
Auto-detects u-law compressed or raw PCM packets.
|
|
|
|
Usage:
|
|
python test_mic_receiver.py
|
|
python test_mic_receiver.py --duration 30 --output recording.wav
|
|
"""
|
|
import argparse
|
|
import socket
|
|
import time
|
|
import wave
|
|
from collections import deque
|
|
|
|
import numpy as np
|
|
|
|
from pipeline.audio import decode_ulaw
|
|
|
|
SAMPLE_RATE = 16000
|
|
CHUNK_SIZE = 512
|
|
CHUNK_BYTES_RAW = CHUNK_SIZE * 2
|
|
CHUNK_BYTES_COMPRESSED = CHUNK_SIZE
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="ESP32 mic UDP receiver")
|
|
parser.add_argument("--port", type=int, default=3000, help="UDP port (default: 3000)")
|
|
parser.add_argument("--output", type=str, default="test_recording.wav", help="Output WAV file")
|
|
parser.add_argument("--duration", type=int, default=10, help="Recording duration in seconds (0 = infinite)")
|
|
args = parser.parse_args()
|
|
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.bind(("0.0.0.0", args.port))
|
|
sock.settimeout(1.0)
|
|
|
|
print(f"Listening on UDP port {args.port}...")
|
|
print(f"Output: {args.output}")
|
|
print("Waiting for packets...\n")
|
|
|
|
audio_frames = []
|
|
packet_count = 0
|
|
total_bytes = 0
|
|
start_time = None
|
|
last_stats_time = None
|
|
rms_history = deque(maxlen=20)
|
|
|
|
try:
|
|
while True:
|
|
try:
|
|
data, addr = sock.recvfrom(2048)
|
|
|
|
if start_time is None:
|
|
start_time = time.time()
|
|
last_stats_time = start_time
|
|
print(f"Receiving from {addr[0]}:{addr[1]}\n")
|
|
|
|
packet_count += 1
|
|
total_bytes += len(data)
|
|
|
|
if len(data) == CHUNK_BYTES_COMPRESSED:
|
|
samples = decode_ulaw(data)
|
|
mode = "u-law"
|
|
elif len(data) == CHUNK_BYTES_RAW:
|
|
samples = np.frombuffer(data, dtype=np.int16)
|
|
mode = "raw"
|
|
else:
|
|
continue
|
|
|
|
audio_frames.append(samples)
|
|
rms = np.sqrt(np.mean(samples.astype(np.float32) ** 2))
|
|
rms_history.append(rms)
|
|
|
|
current_time = time.time()
|
|
if current_time - last_stats_time >= 1.0:
|
|
elapsed = current_time - start_time
|
|
bandwidth_kbps = (total_bytes * 8) / (elapsed * 1000)
|
|
avg_rms = np.mean(list(rms_history))
|
|
print(f"\r[{elapsed:6.1f}s] Packets: {packet_count:4d} | "
|
|
f"Bandwidth: {bandwidth_kbps:5.1f} kbps | "
|
|
f"RMS: {avg_rms:6.0f} | "
|
|
f"Mode: {mode}",
|
|
end="", flush=True)
|
|
last_stats_time = current_time
|
|
|
|
if args.duration > 0 and (current_time - start_time) >= args.duration:
|
|
print("\n\nRecording duration reached.")
|
|
break
|
|
|
|
except socket.timeout:
|
|
if start_time is not None:
|
|
print("\n\nTimeout - no packets for 1 second")
|
|
break
|
|
continue
|
|
|
|
except KeyboardInterrupt:
|
|
print("\n\nInterrupted")
|
|
|
|
if not audio_frames:
|
|
print("No audio received!")
|
|
return
|
|
|
|
audio_data = np.concatenate(audio_frames)
|
|
|
|
with wave.open(args.output, "wb") as wf:
|
|
wf.setnchannels(1)
|
|
wf.setsampwidth(2)
|
|
wf.setframerate(SAMPLE_RATE)
|
|
wf.writeframes(audio_data.tobytes())
|
|
|
|
duration = len(audio_data) / SAMPLE_RATE
|
|
transmitted_kb = total_bytes / 1024
|
|
print(f"\nSaved {duration:.1f}s to {args.output} ({transmitted_kb:.1f} KB received, {packet_count} packets)")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|