File size: 935 Bytes
c3eb7ea | 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 | from enum import Enum
from pydantic import BaseModel, Field
class EmotionalTone(str, Enum):
neutral = "neutral"
satisfied = "satisfied"
frustrated = "frustrated"
upset = "upset"
distressed = "distressed"
class EmotionalIntensity(str, Enum):
low = "low"
medium = "medium"
high = "high"
class NoiseSeverity(str, Enum):
none = "none"
low = "low"
medium = "medium"
high = "high"
class AudioQuality(str, Enum):
clear = "clear"
slightly_impaired = "slightly_impaired"
severely_impaired = "severely_impaired"
class Result(BaseModel):
emotional_tone: EmotionalTone
emotional_intensity: EmotionalIntensity
background_noise_present: bool
background_noise_type: str = ""
background_noise_severity: NoiseSeverity
audio_quality: AudioQuality
speaker_overlap_present: bool
long_silence_present: bool
confidence: float = Field(ge=0.0, le=1.0)
|