feat: count.
This commit is contained in:
234
controllers/ai.go
Normal file
234
controllers/ai.go
Normal file
@ -0,0 +1,234 @@
|
||||
// controllers/ai.go
|
||||
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"context" // 在此处添加 context 导入
|
||||
"fmt"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/sashabaranov/go-openai"
|
||||
"hr_receiver/util"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
// 配置文件 (与 main.go 保持一致)
|
||||
const (
|
||||
BaseURL = "https://api.lkeap.cloud.tencent.com/v1"
|
||||
APIKey = "sk-Y4zjnwulSuSlf60mrzwCxq2ipktHSs4jZHgWeQOArWuWJEOd" // 请替换为实际的 API Key
|
||||
Model = "deepseek-v3"
|
||||
)
|
||||
|
||||
// readDocxContent 读取 .docx 文件并将其转换为结构化文本
|
||||
// 修改为先保存临时文件再读取
|
||||
func readDocxContent(fileHeader *multipart.FileHeader) (string, error) {
|
||||
// 1. 创建临时文件
|
||||
tempFile, err := os.CreateTemp("", "upload_*.docx")
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to create temporary file: %w", err)
|
||||
}
|
||||
defer os.Remove(tempFile.Name()) // 确保函数结束时删除临时文件
|
||||
defer tempFile.Close()
|
||||
|
||||
// 2. 打开上传的文件流
|
||||
src, err := fileHeader.Open()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to open uploaded file: %w", err)
|
||||
}
|
||||
defer src.Close()
|
||||
|
||||
// 3. 将上传的文件内容复制到临时文件
|
||||
_, err = io.Copy(tempFile, src)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to copy file to temporary location: %w", err)
|
||||
}
|
||||
|
||||
// 4. 获取临时文件的完整路径
|
||||
tempFilePath := tempFile.Name()
|
||||
str, err := util.DocxToStructuredPrompt(tempFilePath)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to parse docx with go-docx: %w", err)
|
||||
}
|
||||
// 注意：表格、图片等复杂元素的处理可能需要更复杂的逻辑，这里仅处理简单文本
|
||||
|
||||
return str, nil
|
||||
}
|
||||
|
||||
// readCSVContent 读取 .csv 文件内容
|
||||
// 修改为先保存临时文件再读取
|
||||
func readCSVContent(fileHeader *multipart.FileHeader) (string, error) {
|
||||
// 1. 创建临时文件
|
||||
tempFile, err := os.CreateTemp("", "upload_*.csv")
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to create temporary file: %w", err)
|
||||
}
|
||||
defer os.Remove(tempFile.Name()) // 确保函数结束时删除临时文件
|
||||
defer tempFile.Close()
|
||||
|
||||
// 2. 打开上传的文件流
|
||||
src, err := fileHeader.Open()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to open uploaded file: %w", err)
|
||||
}
|
||||
defer src.Close()
|
||||
|
||||
// 3. 将上传的文件内容复制到临时文件
|
||||
_, err = io.Copy(tempFile, src)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to copy file to temporary location: %w", err)
|
||||
}
|
||||
|
||||
// 4. 读取临时文件内容
|
||||
content, err := ioutil.ReadFile(tempFile.Name())
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to read CSV content from temporary file: %w", err)
|
||||
}
|
||||
|
||||
return string(content), nil
|
||||
}
|
||||
|
||||
// buildAnalysisPrompt 构建发送给 AI 的提示词
|
||||
func buildAnalysisPrompt(teachingPlanContent, heartRateContent string) string {
|
||||
return fmt.Sprintf(`请根据以下体育课堂的教案和心率监测数据，生成一份详细的课堂分析报告：
|
||||
|
||||
## 教案内容：
|
||||
%s
|
||||
|
||||
## 心率监测数据：
|
||||
%s
|
||||
|
||||
这是一份幼儿园体育课的教案和课程心率监测数据，请帮对照分析课程教学效果，运动量和运动负荷情况是否科学，并提出课程设计的优化方案。
|
||||
|
||||
优化方案参考如下格式,教学过程需要详细一些：
|
||||
# 幼儿体育教案（华侨大学版本）
|
||||
|
||||
| 项目 | 内容 |
|
||||
| ------------ | -------------------------------- |
|
||||
| **课程名** | |
|
||||
| **年段** | 小 中 大 |
|
||||
| **教师姓名** | |
|
||||
| **时间** | 年 月 日 |
|
||||
| **地点** | |
|
||||
| **人数** | 男： 女： |
|
||||
| **时长** | 分钟 |
|
||||
| **天气预报** | 晴 雨 阴；温度 ℃ |
|
||||
| **器材准备** | |
|
||||
|
||||
## 教学目标
|
||||
|
||||
| 类型 | 目标 |
|
||||
| -------- | ------------ |
|
||||
| **体能目标** | |
|
||||
| **技能目标** | |
|
||||
| **情感目标** | |
|
||||
|
||||
## 教学过程
|
||||
|
||||
| 阶段 | 阶段 | 项目名称 | 引导语及教学方法 | 队形/站位/留意点 | 目标心率区间 | 时间（分） |
|
||||
| ---------- | -------- | ----------------------------- | ------------------------ | --------------------- | ------------ | ---------- |
|
||||
| **准备部分** | 热身 | | | | | 3 |
|
||||
| | 注意力游戏 | | | | | 3 |
|
||||
| **正课部分** | 基本素质练习及常规意识培养环节 | | | | | 5 |
|
||||
| | 复习环节 | | | | | 5 |
|
||||
| | 新授环节 | | | | | 8 |
|
||||
| **结束部分** | 社会性及情感目标游戏 | | | | | 4 |
|
||||
| | 整理放松 | | | | | 2 |
|
||||
|
||||
请以专业体育教师的视角，提供详细的数据分析和教学建议。`, teachingPlanContent, heartRateContent)
|
||||
}
|
||||
|
||||
// callAIForAnalysis 调用大模型进行分析
|
||||
func callAIForAnalysis(prompt string) (string, error) {
|
||||
config := openai.DefaultConfig(APIKey)
|
||||
config.BaseURL = BaseURL
|
||||
client := openai.NewClientWithConfig(config)
|
||||
|
||||
resp, err := client.CreateChatCompletion(
|
||||
context.Background(),
|
||||
openai.ChatCompletionRequest{
|
||||
Model: Model,
|
||||
Messages: []openai.ChatCompletionMessage{
|
||||
{
|
||||
Role: openai.ChatMessageRoleUser,
|
||||
Content: prompt,
|
||||
},
|
||||
},
|
||||
Temperature: 0.6, // 可调整
|
||||
TopP: 0.6, // 可调整
|
||||
MaxTokens: 4000, // 根据需要调整
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("API call failed: %w", err)
|
||||
}
|
||||
|
||||
if len(resp.Choices) == 0 {
|
||||
return "", fmt.Errorf("no choices returned from API")
|
||||
}
|
||||
|
||||
return resp.Choices[0].Message.Content, nil
|
||||
}
|
||||
|
||||
// AnalyzeByAI Gin 控制器方法
|
||||
func (tc *TrainingController) AnalyzeByAI(c *gin.Context) {
|
||||
// 1. 解析多部分表单请求
|
||||
form, err := c.MultipartForm()
|
||||
if err != nil {
|
||||
log.Printf("Error parsing multipart form: %v", err)
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("Failed to parse form: %v", err)})
|
||||
return
|
||||
}
|
||||
|
||||
// 2. 获取文件列表
|
||||
docxFiles := form.File["teaching_plan"] // 假设前端字段名为 'teaching_plan'
|
||||
csvFiles := form.File["heart_rate_data"] // 假设前端字段名为 'heart_rate_data'
|
||||
|
||||
if len(docxFiles) == 0 || len(csvFiles) == 0 {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "Missing required files: teaching_plan (.docx) or heart_rate_data (.csv)"})
|
||||
return
|
||||
}
|
||||
|
||||
// 3. 读取文件内容
|
||||
// 注意：这里我们只取第一个上传的文件
|
||||
teachingPlanFileHeader := docxFiles[0]
|
||||
heartRateFileHeader := csvFiles[0]
|
||||
|
||||
teachingPlanContent, err := readDocxContent(teachingPlanFileHeader)
|
||||
if err != nil {
|
||||
log.Printf("Error reading teaching plan file (%s): %v", teachingPlanFileHeader.Filename, err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("Failed to process teaching plan file: %v", err)})
|
||||
return
|
||||
}
|
||||
|
||||
heartRateContent, err := readCSVContent(heartRateFileHeader)
|
||||
if err != nil {
|
||||
log.Printf("Error reading heart rate file (%s): %v", heartRateFileHeader.Filename, err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("Failed to process heart rate file: %v", err)})
|
||||
return
|
||||
}
|
||||
|
||||
// 4. 构建 Prompt
|
||||
prompt := buildAnalysisPrompt(teachingPlanContent, heartRateContent)
|
||||
|
||||
// 5. 调用 AI 分析
|
||||
analysisResult, err := callAIForAnalysis(prompt)
|
||||
if err != nil {
|
||||
log.Printf("Error calling AI for analysis: %v", err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("AI analysis failed: %v", err)})
|
||||
return
|
||||
}
|
||||
//outputFile := ".md"
|
||||
//ioutil.WriteFile(outputFile, []byte(analysisResult), 0644)
|
||||
|
||||
// 6. 返回结果
|
||||
// 方式一：返回 JSON 结构
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"status": "success",
|
||||
"data": analysisResult,
|
||||
})
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user