Spaces:
Build error
Build error
File size: 2,031 Bytes
55e29e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
"""AudioChunk value object for streaming audio data."""
from dataclasses import dataclass
from typing import Optional
@dataclass(frozen=True)
class AudioChunk:
"""Value object representing a chunk of audio data for streaming."""
data: bytes
format: str
sample_rate: int
chunk_index: int
is_final: bool = False
timestamp: Optional[float] = None
def __post_init__(self):
"""Validate audio chunk after initialization."""
self._validate()
def _validate(self):
"""Validate audio chunk properties."""
if not isinstance(self.data, bytes):
raise TypeError("Audio data must be bytes")
if not self.data:
raise ValueError("Audio data cannot be empty")
if self.format not in ['wav', 'mp3', 'flac', 'ogg', 'raw']:
raise ValueError(f"Unsupported audio format: {self.format}")
if not isinstance(self.sample_rate, int) or self.sample_rate <= 0:
raise ValueError("Sample rate must be a positive integer")
if not isinstance(self.chunk_index, int) or self.chunk_index < 0:
raise ValueError("Chunk index must be a non-negative integer")
if not isinstance(self.is_final, bool):
raise TypeError("is_final must be a boolean")
if self.timestamp is not None:
if not isinstance(self.timestamp, (int, float)) or self.timestamp < 0:
raise ValueError("Timestamp must be a non-negative number")
@property
def size_bytes(self) -> int:
"""Get the size of audio chunk data in bytes."""
return len(self.data)
@property
def duration_estimate(self) -> float:
"""Estimate duration in seconds based on data size and sample rate."""
# Rough estimation assuming 16-bit audio (2 bytes per sample)
bytes_per_second = self.sample_rate * 2
return len(self.data) / bytes_per_second if bytes_per_second > 0 else 0.0 |