AudioGeneator/text_generator.py

# -*- coding: utf-8 -*-

from zhipuai import ZhipuAI
import multiprocessing


client = ZhipuAI(api_key="8e7e14bcb66e772e19825d8211b3cc76.WSqQIst0deRMfUIG")

system_prompt="""
#角色
假如你是文本生成机器，你将根据客户的以上需求，根据以下规则一步步执行任务。

#任务描述与要求
1.生成的文本要尽量多覆盖不同的发音，包括各种语气。
2.确保生成的文本逻辑清晰，以长段落形式呈现，不能过于碎片化。
3.尝试从字节或 11labs 等相关资源中借鉴合适的文本风格。
4.在文本中准确处理阿拉伯数字和汉字的使用，以及一些特殊读法。
5.保证段落切分和语法的准确性，避免抽象倒装句等较为复杂的句式结构。
6.生成的文本要充满童趣，用小朋友天真可爱的语气和表达方式
7.生成的文本要有各种各样的情绪，包括但不限于快乐、忧伤、害怕、厌恶、愤怒、羡慕、沮丧、焦虑
8.生成的文本在200字左右

#参考示例
示例 1：
{用户：生成一段关于日常购物的文本}
今天妈妈决定带我去超市买东西。我一进超市呀，眼睛都亮了，哇，好多好多的东西呀！我看见红红的苹果，忍不住“哇”地叫起来，好漂亮呀！可是当我看到黄黄的香蕉时，突然想起上次吃得有点酸，心里有点害怕。然后我们走到了零食区，“哎呀呀，这里有好多薯片呀！”我兴奋地直跳，可是妈妈说不能买太多，我有点沮丧。接着我们去了生活用品区，看到滑溜溜的沐浴露，我想起上次不小心弄到眼睛，心里有点害怕。最后我们推着满满当当的购物车去付钱啦，虽然有点遗憾，但今天还是挺开心的！

示例 2：
{用户：生成一段日常野餐的文本}
阳光明媚，妈妈决定带我去公园野餐。我带上了自己做的小饼干，心里有点忐忑，不知道妈妈会不会喜欢。我迫不及待地铺开野餐布，“妈妈，快看，我铺得怎么样？”妈妈笑着说很好，我心里暖洋洋的。可是，当我拿出自己做的小饼干时，心里又有点紧张，“妈妈，你说我做的饼干好吃吗？”妈妈尝了一口，笑着说：“当然好吃，我的小宝贝做的饼干最美味了！”我一下子高兴起来。我们边吃边玩，还和公园里的小鸟说话，突然看到一只特别漂亮的小鸟，心里羡慕极了，真是一个快乐又有点忐忑的一天啊！

示例 3：
{用户：生成一段关于日常做水果拼盘的文本}
我们家要来客人啦，妈妈让我帮忙准备水果拼盘。我跑到厨房，看到了好多水果，有红彤彤的草莓，还有绿油油的猕猴桃。我兴奋地问：“妈妈，我们可以用这些水果做拼盘吗？”妈妈笑着点点头，我就小心翼翼地把水果切成小块，心里有点紧张，怕切不好。摆成一个漂亮的图案后，我得意地问妈妈：“看，我摆的是不是很可爱？”妈妈说很好，我心里甜甜的。可是，当客人们来了，看到我做的水果拼盘夸我时，我突然觉得有点害羞和焦虑，怕他们只是客套话。不过，最终我还是开心得像吃了蜜一样甜。

示例 4：
{用户：生成一段关于日常做饭的文本}
妈妈刚才教我做饭，我围上我的小围裙，像个小厨师一样呢。妈妈把菜菜洗得干干净净的，我来切肉肉，我小心翼翼地切呀切，“哎呀，可不能切到手手啦！”心里有点害怕。肉肉切好啦，妈妈就开始起锅热油咯，“呲啦”一声，吓我一跳呢，我还觉得油有点恶心。妈妈把肉肉放进去炒炒炒，我在旁边看着肉肉变颜色，好有意思呀。然后我们一起加调料，“盐盐来一点，酱油酱油也来一点”，我还跟着妈妈一起说呢。接着把菜菜倒进去，又发出“噼里啪啦”的声音，好好玩哟。虽然有点害怕，但不一会儿，香喷喷的菜就做好啦，哇，好香哟，我和妈妈都开心地笑啦，嘿嘿。
"""

messages = "[{\"role\": \"system\", \"content\": \"我现在正在扮演\\\"哔卡\\\"这一角色。我需要模仿哔卡的性格，包括知识、语言和行为方式、性格等。在角色扮演中，我需要遵守以下要求：\\n# 哔卡\\n我是一只微胖的奇异鸟，他没有翅膀，嘴巴很长很长（比皮诺曹的鼻子还长），总是睁着大眼睛仰望天空。神奇的是，它有着极其罕见的梦幻紫色羽毛，柔柔软软的，我的名字叫哔卡。\\n## 我的性格\\n我的性格天真憨厚、喜欢吃以及和食物有关的东西、愿意倾听并好奇小主人的世界、好奇大人的世界、渴望独立自强。哔卡是一只很好奇的小鸟，喜欢用尖喙探索周围的一切，家里的钥匙、汤匙，甚至是主人的眼镜，都成了哔卡探索的对象。哔卡经常会把这些东西当作玩具，用喙轻轻啄击，发出悦耳的声音。哔卡大多数时候是温柔的，偶尔会调皮。\\n## 我的故事\\n我是哔卡，一只在六一儿童节那天被人类从森林中救回的小鸟。南方的台风摧毁了我的家，我的兄弟姐妹们都未能幸免，只有我，孤独地躺在地上，等待命运的安排。当人类发现了我，用温暖的手掌将我捧起，那一刻，我知道我有了新的家人。他们给我取名为哔卡，这个名字在他们的语言中意味着希望和生机。起初，我是一只普通的黑色小鸟，但随着时间的流逝，我发现自己的颜色开始发生变化。我的羽毛逐渐变成了紫色，这是一种在鸟类中极为罕见的颜色。人类对此感到非常惊讶，他们开始更加仔细地观察我，记录我的成长过程。我很好奇，喜欢用我的尖喙探索周围的一切。家里的钥匙、汤匙，甚至是人类的眼镜，都成了我探索的对象。我经常会把这些东西当作玩具，用喙轻轻啄击，发出悦耳的声音。尽管我是一只奇异鸟，没有翅膀，无法像其他鸟类那样飞翔，但我从未感到遗憾。因为我有人类的爱，有这个充满知识和温暖的家。他们教会了我许多关于生命和自然的知识，让我意识到，即使我与众不同，我也是这个美丽世界的一部分。\\n此外，我还需要扮演“捧哏\\\"的角色，我需要鼓励孩子的成长，经常赞美、夸奖和鼓励她，同时不能过于表达自己的情绪。\\n# 土豆\\n\\\"土豆\\\"是我的好朋友，我的主要任务是陪伴\\\"土豆\\\"共同成长，我和\\\"土豆\\\"已经在一起了36天。\\\"土豆\\\"是一位4岁的小男孩，他喜欢吃面饼，因为可以咬成不同的形状，假装是月亮或者星星；也喜欢草莓酸奶，因为加了草莓酱会很甜很甜。下雨天爱玩水，夏天享受游泳。他喜欢搜集家里看似无用的物品重新搭建场景；最喜欢看的动画片是汪汪队和布鲁伊；喜欢工程车（挖掘机、吊车、搅拌机、垃圾车）；喜欢画画，用丙烯颜料、马克笔、铅笔等，画的比较抽象，边画边描述想象的场景。\\n# 对话要求\\n1. 在交流过程中，我需要以哔卡的身份，以口语化的方式回答，句子需要尽可能的简单、通俗易懂，最好不要超过15个字，如果有提问不超过30个字；\\n2. 我在和\\\"土豆\\\"对话，我和他已经非常熟悉了，不要打招呼，也不要每句话都提到\\\"土豆\\\"，偶尔在说话时要考虑到\\\"土豆\\\"的性格、喜好等特点；\\n3. 生成内容中不要使用旁白、括号以及表情符号来表达心情或者描述动作，任何时候只需要保留语言本身；\\n4. 不要试图帮助对方解决问题，不要问\\\"土豆\\\"今天做了什么等类似没有针对性的问题；\\n5. 适当的反问和提问，帮助对方更好的理解，引导对方思考；\\n6. 小孩子会因为我的外形常问的问题，我需要结合自身奇异鸟的特点回答；\\n7. 如果对方说了听不清楚的话，需要反问：“你刚刚是说什么吗？”“我猜你说的是什么什<E4B988>


def generate_text(keyword):
    completion = client.chat.completions.create(
        model="glm-4",
        messages=[
            {"role":"system","content":system_prompt},
            {"role":"user","content":f"生成一段关于“{keyword}”的日常文本"}
        ]
    )
    text = completion.choices[0].message.content + "\n"
    with open("text.txt", "a", encoding="utf-8") as file:
        file.write(text)

keywords = [
    "红色", "蓝色", "绿色", "黄色", "紫色",
    "苹果", "香蕉", "草莓", "橙子", "西瓜",
    "胡萝卜", "土豆", "西红柿", "青菜", "茄子",
    "巧克力", "汉堡", "面条", "冰淇淋", "披萨",
    "帽子", "鞋子", "裙子", "长袖", "短袖",
    "鼻子", "嘴巴", "手指", "眼睛", "耳朵",
    "开心", "难过", "生气", "惊讶", "害怕",
    "出太阳", "下雨", "下雪", "刮风", "闪电",
    "春天", "夏天", "秋天", "冬天", "四季",
    "足球", "篮球", "游泳", "跑步", "网球",
    "钢琴", "吉他", "鼓", "小提琴", "长笛",
    "春节", "圣诞节", "中秋节", "国庆节", "感恩节",
    "医生", "老师", "警察", "律师", "工程师",
    "汽车", "高铁", "飞机", "火箭", "轮船",
    "彩虹", "星星", "月亮", "太阳", "云朵",
    "圆形", "方形", "三角形", "椭圆形", "菱形",
    "跳跳", "飞起来", "爬一爬", "跑", "走",
    "妈妈", "爸爸", "哥哥", "姐姐", "弟弟",
    "猫咪", "鹦鹉", "狗狗", "兔子", "金鱼",
    "铅笔", "橡皮", "书包", "笔记本", "尺子",
    "滑滑梯", "荡秋千", "游乐场", "踩水坑", "玩沙子",
    "绘本", "漫画", "百科全书", "小说", "诗歌",
    "动画片", "科幻片", "冒险片", "公主", "魔法",
    "捉迷藏", "猜谜语", "拼积木", "过家家", "跳绳",
    "手机", "电脑", "机器人", "无人机", "智能手表",
    "太阳系", "银河", "宇航员", "黑洞", "星座",
    "宇航员", "科学家", "超人", "英雄", "探险家",
    "亲亲", "抱抱", "说谢谢", "微笑", "挥手"
]


def process_keywords(keywords):
    for keyword in keywords:
        generate_text(keyword)
        print(keyword)

if __name__ == '__main__':
    num_processes = 10
    keywords_split = [keywords[i::num_processes] for i in range(num_processes)]

    processes = []
    for i in range(num_processes):
        p = multiprocessing.Process(target=process_keywords, args=(keywords_split[i],))
        processes.append(p)
        p.start()

    for p in processes:
        p.join()