“心悦智伴”——融合多模态语言大模型的智能情感陪伴助手

项目简介

本项目主要需要实现智能心理咨询问答的功能，并将用户与机器人的聊天记录存储到本地，整理成结构化的日记数据，对用户的心理健康状况进行追踪，并参考 CBT 给出相应的练习（五栏表填写）帮助用户维持良好的心理状态，且能够通过聊天文本对用户进行智能的心理辅导，必要时进行心理危机干预。

快速开始

1、配置环境

pip install -r requirements.txt

2、运行交互系统

python server.py 
chainlit run ./web/app.py -w

模型训练

• 本项目基于 Mindformers 的 ChatGLM-6B 模型进行 LoRA 微调

• 训练过程中通过 data2indrecord.py 对数据集进行格式转换，使用生成的 smile_train.mindrecord 数据进行训练

• 使用 task.py 执行低参微调实现 LoRA 微调效果

参考框架：MindFormers、MindNLP、MindSpore

引用

项目采用MeChat使用的SMILE数据集：

@misc{qiu2023smile,
      title={SMILE: Single-turn to Multi-turn Inclusive Language Expansion via ChatGPT for Mental Health Support},
      author={Huachuan Qiu and Hongliang He and Shuai Zhang and Anqi Li and Zhenzhong Lan},
      year={2023},
      eprint={2305.00450},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

项目结构

下面是本项目的结构：

│  README.md
│  requirements.txt
│  server.py
|
+---css
|       diary.css
|       index.css
|       plan.css
|
+---image
|       bottom.png
|       dialog1.png
|       dialog2.png
|       dialog4.png
|       dialog5.png
|       dialog6.png
|       dialog7.png
|       function.png
|       index_logo.png
|
+---question
|   |   text.html
|   |
|   \---img
|           circle (1).png
|           circle (2).png
|           circle (3).png
|           portrait.png
|           profile.png
|           ????.png
|
+---toPaint
|   |   apple-touch-icon.png
|   |   asset-manifest.json
|   |   Cascadia.ttf
|   |   Cascadia.woff2
|   |   chinese.woff2
|   |   favicon.ico
|   |   FG_Virgil.ttf
|   |   FG_Virgil.woff2
|   |   fonts.css
|   |   index.html
|   |   LICENSE
|   |   logo-180x180.png
|   |   manifest.json
|   |   og-image-sm.png
|   |   og-image.png
|   |   robots.txt
|   |   service-worker.js
|   |   service-worker.js.map
|   |   version.json
|   |   Virgil.woff2
|   |   _headers
|   |
|   +---.vscode
|   |       settings.json
|   |
|   +---screenshots
|   |       collaboration.png
|   |       export.png
|   |       illustration.png
|   |       shapes.png
|   |       virtual-whiteboard.png
|   |       wireframe.png
|   |
|   +---static
|   |   +---css
|   |   |       diary.css
|   |   |       main.be8a1cb7.chunk.css
|   |   |       main.be8a1cb7.chunk.css.map
|   |   |
|   |   \---js
|   |           0.7e4d8e9c.chunk.js
|   |           0.7e4d8e9c.chunk.js.LICENSE.txt
|   |           0.7e4d8e9c.chunk.js.map
|   |           1.33f5c530.chunk.js

|   |           bug-issue-template.eefe8674.chunk.js
|   |           bug-issue-template.eefe8674.chunk.js.map
|   |           firebase.f7b13fb5.chunk.js
|   |           firebase.f7b13fb5.chunk.js.LICENSE.txt
|   |           firebase.f7b13fb5.chunk.js.map
|   |           i18n-ar-SA-json.8cd1c0da.chunk.js
|   |           i18n-ar-SA-json.8cd1c0da.chunk.js.map
|   |           i18n-bg-BG-json.7860af68.chunk.js
|   |           i18n-bg-BG-json.7860af68.chunk.js.