feat generate slogan | to product image | slogan 接口部署

2024-05-30 15:01:39 +08:00
parent 1f47d94431
commit 401b76bd95
11 changed files with 280 additions and 174 deletions
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -8,3 +8,15 @@ class GenerateImageModel(BaseModel):
    mode: str
    category: str
    gender: str
+
+
+class GenerateSingleLogoImageModel(BaseModel):
+    tasks_id: str
+    prompt: str
+    image_url: str
+
+
+class GenerateProductImageModel(BaseModel):
+    tasks_id: str
+    prompt: str
+    image_url: str
--- a/app/schemas/slogan.py
+++ b/app/schemas/slogan.py
@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class SloganModel(BaseModel):
+    prompt: str
+    svg: str
+    tasks_id: str