add 图片擦除&局部重绘 (#42)

图片擦除&局部重绘

client/py/yidong/client.py

@@ -23,6 +23,10 @@
     EditorConfig,
     ImageGenerationTask,
     ImageGenerationTaskResult,
+    ImageInpaintTask,
+    ImageInpaintTaskResult,
+    ImageRemoveTask,
+    ImageRemoveTaskResult,
     Pagination,
     PingTask,
     PingTaskResult,
@@ -417,6 +421,23 @@ def image_generation(
         """Generate images based on the given prompt or the reference image."""
         return self._submit_task(locals())
 
+    def image_inpaint(
+        self,
+        image_id: str,
+        mask_base64: str,
+        prompt: str | None = None,
+    ) -> TaskRef[ImageInpaintTask, ImageInpaintTaskResult]:
+        """image inpaint based on the mask image base64 string and the given prompt."""
+        return self._submit_task(locals())
+
+    def image_remove(
+        self,
+        image_id: str,
+        mask_base64: str,
+    ) -> TaskRef[ImageRemoveTask, ImageRemoveTaskResult]:
+        """image remove based on the mask image base64 string."""
+        return self._submit_task(locals())
+
 
 def main():
     rich.print(CLI(YiDong))

client/py/yidong/model.py

@@ -327,6 +327,29 @@ class ImageGenerationTaskResult(BaseModel):
     generated_image_ids: list[str]
 
 
+class ImageInpaintTask(BaseModel):
+    type: Literal["image_inpaint"] = "image_inpaint"
+    image_id: str = Field(..., min_length=1)
+    mask_base64: str = Field(..., min_length=1)
+    prompt: str | None = None
+
+
+class ImageInpaintTaskResult(BaseModel):
+    type: Literal["image_inpaint"] = "image_inpaint"
+    generated_image_ids: list[str]
+
+
+class ImageRemoveTask(BaseModel):
+    type: Literal["image_remove"] = "image_remove"
+    image_id: str = Field(..., min_length=1)
+    mask_base64: str = Field(..., min_length=1)
+
+
+class ImageRemoveTaskResult(BaseModel):
+    type: Literal["image_remove"] = "image_remove"
+    generated_image_ids: list[str]
+
+
 Task = Annotated[
     Union[
         PingTask,
@@ -336,6 +359,8 @@ class ImageGenerationTaskResult(BaseModel):
         VideoConcatTask,
         VideoSnapshotTask,
         ImageGenerationTask,
+        ImageInpaintTask,
+        ImageRemoveTask,
     ],
     Field(discriminator="type"),
 ]
@@ -351,6 +376,8 @@ class ImageGenerationTaskResult(BaseModel):
         VideoConcatTaskResult,
         VideoSnapshotTaskResult,
         ImageGenerationTaskResult,
+        ImageInpaintTaskResult,
+        ImageRemoveTaskResult,
     ],
     Field(discriminator="type"),
 ]