add MCP server, REST API, docstring, gradio[mcp], ~30 min timing

Dockerfile

@@ -52,7 +52,7 @@ RUN curl -fL --retry 3 --retry-delay 5 -o /app/models/ae.safetensors \
     "https://huggingface.co/SeeSee21/Z-Anime/resolve/main/vae/ae.safetensors"
 
 # Install Python deps
-RUN pip3 install --no-cache-dir gradio Pillow
+RUN pip3 install --no-cache-dir "gradio[mcp]" Pillow
 
 COPY app.py /app/app.py
 

app.py

@@ -37,6 +37,17 @@ _proc_lock = threading.Lock()
 # ---------------------------------------------------------------------------
 
 def generate(prompt, negative_prompt, resolution, seed):
+    """Generate an anime image using Z-Anime 6B model.
+
+    Args:
+        prompt: Text description of the image to generate.
+        negative_prompt: Things to avoid in the generated image.
+        resolution: Image resolution (512x512, 768x512, or 512x768).
+        seed: Random seed (-1 for random).
+
+    Returns:
+        tuple: Generated image and status message.
+    """
     global _active_proc
 
     if not prompt or not prompt.strip():
@@ -128,7 +139,7 @@ with gr.Blocks(title="Z-Anime (CPU)") as demo:
     gr.Markdown(
         "**[Z-Anime 6B](https://huggingface.co/SeeSee21/Z-Anime)** S3-DiT Q5_0 GGUF "
         "(distill 4-step) via [sd.cpp](https://github.com/leejet/stable-diffusion.cpp) | "
-        "~42 min at 512x512 on free CPU"
+        "~30 min at 512x512 on free CPU"
     )
     with gr.Row():
         with gr.Column():
@@ -148,7 +159,8 @@ with gr.Blocks(title="Z-Anime (CPU)") as demo:
     gen_btn.click(fn=generate,
         inputs=[prompt_input, neg_input, res_input, seed_input],
         outputs=[output_img, status_box],
-        concurrency_limit=1)
+        concurrency_limit=1,
+        api_name="generate")
 
     def _on_unload():
         with _proc_lock:
@@ -159,4 +171,4 @@ with gr.Blocks(title="Z-Anime (CPU)") as demo:
 
     demo.unload(_on_unload)
 
-demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True, theme="NoCrypt/miku")
+demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True, theme="NoCrypt/miku", mcp_server=True)