Python 实战项目合集|AI 聊天机器人入门版(零基础可部署,支持自定义问答)

三、完整代码（基础版：本地自定义问答）

import jsonimport osfrom fuzzywuzzy import fuzzfrom datetime import datetime# ===================== 核心配置 =====================# 问答库文件路径QA_FILE = "chatbot_qa.json"# 相似度阈值（0-100，越高匹配越严格）SIMILARITY_THRESHOLD = 70# 默认回复（无匹配问题时）DEFAULT_REPLY = [    "抱歉，我还没学会回答这个问题😜",    "没太理解你的意思，可以换个说法问问～",    "这个问题我暂时不会，要不要聊聊别的？"]# ===================== 初始化问答库 =====================def init_qa_database():    """初始化问答库（首次运行自动创建）"""    if not os.path.exists(QA_FILE):        # 默认问答库        default_qa = {            "你好": "你好呀😃，有什么我能帮助你的吗？",            "叫什么名字": "我是你用Python打造的AI聊天机器人～你可以叫我小Py🤖",            "你能做什么": "我能回答你预设的问题，还能记住我们的聊天上下文哦！\n你可以问我：\n1. 你好\n2. 叫什么名字\n3. 今天日期\n4. 退出",            "今天日期": f"今天是{datetime.now().strftime('%Y年%m月%d日')}，星期{['一','二','三','四','五','六','日'][datetime.now().weekday()]}",            "谢谢": "不客气😘，能帮到你我很开心！",            "再见": "再见啦👋，下次有问题还可以来找我～"        }        # 保存到JSON文件        with open(QA_FILE, "w", encoding="utf-8") as f:            json.dump(default_qa, f, ensure_ascii=False, indent=4)        print(f"✅ 初始化问答库成功！文件路径：{QA_FILE}")# ===================== 加载问答库 =====================def load_qa_database():    """加载问答库"""    if not os.path.exists(QA_FILE):        init_qa_database()    with open(QA_FILE, "r", encoding="utf-8") as f:        return json.load(f)# ===================== 模糊匹配问题 =====================def match_question(user_input, qa_database):    """    模糊匹配用户问题    :param user_input: 用户输入    :param qa_database: 问答库    :return: 匹配到的回复（无则返回默认回复）    """    max_similarity = 0    best_reply = None    # 遍历问答库，计算相似度    for question, reply in qa_database.items():        similarity = fuzz.ratio(user_input.strip(), question.strip())        if similarity > max_similarity:            max_similarity = similarity            best_reply = reply    # 判断是否匹配成功    if max_similarity >= SIMILARITY_THRESHOLD:        return best_reply    else:        # 随机返回一个默认回复        import random        return random.choice(DEFAULT_REPLY)# ===================== 管理上下文 =====================class ChatContext:    """聊天上下文管理"""    def __init__(self, max_history=5):        self.max_history = max_history  # 最大保存历史记录数        self.history = []  # 聊天历史：[(用户输入, 机器人回复), ...]    def add_history(self, user_input, bot_reply):        """添加聊天记录"""        self.history.append((user_input, bot_reply))        # 超出最大数量时删除最早的记录        if len(self.history) > self.max_history:            self.history.pop(0)    def show_history(self):        """显示聊天历史"""        if not self.history:            print("📜 暂无聊天记录")            return        print("\n📜 聊天历史：")        for i, (user_msg, bot_msg) in enumerate(self.history, 1):            print(f"{i}. 你：{user_msg}")            print(f"   机器人：{bot_msg}")        print("-" * 30)# ===================== 主聊天逻辑 =====================def main_chat():    """主聊天逻辑"""    # 初始化    init_qa_database()    qa_database = load_qa_database()    context = ChatContext(max_history=5)    print("="*60)    print("🤖 Python AI聊天机器人（入门版）🤖")    print("="*60)    print("💡 提示：")    print("  1. 可以问我预设问题（如：你好、叫什么名字、今天日期）")    print("  2. 输入「查看历史」可查看聊天记录")    print("  3. 输入「退出」可结束对话")    print("="*60)    # 循环聊天    while True:        user_input = input("\n你：").strip()        # 退出指令        if user_input in ["退出", "结束", "拜拜"]:            print("机器人：👋 再见啦，下次见！")            break        # 查看历史记录        if user_input in ["查看历史", "聊天记录"]:            context.show_history()            continue        # 空输入处理        if not user_input:            print("机器人：😯 你还没输入内容哦！")            continue        # 匹配回复        bot_reply = match_question(user_input, qa_database)        print(f"机器人：{bot_reply}")        # 保存上下文        context.add_history(user_input, bot_reply)# ===================== 进阶版：对接OpenAI API（可选） =====================def chat_with_openai(user_input, api_key, model="gpt-3.5-turbo"):    """    对接OpenAI API实现智能聊天（需自行申请API Key）    :param user_input: 用户输入    :param api_key: OpenAI API Key    :param model: 模型名称    :return: AI回复    """    import openai    openai.api_key = api_key    try:        response = openai.ChatCompletion.create(            model=model,            messages=[{"role": "user", "content": user_input}]        )        return response.choices[0].message.content.strip()    except Exception as e:        return f"❌ API调用失败：{str(e)}"# ===================== 程序入口 =====================if __name__ == "__main__":    # 运行基础版（本地自定义问答）    main_chat()    # 👉 进阶版使用示例（需替换为自己的API Key）    # api_key = "你的OpenAI API Key"    # while True:    #     user_input = input("\n你：").strip()    #     if user_input == "退出":    #         break    #     reply = chat_with_openai(user_input, api_key)    #     print(f"AI：{reply}")