From a12f05127a25ecc869e40dc2dac2fd915aebbbd9 Mon Sep 17 00:00:00 2001
From: imeepos <imeepos@sker.com>
Date: Wed, 30 Jul 2025 18:36:34 +0800
Subject: [PATCH] =?UTF-8?q?=20=E6=B7=BB=E5=8A=A0=E7=A9=BF=E6=90=AD?=
 =?UTF-8?q?=E5=9B=BE=E7=89=87=E7=94=9F=E6=88=90=E5=AE=8C=E6=88=90=E5=90=8E?=
 =?UTF-8?q?=E8=87=AA=E5=8A=A8=E5=85=B3=E9=97=AD=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

 功能改进：
- OutfitImageGenerator 组件添加 onClose 属性
- 生成成功后自动关闭模态框，提升用户体验
- 保持表单清空逻辑，确保下次使用时状态干净

 修改内容：
- OutfitImageGenerator.tsx: 添加 onClose 回调属性
- OutfitImageGenerationModal.tsx: 传递 onClose 属性
- handleGenerate 方法: 成功后调用 onClose 关闭模态框

这样用户在生成穿搭图片后，模态框会自动关闭，无需手动点击关闭按钮。
---
 .../src/components/OutfitImageGenerationModal.tsx        | 1 +
 apps/desktop/src/components/OutfitImageGenerator.tsx     | 9 ++++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/apps/desktop/src/components/OutfitImageGenerationModal.tsx b/apps/desktop/src/components/OutfitImageGenerationModal.tsx
index 38712ce..e9b1854 100644
--- a/apps/desktop/src/components/OutfitImageGenerationModal.tsx
+++ b/apps/desktop/src/components/OutfitImageGenerationModal.tsx
@@ -90,6 +90,7 @@ export const OutfitImageGenerationModal: React.FC<OutfitImageGenerationModalProp
               modelId={model.id}
               modelPhotos={model.photos}
               onGenerate={onGenerate}
+              onClose={onClose}
               isGenerating={isGenerating}
             />
           </div>
diff --git a/apps/desktop/src/components/OutfitImageGenerator.tsx b/apps/desktop/src/components/OutfitImageGenerator.tsx
index 2dc9923..f7b233f 100644
--- a/apps/desktop/src/components/OutfitImageGenerator.tsx
+++ b/apps/desktop/src/components/OutfitImageGenerator.tsx
@@ -17,6 +17,7 @@ interface OutfitImageGeneratorProps {
   modelId: string;
   modelPhotos: ModelPhoto[];
   onGenerate: (request: OutfitImageGenerationRequest) => Promise<void>;
+  onClose?: () => void; // 添加关闭回调
   isGenerating?: boolean;
   disabled?: boolean;
 }
@@ -31,6 +32,7 @@ export const OutfitImageGenerator: React.FC<OutfitImageGeneratorProps> = ({
   modelId,
   modelPhotos,
   onGenerate,
+  onClose,
   isGenerating = false,
   disabled = false
 }) => {
@@ -154,11 +156,16 @@ export const OutfitImageGenerator: React.FC<OutfitImageGeneratorProps> = ({
       setSelectedModelImageId('');
       setProductImages([]);
       setGenerationPrompt('');
+
+      // 生成成功后关闭模态框
+      if (onClose) {
+        onClose();
+      }
     } catch (error) {
       console.error('生成穿搭图片失败:', error);
       setError(`生成穿搭图片失败: ${error}`);
     }
-  }, [modelId, selectedModelImageId, productImages, generationPrompt, onGenerate]);
+  }, [modelId, selectedModelImageId, productImages, generationPrompt, onGenerate, onClose]);
 
   const canGenerate = selectedModelImageId && productImages.length > 0 && !isGenerating && !disabled;