├── .env.example ├── .gitignore ├── Docs ├── BlockDiagram.drawio ├── BlockDiagram.drawio.svg └── Models.md ├── Evaluate.py ├── ExamplePrompts ├── Example1.txt ├── Example1 │ ├── Output_2024-06-20.md │ ├── Output_2024-06-24.md │ ├── Output_2024-06-26.md │ ├── Output_2024-06-27.md │ └── Prompt.txt ├── Example2 │ ├── Output_2024-06-27_Gemini1.5Flash.md │ ├── Output_2024-06-27_Gemini1.5Pro.md │ └── Prompt.txt └── ShortDebuggingStory │ ├── Prompt.txt │ ├── PromptFrench.txt │ └── TranslationExample.md ├── LICENSE ├── README.md ├── Todo.md ├── Tools └── Test.py ├── Write.py ├── Writer ├── Chapter │ ├── ChapterDetector.py │ ├── ChapterGenSummaryCheck.py │ └── ChapterGenerator.py ├── Config.py ├── Interface │ ├── OpenRouter.py │ ├── Wrapper.py │ └── ZAI.py ├── LLMEditor.py ├── NovelEditor.py ├── Outline │ └── StoryElements.py ├── OutlineGenerator.py ├── PrintUtils.py ├── Prompts.py ├── Scene │ ├── ChapterByScene.py │ ├── ChapterOutlineToScenes.py │ ├── SceneOutlineToScene.py │ └── ScenesToJSON.py ├── Scrubber.py ├── Statistics.py ├── StoryInfo.py └── Translator.py └── requirements.txt /.env.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/.env.example -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/.gitignore -------------------------------------------------------------------------------- /Docs/BlockDiagram.drawio: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Docs/BlockDiagram.drawio -------------------------------------------------------------------------------- /Docs/BlockDiagram.drawio.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Docs/BlockDiagram.drawio.svg -------------------------------------------------------------------------------- /Docs/Models.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Docs/Models.md -------------------------------------------------------------------------------- /Evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Evaluate.py -------------------------------------------------------------------------------- /ExamplePrompts/Example1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example1.txt -------------------------------------------------------------------------------- /ExamplePrompts/Example1/Output_2024-06-20.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example1/Output_2024-06-20.md -------------------------------------------------------------------------------- /ExamplePrompts/Example1/Output_2024-06-24.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example1/Output_2024-06-24.md -------------------------------------------------------------------------------- /ExamplePrompts/Example1/Output_2024-06-26.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example1/Output_2024-06-26.md -------------------------------------------------------------------------------- /ExamplePrompts/Example1/Output_2024-06-27.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example1/Output_2024-06-27.md -------------------------------------------------------------------------------- /ExamplePrompts/Example1/Prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example1/Prompt.txt -------------------------------------------------------------------------------- /ExamplePrompts/Example2/Output_2024-06-27_Gemini1.5Flash.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example2/Output_2024-06-27_Gemini1.5Flash.md -------------------------------------------------------------------------------- /ExamplePrompts/Example2/Output_2024-06-27_Gemini1.5Pro.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example2/Output_2024-06-27_Gemini1.5Pro.md -------------------------------------------------------------------------------- /ExamplePrompts/Example2/Prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/Example2/Prompt.txt -------------------------------------------------------------------------------- /ExamplePrompts/ShortDebuggingStory/Prompt.txt: -------------------------------------------------------------------------------- 1 | Please write a very short (2 chapter) story about a sci-fi city. -------------------------------------------------------------------------------- /ExamplePrompts/ShortDebuggingStory/PromptFrench.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/ShortDebuggingStory/PromptFrench.txt -------------------------------------------------------------------------------- /ExamplePrompts/ShortDebuggingStory/TranslationExample.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/ExamplePrompts/ShortDebuggingStory/TranslationExample.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/README.md -------------------------------------------------------------------------------- /Todo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Todo.md -------------------------------------------------------------------------------- /Tools/Test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Tools/Test.py -------------------------------------------------------------------------------- /Write.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Write.py -------------------------------------------------------------------------------- /Writer/Chapter/ChapterDetector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Chapter/ChapterDetector.py -------------------------------------------------------------------------------- /Writer/Chapter/ChapterGenSummaryCheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Chapter/ChapterGenSummaryCheck.py -------------------------------------------------------------------------------- /Writer/Chapter/ChapterGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Chapter/ChapterGenerator.py -------------------------------------------------------------------------------- /Writer/Config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Config.py -------------------------------------------------------------------------------- /Writer/Interface/OpenRouter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Interface/OpenRouter.py -------------------------------------------------------------------------------- /Writer/Interface/Wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Interface/Wrapper.py -------------------------------------------------------------------------------- /Writer/Interface/ZAI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Interface/ZAI.py -------------------------------------------------------------------------------- /Writer/LLMEditor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/LLMEditor.py -------------------------------------------------------------------------------- /Writer/NovelEditor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/NovelEditor.py -------------------------------------------------------------------------------- /Writer/Outline/StoryElements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Outline/StoryElements.py -------------------------------------------------------------------------------- /Writer/OutlineGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/OutlineGenerator.py -------------------------------------------------------------------------------- /Writer/PrintUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/PrintUtils.py -------------------------------------------------------------------------------- /Writer/Prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Prompts.py -------------------------------------------------------------------------------- /Writer/Scene/ChapterByScene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Scene/ChapterByScene.py -------------------------------------------------------------------------------- /Writer/Scene/ChapterOutlineToScenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Scene/ChapterOutlineToScenes.py -------------------------------------------------------------------------------- /Writer/Scene/SceneOutlineToScene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Scene/SceneOutlineToScene.py -------------------------------------------------------------------------------- /Writer/Scene/ScenesToJSON.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Scene/ScenesToJSON.py -------------------------------------------------------------------------------- /Writer/Scrubber.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Scrubber.py -------------------------------------------------------------------------------- /Writer/Statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Statistics.py -------------------------------------------------------------------------------- /Writer/StoryInfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/StoryInfo.py -------------------------------------------------------------------------------- /Writer/Translator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/datacrystals/AIStoryWriter/HEAD/Writer/Translator.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | ollama 2 | termcolor 3 | google.generativeai 4 | python-dotenv --------------------------------------------------------------------------------