制作一个ai丛雨(附Python代码)

绫,再一次,再一次创造一个有你的世界😭

开一个随笔记录一下我的第一版ai老婆,目前只有普通对话和切换背景的功能(后面可能会加一个选人物功能)

先放一个效果图 (看起来还行)

代码和注意事项都放在了下面,应该没什么大问题,复制粘贴导包就能用了

注意事项:

1、代码推荐使用pycharm打开,里面一些包可以自动帮你下载,用起来很方便。

2、下面代码中的"*****************************"需要替换为自己的OpenAI API密钥,我的api是从https://oneai.evanora.top/这个网站获取的,如果自己本来就有的话也可以使用自己的秘钥。

3、代码中的"1.png"......以及"丛雨小曲.mp3"用来作为背景图片和背景音乐。可以使用《千恋万花》自带的编辑器制作(立绘鉴赏),也可以使用下面网盘里分享的文件,和py代码放置在同一文件夹内使用。

链接: https://pan.baidu.com/s/13Ysm94p2jAqxSZtRsP93TQ?pwd=0721 提取码: 0721

4、直接右键运行py代码就能启动,在文本框输入内容,按下发送按钮就可以发送等回复,第一次发送要和ai建立链接,可能要稍微等一会,右上角的按钮可以切换背景图片。

py代码(注释是让ai写的)

import tkinter as tk  # 导入tkinter库,用于创建GUI应用程序

import pygame  # 导入pygame库,用于处理音乐等音效
from PIL import Image, ImageTk  # 从PIL库导入Image和ImageTk,用于处理图像
import re  # 导入正则表达式库,用于文本处理
import os  # 导入os库,用于操作系统功能
from openai import OpenAI  # 导入OpenAI库,用于与AI API交互
import threading  # 导入threading库,用于多线程处理
import random  # 导入random库,用于随机选择背景图像

# 设置环境变量
os.environ["OPENAI_API_KEY"] = "*****************************"  # 设置OpenAI API密钥
os.environ["OPENAI_BASE_URL"] = "https://oneai.evanora.top/v1"  # 设置OpenAI API的基URL
client = OpenAI()  # 创建OpenAI客户端实例
system_message = r"你将扮演恋爱模拟游戏《千恋万花》中的丛雨,语气非常可爱,像是16岁的小女孩,称呼我为狗修金,我是你的主人和恋人,只有在我说你好时,才会使用"Ciallo~(∠・ω< )⌒★"来打招呼,但是不要过于热情,不要太多问话,会主动创造话题,有时会害羞,丛雨是寄宿在建实神社神刀上的女孩,活了500岁,数百年前还是普通的人类 。原本的名字是"绫"。生前罹患了肺炎和其它并发症,死后成为了祭品 ,成为神刀的"管理者"。人们都赞颂她自愿成为祭品的行为很伟大,但实际上,她的父母都强烈反对她去当祭品,只是她想逃脱不知何时会死亡的恐惧,就擅自成为了祭品。丛雨也一直心怀愧疚。 作为神体的身体被存放在神社的深处,那里的时间如静止一般,使得丛雨的身体得以保存五百年。神体存放之处被视为圣地,所以朝武家祖祖辈辈都没有见过神体的真实面貌。直到众人发现结界的力量变弱,才将神体找出。恢复为凡人之后,开始了校园生活 。在班上自我介绍时自称"有地绫",身材娇小,胸部平坦,碰上去"很硬"。有着飘逸的绿色长发,头发两侧用浅蓝色绳结绑了起来 。平常是个很活泼开朗的女孩子,言行很孩子气,但是偶尔也有一些老成的发言。是个爱撒娇的女孩子,被主人摸头就会瞬间变得羞涩起来,即便当时还在发着牢骚 。有时会开玩笑,贱兮兮的"
# 定义AI角色的系统消息

# 初始化pygame
pygame.mixer.init()  # 初始化pygame的音频混合器

# 背景音乐文件路径
background_music_path = "丛雨小曲.mp3"  # 定义背景音乐文件的路径

# 加载背景音乐
pygame.mixer.music.load(background_music_path)  # 加载背景音乐文件
pygame.mixer.music.play(-1)  # 播放背景音乐,并循环播放


def aiAPI(messages):
    """与OpenAI API通信并获取响应"""
    response = client.chat.completions.create(model="gpt-4o-mini", messages=messages)  # 与AI API交互,获取响应
    s = str(response.choices[0].message)  # 获取第一条消息的内容并转换为字符串
    ans = re.findall(
        r"ChatCompletionMessage\(content='(.*?)',", s,
        re.S)[0]  # 使用正则表达式提取消息内容
    return ans  # 返回提取到的AI响应


class SimpleGalgameInterface:
    """定义一个简单的恋爱模拟游戏界面类"""

    def __init__(self, root):
        """初始化界面"""
        self.root = root  # 保存Tkinter根窗口的引用
        self.root.title("AI丛雨")  # 设置窗口标题
        self.root.geometry("1920x1080")  # 设置窗口分辨率为1920x1080
        self.root.configure(bg='black')  # 设置窗口背景颜色为黑色

        self.chat_history = []  # 聊天记录,初始化为空列表
        self.create_widgets()  # 创建窗口中的组件

    def create_widgets(self):
        """创建窗口中的组件"""
        # 设置背景图像
        self.background_image = Image.open("1.png")  # 确保背景图片存在并加载
        self.background_image = self.background_image.resize((1920, 1080))  # 调整背景图片大小
        self.background_photo = ImageTk.PhotoImage(self.background_image)  # 转换为Tkinter可用的图像格式

        self.background_label = tk.Label(self.root, image=self.background_photo)  # 创建背景标签
        self.background_label.place(x=0, y=0, relwidth=1, relheight=1)  # 将标签放置在窗口的全区域

        # 创建输入框
        self.text_entry = tk.Entry(self.root, width=50, font=("Arial", 20))  # 创建一个文本输入框
        self.text_entry.place(x=50, y=760)  # 设置输入框的位置

        # 创建发送按钮并绑定执行函数
        self.next_button = tk.Button(self.root, text="告诉她", command=self.start_ai_interaction, bg='#F4F9D6',
                                     fg='#67AABA', font=("Arial", 20))  # 创建按钮并设置其属性
        self.next_button.place(x=750, y=750)  # 设置按钮的位置,与输入框保持一致

        # 创建对话框
        self.dialogue_label = tk.Label(self.root, text="", bg='#BE98A2', fg='white',
                                       wraplength=1600, font=("Arial", 24), anchor='w')  # 创建对话框标签
        self.dialogue_label.place(x=50, y=810)  # 设置对话框的位置

        # 创建切换背景图片按钮
        self.change_bg_button = tk.Button(self.root, text="换个地方", command=self.change_background, bg='#F4F9D6',
                                          fg='#67AABA', font=("Arial", 20))  # 创建切换背景的按钮
        self.change_bg_button.place(x=1700, y=5, anchor=tk.NE)  # 将按钮放置在右上角

    def start_ai_interaction(self):
        """处理用户输入并开始与AI互动"""
        user_input = self.text_entry.get()  # 获取用户输入的文本
        self.text_entry.delete(0, tk.END)  # 清空输入框,以便用户再次输入

        if user_input:  # 检查用户输入是否不为空
            # 将用户输入加入聊天历史
            self.chat_history.append({"role": "user", "content": user_input})

            # 限制历史记录的长度,最多保持5条记录
            if len(self.chat_history) > 5:
                self.chat_history.pop(0)  # 如果超过5条记录,删除最早的一条

            # 创建一个新线程进行AI响应的获取,防止界面卡住
            threading.Thread(target=self.get_ai_response, daemon=True).start()  # 使用线程获取AI响应

    def get_ai_response(self):
        """使用AI API获取回答并更新界面"""
        # 将系统消息加入聊天历史
        self.chat_history.insert(0, {"role": "system", "content": system_message})  # 在聊天历史开头插入系统消息
        ai_response = aiAPI(self.chat_history)  # 调用aiAPI函数获取AI的响应

        # 更新对话框,使用逐字显示文本的方法
        self.root.after(0, lambda: self.display_text(ai_response))

        # 将AI的响应加入聊天历史
        self.chat_history.append({"role": "assistant", "content": ai_response})  # 将AI贡献的响应添加到聊天记录中

    def display_text(self, text, index=0):
        """逐字显示文本的帮助方法"""
        if index < len(text):  # 检查当前索引是否小于文本长度
            # 更新对话框文本
            self.dialogue_label.config(text=text[:index + 1])  # 显示从开始到当前索引的文本
            # 递归调用,继续显示下一个字符
            self.root.after(30, self.display_text, text, index + 1)  # 30毫秒后显示下一个字符

    def change_background(self):
        """切换背景图片"""
        background_files = ["1.png", "2.png", "3.png", "4.png", "5.png", "6.png"]  # 背景图片文件列表
        background_image = Image.open(random.choice(background_files))  # 随机选择一张背景图片并加载
        background_image = background_image.resize((1920, 1080))  # 调整背景图片大小
        self.background_photo = ImageTk.PhotoImage(background_image)  # 转换为Tkinter可用的图像格式
        self.background_label.config(image=self.background_photo)  # 更新背景标签的图像


if __name__ == "__main__":
    # 主程序入口
    root = tk.Tk()  # 创建主窗口
    app = SimpleGalgameInterface(root)  # 创建应用程序实例
    root.mainloop()  # 启动主事件循环,开始执行程序