feat: add CoverCraft LoRA + dropdown preset (字体参考)

app.py

@@ -140,11 +140,33 @@ try:
         weight_name="FireRed-Image-Edit-1.0-Lightning-8steps-v1.1.safetensors",
         adapter_name="lightning",
     )
-    pipe.set_adapters(["lightning"], adapter_weights=[1.0])
     print("Lightning LoRA (FireRed 8steps v1.1) loaded successfully.")
 except Exception as e:
     print(f"Warning: Could not load Lightning LoRA: {e}")
 
+# ── CoverCraft LoRA（FireRed 官方字体参考 LoRA）──
+try:
+    pipe.load_lora_weights(
+        "FireRedTeam/FireRed-Image-Edit-LoRA-Zoo",
+        weight_name="FireRed-Image-Edit-Covercraft.safetensors",
+        adapter_name="covercraft",
+    )
+    print("CoverCraft LoRA (字体参考) loaded successfully.")
+except Exception as e:
+    print(f"Warning: Could not load CoverCraft LoRA: {e}")
+
+# 默认仅启用 Lightning
+try:
+    pipe.set_adapters(["lightning"], adapter_weights=[1.0])
+except Exception as e:
+    print(f"Warning: Could not set default adapters: {e}")
+
+# LoRA 预设映射（UI 下拉选项 → adapter 配置）
+LORA_PRESETS = {
+    "无": (["lightning"], [1.0]),
+    "字体参考": (["lightning", "covercraft"], [1.0, 1.0]),
+}
+
 # ── RMBG 2.0 抠图模型 ──
 rmbg = AutoModelForImageSegmentation.from_pretrained(
     "briaai/RMBG-2.0",
@@ -487,6 +509,7 @@ def infer(
     guidance_scale,
     steps,
     color_match,
+    lora_preset="无",
     out_width=0,
     out_height=0,
     progress=gr.Progress(track_tqdm=True),
@@ -500,6 +523,13 @@ def infer(
     if not prompt or not prompt.strip():
         raise gr.Error("⚠️ Please enter an edit prompt.")
 
+    # 切换 LoRA 预设
+    preset_cfg = LORA_PRESETS.get(lora_preset, LORA_PRESETS["无"])
+    try:
+        pipe.set_adapters(preset_cfg[0], adapter_weights=preset_cfg[1])
+    except Exception as e:
+        print(f"Warning: could not switch LoRA preset '{lora_preset}': {e}")
+
     # 提取原图（兼容 ImageEditor dict 和普通路径）
     try:
         src_pil = extract_pil_from_source(source_image)
@@ -715,6 +745,12 @@ with gr.Blocks(
 
                     color_match_input = gr.Checkbox(label="Color Match — 色彩对齐原图", value=True)
 
+                    lora_preset_input = gr.Dropdown(
+                        label="LoRA 预设",
+                        choices=list(LORA_PRESETS.keys()),
+                        value="无",
+                    )
+
                     with gr.Accordion("Advanced Settings", open=False):
                         seed_input = gr.Slider(
                             label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0,
@@ -745,7 +781,7 @@ with gr.Blocks(
                 inputs=[
                     source_input, ref_input, prompt_input, negative_prompt_input,
                     seed_input, randomize_seed_input, guidance_scale_input, steps_input,
-                    color_match_input,
+                    color_match_input, lora_preset_input,
                 ],
                 outputs=[output_image, seed_input],
             ).then(