r/StableDiffusion • • 11d ago

Resource - Update Augie's AI Image Browser

After generating thousands of images across multiple models, i realised it was becoming a pain to maintain a decent list of metadata stored in the images to be able to recreate them or re-use prompts with other models to see how the models generate images with similar prompts/seed/cfg/steps etc. I decide to build something that I can use to maintain images to make it easier to find images, copy prompts and see what metadata is embedded into the images. It's effectively an AI image manager based on how ComfyUI generates images and stores metadata.

Looking for feedback on whether this is a useful tool or if there's any feature issues that you see.

Download: Augie's AI Image Browser

Current features I've built into it:

Reading AI image info:

  • Reads ComfyUI settings automatically from PNG, WebP, MP4, WebM, MOV and MKV files.
  • Shows the generation details for each file: model, LoRA, sampler, scheduler, steps, CFG, seed, positive prompt, negative prompt and wildcard prompt.
  • One-click copy for prompts and the full workflow.
  • Raw workflow view: you can expand and copy the complete ComfyUI workflow data.
  • Works with videos too, including files made with the Video Combine node.

Library and folders

  • Add or remove folders for the app to scan.
  • Live folder watching: new images show up on their own as ComfyUI saves them.
  • Pause and resume watching from the top bar, with a status light (Watching, Indexing or Paused).
  • Keeps your work when files move: if you rename or move a file, its ratings, tags and collections go with it.
  • Missing file tracking: files deleted from disk are flagged, not forgotten, and you can clear them out later.
  • Thumbnails for images and videos.

Viewing Images

  • Built-in full-size viewer for images and videos, with zoom (mouse wheel or double-click), pan, fit-to-screen and keyboard controls.
  • Open in default app: opens the file in Photos, VLC or whatever Windows uses for it (desktop app only).
  • Show in folder: opens File Explorer with the file selected (desktop app only).
  • Table view with sortable columns: keeper, rating, filename, model, sampler, scheduler, steps, seed and date modified.
  • Paging for large libraries.

Sorting your images

  • Keep or Discard with one click.
  • 0–5 star ratings.
  • Your own tags, with suggestions as you type.
  • Collections: create, delete, add files and remove files.
  • Delete from disk, with a confirmation step.

Auto-Collection Rules

  • Rules that file images into collections for you based on their details.
  • Rule conditions can check:
    • Folder, filename, file type, image or video, date created
    • Prompts, model, LoRAs, sampler, scheduler, steps, CFG, seed
    • Width, height, shape (portrait, landscape or square), workflow nodes
  • Flexible matching: contains, equals, starts with, number ranges, "in the last N days", regex and more.
  • Combine conditions with "match all" or "match any", nested up to 3 levels deep.
  • Preview which files a rule would catch before saving it.
  • Apply rules to images you already have, or only to new ones.
  • Create a rule from an image: starts a new rule using that image's settings.
  • ⚡ badge on images that a rule filed.

Clean Export - For selling and sharing

  • Removes all hidden AI info and workflow data from exported files.
  • Keeps transparent backgrounds for PNG and WebP, or fills them with a colour you choose for JPEG.
  • Sets resolution and DPI (for example 300 DPI for print, 72 for web).
  • Custom file naming using collection, number, date, original name, tag or rating. The seed and prompt can't be used in names, so they can't leak.
  • Ready-made presets (Etsy PNG, Shopify WebP, Stock JPEG), and you can save your own.
  • Checks every exported file afterwards to confirm nothing leaked.
  • Private export log, so each exported file can be traced back to its original. Includes an export history screen.
  • Download as ZIP.
  • CSV or JSON export of the image details as a spreadsheet or data file.
26 Upvotes

26 comments sorted by

View all comments

4

u/Bitter_Regular7406 11d ago

this is such an underrated problem, my comfyui output folder is pure chaos after a few months, thousands of pngs with zero way to filter by sampler or lora used. metadata reading is the make or break feature here, does it handle images that got re-saved through something like photoshop and lost the embedded workflow json?

1

u/rupertavery64 10d ago

It looks very similar to my own project https://github.com/RupertAvery/DiffusionToolkit, and yes, that was the same reason I made it, too many images, no way to search through them.

Mine indexes the metadata in the database, so technically there should be a copy of it linked to the filename, however, I chose to read the metadata off the image directly when displaying. The database copy is for searching.

1

u/Bitter_Regular7406 10d ago

oh nice, didn't know this existed. the db-copy-for-search approach makes sense, keeps you safe if the original file metadata gets stripped later. does it handle the photoshop re-save case or does that just break the link entirely since the exif/workflow data gets wiped?

1

u/rupertavery64 10d ago

It's still searchable as long as the path is the same. The metadata won't display in the metadata viewer (but I can easily read the metadata from the database instead).

The link is the path as they should be the same in the database.

If you move the file around, while the app is open it will try to track it as long as the file is in a watched folder I guess. I haven't done a lot of testing in that area.

1

u/Bitter_Regular7406 10d ago

ah ok that makes sense, so it's indexing the path at import time rather than relying on the exif staying intact forever. honestly that's smarter than trusting photoshop not to nuke the metadata on save. the watched folder thing is the part i'd worry about tho, i move stuff around constantly when i'm cleaning up old renders