Skip to main content

A comprehensive toolkit for generating engaging and coherent stories with optional LLM support

Project description

📚 Story Development Toolkit

Tests Python License Version Author

A comprehensive Python toolkit for generating engaging and coherent stories with optional LLM support.

Story Development Toolkit provides tools for character creation, plot generation, dialogue writing, world building, and story coherence analysis — all in one package. NEW in v2.2.2: Complete CLI tool and pre-built story templates!


🌐 Language

This README is also available in:

Language File
🇮🇷 Persian (فارسی) README_FA.md

✨ Features

Feature Description
🎭 Character Creation Build complex characters with traits, goals, skills, fears, relationships
📚 Plot Generation Generate story structures for fantasy, mystery, romance, adventure, sci-fi
💬 Dialogue Writing Create natural dialogues with context-aware templates
🌍 World Building Design detailed fictional worlds with locations, cultures, rules, factions
🔍 Coherence Checking Identify plot holes, character inconsistencies, timeline issues
📊 Text Analysis Analyze readability, pacing, dialogue balance, vocabulary richness
🤖 LLM Support Optional integration with OpenAI, Anthropic, and local models
💾 Long-term Memory SQLite database for story persistence (v2.1.0)
📄 Export Formats PDF, EPUB, HTML, JSON, Markdown (v2.2.0)
📋 Pre-built Templates Hero's Journey, 3-Act, Mystery, Romance, Horror (v2.2.1)
💻 CLI Tool Command-line interface for easy usage (v2.2.2)

📦 Installation

# Basic installation (no LLM)
pip install story-toolkit

# With OpenAI support (GPT-4, GPT-3.5)
pip install story-toolkit[openai]

# With Anthropic support (Claude)
pip install story-toolkit[anthropic]

# With local LLM support (Ollama, llama.cpp)
pip install story-toolkit[local]

# With export formats (PDF, EPUB)
pip install story-toolkit[export]

# Full installation (all features)
pip install story-toolkit[all]

# Or install from source
git clone https://github.com/miladrezanezhad/story-toolkit.git
cd story-toolkit
pip install -e .

🚀 Quick Start

Basic Usage (No LLM)

from story_toolkit import StoryToolkit

# Create toolkit instance
toolkit = StoryToolkit()

# Create a story
story = toolkit.create_story(genre="fantasy", theme="courage")

# Add a hero
hero = toolkit.add_character_to_story(story, "Kai", "protagonist")
hero.add_trait("brave")
hero.add_goal("Save the kingdom")

# Generate dialogue
dialogue = toolkit.dialogue_gen.generate_dialogue(
    "Kai", "Villain", context="conflict"
)
for line in dialogue:
    print(line)

# Check coherence
report = toolkit.check_story_coherence(story)
print(f"Coherence Score: {report['overall_score']:.0%}")

Output:

Kai: I can't believe you would do this!
Villain: You left me no choice.
Kai: There's always a choice. You just chose wrong.
Coherence Score: 100%

Using Pre-built Templates (v2.2.1)

from story_toolkit import StoryToolkit

toolkit = StoryToolkit()

# Create story using Hero's Journey template (12 stages)
story = toolkit.use_template("hero_journey", genre="fantasy", theme="redemption")

# List all available templates
templates = toolkit.list_templates()
for t in templates:
    print(f"{t['name']}: {t['stage_count']} stages")

Export to PDF (v2.2.0)

from story_toolkit import StoryToolkit
from story_toolkit.exporters import PDFExporter, ExportConfig, PDFStyle

toolkit = StoryToolkit()
story = toolkit.create_story("fantasy", "courage")

# Export as PDF
config = ExportConfig(title="My Story", author="Me", pdf_style=PDFStyle.PRINT)
exporter = PDFExporter(config)
exporter.export(story, "my_story.pdf")

Using CLI Tool (v2.2.2)

# Create a new story
story-toolkit story new --genre fantasy --theme courage

# List available templates
story-toolkit template list

# Use a template
story-toolkit template use hero_journey --output my_story.json

Advanced Usage with LLM

from story_toolkit import StoryToolkit
from story_toolkit.llm import LLMFactory, LLMProvider

# Create LLM backend (Mock for testing)
llm = LLMFactory.create_backend(provider=LLMProvider.MOCK)
toolkit = StoryToolkit(llm_backend=llm)

# Generate advanced dialogue
dialogue = toolkit.dialogue_gen.generate_dialogue(
    "Kai", "Villain",
    context="final_battle",
    use_advanced=True,
    style="dramatic",
    num_lines=8
)

# Check LLM status
print(f"LLM Status: {toolkit.get_llm_status()}")

🤖 LLM Backends

v2.0.0+ supports multiple LLM providers:

Provider Installation API Key Required
Mock Included ❌ No (for testing)
OpenAI pip install story-toolkit[openai] ✅ Yes
Anthropic pip install story-toolkit[anthropic] ✅ Yes
Local (Ollama) pip install story-toolkit[local] ❌ No (free)

📋 Pre-built Templates (v2.2.1)

Template Name Genre Stages Description
hero_journey Fantasy/Adventure 12 Campbell's monomyth
three_act General 3 Standard 3-act structure
mystery_clues Mystery 5 Detective story structure
romance_beat Romance 15 15-beat romance structure
horror_cycle Horror 6 Classic horror structure

📄 Export Formats (v2.2.0)

Format Description Usage
PDF Print, Manuscript, eBook styles exporter.export(story, "file.pdf")
EPUB eBook for Amazon/Kobo/Apple exporter.export(story, "file.epub")
HTML 4 templates (Modern, Classic, Dark, Minimal) exporter.export(story, "file.html")
JSON Raw data save_story(story, "file.json")
Markdown Readable text export_to_markdown(story, "file.md")

💾 Memory Storage (v2.1.0)

from story_toolkit import StoryToolkit

# Enable SQLite memory
toolkit = StoryToolkit(memory_backend="sqlite", db_path="stories.db")

# Create story with auto-save
story = toolkit.create_story("fantasy", "courage", save_to_memory=True)

# Add events to timeline
toolkit.add_event(1, "Hero discovers the map", "plot", 9)

# View timeline
for event in toolkit.get_timeline():
    print(f"Chapter {event.chapter}: {event.description}")

# List all saved stories
stories = toolkit.list_stored_stories()

📁 Project Structure

story_toolkit/
├── story_toolkit/          # Main Python package
│   ├── core/               # Story engine, Character, Plot, WorldBuilder
│   ├── generators/         # Character, Plot, Dialogue, Story generators
│   ├── nlp/                # Coherence checker and Text analyzer
│   ├── llm/                # LLM layer (OpenAI, Anthropic, Local, Mock)
│   ├── memory/             # SQLite long-term memory
│   ├── exporters/          # PDF, EPUB, HTML, Bionic exporters
│   ├── templates/          # Pre-built story templates
│   ├── cli/                # Command-line interface
│   └── utils/              # Helper functions
├── docs/                   # Documentation (English & Persian)
├── examples/               # Usage examples
├── tests/                  # Unit tests
├── requirements.txt        # Dependencies
└── setup.py                # Package setup

📖 Documentation

Full documentation is available in two languages:

Language Link
🇬🇧 English docs/eng/index.html
🇮🇷 فارسی docs/fa-ir/index.html

Documentation Pages

  • Quick Start Guide — Build your first story in 5 minutes
  • API Reference — Complete documentation for all classes and methods
  • LLM Integration Guide — How to use OpenAI, Anthropic, and local models
  • CLI Guide — Command-line interface usage
  • Examples — Simple, complete, and advanced usage examples

🧪 Running Tests

# Run all tests
pytest tests/ -v

# Core module tests
python -m tests.test_core

# Generator tests
python -m tests.test_generators

# NLP tool tests
python -m tests.test_nlp

# LLM layer tests
python -m tests.test_llm_quick.test_quick_verify

# Comprehensive test suite
python tests/test_suite.py

All tests should pass:

✅ StoryEngine tests passed!
✅ Character tests passed!
✅ Plot tests passed!
✅ WorldBuilder tests passed!
✅ LLM layer tests passed!
✅ Memory tests passed!
✅ Exporter tests passed!
✅ CLI tests passed!

🎮 Usage Examples

# Simple example
python -m examples.simple_example

# Complete demo
python -m examples.example

# Advanced features (with LLM)
python -m examples.advanced_example

🛠️ Core Components

Character Development

from story_toolkit.core.character import Character

hero = Character(name="Elena", age=32, role="protagonist")
hero.add_trait("brave")
hero.add_skill("sword_mastery")
hero.add_relationship("Villain", "enemy", strength=9)
hero.advance_arc()  # initial → challenged → transformation → new_equilibrium

World Building

from story_toolkit.core.world_builder import WorldBuilder

world = WorldBuilder()
world.create_world("Eldoria", "fantasy")
world.add_location("Crystal City", "Ancient metropolis", "city")
world.add_rule("magical", "Only eclipse-born can wield magic")
world.add_faction("Shadow Guild", "Secret organization", goals=["control_magic"])

Plot Generation

from story_toolkit.generators.plot_generator import PlotGenerator

gen = PlotGenerator()
plot = gen.generate_plot("mystery", complexity=4)
print(f"Estimated chapters: {plot['estimated_length']['estimated_chapters']}")
print(f"Estimated words: {plot['estimated_length']['estimated_words']:,}")

Coherence Checking

from story_toolkit.nlp.coherence_checker import CoherenceChecker

checker = CoherenceChecker()
report = checker.generate_coherence_report(story_data)

if report['plot_holes']:
    print("Plot holes found:")
    for hole in report['plot_holes']:
        print(f"  - {hole}")

for rec in report['recommendations']:
    print(f"💡 {rec}")

🔧 Requirements

  • Python 3.11 or higher
  • Dependencies listed in requirements.txt:
    • nltk>=3.8.1
    • spacy>=3.7.0
    • textblob>=0.17.1
    • pydantic>=2.5.0
    • pyyaml>=6.0

Optional Dependencies

Backend/Feature Package
OpenAI openai>=1.0.0
Anthropic anthropic>=0.18.0
Local LLM ollama>=0.1.0
PDF Export reportlab>=4.0
EPUB Export ebooklib>=0.18

🔄 Version History

Version Features Python
2.2.2 CLI Tool, Complete integration 3.11+
2.2.1 Pre-built Templates (5 templates) 3.11+
2.2.0 Exporters (PDF, EPUB, HTML, Bionic) 3.11+
2.1.0 SQLite Memory, Timeline, Storage 3.11+
2.0.0 LLM Layer (OpenAI, Anthropic, Local) 3.11+
1.0.0 Core Features (Character, Plot, World) 3.8+

📄 License

This project is licensed under the MIT License - see the LICENSE file for details.


👤 Author

Milad Rezanezhad


🤝 Contributing

Contributions are welcome! Feel free to:

  1. Fork the repository
  2. Create a feature branch (git checkout -b feature/amazing-feature)
  3. Commit your changes (git commit -m 'Add amazing feature')
  4. Push to the branch (git push origin feature/amazing-feature)
  5. Open a Pull Request

🌟 Support

If you find this project useful, please consider giving it a ⭐️ on GitHub!


Built with ❤️ for writers and developers


Version 2.2.2 Highlights

Feature Description
🎯 Badges Python 3.11+, License, Version 2.2.2, Author
🤖 LLM Support OpenAI, Anthropic, Local (Ollama)
💾 Memory SQLite long-term persistence
📄 Exporters PDF, EPUB, HTML (4 templates), Bionic Reading
📋 Templates 5 pre-built story structures
💻 CLI Tool Full command-line interface
📖 Docs English & Persian documentation
🧪 Tests 100% pass rate (16+ tests)
🔄 Backward Compatible v1.0.0 code works unchanged

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

story_toolkit-2.2.2.tar.gz (86.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

story_toolkit-2.2.2-py3-none-any.whl (105.8 kB view details)

Uploaded Python 3

File details

Details for the file story_toolkit-2.2.2.tar.gz.

File metadata

  • Download URL: story_toolkit-2.2.2.tar.gz
  • Upload date:
  • Size: 86.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.2

File hashes

Hashes for story_toolkit-2.2.2.tar.gz
Algorithm Hash digest
SHA256 b71ca95f0e56fa6d20e1c6ef7984669f6fd73a284ccca7b3c657847e13c44358
MD5 20028e5978eac54ec1078378839baca8
BLAKE2b-256 c061aa221d2514a31e0c161da039507d8cd4826fa3fec745aa79fff306ba592e

See more details on using hashes here.

File details

Details for the file story_toolkit-2.2.2-py3-none-any.whl.

File metadata

  • Download URL: story_toolkit-2.2.2-py3-none-any.whl
  • Upload date:
  • Size: 105.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.2

File hashes

Hashes for story_toolkit-2.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 19bb9fc3cba26f814390dcd2eea3ae6806a735911ce330c4d863d33e2e7eec61
MD5 c5880e2c0186667e83fb355dbd145b6a
BLAKE2b-256 58196c2055a7db50053ea70d1118f7e64314cdd01cb8238989413d4c123606da

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page