diff --git a/src/renderer/src/components/views/ai-drawing/AIDrawingImgCurrent.vue b/src/renderer/src/components/views/ai-drawing/AIDrawingImgCurrent.vue
index 9cec7ef..7f37f0c 100644
--- 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
index f871ec9..1268852 100644
--- 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
+      }
     }
   })
 }