Hyingerrr / spirited

AI-agent应用,基于GPT、langchain、function calling、Stable diffusion等的AI儿童绘本生成

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SPirited 一款AI儿童绘本生成程序

img.png

项目简介

一句话生成一本书。

这是一个基于 GPT、Langchain、Function Calling 和 Stable Diffusion 技术实现的 AI 代理应用程序,旨在生成儿童绘本故事。

功能特点

  • 一句话生成一本书:用户只需输入一句话便可生成一本丰富多彩的儿童故事书。

  • 随声朗读:Coming soon。(生成符合故事情节的文字、语音朗读,有种身临其境的感觉。)

  • GPT 驱动:利用 GPT(OpenAI、LLama2、ChatGLM等) 技术,生成富有创意的绘本故事,使其充满乐趣和想象力。

  • Function Calling 支持:使用 Function Calling 技术,可以自定义绘本中的交互和功能,增加互动性。

  • Stable Diffusion 稳定扩散:确保生成的故事内容具有一致性和连贯性,提供高质量的绘本。

如何使用

以下是使用本项目的基本步骤:

  1. 安装依赖
    pip install -r requirements.txt
  1. 配置参数:在配置文件conf.env中设置项目所需的参数,ACTIVELOOP_TOKEN、DATASET_PATH、STABILITY等。

  2. 运行程序: 运行程序并生成儿童绘本故事。启动命令:

    streamlit run app.py

示例

# 示例代码,展示如何使用程序生成绘本
from book_maker import BookMaker
from dotenv import load_dotenv

load_dotenv('conf.env')

inputs = 'a cute girl is running near the river and singing and jumping'
    model_name = 'gpt-3.5-turbo-0613'
    style = 'Monet, impressionist art style, loose brushstrokes, vibrant colors, painted, painted light'
    app = BookMaker(inputs, model_name, style)

    story_pages = ["Title: Lily's Joyful Adventure\n\n",
                   "\nLily skips along the river's bend,\nHer laughter echoing with every bend.\nShe twirls and sings "
                   "a merry tune,\nUnder the golden sun and a sky so blue.\n\n",
                   "\nThe river sparkles with a magical glow,\nAs Lily's feet dance to and fro.\nShe jumps over "
                   "pebbles, oh so small,\nAnd splashes in the water, having a ball.\n\nAmidst the green trees and "
                   "flowers so bright,\nLily's joy shines with all her might.\nShe sings with the birds, "
                   "their melodies entwined,\nAs she explores the wonders she's destined to find.\n\n",
                   "\nThe wind whispers secrets through the trees,\nAs Lily's laughter floats on the breeze.\nShe "
                   "leaps over rocks, as graceful as a fawn,\nAnd makes friends with creatures, big and small.\n\nA "
                   "bunny hops by, its fluffy tail so spry,\nWhile a squirrel chatters, scampering up high.\nLily "
                   "giggles and joins in their playful game,\nAs they all frolic together, without any shame.\n\nWith "
                   "each bound and skip, she fills the air,\nWith joy and happiness, beyond compare.\nLily's "
                   "adventure, so full of glee,\nIs a reminder for all to be wild and free.\n\nEnd of story."
                   ]
    # app.make_pages_prompt(story_pages)
    app.runner()

贡献

我们欢迎贡献者加入项目并提出改进意见。如果你有任何想法或问题,请提交 Issue 或发起 Pull Request。

许可证

此项目采用 MIT 许可证。

如果您对这个项目有任何问题或建议,请随时联系我们。

About

AI-agent应用,基于GPT、langchain、function calling、Stable diffusion等的AI儿童绘本生成

License:MIT License


Languages

Language:Python 100.0%