INPAINT

trueai-org · Jul 10, 2024 · ff5ad76 · ff5ad76
1 parent 9a630b7
commit ff5ad76
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/Midjourney.Infrastructure/Services/TaskServiceImpl.cs b/src/Midjourney.Infrastructure/Services/TaskServiceImpl.cs
@@ -178,6 +178,13 @@ public SubmitResultVO SubmitAction(TaskInfo task, SubmitActionDTO submitAction)
             if (submitAction.CustomId.StartsWith("MJ::CustomZoom::")
                 || submitAction.CustomId.StartsWith("MJ::Inpaint::"))
             {
+
+                // 如果是局部重绘，则设置任务状态为进行中
+                if (task.Action == TaskAction.INPAINT)
+                {
+                    task.Status = TaskStatus.IN_PROGRESS;
+                }
+
                 task.SetProperty(Constants.TASK_PROPERTY_MESSAGE_ID, targetTask.MessageId);
                 task.SetProperty(Constants.TASK_PROPERTY_FLAGS, messageFlags);