Upload folder using huggingface_hub

src/routes/api/download-zip.ts

@@ -24,6 +24,7 @@ const PostSchema = z.object({
 const BodySchema = z.object({
   siteName: z.string().default("grabber"),
   posts: z.array(PostSchema).min(1).max(200),
+  excludeTags: z.array(z.string()).default([]),
 });
 
 function sanitize(s: string) {
@@ -38,7 +39,7 @@ function humanSize(n?: number): string {
   return `${(n / 1024 / 1024 / 1024).toFixed(2)} GB`;
 }
 
-function buildMarkdown(post: z.infer<typeof PostSchema>): string {
+function buildMarkdown(post: z.infer<typeof PostSchema>, excludeSet: Set<string>): string {
   const dims = post.width && post.height ? `${post.width} x ${post.height}` : "";
   const rows: [string, string][] = [
     ["Source", post.source ?? ""],
@@ -56,7 +57,8 @@ function buildMarkdown(post: z.infer<typeof PostSchema>): string {
   ];
 
   const meta = rows.map(([k, v]) => `- **${k}:** ${v}`).join("\n");
-  const tags = post.tags.length ? post.tags.join(" ") : "";
+  const filteredTags = post.tags.filter((t) => !excludeSet.has(t.toLowerCase()));
+  const tags = filteredTags.length ? filteredTags.join(" ") : "";
   let raw = "{}";
   if (post.rawJson) {
     try {
@@ -99,6 +101,7 @@ export const Route = createFileRoute("/api/download-zip")({
         const files: Record<string, Uint8Array> = {};
         const errors: string[] = [];
         const siteName = sanitize(body.siteName);
+        const excludeSet = new Set((body.excludeTags || []).map((t) => t.toLowerCase()));
 
         await Promise.all(
           body.posts.map(async (post) => {
@@ -106,11 +109,14 @@ export const Route = createFileRoute("/api/download-zip")({
             const extSafe = sanitize(post.ext || "jpg");
             const base = `${siteName}_${idSafe}`;
 
+            // Filter out tags that user specified to be missing
+            const filteredTags = post.tags.filter((t) => !excludeSet.has(t.toLowerCase()));
+
             // tags .txt — same basename as the image
-            files[`${base}.txt`] = strToU8(post.tags.join(" "));
+            files[`${base}.txt`] = strToU8(filteredTags.join(" "));
 
             // per-post markdown metadata
-            files[`${idSafe}.md`] = strToU8(buildMarkdown(post));
+            files[`${idSafe}.md`] = strToU8(buildMarkdown(post, excludeSet));
 
             // fetch the image itself
             try {

src/routes/index.tsx

@@ -47,6 +47,7 @@ function Grabber() {
   const [addOpen, setAddOpen] = useState(false);
   const [lightbox, setLightbox] = useState<NormalizedPost | null>(null);
   const [downloading, setDownloading] = useState(false);
+  const [excludeTags, setExcludeTags] = useState("");
 
   const site = useMemo(() => sites.find((s) => s.id === siteId) ?? sites[0], [sites, siteId]);
 
@@ -90,6 +91,10 @@ function Grabber() {
         body: JSON.stringify({
           siteName: site.id,
           posts: chosen,
+          excludeTags: excludeTags
+            .split(/\s+|,/)
+            .map((t) => t.trim().toLowerCase())
+            .filter(Boolean),
         }),
       });
       if (!res.ok) {
@@ -193,6 +198,12 @@ function Grabber() {
             {selected.size === posts.length && posts.length > 0 ? "Deselect all" : "Select all"}
           </Button>
           <span className="text-sm text-muted-foreground">selected: {selected.size}</span>
+          <Input
+            placeholder="Exclude tags from ZIP (e.g. fluffy, safety)"
+            value={excludeTags}
+            onChange={(e) => setExcludeTags(e.target.value)}
+            className="w-64 max-w-xs"
+          />
           <Button onClick={downloadZip} disabled={!selected.size || downloading}>
             {downloading ? (
               <Loader2 className="w-4 h-4 mr-1 animate-spin" />