Add agent skill download endpoint

app.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+import shutil
 
 import gradio as gr
 
@@ -207,6 +208,13 @@ structured_json_templates = {
 
 default_structured_json_template = "Invoice with line items"
 
+def download_agentskill():
+    return shutil.make_archive(
+        "extract-image-data",
+        "zip",
+        root_dir=".agents/skills",
+        base_dir="extract-image-data",
+    )
 
 with gr.Blocks(title="Image Data Extractor") as demo:
     gr.Markdown(
@@ -214,6 +222,15 @@ with gr.Blocks(title="Image Data Extractor") as demo:
         "Extract structured JSON from an image and optional text using "
         "[numind/NuExtract3](https://huggingface.co/numind/NuExtract3)."
     )
+    gr.Markdown(
+        "## Install the agent skill\n"
+        "Download the ZIP, extract it, and place the `extract-image-data` folder "
+        "in your project's `.agents/skills/` directory. Agents should read its "
+        "`SKILL.md` and use it whenever a task requires structured extraction "
+        "from an image."
+    )
+    download_agentskill_button = gr.Button("Download agent skill")
+    download_agentskill_file = gr.File(label="Agent skill ZIP")
     with gr.Row():
         with gr.Column():
             image = gr.Image(label="Document image", type="pil", height=430)
@@ -230,7 +247,9 @@ with gr.Blocks(title="Image Data Extractor") as demo:
                 info="Choose a starting structure, then edit it below.",
             )
             generate_template_button = gr.Button("Generate template from Image")
-            with gr.Accordion("Structured JSON template", open=False):
+            with gr.Accordion(
+                "Structured JSON template", open=False
+            ) as template_accordion:
                 template = gr.Textbox(
                     label="Structured JSON template",
                     value=structured_json_templates[default_structured_json_template],
@@ -318,7 +337,18 @@ with gr.Blocks(title="Image Data Extractor") as demo:
         api_name="select_structured_json_template",
     )
 
+    download_agentskill_button.click(
+        download_agentskill,
+        outputs=download_agentskill_file,
+        api_name="download_agentskill",
+    )
+
     generate_template_button.click(
+        lambda: gr.Accordion(open=True),
+        outputs=template_accordion,
+        api_visibility="private",
+        show_progress="hidden",
+    ).then(
         generate_template,
         inputs=image,
         outputs=template,