Update app.py

app.py

@@ -3,10 +3,11 @@ import spaces
 import gradio as gr
 import time
 import re
-from diffusers import DiffusionPipeline
+import random
+from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 import warnings
 
-# 忽略一些警告
+# 忽略警告
 warnings.filterwarnings("ignore")
 
 # ==================== 分辨率配置 ====================
@@ -58,609 +59,258 @@ def get_resolution(resolution_str):
     if match:
         width = int(match.group(1))
         height = int(match.group(2))
-        # 确保是8的倍数
-        width = width - width % 8
-        height = height - height % 8
-        return width, height
+        return width - width % 8, height - height % 8
     return 1024, 1024
 
-# ==================== 模型加载优化 ====================
-print("🚀 Loading Z-Image-Turbo pipeline...")
+# ==================== 模型加载与优化 ====================
+print("🚀 Loading Z-Image-Turbo pipeline with optimizations...")
 start_time = time.time()
 
-# 注意：Z-Image-Turbo 没有 fp16 变体，移除 variant 参数
+model_id = "Tongyi-MAI/Z-Image-Turbo"
+
+# 1. 加载管道
 pipe = DiffusionPipeline.from_pretrained(
-    "Tongyi-MAI/Z-Image-Turbo",
+    model_id,
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
     use_safetensors=True,
 )
 
-# 快速移动到GPU
+# 2. 调度器优化 
+# 参考官方代码 logic
+pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
+    pipe.scheduler.config, 
+    shift=3.0
+)
+
+# 3. 移动到 GPU
 pipe.to("cuda")
 
-# 尝试启用内存优化
+# 4. 显存优化 (xFormers)
 try:
-    if hasattr(pipe, "enable_xformers_memory_efficient_attention"):
-        pipe.enable_xformers_memory_efficient_attention()
-        print("✅ XFormers enabled")
+    pipe.enable_xformers_memory_efficient_attention()
+    print("✅ XFormers enabled")
 except Exception as e:
     print(f"⚠️ XFormers not available: {e}")
 
-# 启用VAE切片减少内存
+# 5. VAE 优化
 try:
-    if hasattr(pipe.vae, "enable_slicing"):
-        pipe.vae.enable_slicing()
-        print("✅ VAE slicing enabled")
+    pipe.vae.enable_slicing()
+    # pipe.vae.enable_tiling() # 如果显存非常紧张可开启此项，但会轻微降低速度
 except:
     pass
 
+# 6. 编译优化
+# 注意：第一次生成时会进行编译，耗时较长，之后速度会大幅提升
+print("⚡ Compiling transformer model (this may take a moment during first run)...")
+try:
+    pipe.transformer = torch.compile(
+        pipe.transformer, 
+        mode="max-autotune-no-cudagraphs", 
+        fullgraph=False
+    )
+    print("✅ Torch Compile enabled")
+except Exception as e:
+    print(f"⚠️ Torch compile skipped: {e}")
+
 load_time = time.time() - start_time
-print(f"✅ Pipeline loaded in {load_time:.2f} seconds!")
+print(f"✅ Pipeline ready in {load_time:.2f} seconds!")
 
 # ==================== 生成函数 ====================
 @spaces.GPU
 def generate_image(
     prompt, 
     resolution_choice,
-    custom_height=None,
-    custom_width=None,
-    use_custom_res=False,
-    num_inference_steps=8, 
-    seed=42, 
-    randomize_seed=True,
-    negative_prompt_text="",
+    custom_height,
+    custom_width,
+    use_custom_res,
+    num_inference_steps, 
+    seed, 
+    randomize_seed,
+    negative_prompt_text,
+    gallery_history, # 接收之前的图片列表
     progress=gr.Progress(track_tqdm=True)
 ):
-    """优化后的图像生成函数"""
+    """支持画廊历史记录的生成函数"""
+    if gallery_history is None:
+        gallery_history = []
+        
     try:
-        # 输入验证
         if not prompt or len(prompt.strip()) < 2:
-            return None, 0, "❌ Please enter a meaningful prompt"
+            raise gr.Error("❌ Please enter a meaningful prompt")
         
         prompt = prompt.strip()
         negative_prompt = negative_prompt_text.strip() if negative_prompt_text else None
         
         # 确定分辨率
         if use_custom_res and custom_height and custom_width:
-            # 使用自定义分辨率
             height = max(512, int(custom_height) - int(custom_height) % 8)
             width = max(512, int(custom_width) - int(custom_width) % 8)
         else:
-            # 使用预设分辨率
             width, height = get_resolution(resolution_choice)
         
-        # 限制最大尺寸防止OOM（T4 GPU限制）
-        MAX_SIZE = 1536
-        if height > MAX_SIZE or width > MAX_SIZE:
-            height = min(height, MAX_SIZE)
-            width = min(width, MAX_SIZE)
-        
-        # 检查宽高积是否过大
+        # 限制最大尺寸
         if height * width > 2048 * 2048:
-            return None, 0, "❌ Resolution too large! Maximum total pixels: 2048x2048"
+            raise gr.Error("❌ Resolution too large! Max pixels: 2048x2048")
         
-        # 生成随机种子
+        # 处理种子
         if randomize_seed:
-            seed = torch.randint(0, 2**32 - 1, (1,)).item()
-        
+            seed = random.randint(0, 2**32 - 1)
         seed = int(seed)
         
-        # 记录开始时间
         gen_start = time.time()
         
-        # 创建生成器
         generator = torch.Generator("cuda").manual_seed(seed)
         
-        # 清除CUDA缓存
-        torch.cuda.empty_cache()
-        
-        # 生成图像
+        # 生成
         with torch.cuda.amp.autocast(dtype=torch.bfloat16):
             image = pipe(
                 prompt=prompt,
                 height=height,
                 width=width,
                 num_inference_steps=int(num_inference_steps),
-                guidance_scale=0.0,
+                guidance_scale=0.0, # Turbo 模型通常不需要 guidance
                 generator=generator,
                 negative_prompt=negative_prompt,
-                output_type="pil",
+                max_sequence_length=512, # 官方设置
             ).images[0]
-        
-        # 再次清除缓存
-        torch.cuda.empty_cache()
-        
-        # 计算生成时间
+            
         gen_time = time.time() - gen_start
+        info = f"✅ {width}x{height} | {gen_time:.2f}s | Seed: {seed}"
         
-        # 生成信息
-        info = f"✅ Generated {width}x{height} in {gen_time:.1f}s ({num_inference_steps} steps)"
+        # 将新图片插入到列表最前面
+        # Gallery 接收的格式通常是 [(image, label), ...] 或者直接 image list
+        gallery_history.insert(0, (image, info))
         
-        return image, seed, info
+        return gallery_history, seed, info
         
-    except torch.cuda.OutOfMemoryError:
-        return None, seed, "💥 Out of memory! Try smaller image size"
     except Exception as e:
-        error_msg = str(e)[:100]
-        return None, seed, f"❌ Error: {error_msg}"
-
-# ==================== 主题配置 ====================
-custom_theme = gr.themes.Soft(
-    primary_hue="yellow",
-    secondary_hue="amber",
-    neutral_hue="slate",
-    font=gr.themes.GoogleFont("Inter"),
-    text_size="lg",
-    spacing_size="md",
-    radius_size="lg"
-).set(
-    button_primary_background_fill="*primary_500",
-    button_primary_background_fill_hover="*primary_600",
-    button_primary_border_color="*primary_500",
-    button_primary_text_color="white",
-    block_title_text_weight="600",
-    block_label_text_weight="500",
-    checkbox_label_background_fill_selected="*primary_500",
-    slider_color="*primary_500",
-)
-
-# ==================== 全局样式 ====================
-global_styles = """
-/* 全局样式 */
-body {
-    font-family: 'Inter', sans-serif !important;
-}
-
-.gradio-container {
-    max-width: 1200px !important;
-    margin: 0 auto !important;
-    padding: 1rem !important;
-}
-
-/* 头部样式 */
-#header h1 {
-    margin-bottom: 0.5rem !important;
-}
-
-/* 输入框 */
-#prompt-box, #negative-prompt-box {
-    border-radius: 10px !important;
-    border: 2px solid #e2e8f0 !important;
-    transition: all 0.2s ease !important;
-}
-
-#prompt-box:focus, #negative-prompt-box:focus {
-    border-color: #f59e0b !important;
-    box-shadow: 0 0 0 3px rgba(245, 158, 11, 0.1) !important;
-}
-
-/* 按钮 */
-button.primary {
-    background: linear-gradient(135deg, #fbbf24 0%, #f59e0b 100%) !important;
-    border: none !important;
-    font-weight: 600 !important;
-    transition: all 0.2s ease !important;
-}
-
-button.primary:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 4px 12px rgba(245, 158, 11, 0.3) !important;
-}
-
-button.secondary {
-    transition: all 0.2s ease !important;
-}
-
-button.secondary:hover {
-    transform: translateY(-1px) !important;
-}
-
-/* 输出图片 */
-#output-image {
-    border-radius: 12px !important;
-    overflow: hidden !important;
-    border: 1px solid #e2e8f0 !important;
-}
-
-#output-image img {
-    border-radius: 10px !important;
-    transition: transform 0.3s ease !important;
-}
-
-#output-image img:hover {
-    transform: scale(1.01) !important;
-}
-
-/* 信息显示 */
-#info-display {
-    background: rgba(71,85,105) !important;
-    border: 1px solid #e2e8f0 !important;
-    border-radius: 8px !important;
-    font-size: 0.9rem !important;
-    padding: 0.75rem !important;
-}
-
-/* 进度条 */
-.progress-bar {
-    background: linear-gradient(90deg, #fbbf24 0%, #f59e0b 100%) !important;
-}
-
-/* 分辨率选择器 */
-.resolution-option {
-    border: 1px solid #e2e8f0 !important;
-    border-radius: 8px !important;
-    padding: 0.5rem !important;
-    margin: 0.25rem !important;
-    cursor: pointer !important;
-    transition: all 0.2s ease !important;
-}
-
-.resolution-option:hover {
-    background: #fefce8 !important;
-    border-color: #fbbf24 !important;
-}
-
-.resolution-option.selected {
-    background: linear-gradient(135deg, #fef3c7 0%, #fde68a 100%) !important;
-    border-color: #f59e0b !important;
-    font-weight: 600 !important;
-}
-
-/* 移动端适配 */
-@media (max-width: 768px) {
-    .gradio-container {
-        padding: 0.5rem !important;
-    }
-    
-    .gr-row {
-        flex-direction: column !important;
-    }
-    
-    .gr-column {
-        min-width: 100% !important;
-    }
-    
-    #header h1 {
-        font-size: 2rem !important;
-    }
-    
-    #output-image {
-        height: 350px !important;
-    }
-    
-    button {
-        width: 100% !important;
-        margin-bottom: 0.5rem !important;
-    }
-}
-
-/* 示例样式 */
-.example {
-    cursor: pointer !important;
-    transition: all 0.2s ease !important;
-    border: 1px solid #e2e8f0 !important;
-    border-radius: 8px !important;
-    padding: 0.75rem !important;
-    margin-bottom: 0.5rem !important;
-}
-
-.example:hover {
-    background: #fefce8 !important;
-    border-color: #fbbf24 !important;
-    transform: translateY(-1px) !important;
-}
-
-/* 加载动画 */
-.spinner {
-    display: inline-block;
-    width: 16px;
-    height: 16px;
-    border: 2px solid rgba(245, 158, 11, 0.3);
-    border-radius: 50%;
-    border-top-color: #f59e0b;
-    animation: spin 1s ease-in-out infinite;
-    margin-right: 8px;
-}
-
-@keyframes spin {
-    to { transform: rotate(360deg); }
-}
-
-/* 成功/错误消息 */
-.success {
-    color: #059669 !important;
-}
-
-.error {
-    color: #dc2626 !important;
+        raise gr.Error(f"Error: {str(e)}")
+
+# ==================== 界面样式 ====================
+css = """
+body { font-family: 'Inter', sans-serif; }
+.gradio-container { max-width: 1280px !important; margin: 0 auto; }
+#gallery-container { min-height: 500px; }
+.generate-btn { 
+    background: linear-gradient(90deg, #f59e0b 0%, #d97706 100%) !important; 
+    border: none !important; 
+    color: white !important; 
 }
 """
 
-# ==================== 示例提示词 ====================
-examples = [
-    ["A beautiful Chinese woman in traditional red Hanfu, intricate embroidery, cinematic lighting, photorealistic"],
-    ["Cyberpunk city at night, neon lights, rainy streets, futuristic architecture, Blade Runner style"],
-    ["Majestic dragon flying over ancient Chinese palace, sunset, epic fantasy art"],
-    ["Cute anime girl with pink hair, cyberpunk street background, vibrant colors"],
-    ["Fantasy landscape with floating islands, waterfalls, magical creatures, digital painting"],
-    ["Portrait of a wise old samurai, detailed armor, cherry blossoms, studio lighting"],
-    ["Steampunk airship flying over Victorian London, gears and cogs, detailed"],
-]
-
-# ==================== Gradio界面 ====================
-with gr.Blocks(
-    theme=custom_theme,
-    title="Z-Image-Turbo • Ultra-Fast AI Image Generator",
-    fill_height=True,
-    css=global_styles
-) as demo:
+# ==================== Gradio Blocks ====================
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="amber"), css=css, title="Z-Image-Turbo Fast") as demo:
     
-    # 头部
     gr.Markdown(
         """
-        <div style="text-align: center;">
-        <h1 style="font-size: 2.5rem; font-weight: 800; margin-bottom: 0.5rem; background: linear-gradient(135deg, #fbbf24 0%, #f59e0b 50%, #d97706 100%); -webkit-background-clip: text; -webkit-text-fill-color: transparent; background-clip: text;">
-        🎨 Z-Image-Turbo</h1>
-        <p style="font-size: 1.1rem; color: #64748b; margin-bottom: 1.5rem;">
-        Generate stunning images in <strong>8 steps</strong> • Optimized for speed • Multiple resolutions</p>
+        <div style="text-align: center; margin-bottom: 1rem">
+            <h1 style="font-size: 2.5rem; color: #d97706; margin: 0">⚡ Z-Image-Turbo Gallery</h1>
+            <p>Super fast generation with history • Optimized with Torch Compile</p>
         </div>
-        """,
-        elem_id="header"
+        """
     )
     
-    with gr.Row(equal_height=False, variant="panel"):
-        # 左侧控制面板
-        with gr.Column(scale=1, min_width=350):
+    with gr.Row():
+        # 左侧控制栏
+        with gr.Column(scale=4, min_width=300):
+            prompt = gr.Textbox(
+                label="Prompt", 
+                placeholder="Describe your image...", 
+                lines=3,
+                autofocus=True
+            )
+            negative_prompt = gr.Textbox(
+                label="Negative Prompt", 
+                placeholder="Low quality, blurry...", 
+                lines=1
+            )
+            
             with gr.Group():
-                prompt = gr.Textbox(
-                    label="✨ Your Prompt",
-                    placeholder="Describe the image you want to create...",
-                    lines=4,
-                    max_lines=6,
-                    autofocus=True,
-                    elem_id="prompt-box"
-                )
-                
-                negative_prompt = gr.Textbox(
-                    label="🚫 Negative Prompt (Optional)",
-                    placeholder="What you don't want in the image...",
-                    lines=2,
-                    max_lines=4,
-                    elem_id="negative-prompt-box"
-                )
-                
                 with gr.Row():
-                    generate_btn = gr.Button(
-                        "🚀 Generate Image", 
-                        variant="primary", 
-                        scale=2,
-                        size="lg"
-                    )
-                    clear_btn = gr.Button(
-                        "🗑️ Clear", 
-                        variant="secondary",
+                    res_preset = gr.Dropdown(
+                        choices=["1024", "1280", "1536"], 
+                        value="1024", 
+                        label="Size Class",
                         scale=1
                     )
-            
-            # 分辨率设置
-            with gr.Accordion("🖼️ Resolution Settings", open=True):
-                # 分辨率预设选择
-                resolution_preset = gr.Radio(
-                    label="Preset Resolutions",
-                    choices=["1024", "1280", "1536"],
-                    value="1024",
-                    type="value",
-                    elem_classes="resolution-preset"
-                )
-                
-                # 分辨率选项动态更新
-                resolution_choices = gr.Dropdown(
-                    label="Select Resolution",
-                    choices=RES_CHOICES["1024"],
-                    value="720x1280 (9:16)",
-                    elem_classes="resolution-choices"
-                )
-                
-                # 自定义分辨率选项
-                use_custom_res = gr.Checkbox(
-                    label="Use Custom Resolution",
-                    value=False,
-                    info="Manually set width and height"
-                )
-                
-                with gr.Row(visible=False) as custom_res_row:
-                    custom_width = gr.Slider(
-                        label="Width",
-                        minimum=512,
-                        maximum=1536,
-                        value=1024,
-                        step=64,
-                        info="512-1536 pixels"
-                    )
-                    custom_height = gr.Slider(
-                        label="Height",
-                        minimum=512,
-                        maximum=1536,
-                        value=1024,
-                        step=64,
-                        info="512-1536 pixels"
+                    res_choice = gr.Dropdown(
+                        choices=RES_CHOICES["1024"], 
+                        value=RES_CHOICES["1024"][0], 
+                        label="Resolution", 
+                        scale=2
                     )
             
-            # 高级设置
-            with gr.Accordion("⚙️ Advanced Settings", open=False):
-                num_inference_steps = gr.Slider(
-                    label="Inference Steps",
-                    minimum=4,
-                    maximum=20,
-                    value=8,
-                    step=1,
-                    info="8 steps recommended (fastest)"
-                )
-                
+            # 自定义分辨率折叠栏
+            with gr.Accordion("📐 Custom Resolution", open=False):
+                use_custom = gr.Checkbox(label="Use Custom Size", value=False)
                 with gr.Row():
-                    randomize_seed = gr.Checkbox(
-                        label="🎲 Random Seed", 
-                        value=True,
-                        scale=1
-                    )
-                    seed = gr.Number(
-                        label="Custom Seed",
-                        value=42,
-                        precision=0,
-                        visible=False,
-                        scale=2
-                    )
+                    w_slider = gr.Slider(512, 1536, value=1024, step=64, label="Width")
+                    h_slider = gr.Slider(512, 1536, value=1024, step=64, label="Height")
+
+            with gr.Accordion("⚙️ Settings", open=False):
+                steps = gr.Slider(4, 20, value=8, step=1, label="Steps (8 is optimal)")
+                with gr.Row():
+                    use_random_seed = gr.Checkbox(label="Random Seed", value=True)
+                    seed_input = gr.Number(label="Seed", value=42, precision=0, visible=False)
+
+            generate_btn = gr.Button("🚀 GENERATE", elem_classes="generate-btn", size="lg")
             
-            # 示例提示词
-            gr.Examples(
-                examples=examples,
-                inputs=[prompt],
-                label="💡 Try These Prompts",
-                examples_per_page=7
-            )
-        
-        # 右侧输出面板
-        with gr.Column(scale=1, min_width=350):
-            output_image = gr.Image(
-                label="Generated Image",
-                type="pil",
-                show_label=False,
-                height=450,
-                show_download_button=True,
+            # 状态信息
+            last_seed = gr.Number(label="Last Seed", interactive=False, visible=True)
+            status_info = gr.Textbox(label="Status", interactive=False)
+
+        # 右侧画廊
+        with gr.Column(scale=6, min_width=500):
+            # Gallery 组件用于展示多图
+            output_gallery = gr.Gallery(
+                label="History",
+                columns=[2],
+                rows=[2],
+                height="auto",
+                object_fit="contain",
+                elem_id="gallery-container",
                 show_share_button=True,
-                elem_id="output-image"
-            )
-            
-            # 生成信息
-            info_display = gr.Textbox(
-                label="ℹ️ Generation Info",
-                interactive=False,
-                value="Ready to generate! Enter a prompt above.",
-                elem_id="info-display"
-            )
-            
-            with gr.Row():
-                used_seed = gr.Number(
-                    label="🎲 Seed Used",
-                    value=0,
-                    interactive=False,
-                    scale=2
-                )
-                copy_seed_btn = gr.Button(
-                    "📋 Copy", 
-                    size="sm", 
-                    variant="secondary",
-                    scale=1
-                )
-            
-            # 建议提示
-            gr.Markdown(
-                """
-                <div style="background: rgba(71,85,105); border-left: 4px solid #fbbf24; padding: 0.75rem; border-radius: 0.5rem; margin-top: 1rem;">
-                <strong>💡 Tips for best results:</strong>
-                <ul style="margin: 0.5rem 0 0 1rem; padding-left: 0.5rem;">
-                <li>Use descriptive prompts</li>
-                <li>Start with 8 steps for speed</li>
-                <li>1024x1024 is optimal for most cases</li>
-                <li>Higher resolutions require more VRAM</li>
-                <li>Use negative prompts to exclude unwanted elements</li>
-                </ul>
-                </div>
-                """,
-                elem_id="tips-box"
+                show_download_button=True,
+                interactive=False
             )
+            clear_btn = gr.Button("🗑️ Clear History", variant="secondary")
+
+    # ==================== 事件交互 ====================
     
-    # 页脚
-    gr.Markdown(
-        """
-        <div style="text-align: center; margin-top: 2rem; padding-top: 1.5rem; border-top: 1px solid #e2e8f0; color: #64748b; font-size: 0.9rem;">
-        <p style="margin-bottom: 0.5rem;">
-        <strong>Model:</strong> <a href="https://huggingface.co/Tongyi-MAI/Z-Image-Turbo" target="_blank" style="color: #f59e0b; text-decoration: none;">Z-Image-Turbo</a> • 
-        <strong>Demo:</strong> Optimized for Hugging Face Spaces
-        </p>
-        </div>
-        """,
-        elem_id="footer"
-    )
-    
-    # ==================== 事件处理 ====================
-    
-    # 更新分辨率选项
-    def update_resolution_choices(preset):
+    # 联动：分辨率预设 -> 分辨率列表
+    def update_res_list(preset):
         return gr.Dropdown(choices=RES_CHOICES[preset], value=RES_CHOICES[preset][0])
     
-    resolution_preset.change(
-        update_resolution_choices,
-        inputs=[resolution_preset],
-        outputs=[resolution_choices]
-    )
-    
-    # 显示/隐藏自定义分辨率
-    def toggle_custom_resolution(use_custom):
-        return gr.Row(visible=use_custom), gr.Dropdown(visible=not use_custom)
+    res_preset.change(update_res_list, inputs=res_preset, outputs=res_choice)
     
-    use_custom_res.change(
-        toggle_custom_resolution,
-        inputs=[use_custom_res],
-        outputs=[custom_res_row, resolution_choices]
+    # 联动：自定义分辨率显示
+    use_custom.change(
+        lambda x: (gr.update(visible=x), gr.update(visible=not x)),
+        inputs=use_custom,
+        outputs=[gr.Row(visible=False), res_choice] # 这里需要逻辑稍微调整，但为了简洁，主要通过use_custom参数在generate中控制
     )
     
-    # 显示/隐藏种子输入
-    def toggle_seed_visibility(randomize):
-        return gr.Number(visible=not randomize)
-    
-    randomize_seed.change(
-        toggle_seed_visibility,
-        inputs=[randomize_seed],
-        outputs=[seed]
+    # 联动：种子输入框显示
+    use_random_seed.change(
+        lambda x: gr.update(visible=not x),
+        inputs=use_random_seed,
+        outputs=seed_input
     )
     
-    # 复制种子按钮
-    def copy_seed_to_clipboard(s):
-        if s != 0:
-            gr.Info(f"Seed {s} copied to clipboard!")
-        return s
-    
-    copy_seed_btn.click(
-        copy_seed_to_clipboard,
-        inputs=[used_seed],
-        outputs=[]
-    )
-    
-    # 生成按钮点击
+    # 生成事件
     generate_btn.click(
         fn=generate_image,
-        inputs=[prompt, resolution_choices, custom_height, custom_width, use_custom_res, 
-                num_inference_steps, seed, randomize_seed, negative_prompt],
-        outputs=[output_image, used_seed, info_display],
-        api_name="generate"
-    )
-    
-    # Enter键提交
-    prompt.submit(
-        fn=generate_image,
-        inputs=[prompt, resolution_choices, custom_height, custom_width, use_custom_res,
-                num_inference_steps, seed, randomize_seed, negative_prompt],
-        outputs=[output_image, used_seed, info_display]
+        inputs=[
+            prompt, res_choice, h_slider, w_slider, use_custom,
+            steps, seed_input, use_random_seed, negative_prompt,
+            output_gallery # 传入当前的画廊状态
+        ],
+        outputs=[output_gallery, last_seed, status_info] # 更新画廊
     )
     
-    # 清除按钮
-    def clear_all():
-        return None, 0, "Cleared! Enter a new prompt..."
-    
-    clear_btn.click(
-        fn=clear_all,
-        outputs=[output_image, used_seed, info_display]
-    )
+    # 清除历史
+    clear_btn.click(lambda: ([], None, ""), outputs=[output_gallery, last_seed, status_info])
 
-# ==================== 启动应用 ====================
 if __name__ == "__main__":
-    demo.launch(
-        debug=False,
-        show_error=True,
-        share=False,
-        server_name="0.0.0.0",
-        server_port=7860,
-        quiet=True
-    )
+    demo.launch()