Image Caption Tool

A Python app for captioning images with both web and command-line interfaces. This will result in text files output with the same name as the image. This is handy for prepping for image training.

Setup

uv sync

Usage

Web Interface (Default)

uv run python main.py

The web interface is available on http://127.0.0.1:7860

• Upload multiple images
• View thumbnails
• Add captions for each image
• Set trigger words/tags for all images
• Save captions as .txt files

Command Line Interface

# Caption all images in a directory
uv run python main.py --cli --dir /path/to/images

# Caption specific files
uv run python main.py --cli --files image1.jpg image2.png

# Add trigger words to all captions
uv run python main.py --cli --dir /path/to/images --trigger "anime, masterpiece"

Features

• Supports common image formats (jpg, png, bmp, gif, tiff)
• Saves captions as .txt files with matching filenames
• Optional trigger words/tags for all images
• Both GUI and CLI interfaces