Update app.py

app.py

@@ -179,28 +179,28 @@ def generate_poster(
     seed_input,
     progress=gr.Progress(track_tqdm=True),
 ):
-    """使用预加载的模型生成海报"""
+    """Generate poster using preloaded models"""
     if not original_prompt or not original_prompt.strip():
-        return None, "❌ 提示词不能为空！", ""
+        return None, "❌ Prompt cannot be empty!", ""
 
     try:
         if not HF_TOKEN:
-            return None, "❌ 错误：未找到 HF_TOKEN，请配置认证。", ""
+            return None, "❌ Error: HF_TOKEN not found, please configure authentication.", ""
 
-        progress(0.1, desc="开始生成...")
+        progress(0.1, desc="Starting generation...")
 
-        # 确定最终提示词
+        # Determine final prompt
         final_prompt = original_prompt
         if enable_recap:
-            progress(0.2, desc="增强提示词...")
+            progress(0.2, desc="Enhancing prompt...")
             final_prompt = enhance_prompt_with_qwen(original_prompt)
         
-        # 确定种子
+        # Determine seed
         actual_seed = int(seed_input) if seed_input and seed_input != -1 else random.randint(1, 2**32 - 1)
         
-        progress(0.3, desc="生成图像...")
+        progress(0.3, desc="Generating image...")
         
-        # 使用预加载的 FLUX 管道生成图像
+        # Use preloaded FLUX pipeline to generate image
         generator = torch.Generator("cuda").manual_seed(actual_seed)
         
         with torch.inference_mode():
@@ -213,99 +213,62 @@ def generate_poster(
                 height=int(height)
             ).images[0]
         
-        progress(1.0, desc="完成！")
-        status_log = f"✅ 生成完成！种子：{actual_seed}"
+        progress(1.0, desc="Complete!")
+        status_log = f"✅ Generation complete! Seed: {actual_seed}"
         return image, status_log, final_prompt
 
     except Exception as e:
-        logging.error(f"生成失败：{e}")
-        return None, f"❌ 生成失败：{str(e)}", ""
+        logging.error(f"Generation failed: {e}")
+        return None, f"❌ Generation failed: {str(e)}", ""
 
 # ─────────────────────────────────────────────
-# Gradio 界面
+# Gradio Interface
 # ─────────────────────────────────────────────
 def create_interface():
-    """创建 Gradio 界面"""
+    """Create Gradio interface"""
     
-    with gr.Blocks(
-        title="PosterCraft-v1.0",
-        theme=gr.themes.Soft(),
-        css="""
-        .main-container { max-width: 1200px; margin: 0 auto; }
-        .status-box { padding: 10px; border-radius: 5px; margin: 10px 0; }
-        """
-    ) as demo:
-        
-        gr.HTML("""
-        <div class="main-container">
-            <h1 style="text-align: center; margin-bottom: 20px;">🎨 PosterCraft-v1.0</h1>
-            <p style="text-align: center; color: #666; margin-bottom: 30px;">
-                专业海报生成工具，基于 FLUX.1-dev 和定制微调权重
-            </p>
-        </div>
-        """)
-        
-        with gr.Row():
-            gr.Markdown(f"**基础模型：** `{DEFAULT_PIPELINE_PATH}`")
-            gr.Markdown(f"**认证状态：** {auth_status}")
-        
-        gr.HTML("""
-        <div class="status-box">
-            <p><strong>⚡ 首次生成需要加载模型（5-10分钟），后续生成会非常快！</strong></p>
-        </div>
-        """)
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        gr.Markdown("# PosterCraft-v1.0")
+        gr.Markdown(f"Running on: **{device if 'device' in globals() else 'cuda' if torch.cuda.is_available() else 'cpu'}** | Base Pipeline: **{DEFAULT_PIPELINE_PATH}**")
 
         with gr.Row():
             with gr.Column(scale=1):
-                gr.Markdown("### 1. 配置")
+                gr.Markdown("### 1. Configuration")
                 prompt_input = gr.Textbox(
-                    label="海报提示词", 
+                    label="Prompt", 
                     lines=3, 
-                    placeholder="输入您的海报描述...",
-                    value="复古科幻电影海报，霓虹色彩和飞行汽车"
+                    placeholder="Enter your creative prompt...",
+                    value="Retro sci-fi movie poster with neon colors and flying cars"
                 )
                 enable_recap_checkbox = gr.Checkbox(
-                    label="启用提示词增强 (Qwen3-8B)", 
+                    label="Enable Prompt Recap", 
                     value=True, 
-                    info="使用 AI 增强和扩展您的提示词"
+                    info=f"Uses {DEFAULT_QWEN_MODEL_PATH} for rewriting."
                 )
                 
                 with gr.Row():
-                    # 默认宽度 832，高度 1216
-                    width_input = gr.Slider(label="宽度", minimum=256, maximum=MAX_IMAGE_SIZE, value=832, step=32)
-                    height_input = gr.Slider(label="高度", minimum=256, maximum=MAX_IMAGE_SIZE, value=1216, step=32)
+                    width_input = gr.Slider(label="Width", minimum=256, maximum=2048, value=832, step=64)
+                    height_input = gr.Slider(label="Height", minimum=256, maximum=2048, value=1216, step=64)
+                gr.Markdown("Tip: Recommended size is 832x1216 for best results.")
                 
-                # 默认步数 28
-                num_inference_steps_input = gr.Slider(label="推理步数", minimum=1, maximum=100, value=28, step=1)
-                # 默认 CFG 28，为了能取到 28，将最大值提高到 40
-                guidance_scale_input = gr.Slider(label="引导强度", minimum=0.0, maximum=40.0, value=28, step=0.1)
-                seed_number_input = gr.Number(label="种子 (-1 随机)", value=-1, minimum=-1, step=1)
-                generate_button = gr.Button("🎨 生成海报", variant="primary", size="lg")
+                num_inference_steps_input = gr.Slider(label="Inference Steps", minimum=1, maximum=100, value=28, step=1)
+                guidance_scale_input = gr.Slider(label="Guidance Scale (CFG)", minimum=0.0, maximum=20.0, value=3.5, step=0.1)
+                seed_number_input = gr.Number(label="Seed", value=None, minimum=-1, step=1, info="Leave blank or set to -1 for a random seed.")
+                generate_button = gr.Button("Generate Image", variant="primary")
 
             with gr.Column(scale=1):
-                gr.Markdown("### 2. 结果")
-                image_output = gr.Image(label="生成的海报", type="pil", height=600)
-                status_output = gr.Textbox(label="生成状态", lines=2, interactive=False)
-                recapped_prompt_output = gr.Textbox(label="增强后的提示词", lines=5, interactive=False, info="用于生成的最终提示词")
+                gr.Markdown("### 2. Results")
+                image_output = gr.Image(label="Generated Image", type="pil", show_download_button=True, height=512)
+                recapped_prompt_output = gr.Textbox(label="Final Prompt Used", lines=5, interactive=False)
+                status_output = gr.Textbox(label="Status Log", lines=4, interactive=False)
 
         inputs_list = [
             prompt_input, enable_recap_checkbox, height_input, width_input,
             num_inference_steps_input, guidance_scale_input, seed_number_input
         ]
-        outputs_list = [image_output, status_output, recapped_prompt_output]
+        outputs_list = [image_output, recapped_prompt_output, status_output]
         
         generate_button.click(fn=generate_poster, inputs=inputs_list, outputs=outputs_list)
-        
-        # 示例
-        gr.Examples(
-            examples=[
-                ["复古科幻电影海报，霓虹色彩和飞行汽车"],
-                ["优雅的装饰艺术风格豪华酒店海报"],
-                ["简约音乐会海报，粗体字体"],
-                ["有机咖啡的复古广告"],
-            ],
-            inputs=[prompt_input]
-        )
     
     return demo