feat: count.
This commit is contained in:
188
test/main.go
Normal file
188
test/main.go
Normal file
@ -0,0 +1,188 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/sashabaranov/go-openai"
|
||||
)
|
||||
|
||||
// 配置文件
|
||||
const (
|
||||
BaseURL = "https://api.lkeap.cloud.tencent.com/v1"
|
||||
APIKey = "sk-Y4zjnwulSuSlf60mrzwCxq2ipktHSs4jZHgWeQOArWuWJEOd" // 替换为实际的API Key
|
||||
Model = "deepseek-v3" // 推荐使用terminus版本
|
||||
)
|
||||
|
||||
// 读取文件内容
|
||||
func readFileContent(filename string) (string, error) {
|
||||
content, err := ioutil.ReadFile(filename)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("读取文件 %s 失败: %v", filename, err)
|
||||
}
|
||||
return string(content), nil
|
||||
}
|
||||
|
||||
// 构建分析提示词
|
||||
func buildAnalysisPrompt(teachingPlanContent, heartRateContent string) string {
|
||||
return fmt.Sprintf(`请根据以下体育课堂的教案和心率监测数据，生成一份详细的课堂分析报告：
|
||||
|
||||
## 教案内容：
|
||||
%s
|
||||
|
||||
## 心率监测数据：
|
||||
%s
|
||||
|
||||
这是一份幼儿园体育课的教案和课程心率监测数据，请帮对照分析课程教学效果，运动量和运动负荷情况是否科学，并提出课程设计的优化方案。
|
||||
优化方案参考如下格式,教学过程需要详细一些：
|
||||
# 幼儿体育教案（华侨大学版本）
|
||||
|
||||
| 项目 | 内容 |
|
||||
| ------------ | -------------------------------- |
|
||||
| **课程名** | |
|
||||
| **年段** | 小 中 大 |
|
||||
| **教师姓名** | |
|
||||
| **时间** | 年 月 日 |
|
||||
| **地点** | |
|
||||
| **人数** | 男： 女： |
|
||||
| **时长** | 分钟 |
|
||||
| **天气预报** | 晴 雨 阴；温度 ℃ |
|
||||
| **器材准备** | |
|
||||
|
||||
## 教学目标
|
||||
|
||||
| 类型 | 目标 |
|
||||
| -------- | ------------ |
|
||||
| **体能目标** | |
|
||||
| **技能目标** | |
|
||||
| **情感目标** | |
|
||||
|
||||
## 教学过程
|
||||
|
||||
| 阶段 | 阶段 | 项目名称 | 引导语及教学方法 | 队形/站位/留意点 | 目标心率区间 | 时间（分） |
|
||||
| ---------- | -------- | ----------------------------- | ------------------------ | --------------------- | ------------ | ---------- |
|
||||
| **准备部分** | 热身 | | | | | 3 |
|
||||
| | 注意力游戏 | | | | | 3 |
|
||||
| **正课部分** | 基本素质练习及常规意识培养环节 | | | | | 5 |
|
||||
| | 复习环节 | | | | | 5 |
|
||||
| | 新授环节 | | | | | 8 |
|
||||
| **结束部分** | 社会性及情感目标游戏 | | | | | 4 |
|
||||
| | 整理放松 | | | | | 2 |
|
||||
|
||||
|
||||
请以专业体育教师的视角，提供详细的数据分析和教学建议。`, teachingPlanContent, heartRateContent)
|
||||
}
|
||||
|
||||
// 调用大模型进行分析
|
||||
func analyzeClassData(teachingPlanFile, heartRateFile string) (string, error) {
|
||||
// 读取文件内容
|
||||
teachingPlanContent, err := readFileContent(teachingPlanFile)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
heartRateContent, err := readFileContent(heartRateFile)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 构建客户端
|
||||
config := openai.DefaultConfig(APIKey)
|
||||
config.BaseURL = BaseURL
|
||||
client := openai.NewClientWithConfig(config)
|
||||
|
||||
// 构建提示词
|
||||
prompt := buildAnalysisPrompt(teachingPlanContent, heartRateContent)
|
||||
|
||||
// 调用API
|
||||
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: 5000, // 适当限制输出长度
|
||||
},
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("API调用失败: %v", err)
|
||||
}
|
||||
|
||||
if len(resp.Choices) == 0 {
|
||||
return "", fmt.Errorf("未收到有效响应")
|
||||
}
|
||||
|
||||
return resp.Choices[0].Message.Content, nil
|
||||
}
|
||||
|
||||
// 保存分析结果到文件
|
||||
func saveAnalysisResult(result, outputFile string) error {
|
||||
// 添加时间戳和分隔符
|
||||
timestamp := "生成时间: " + getCurrentTime()
|
||||
separator := strings.Repeat("=", 80)
|
||||
|
||||
formattedResult := fmt.Sprintf("%s\n%s\n%s\n%s\n%s\n",
|
||||
separator, timestamp, separator, result, separator)
|
||||
|
||||
return ioutil.WriteFile(outputFile, []byte(formattedResult), 0644)
|
||||
}
|
||||
|
||||
// 获取当前时间（简化版）
|
||||
func getCurrentTime() string {
|
||||
// 实际使用时可以导入time包
|
||||
return time.Now().Format("2006-01-02 15:04:05") // 替换为 time.Now().Format("2006-01-02 15:04:05")
|
||||
}
|
||||
|
||||
func main() {
|
||||
// 文件路径配置
|
||||
teachingPlanFile := "D:\\projects\\IdeaProjects\\hr_receiver\\test\\b.md"
|
||||
heartRateFile := "D:\\projects\\IdeaProjects\\hr_receiver\\test\\b.csv"
|
||||
outputFile := "小班.md"
|
||||
|
||||
// 检查文件是否存在
|
||||
if _, err := os.Stat(teachingPlanFile); os.IsNotExist(err) {
|
||||
log.Fatalf("教案文件不存在: %s", teachingPlanFile)
|
||||
}
|
||||
if _, err := os.Stat(heartRateFile); os.IsNotExist(err) {
|
||||
log.Fatalf("心率数据文件不存在: %s", heartRateFile)
|
||||
}
|
||||
|
||||
fmt.Println("开始分析体育课堂数据...")
|
||||
fmt.Printf("教案文件: %s\n", teachingPlanFile)
|
||||
fmt.Printf("心率数据: %s\n", heartRateFile)
|
||||
|
||||
// 进行分析
|
||||
result, err := analyzeClassData(teachingPlanFile, heartRateFile)
|
||||
if err != nil {
|
||||
log.Fatalf("分析失败: %v", err)
|
||||
}
|
||||
|
||||
// 保存结果
|
||||
err = saveAnalysisResult(result, outputFile)
|
||||
if err != nil {
|
||||
log.Fatalf("保存结果失败: %v", err)
|
||||
}
|
||||
|
||||
fmt.Printf("分析完成！结果已保存到: %s\n", outputFile)
|
||||
fmt.Println("\n分析报告摘要:")
|
||||
fmt.Println(strings.Repeat("-", 50))
|
||||
|
||||
// 显示前200字符作为预览
|
||||
preview := result
|
||||
if len(result) > 200 {
|
||||
preview = result[:200] + "..."
|
||||
}
|
||||
fmt.Println(preview)
|
||||
}
|
||||
86
test/util.go
Normal file
86
test/util.go
Normal file
@ -0,0 +1,86 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
docx "github.com/fumiama/go-docx"
|
||||
)
|
||||
|
||||
func DocxToStructuredPrompt(filename string) (string, error) {
|
||||
f, err := os.Open(filename)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
fi, err := f.Stat()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
doc, err := docx.Parse(f, fi.Size())
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
var sb strings.Builder
|
||||
sb.WriteString(fmt.Sprintf("# 文件：%s\n\n", filename))
|
||||
|
||||
for _, item := range doc.Document.Body.Items {
|
||||
switch v := item.(type) {
|
||||
case *docx.Paragraph:
|
||||
// 直接用 fmt.Sprint 利用庫的 Stringer
|
||||
text := fmt.Sprint(v)
|
||||
text = strings.TrimSpace(text)
|
||||
if text != "" {
|
||||
sb.WriteString(text + "\n\n")
|
||||
}
|
||||
|
||||
case *docx.Table:
|
||||
sb.WriteString("## 表格\n")
|
||||
|
||||
// 先印表頭（可選）
|
||||
sb.WriteString("| ")
|
||||
|
||||
// 假設第一行是表頭（很多文件如此），或全部當內容
|
||||
for i, row := range v.TableRows {
|
||||
var cells []string
|
||||
for _, cell := range row.TableCells {
|
||||
// 這裡是重點：cell 本身沒有 String()，但可以遍歷它的 Paragraphs
|
||||
var cellText strings.Builder
|
||||
for _, p := range cell.Paragraphs {
|
||||
cellText.WriteString(fmt.Sprint(p))
|
||||
cellText.WriteString(" ")
|
||||
}
|
||||
cells = append(cells, strings.TrimSpace(cellText.String()))
|
||||
}
|
||||
|
||||
sb.WriteString(strings.Join(cells, " | "))
|
||||
sb.WriteString(" |\n")
|
||||
|
||||
// 如果想加 markdown 表頭分隔線（只在第一行後加）
|
||||
if i == 0 {
|
||||
sb.WriteString("| " + strings.Repeat("--- | ", len(cells)) + "\n")
|
||||
}
|
||||
}
|
||||
sb.WriteString("\n")
|
||||
|
||||
default:
|
||||
// 忽略圖片、頁首等
|
||||
}
|
||||
}
|
||||
|
||||
return sb.String(), nil
|
||||
}
|
||||
|
||||
func main1() {
|
||||
// 測試用
|
||||
prompt, err := docxToStructuredPrompt("D:\\myDocument\\tencent\\weChat\\WeChat Files\\wxid_pv6rg3z2l28y22\\FileStorage\\File\\2026-01\\（改）小班体育活动《蚂蚁运粮》(泉秀实幼吴思莹）.docx")
|
||||
if err != nil {
|
||||
fmt.Println("錯誤:", err)
|
||||
return
|
||||
}
|
||||
fmt.Println(prompt)
|
||||
}
|
||||
Reference in New Issue
Block a user