You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

147 lines
6.6 KiB

1 month ago
package com.dsideal.base.AI.Model;
import cn.hutool.json.JSONUtil;
import com.dsideal.base.AI.Generator.HtmlGenerator;
import com.dsideal.base.AI.Generator.WordGenerator;
import com.dsideal.base.DataEase.Model.DataEaseModel;
import com.dsideal.base.Util.CallDeepSeek;
import com.jfinal.plugin.activerecord.Db;
import com.jfinal.plugin.activerecord.Record;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
public class YunNanModel {
/**
*
*
* @param regions
* @return
*/
public String collectEducationData(String[] regions) {
// 查询教育资源配置发展预测相关表
String sql = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'dataease' AND TABLE_NAME LIKE 'excel_报告-教育资源配置发展预测%';";
List<com.jfinal.plugin.activerecord.Record> tableList = Db.find(sql);
StringBuilder dataContent = new StringBuilder();
// 构建数据标题
dataContent.append("教育资源配置发展预测数据对比分析\n\n");
dataContent.append("对比州市:").append(String.join(" vs ", regions)).append("\n\n");
// 遍历所有相关数据表
for (com.jfinal.plugin.activerecord.Record record : tableList) {
String tableName = record.getStr("TABLE_NAME");
dataContent.append("数据表:").append(tableName).append("\n");
// 为每个地区收集数据
for (String region : regions) {
sql = "select * from `" + tableName + "` where `行政区划`=?";
List<com.jfinal.plugin.activerecord.Record> listContent = Db.use(DataEaseModel.DB_NAME).find(sql, region);
if (!listContent.isEmpty()) {
dataContent.append("\n").append(region).append("数据:\n");
for (Record dataRecord : listContent) {
dataContent.append(JSONUtil.toJsonPrettyStr(dataRecord.getColumns())).append("\n");
}
} else {
dataContent.append("\n").append(region).append(":无相关数据\n");
}
}
dataContent.append("\n----------------------------------------\n\n");
}
return dataContent.toString();
}
/**
*
*
* @param dataContent
* @param regions
* @return
*/
public String createAnalysisPrompt(String dataContent, String[] regions) {
String regionNames = String.join("和", regions);
return "你是一位专业的教育数据分析专家,请基于以下数据对" + regionNames + "的教育资源配置情况进行深入对比分析。\n\n" +
1 month ago
"重要要求请生成结构化的Markdown格式报告使用标准的Markdown语法包括标题、列表、粗体等格式。\n\n" +
1 month ago
"请按照以下结构进行分析并生成专业的分析报告:\n\n" +
1 month ago
"## 1. 执行摘要\n" +
"简要概述两州教育资源配置的整体情况和主要发现\n\n" +
"## 2. 数据概览\n" +
"两州基本教育数据对比和关键指标汇总\n\n" +
"## 3. 详细对比分析\n" +
"教育资源配置水平对比、发展趋势分析、优势与不足分析\n\n" +
"## 4. 问题识别\n" +
"存在的主要问题和资源配置不均衡情况\n\n" +
"## 5. 建议与对策\n" +
"针对性改进建议和资源优化配置方案\n\n" +
"## 6. 结论\n" +
"总体评价和未来发展方向\n\n" +
1 month ago
"请确保分析客观、专业,数据引用准确,建议具有可操作性。\n\n" +
"=== 原始数据 ===\n" + dataContent;
}
/**
* Word
*/
1 month ago
public void generateWordReport(String analysisPrompt, String[] regions, String path) {
1 month ago
// 生成输出文件名(不包含路径)
String timestamp = new SimpleDateFormat("yyyyMMdd_HHmmss").format(new Date());
// 构建区域名称
String regionName = String.join("与", regions);
// 调用DeepSeek进行分析
CallDeepSeek.callDeepSeekStream(analysisPrompt, new CallDeepSeek.SSEListener() {
private StringBuilder fullResponse = new StringBuilder();
@Override
public void onData(String data) {
System.out.print(data);
System.out.flush();
fullResponse.append(data);
}
@Override
public void onComplete(String response) {
System.out.println("\n\n=== Word文档分析完成 ===");
try {
// 使用完整的分析结果
String analysisResult = fullResponse.toString();
// 生成Word文档
1 month ago
String wordOutputPath = path + "/" + regionName + "_教育分析报告_" + timestamp + ".docx";
1 month ago
WordGenerator.generateWordDocument(analysisResult, wordOutputPath, regions);
System.out.println("Word文档生成完成: " + wordOutputPath);
1 month ago
// 生成HTML报告
1 month ago
// String htmlOutputPath = path + "/" + regionName + "_教育分析报告_" + timestamp + ".html";
// String htmlFilePath = HtmlGenerator.generateHtmlReport(analysisResult, htmlOutputPath, regionName);
// if (htmlFilePath != null) {
// System.out.println("HTML报告生成完成: " + htmlFilePath);
// }
// // 生成PPT
// String token = PptAIKit.createApiToken("dsideal", 1000);
// String pptFilePath = PptGenerator.generatePptPresentation(analysisResult, regions, token);
// if (pptFilePath != null) {
// System.out.println("PPT演示文稿生成完成: " + pptFilePath);
// }
1 month ago
} catch (Exception e) {
System.err.println("保存文件时出错: " + e.getMessage());
e.printStackTrace();
}
}
@Override
public void onError(String error) {
System.err.println("DeepSeek分析出错: " + error);
}
});
}
}