Spaces:
Sleeping
Sleeping
File size: 2,351 Bytes
06a06a0 d1c4aa1 06a06a0 d1c4aa1 06a06a0 d1c4aa1 06a06a0 d1c4aa1 |
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 |
"""
Pydantic models for request/response validation and API documentation.
"""
from typing import Optional
from pydantic import BaseModel, Field
class TTSRequest(BaseModel):
"""Request model for TTS generation with optional voice cloning."""
text: str = Field(..., description="Text to convert to speech", max_length=5000)
voice_prompt_base64: Optional[str] = Field(None, description="Base64 encoded voice prompt audio")
class FullTextTTSRequest(BaseModel):
"""Request model for full-text TTS generation with server-side processing."""
text: str = Field(..., description="Full text to convert to speech (any length)")
voice_prompt_base64: Optional[str] = Field(None, description="Base64 encoded voice prompt audio")
max_chunk_size: Optional[int] = Field(800, description="Maximum characters per chunk")
silence_duration: Optional[float] = Field(0.5, description="Silence duration between chunks (seconds)")
fade_duration: Optional[float] = Field(0.1, description="Fade in/out duration (seconds)")
overlap_sentences: Optional[int] = Field(0, description="Number of sentences to overlap between chunks")
class TTSResponse(BaseModel):
"""Response model for TTS generation with JSON output."""
success: bool = Field(..., description="Whether the request was successful")
message: str = Field(..., description="Status message")
audio_base64: Optional[str] = Field(None, description="Base64 encoded audio data")
duration_seconds: Optional[float] = Field(None, description="Duration of generated audio in seconds")
class FullTextTTSResponse(BaseModel):
"""Response model for full-text TTS generation."""
success: bool = Field(..., description="Whether the request was successful")
message: str = Field(..., description="Status message")
audio_base64: Optional[str] = Field(None, description="Base64 encoded audio data")
duration_seconds: Optional[float] = Field(None, description="Duration of generated audio in seconds")
processing_info: Optional[dict] = Field(None, description="Information about the processing (chunks, etc.)")
class HealthResponse(BaseModel):
"""Response model for health check endpoint."""
status: str = Field(..., description="Service status")
model_loaded: bool = Field(..., description="Whether the TTS model is loaded")
|