feat: 初始提交

backend/app/schemas/novel.py

@@ -0,0 +1,170 @@
+from enum import Enum
+from typing import Any, Dict, List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class ChoiceOption(BaseModel):
+    """前端选择项描述，用于动态 UI 控件。"""
+
+    id: str
+    label: str
+
+
+class UIControl(BaseModel):
+    """描述前端应渲染的组件类型与配置。"""
+
+    type: str = Field(..., description="控件类型，如 single_choice/text_input")
+    options: Optional[List[ChoiceOption]] = Field(default=None, description="可选项列表")
+    placeholder: Optional[str] = Field(default=None, description="输入提示文案")
+
+
+class ConverseResponse(BaseModel):
+    """概念对话接口的统一返回体。"""
+
+    ai_message: str
+    ui_control: UIControl
+    conversation_state: Dict[str, Any]
+    is_complete: bool = False
+    ready_for_blueprint: Optional[bool] = None
+
+
+class ConverseRequest(BaseModel):
+    """概念对话接口的请求体。"""
+
+    user_input: Dict[str, Any]
+    conversation_state: Dict[str, Any]
+
+
+class ChapterGenerationStatus(str, Enum):
+    NOT_GENERATED = "not_generated"
+    GENERATING = "generating"
+    EVALUATING = "evaluating"
+    SELECTING = "selecting"
+    FAILED = "failed"
+    EVALUATION_FAILED = "evaluation_failed"
+    WAITING_FOR_CONFIRM = "waiting_for_confirm"
+    SUCCESSFUL = "successful"
+
+
+class ChapterOutline(BaseModel):
+    chapter_number: int
+    title: str
+    summary: str
+
+
+class Chapter(ChapterOutline):
+    real_summary: Optional[str] = None
+    content: Optional[str] = None
+    versions: Optional[List[str]] = None
+    evaluation: Optional[str] = None
+    generation_status: ChapterGenerationStatus = ChapterGenerationStatus.NOT_GENERATED
+
+
+class Relationship(BaseModel):
+    character_from: str
+    character_to: str
+    description: str
+
+
+class Blueprint(BaseModel):
+    title: str
+    target_audience: str = ""
+    genre: str = ""
+    style: str = ""
+    tone: str = ""
+    one_sentence_summary: str = ""
+    full_synopsis: str = ""
+    world_setting: Dict[str, Any] = {}
+    characters: List[Dict[str, Any]] = []
+    relationships: List[Relationship] = []
+    chapter_outline: List[ChapterOutline] = []
+
+
+class NovelProject(BaseModel):
+    id: str
+    user_id: int
+    title: str
+    initial_prompt: str
+    conversation_history: List[Dict[str, Any]] = []
+    blueprint: Optional[Blueprint] = None
+    chapters: List[Chapter] = []
+
+    class Config:
+        from_attributes = True
+
+
+class NovelProjectSummary(BaseModel):
+    id: str
+    title: str
+    genre: str
+    last_edited: str
+    completed_chapters: int
+    total_chapters: int
+
+
+class BlueprintGenerationResponse(BaseModel):
+    blueprint: Blueprint
+    ai_message: str
+
+
+class ChapterGenerationResponse(BaseModel):
+    ai_message: str
+    chapter_versions: List[Dict[str, Any]]
+
+
+class NovelSectionType(str, Enum):
+    OVERVIEW = "overview"
+    WORLD_SETTING = "world_setting"
+    CHARACTERS = "characters"
+    RELATIONSHIPS = "relationships"
+    CHAPTER_OUTLINE = "chapter_outline"
+    CHAPTERS = "chapters"
+
+
+class NovelSectionResponse(BaseModel):
+    section: NovelSectionType
+    data: Dict[str, Any]
+
+
+class GenerateChapterRequest(BaseModel):
+    chapter_number: int
+    writing_notes: Optional[str] = Field(default=None, description="章节额外写作指令")
+
+
+class SelectVersionRequest(BaseModel):
+    chapter_number: int
+    version_index: int
+
+
+class EvaluateChapterRequest(BaseModel):
+    chapter_number: int
+
+
+class UpdateChapterOutlineRequest(BaseModel):
+    chapter_number: int
+    title: str
+    summary: str
+
+
+class DeleteChapterRequest(BaseModel):
+    chapter_numbers: List[int]
+
+
+class GenerateOutlineRequest(BaseModel):
+    start_chapter: int
+    num_chapters: int
+
+
+class BlueprintPatch(BaseModel):
+    one_sentence_summary: Optional[str] = None
+    full_synopsis: Optional[str] = None
+    world_setting: Optional[Dict[str, Any]] = None
+    characters: Optional[List[Dict[str, Any]]] = None
+    relationships: Optional[List[Relationship]] = None
+    chapter_outline: Optional[List[ChapterOutline]] = None
+
+
+class EditChapterRequest(BaseModel):
+    chapter_number: int
+    content: str