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;