可锐资源网

技术资源分享平台,提供编程学习、网站建设、脚本开发教程

基于maxkb实现对话一键生成docx分析报告,支持图片和表格

通过扩展maxkb的知识库管理和ai对话能力,实现一键生成分析报告及word文档;

本次优化点在于:

  1. 实现用户输入多个问题,一键生成分析报告,并提供word文件将内容下载下来
  2. word中支持markdown格式,能下载各种图片、报表等复杂格式
  3. 能够支持动态的提示词,即每个问题都可以自带提示词,来微调大模型回复;

先看下操作界面:

如上图所示,可以输入多个问题,一键生成分析报告;

将大模型分析的结果变成word文档,提供给用户下载

下面来说下实现方案:

第一步:修改maxkb的前端代码,增加一个页面和菜单; 由于maxkb是docker部署,所以第一步是进入docker文件目录中;可以通过以下方法进去:

  1. docker ps #查看maxkb的containerId
  2. docker exec -it dbb76df3d062 /bin/bash #dbb76df3d062是maxkb的containerId
  3. 上述命令进入了/opt/maxkb/app 目录,这里有所有maxkb的源码信息,我们把ui目录拷贝出来
  4. 在ui的src目录更新代码,更新完成后,执行 npm build (注意:node版本用最新版)
  5. 将生成的dist 目录,上传到 docker的 /opt/maxkb/app/ui 里面,替换原来的dist
  6. 执行 rm -rf /opt/maxkb/app/data/cache/default_file_cache 删除缓存
  7. 重启maxkb或者容器,重启 maxkb 命令为 mkctl restart ; 重启容器可以用 docker restart dbb76df3d062 #dbb76df3d062是maxkb的containerId

第二步:实现提示词动态输入,这个要改下后端代码:

  1. 在chat_views.py 中增加个参数支持传入 self_tip
  1. 在chat_api.py 中增加参数 self_tip
  1. 在 chat_message_serializers.py 中增加动态修改逻辑

这样改动后,maxkb可以允许前端传入提示词,并指导大模型回答了

第三步:将回答内容变成word并支持markdown格式的样式输出;

这一步我扩展了个api接口,用了php的代码来做的(当然也可以用其他语言),具体实现方法如下:


// 引入 autoload.php 文件,用于自动加载 PHPOffice 类
require_once 'vendor/autoload.php';

use \PhpOffice\PhpWord\IOFactory;

// 模拟数据
$data = $ary;


// 创建Word文档
$phpWord = new PhpOffice\PhpWord\PhpWord();
$section = $phpWord->addSection();

$html_o = '
<div style="text-align:center;width:100%;font-size:32px;font-weight:bold">XXX矿区研究报告</div>
';

// 将HTML添加到Word文档
\PhpOffice\PhpWord\Shared\Html::addHtml($section, $html_o);
$section = $phpWord->addSection();
// 将HTML添加到Word文档
\PhpOffice\PhpWord\Shared\Html::addHtml($section, $html);
$j = 0;
$html = '<html><header>
<style>img {width:100%;}</style>
</header>
<body>';
foreach ($data as $k => $v) {
    $j = $j + 1;
    // Markdown内容
    $markdown = <<<MD
 $v
MD;
    // 解析Markdown为HTML
    $html .= '<span style="font-size:16px;font-weight:bold">' . $j . '、' . $k . '</span><br/>';
    // 解析Markdown为HTML
    $parsedown = new Parsedown();
    $html .= $parsedown->text($markdown);
    $html .= '<br/><br/>';

}
$html.= '</body></html>';
$html = str_replace('<img src="/api/image/','<img width="600" src="'.$host.'/api/image/', $html);
$html = str_replace('<img src="https://api/image/','<img  width="600"  src="'.$host.'/api/image/', $html);

// 将HTML添加到Word文档
\PhpOffice\PhpWord\Shared\Html::addHtml($section, $html);

// 保存生成的新文档,文件名为 output.docx
$path = 'uploads/report_' . time() . '.docx';

// 保存Word文档
$xmlWriter = IOFactory::createWriter($phpWord, 'Word2007');
$xmlWriter->save($path);

以上就完成了整个文档的输出; 如果您有类似需求,通过ai自动生成文档或者分析报告,可以跟我联系;

控制面板
您好,欢迎到访网站!
  查看权限
网站分类
最新留言