A TUI app that generates short stories and AI images from prompts using a language model.
Project description
StoryForge
StoryForge is a command-line tool that generates illustrated children's stories using AI language models. Simply provide a story prompt, and StoryForge will create both a short story and accompanying AI-generated images.
Supported AI Backends
- Google Gemini - Fully supported for story and image generation
- OpenAI - Fully supported for story and image generation
- Anthropic - Supported for story (text) generation only; image generation is not currently supported
Features
- 📖 Generate custom children's stories from simple prompts
- 🎨 Create AI illustrations with multiple art styles (chibi, realistic, cartoon, watercolor, sketch)
- ⚙️ Flexible story customization (age range, length, tone, theme, learning focus)
- 💾 Save stories and images with organized output directories
- 🖥️ Interactive terminal interface or direct CLI usage
- 📚 Context system for character consistency across stories
- ⏯️ Checkpoint system for resuming interrupted sessions
- 📝 Story extension - Continue existing stories with wrap-up or cliffhanger endings
- 🔄 Intelligent context summarization for efficient token usage with large context files
Configuration
For detailed configuration options, defaults, and examples see the full configuration reference: Configuration Documentation
StoryForge supports model configuration for OpenAI backends via the config file:
openai_story_model(default:gpt-5.2) - Model for story generationopenai_image_model(default:gpt-image-1.5) - Model for image generationimage_count(default:3) - Number of images to generate per story (1-5)
See the Configuration Documentation for all available options.
Generate a default config file
# Create config file in default location
storyforge config init
# Force overwrite an existing config file
storyforge config init --force
# Create config at custom location
storyforge config init --path /path/to/config.ini
The config file will be created at ~/.config/storyforge/storyforge.ini by default. You can override the location using the STORYFORGE_CONFIG environment variable.
Command Alias
For convenience, sf can be used as a shorter alias for all storyforge commands:
# These are equivalent
storyforge "A brave knight befriends a dragon"
sf "A brave knight befriends a dragon"
# Works with all commands
sf config init
sf continue
sf extend
sf --help
Requirements
Checkpoint System
StoryForge automatically saves your progress during story generation, allowing you to resume from any point if the process is interrupted or if you want to retry different options.
Resume from Previous Sessions
# Resume from a previous session (interactive selection)
storyforge continue
# Or use the --continue flag with main command
storyforge main --continue
This will show you the last 5 sessions and let you choose:
- For interrupted sessions: Resume from where you left off
- For completed sessions: Choose to:
- Generate new images with the same story
- Modify and regenerate the story
- Save the story as context for future use
- Start completely over with the same parameters
Checkpoint Storage
Checkpoint files are automatically stored in:
- Linux/macOS:
~/.local/share/storyforge/checkpoints/ - Windows:
%APPDATA%\storyforge\storyforge\checkpoints\
The system automatically cleans up old checkpoints, keeping the 15 most recent sessions. Stale active sessions (older than 24 hours) are automatically marked as failed/abandoned.
Example Workflow
# Start a story generation
storyforge "A dragon learning to dance"
# If interrupted, resume later with:
storyforge continue
# Select your session and choose where to resume
Story Extension
Create continuations of previously generated stories with the extend command. This is perfect for creating sequels or adding new chapters to existing stories.
Extend an Existing Story
# Interactive story selection from recent stories
storyforge extend
# The extend command will:
# 1. Show you a list of recently generated stories
# 2. Let you select which story to continue
# 3. Ask if you want to wrap up or leave a cliffhanger
# 4. Generate a continuation based on your choice
Example Extend Workflow
# First, generate a story
storyforge "A brave mouse named Max finds a magic acorn"
# Later, extend it with a continuation
storyforge extend
# Output:
# Recent stories:
# 1. "A brave mouse named Max finds a magic acorn" (2025-10-26 14:30)
# 2. "Two robots become friends" (2025-10-25 10:15)
# ...
#
# Select story to extend [1-5]: 1
#
# How should the continuation end?
# 1. Wrap up the story (satisfying conclusion)
# 2. Leave a cliffhanger (sets up next adventure)
#
# Select ending type [1/2]: 2
#
# Generating continuation...
The extended story will:
- Continue from where the original story left off
- Maintain character consistency and story context
- Be saved in a new timestamped output directory
- Include the original story context for reference
Story Chain Tracking
When you extend stories multiple times, StoryForge automatically tracks the complete "chain" of related stories. This makes it easy to see the full lineage and export complete story sagas.
View Story Chains
# When extending a story, StoryForge shows the full chain
storyforge extend
# Example output:
# Story Chain (3 parts):
# 1. "A wizard finds a mysterious artifact" (2025-11-05 10:00)
# 2. "The artifact reveals its power" (2025-11-05 12:00)
# 3. "The final confrontation" (2025-11-05 14:00) ← You are here
Export Complete Story Chains
Combine all parts of a story chain into a single readable file:
# Interactive selection from available chains
storyforge export-chain
# Export a specific chain by name match
storyforge export-chain -c wizard_artifact
# Specify custom output location
storyforge export-chain -c wizard_story -o my_complete_saga.txt
The exported file will contain:
- All story parts in chronological order
- Clear section dividers between parts
- Metadata about when each part was created
For detailed information about story chain tracking, see: Story Chain Tracking Documentation
Installation
Recommended: Using uv
uv tool install StoryForge
If you don't have uv:
curl -LsSf https://astral.sh/uv/install.sh | sh
Alternative: Using pipx
pipx install StoryForge
If you don't have pipx:
# macOS: brew install pipx
# Ubuntu/Debian: sudo apt install pipx
# Or: pip install pipx
Setup
Choose one of the supported AI backends and configure the corresponding API key:
Google Gemini
- Visit Google AI Studio to get your free Gemini API key
- Set the environment variable:
export GEMINI_API_KEY=your_api_key_here
OpenAI
- Get your API key from OpenAI Platform
- Set the environment variable:
export OPENAI_API_KEY=your_api_key_here
Anthropic (Experimental)
- Get your API key from Anthropic Console
- Set the environment variable:
export ANTHROPIC_API_KEY=your_api_key_here
Environment Variables
| Variable | Backend | Status | Description |
|---|---|---|---|
GEMINI_API_KEY |
Google Gemini | ✅ Fully Supported | Required for Gemini backend |
GEMINI_IMAGE_MODEL |
Google Gemini | 🔧 Optional | Override image model (e.g., gemini-2.5-flash-image) |
OPENAI_API_KEY |
OpenAI | ✅ Fully Supported | Required for OpenAI backend |
ANTHROPIC_API_KEY |
Anthropic | 🧪 Experimental | Required for Anthropic backend |
LLM_BACKEND |
All | Optional | Force specific backend (gemini, openai, anthropic) |
Note: StoryForge will automatically detect which backend to use based on available API keys. If multiple keys are set, you can specify which backend to use with the LLM_BACKEND environment variable.
Add environment variables to your shell profile (.bashrc, .zshrc, etc.) to make them permanent:
# Example for Gemini
echo 'export GEMINI_API_KEY=your_api_key_here' >> ~/.bashrc
source ~/.bashrc
# Example for OpenAI
echo 'export OPENAI_API_KEY=your_api_key_here' >> ~/.bashrc
source ~/.bashrc
Usage
Basic Story Generation
# Generate a story from a simple prompt
storyforge "Tell me a story about a robot learning to make friends"
Continue an Existing Story
# Extend a previously generated story
storyforge extend
Resume a Previous Session
# Resume from an interrupted or completed session
storyforge continue
Interactive prompts
The CLI is interactive and will ask for confirmation and decisions during the run (for images, story refinements, etc.).
Advanced Options
storyforge "A brave mouse goes on an adventure" \
--age-range preschool \
--length short \
--tone exciting \
--image-style cartoon \
--output-dir my_story \
-n 3
Available Options
- Age Range:
toddler,preschool,early_reader,middle_grade - Length:
flash,short,medium,bedtime - Style:
adventure,comedy,fantasy,fairy_tale,friendship,random - Tone:
gentle,exciting,silly,heartwarming,magical,random - Theme:
courage,kindness,teamwork,problem_solving,creativity,random - Image Style:
chibi,realistic,cartoon,watercolor,sketch
All Available Commands
# Generate a new story
storyforge "Your story prompt here" [options]
# Continue/extend an existing story
storyforge extend
# Export a complete story chain
storyforge export-chain [-c CONTEXT_NAME] [-o OUTPUT_FILE]
# Resume a previous session
storyforge continue
# Initialize configuration file
storyforge config init [--force] [--path PATH]
# Show help
storyforge --help
storyforge extend --help
storyforge export-chain --help
storyforge continue --help
storyforge config --help
Tab Completion
Enable tab completion for easier CLI usage:
storyforge --install-completion
Or manually for bash/zsh:
eval "$(storyforge --show-completion)"
Output
StoryForge creates timestamped directories containing:
story.txt- The generated story*.png- AI-generated illustrations- Organized by creation date/time
Development
For development setup, testing, and contributing guidelines, see DEV.md.
License
MIT License - see LICENSE file for details.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file storyforge-0.0.6.tar.gz.
File metadata
- Download URL: storyforge-0.0.6.tar.gz
- Upload date:
- Size: 93.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.5.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a6c2cb99ebf2dc7f1c89b7613da1b96845595e7c5d6a1010b1606fd27009dd9e
|
|
| MD5 |
5244c44c57425da75ed644e018b54c67
|
|
| BLAKE2b-256 |
4a040208c60dc68448af525f9dc104d5ed41a84634b539e957ca59b8013fcf24
|
File details
Details for the file storyforge-0.0.6-py3-none-any.whl.
File metadata
- Download URL: storyforge-0.0.6-py3-none-any.whl
- Upload date:
- Size: 66.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.5.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
3b866376283f957e5666ca02d500c72d133517fea940ce52bc4a9146edba0f64
|
|
| MD5 |
d966b9a967de27eec4fe4110a7fa611d
|
|
| BLAKE2b-256 |
0e69d42b1f968d8ed8b08a7bcbfd66990e81290af46f31bb30db8f5c036a4703
|