first commit

schemas.py

@@ -0,0 +1,48 @@
+# -*- coding:utf-8 -*-
+
+from datetime import datetime
+from typing import Any, List, Optional, Union
+
+from pydantic import BaseModel, Field
+
+
+class Response(BaseModel):
+    code: Optional[int] = 0
+    msg: Optional[str] = "success"
+    data: Optional[Any] = None
+
+
+class CustomModeGenerateParam(BaseModel):
+    """Generate with Custom Mode"""
+
+    prompt: str = Field(..., description="lyrics")
+    mv: str = Field(
+        ...,
+        description="model version, default: chirp-v3-0",
+        examples=["chirp-v3-0"],
+    )
+    title: str = Field(..., description="song title")
+    tags: str = Field(..., description="style of music")
+    continue_at: Optional[int] = Field(
+        default=None,
+        description="continue a new clip from a previous song, format number",
+        examples=[120],
+    )
+    continue_clip_id: Optional[str] = None
+
+
+class DescriptionModeGenerateParam(BaseModel):
+    """Generate with Song Description"""
+
+    gpt_description_prompt: str
+    make_instrumental: bool = False
+    mv: str = Field(
+        default='chirp-v3-0',
+        description="model version, default: chirp-v3-0",
+        examples=["chirp-v3-0"],
+    )
+    
+    prompt: str = Field(
+        default="",
+        description="Placeholder, keep it as an empty string, do not modify it",
+    )