Add Collection Cloner feature using Hugging Face

app.py

@@ -0,0 +1,85 @@
+from huggingface_hub import HfApi
+import gradio as gr
+
+
+def clone_collection(
+    source_slug, dest_namespace, dest_title, token, private=False, exist_ok=False
+):
+    api = HfApi(token=token)
+    collection = api.get_collection(source_slug)
+    if not collection:
+        raise gr.Error(
+            f"Collection {source_slug} does not exist or you do not have access to it."
+        )
+    description = f"Copy of {collection.title}"
+
+    new_collection = api.create_collection(
+        dest_title,
+        namespace=dest_namespace,
+        exists_ok=exist_ok,
+        private=private,
+        description=description,
+        token=token,
+    )
+    for item in collection.items:
+        api.add_collection_item(
+            new_collection.slug, item.item_id, item_type=item.item_type
+        )
+    return f"[Collection]({collection.url}) has been cloned into [new_collection.slug]({new_collection.url})"
+
+
+title = (
+    """<h1 style='text-align: center;'> &#129516; Collection Cloner &#129516;</h1>"""
+)
+description = """<div style='text-align: center;'>
+<p>This space allows you to clone a [Collection](https://huggingface.co/docs/hub/collections) from the Hugging Face Hub into your own namespace.
+You can then edit the collection to your liking!</p>
+</div>"""
+
+with gr.Blocks() as demo:
+    gr.HTML(title)
+    gr.Markdown(description)
+    with gr.Row():
+        token = gr.Textbox(
+            label="Token",
+            type="password",
+        )
+    with gr.Column():
+        gr.Markdown("## Source Collection")
+        source_slug = gr.Textbox(
+            label="Source Collection Slug",
+        )
+        gr.Markdown("## Destination Collection info")
+        dest_namespace = gr.Textbox(
+            label="Destination Namespace",
+            interactive=True,
+        )
+        dest_title = gr.Textbox(
+            label="Destination Title",
+        )
+
+        with gr.Row():
+            private = gr.Checkbox(
+                False,
+                label="Private",
+            )
+            overwrite = gr.Checkbox(
+                False,
+                label="Overwrite existing collection?",
+            )
+    submit_btn = gr.Button("Clone Collection")
+    response = gr.Markdown()
+    submit_btn.click(
+        clone_collection,
+        [
+            source_slug,
+            dest_namespace,
+            dest_title,
+            token,
+            private,
+            overwrite,
+        ],
+        response,
+    )
+
+demo.launch()