Fix mcp

README.md

@@ -1,5 +1,5 @@
 ---
-title: "SD Dataset Automaker: Fancaps → YOLO → Siamese → WD Tag → ZIP"
+title: "SD Dataset Automaker: Fancaps → Face Crop (YOLO) → Similarity (Siamese) → WD Tagger → ZIP"
 emoji: 🎴💆📦
 colorFrom: indigo
 colorTo: pink
@@ -96,4 +96,4 @@ print(f"ZIP: {zip_path}")
 </details>
 
 ---
-*CPU runtime ~5-10 min*
+*CPU runtime ~5-10 min* | Based on [Maximax67/LoRA-Dataset-Automaker](https://github.com/Maximax67/LoRA-Dataset-Automaker)

app.py

@@ -536,8 +536,8 @@ css = """
 .compact-group { margin-bottom: 8px !important; }
 """
 
-with gr.Blocks(title="SD Dataset Automaker: Fancaps → YOLO → Siamese → WD Tag → ZIP") as demo:
-    gr.Markdown("### SD Dataset Automaker: Fancaps → YOLO → Siamese → WD → ZIP")
+with gr.Blocks(title="SD Dataset Automaker: Fancaps → Face Crop (YOLO) → Similarity (Siamese) → WD Tagger → ZIP") as demo:
+    gr.Markdown("### SD Dataset Automaker: Fancaps → Face Crop (YOLO) → Similarity (Siamese) → WD Tagger → ZIP")
 
     with gr.Row():
         with gr.Column(scale=3):
@@ -623,6 +623,49 @@ with gr.Blocks(title="SD Dataset Automaker: Fancaps → YOLO → Siamese → WD
 
     run_btn.click(process, [query, char, ref_imgs, ref_files, max_img, thresh, act_tag, anime_cb, movies_cb, tv_cb], [out_file, out_log])
 
+# MCP-only function (no Gallery - avoids $ref schema bug)
+def create_dataset(url: str, character: str, reference_images, max_images: int = 200, similarity_threshold: int = 32, trigger_tag: str = ""):
+    """Create anime character dataset for LoRA/SD training.
+
+    Args:
+        url: Fancaps.net URL (e.g. https://fancaps.net/anime/showimages.php?3092-Cowboy_Bebop)
+        character: Character name for tagging (e.g. spike_spiegel)
+        reference_images: 1-5 face reference images of target character
+        max_images: Maximum frames to download (50-500)
+        similarity_threshold: Face matching threshold, lower=stricter (25-40, default 32)
+        trigger_tag: Optional trigger word to prepend to captions
+
+    Returns:
+        Tuple of (zip_file_path, log_text)
+    """
+    paths = []
+    if reference_images:
+        for f in (reference_images if isinstance(reference_images, list) else [reference_images]):
+            fp = f.name if hasattr(f, 'name') else str(f)
+            if fp and os.path.exists(fp): paths.append(fp)
+    if not paths:
+        return None, "ERROR: No reference images provided. Upload 1-5 face images."
+    if not url or 'fancaps.net' not in url:
+        return None, "ERROR: Invalid URL. Provide a fancaps.net URL."
+    zp, log = run(url, character, paths, max_images, similarity_threshold, trigger_tag, True, False, False, None)
+    return zp, log
+
+# Create MCP interface (File input only - no Gallery)
+mcp_interface = gr.Interface(
+    fn=create_dataset,
+    inputs=[
+        gr.Textbox(label="Fancaps URL"),
+        gr.Textbox(label="Character Name"),
+        gr.File(label="Reference Images", file_count="multiple", file_types=["image"]),
+        gr.Slider(50, 500, 200, step=50, label="Max Images"),
+        gr.Slider(25, 40, 32, step=1, label="Similarity Threshold"),
+        gr.Textbox(label="Trigger Tag"),
+    ],
+    outputs=[gr.File(label="Dataset ZIP"), gr.Textbox(label="Log")],
+    api_name="create_dataset",
+    title="SD Dataset Automaker (MCP)",
+)
+
 def run_cli():
     """CLI mode with cloudscraper for Cloudflare bypass"""
     # Use cloudscraper for CLI (bypasses Cloudflare on local/residential IPs)
@@ -680,13 +723,17 @@ if __name__ == "__main__":
     if len(sys.argv) > 1:
         run_cli()
     else:
-        # Gradio UI mode
+        # Gradio UI mode - combine main UI and MCP interface
         allowed_dir = os.path.dirname(os.path.abspath(__file__))
-        demo.launch(
+        app = gr.TabbedInterface(
+            [demo, mcp_interface],
+            ["Dataset Maker", "MCP API"],
+            title="SD Dataset Automaker"
+        )
+        app.launch(
             server_name="0.0.0.0",
             server_port=7860,
             mcp_server=True,
             show_error=True,
             allowed_paths=[allowed_dir],
-            css=css,
         )