refactor: rename canto-backend → backend, canto-frontend → frontend
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
0
backend/schemas/__init__.py
Normal file
0
backend/schemas/__init__.py
Normal file
165
backend/schemas/audiobook.py
Normal file
165
backend/schemas/audiobook.py
Normal file
@@ -0,0 +1,165 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional, List, Dict, Any
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
|
||||
class AudiobookProjectCreate(BaseModel):
|
||||
title: str
|
||||
source_type: str
|
||||
source_text: Optional[str] = None
|
||||
|
||||
|
||||
class SynopsisGenerationRequest(BaseModel):
|
||||
title: str
|
||||
genre: str
|
||||
subgenre: str = ""
|
||||
protagonist_type: str = ""
|
||||
tone: str = ""
|
||||
conflict_scale: str = ""
|
||||
num_characters: int = 5
|
||||
num_chapters: int = 8
|
||||
violence_level: int = 0
|
||||
eroticism_level: int = 0
|
||||
|
||||
|
||||
class ScriptGenerationRequest(BaseModel):
|
||||
title: str
|
||||
genre: str
|
||||
subgenre: str = ""
|
||||
premise: str
|
||||
style: str = ""
|
||||
num_characters: int = 5
|
||||
num_chapters: int = 8
|
||||
violence_level: int = 0
|
||||
eroticism_level: int = 0
|
||||
|
||||
|
||||
class AudiobookProjectResponse(BaseModel):
|
||||
id: int
|
||||
user_id: int
|
||||
title: str
|
||||
source_type: str
|
||||
status: str
|
||||
llm_model: Optional[str] = None
|
||||
error_message: Optional[str] = None
|
||||
script_config: Optional[Dict[str, Any]] = None
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
segment_total: int = 0
|
||||
segment_done: int = 0
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class AudiobookCharacterResponse(BaseModel):
|
||||
id: int
|
||||
project_id: int
|
||||
name: str
|
||||
gender: Optional[str] = None
|
||||
description: Optional[str] = None
|
||||
instruct: Optional[str] = None
|
||||
voice_design_id: Optional[int] = None
|
||||
voice_design_name: Optional[str] = None
|
||||
voice_design_speaker: Optional[str] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class AudiobookChapterResponse(BaseModel):
|
||||
id: int
|
||||
project_id: int
|
||||
chapter_index: int
|
||||
title: Optional[str] = None
|
||||
status: str
|
||||
error_message: Optional[str] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class AudiobookProjectDetail(AudiobookProjectResponse):
|
||||
characters: List[AudiobookCharacterResponse] = []
|
||||
chapters: List[AudiobookChapterResponse] = []
|
||||
|
||||
|
||||
class ContinueScriptRequest(BaseModel):
|
||||
additional_chapters: int = 4
|
||||
|
||||
|
||||
class AudiobookAnalyzeRequest(BaseModel):
|
||||
turbo: bool = False
|
||||
|
||||
|
||||
class AudiobookGenerateRequest(BaseModel):
|
||||
chapter_index: Optional[int] = None
|
||||
force: bool = False
|
||||
|
||||
|
||||
class AudiobookCharacterUpdate(BaseModel):
|
||||
voice_design_id: int
|
||||
|
||||
|
||||
class AudiobookCharacterEdit(BaseModel):
|
||||
name: Optional[str] = None
|
||||
gender: Optional[str] = None
|
||||
description: Optional[str] = None
|
||||
instruct: Optional[str] = None
|
||||
voice_design_id: Optional[int] = None
|
||||
|
||||
|
||||
class AudiobookSegmentResponse(BaseModel):
|
||||
id: int
|
||||
project_id: int
|
||||
chapter_index: int
|
||||
segment_index: int
|
||||
character_id: int
|
||||
character_name: Optional[str] = None
|
||||
text: str
|
||||
emo_text: Optional[str] = None
|
||||
emo_alpha: Optional[float] = None
|
||||
audio_path: Optional[str] = None
|
||||
status: str
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class AudiobookSegmentUpdate(BaseModel):
|
||||
text: str
|
||||
emo_text: Optional[str] = None
|
||||
emo_alpha: Optional[float] = None
|
||||
|
||||
|
||||
class LLMConfigUpdate(BaseModel):
|
||||
base_url: str
|
||||
api_key: str
|
||||
model: str
|
||||
|
||||
|
||||
class LLMConfigResponse(BaseModel):
|
||||
base_url: Optional[str] = None
|
||||
model: Optional[str] = None
|
||||
has_key: bool
|
||||
|
||||
|
||||
class NsfwSynopsisGenerationRequest(BaseModel):
|
||||
title: str
|
||||
genre: str
|
||||
subgenre: str = ""
|
||||
protagonist_type: str = ""
|
||||
tone: str = ""
|
||||
conflict_scale: str = ""
|
||||
num_characters: int = 5
|
||||
num_chapters: int = 8
|
||||
violence_level: int = 0
|
||||
eroticism_level: int = 0
|
||||
|
||||
|
||||
class NsfwScriptGenerationRequest(BaseModel):
|
||||
title: str
|
||||
genre: str
|
||||
subgenre: str = ""
|
||||
premise: str
|
||||
style: str = ""
|
||||
num_characters: int = 5
|
||||
num_chapters: int = 8
|
||||
violence_level: int = 0
|
||||
eroticism_level: int = 0
|
||||
15
backend/schemas/cache.py
Normal file
15
backend/schemas/cache.py
Normal file
@@ -0,0 +1,15 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional, Dict, Any
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
class CacheEntry(BaseModel):
|
||||
id: int
|
||||
user_id: int
|
||||
ref_audio_hash: str
|
||||
cache_path: str
|
||||
meta_data: Optional[Dict[str, Any]] = None
|
||||
created_at: datetime
|
||||
last_accessed: datetime
|
||||
access_count: int
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
25
backend/schemas/job.py
Normal file
25
backend/schemas/job.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional, Dict, Any, List
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
class JobBase(BaseModel):
|
||||
job_type: str
|
||||
|
||||
class JobCreate(JobBase):
|
||||
input_data: Dict[str, Any]
|
||||
|
||||
class Job(JobBase):
|
||||
id: int
|
||||
user_id: int
|
||||
status: str
|
||||
output_path: Optional[str] = None
|
||||
download_url: Optional[str] = None
|
||||
error_message: Optional[str] = None
|
||||
created_at: datetime
|
||||
completed_at: Optional[datetime] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
class JobList(BaseModel):
|
||||
total: int
|
||||
jobs: List[Job]
|
||||
60
backend/schemas/tts.py
Normal file
60
backend/schemas/tts.py
Normal file
@@ -0,0 +1,60 @@
|
||||
from typing import Optional, List
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
class TTSRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
ref_audio: Optional[str] = None
|
||||
ref_text: Optional[str] = None
|
||||
language: str = Field(default="en")
|
||||
speed: float = Field(default=1.0, ge=0.5, le=2.0)
|
||||
|
||||
class TTSResponse(BaseModel):
|
||||
job_id: int
|
||||
status: str
|
||||
audio_url: Optional[str] = None
|
||||
|
||||
|
||||
class CustomVoiceRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
language: str = Field(default="Auto")
|
||||
speaker: str
|
||||
instruct: Optional[str] = Field(default="")
|
||||
max_new_tokens: Optional[int] = Field(default=2048, ge=128, le=4096)
|
||||
temperature: Optional[float] = Field(default=0.9, ge=0.1, le=2.0)
|
||||
top_k: Optional[int] = Field(default=50, ge=1, le=100)
|
||||
top_p: Optional[float] = Field(default=1.0, ge=0.0, le=1.0)
|
||||
repetition_penalty: Optional[float] = Field(default=1.05, ge=1.0, le=2.0)
|
||||
backend: Optional[str] = Field(default=None)
|
||||
|
||||
|
||||
class VoiceDesignRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
language: str = Field(default="Auto")
|
||||
instruct: Optional[str] = Field(default=None, min_length=1)
|
||||
saved_design_id: Optional[int] = None
|
||||
max_new_tokens: Optional[int] = Field(default=2048, ge=128, le=4096)
|
||||
temperature: Optional[float] = Field(default=0.9, ge=0.1, le=2.0)
|
||||
top_k: Optional[int] = Field(default=50, ge=1, le=100)
|
||||
top_p: Optional[float] = Field(default=1.0, ge=0.0, le=1.0)
|
||||
repetition_penalty: Optional[float] = Field(default=1.05, ge=1.0, le=2.0)
|
||||
backend: Optional[str] = Field(default=None)
|
||||
|
||||
|
||||
class VoiceCloneRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
language: str = Field(default="Auto")
|
||||
ref_text: Optional[str] = Field(default=None, max_length=500)
|
||||
use_cache: bool = Field(default=True)
|
||||
x_vector_only_mode: bool = Field(default=False)
|
||||
max_new_tokens: Optional[int] = Field(default=2048, ge=128, le=4096)
|
||||
temperature: Optional[float] = Field(default=0.9, ge=0.1, le=2.0)
|
||||
top_k: Optional[int] = Field(default=50, ge=1, le=100)
|
||||
top_p: Optional[float] = Field(default=1.0, ge=0.0, le=1.0)
|
||||
repetition_penalty: Optional[float] = Field(default=1.05, ge=1.0, le=2.0)
|
||||
|
||||
|
||||
class IndexTTS2FromDesignRequest(BaseModel):
|
||||
voice_design_id: int
|
||||
text: str
|
||||
emo_text: Optional[str] = None
|
||||
emo_alpha: float = Field(default=0.6, ge=0.0, le=1.0)
|
||||
128
backend/schemas/user.py
Normal file
128
backend/schemas/user.py
Normal file
@@ -0,0 +1,128 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
from pydantic import BaseModel, EmailStr, Field, field_validator, model_validator, ConfigDict
|
||||
import re
|
||||
|
||||
class UserBase(BaseModel):
|
||||
username: str = Field(..., min_length=3, max_length=50)
|
||||
email: EmailStr
|
||||
|
||||
@field_validator('username')
|
||||
@classmethod
|
||||
def validate_username(cls, v: str) -> str:
|
||||
if not re.match(r'^[a-zA-Z0-9_-]+$', v):
|
||||
raise ValueError('Username must contain only alphanumeric characters, underscores, and dashes')
|
||||
return v
|
||||
|
||||
class UserCreate(UserBase):
|
||||
password: str = Field(..., min_length=8, max_length=128)
|
||||
|
||||
@field_validator('password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: str) -> str:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
class User(UserBase):
|
||||
id: int
|
||||
is_active: bool
|
||||
is_superuser: bool
|
||||
can_use_local_model: bool
|
||||
can_use_nsfw: bool = False
|
||||
created_at: datetime
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
class UserCreateByAdmin(UserBase):
|
||||
password: str = Field(..., min_length=8, max_length=128)
|
||||
is_superuser: bool = False
|
||||
can_use_local_model: bool = False
|
||||
can_use_nsfw: bool = False
|
||||
|
||||
@field_validator('password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: str) -> str:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
class UserUpdate(BaseModel):
|
||||
username: Optional[str] = Field(None, min_length=3, max_length=50)
|
||||
email: Optional[EmailStr] = None
|
||||
password: Optional[str] = Field(None, min_length=8, max_length=128)
|
||||
is_active: Optional[bool] = None
|
||||
is_superuser: Optional[bool] = None
|
||||
can_use_local_model: Optional[bool] = None
|
||||
can_use_nsfw: Optional[bool] = None
|
||||
|
||||
@field_validator('username')
|
||||
@classmethod
|
||||
def validate_username(cls, v: Optional[str]) -> Optional[str]:
|
||||
if v is not None and not re.match(r'^[a-zA-Z0-9_-]+$', v):
|
||||
raise ValueError('Username must contain only alphanumeric characters, underscores, and dashes')
|
||||
return v
|
||||
|
||||
@field_validator('password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: Optional[str]) -> Optional[str]:
|
||||
if v is not None:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
class UserListResponse(BaseModel):
|
||||
users: list[User]
|
||||
total: int
|
||||
skip: int
|
||||
limit: int
|
||||
|
||||
class Token(BaseModel):
|
||||
access_token: str
|
||||
token_type: str
|
||||
|
||||
class TokenData(BaseModel):
|
||||
username: Optional[str] = None
|
||||
|
||||
class PasswordChange(BaseModel):
|
||||
current_password: str = Field(..., min_length=1)
|
||||
new_password: str = Field(..., min_length=8, max_length=128)
|
||||
confirm_password: str = Field(..., min_length=8, max_length=128)
|
||||
|
||||
@field_validator('new_password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: str) -> str:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
@model_validator(mode='after')
|
||||
def passwords_match(self) -> 'PasswordChange':
|
||||
if self.new_password != self.confirm_password:
|
||||
raise ValueError('Passwords do not match')
|
||||
return self
|
||||
|
||||
class UserPreferences(BaseModel):
|
||||
default_backend: str = Field(default="local", pattern="^local$")
|
||||
onboarding_completed: bool = Field(default=False)
|
||||
|
||||
class UserPreferencesResponse(BaseModel):
|
||||
default_backend: str = Field(default="local")
|
||||
onboarding_completed: bool = Field(default=False)
|
||||
available_backends: list[str] = Field(default=["local"])
|
||||
28
backend/schemas/voice_design.py
Normal file
28
backend/schemas/voice_design.py
Normal file
@@ -0,0 +1,28 @@
|
||||
from typing import Optional, Dict, Any, List
|
||||
from datetime import datetime
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
class VoiceDesignCreate(BaseModel):
|
||||
name: str = Field(..., min_length=1, max_length=100)
|
||||
instruct: str = Field(..., min_length=1)
|
||||
meta_data: Optional[Dict[str, Any]] = None
|
||||
preview_text: Optional[str] = None
|
||||
|
||||
class VoiceDesignResponse(BaseModel):
|
||||
id: int
|
||||
user_id: int
|
||||
name: str
|
||||
instruct: str
|
||||
meta_data: Optional[Dict[str, Any]]
|
||||
preview_text: Optional[str]
|
||||
ref_audio_path: Optional[str] = None
|
||||
created_at: datetime
|
||||
last_used: datetime
|
||||
use_count: int
|
||||
|
||||
class Config:
|
||||
from_attributes = True
|
||||
|
||||
class VoiceDesignListResponse(BaseModel):
|
||||
designs: List[VoiceDesignResponse]
|
||||
total: int
|
||||
Reference in New Issue
Block a user