Bug fixes

classfang · Apr 8, 2024 · c15e86b · c15e86b
1 parent 70b821a
commit c15e86b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/renderer/src/components/views/ai-drawing/AIDrawingImgCurrent.vue b/src/renderer/src/components/views/ai-drawing/AIDrawingImgCurrent.vue
@@ -79,6 +79,7 @@ const collectImage = () => {
 <style scoped lang="less">
 .ai-drawing-img-current {
   display: flex;
+  flex-direction: column;
   align-items: center;
   justify-content: center;
   cursor: pointer;

diff --git a/src/renderer/src/components/views/ai-drawing/AIDrawingImgList.vue b/src/renderer/src/components/views/ai-drawing/AIDrawingImgList.vue
@@ -44,6 +44,9 @@ const deleteDrawingTask = (taskId: string) => {
     cancelText: t('common.cancel'),
     onOk: () => {
       drawingStore.drawingTaskList = drawingStore.drawingTaskList.filter((t) => t.id !== taskId)
+      if (drawingStore.currentTaskId === taskId) {
+        drawingStore.currentTaskId = null
+      }
     }
   })
 }