400 lines
14 KiB
Go
400 lines
14 KiB
Go
package spark
|
||
|
||
import (
|
||
"StuAcaWorksAI/dao"
|
||
"StuAcaWorksAI/proto"
|
||
"StuAcaWorksAI/service"
|
||
"StuAcaWorksAI/worker"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
func SparkPPTThemeList(model *proto.ModelParam, req proto.GetSparkPPTThemeReq) (proto.SparkPPTThemeListResponse, error) {
|
||
url := "https://zwapi.xfyun.cn/api/ppt/v2/template/list"
|
||
headers, err := getSparkRequestHeaders(model.APPID, model.APISecret)
|
||
if err != nil {
|
||
log.Println("Spark PPT Theme Error getting headers:", err)
|
||
return proto.SparkPPTThemeListResponse{}, err
|
||
}
|
||
//log.Println("Spark PPT Theme headers:", headers)
|
||
|
||
req_ := proto.PPTThemeRequest{
|
||
Style: req.Style,
|
||
Color: req.Color,
|
||
Industry: req.Industry,
|
||
PageNum: 1,
|
||
PageSize: 10,
|
||
}
|
||
|
||
reqStr, err := json.Marshal(req_)
|
||
if err != nil {
|
||
log.Println("Spark PPT Theme Error encoding request:", err)
|
||
return proto.SparkPPTThemeListResponse{}, err
|
||
}
|
||
log.Println("Spark PPT Theme req:", string(reqStr))
|
||
|
||
err, resp := worker.DoPostRequestJSON(url, reqStr, headers)
|
||
var response proto.SparkPPTThemeListResponse
|
||
if err != nil {
|
||
log.Println("Spark PPT Theme Error:", err)
|
||
} else {
|
||
err = json.Unmarshal(resp, &response)
|
||
if err != nil {
|
||
log.Println("Spark PPT Theme Error decoding response:", err)
|
||
}
|
||
}
|
||
log.Println("Spark PPT Theme Response:", string(resp))
|
||
//log.Println("Spark PPT Theme Response:", response)
|
||
return response, err
|
||
}
|
||
|
||
func getSparkRequestHeaders(appId, apiSecret string) (map[string]string, error) {
|
||
headers := map[string]string{
|
||
"appId": "d3a4647c",
|
||
"timestamp": strconv.FormatInt(time.Now().Unix(), 10),
|
||
}
|
||
algorithm := ApiAuthAlgorithm{}
|
||
signature := algorithm.GetSignature(appId, apiSecret, time.Now().Unix())
|
||
headers["signature"] = signature
|
||
return headers, nil
|
||
}
|
||
|
||
// 根据用户输入及文档生成大纲,文档可以是docx、doc、txt、pdf格式
|
||
func SparkCreateOutlineByDoc(model *proto.ModelParam, userReq *proto.SparkCreateOutlineRequest) (proto.SparkCreateOutlineResponse, error) {
|
||
url := "https://zwapi.xfyun.cn/api/ppt/v2/createOutlineByDoc"
|
||
headers, err := getSparkRequestHeaders(model.APPID, model.APISecret)
|
||
if err != nil {
|
||
log.Println("Spark PPT Theme Error getting headers:", err)
|
||
return proto.SparkCreateOutlineResponse{}, err
|
||
}
|
||
log.Println("Spark create outline by doc headers:", headers)
|
||
|
||
req := proto.SparkCreateOutlineByDocRequest{
|
||
Query: userReq.Query,
|
||
FileUrl: userReq.FileUrl,
|
||
FileName: userReq.FileName,
|
||
}
|
||
|
||
reqStr, err := json.Marshal(req)
|
||
if err != nil {
|
||
log.Println("Spark create outline by doc Error encoding request:", err)
|
||
}
|
||
log.Println("Spark create outline by doc req:", string(reqStr))
|
||
|
||
err, resp := worker.DoPostRequestForm(url, reqStr, headers)
|
||
//log.Println("Spark create outline by doc Response:", string(resp))
|
||
var response proto.SparkCreateOutlineResponse
|
||
if err != nil {
|
||
log.Println("Spark create outline by doc Error:", err)
|
||
} else {
|
||
err = json.Unmarshal(resp, &response)
|
||
if err != nil {
|
||
log.Println("Spark create outline by doc Error decoding response:", err)
|
||
}
|
||
}
|
||
log.Println("Spark create outline by doc Response struct:", response)
|
||
return response, err
|
||
}
|
||
|
||
// 根据query生成大纲
|
||
func SparkCreateOutline(model *proto.ModelParam, userReq *proto.SparkCreateOutlineRequest) (proto.SparkCreateOutlineResponse, error) {
|
||
url := "https://zwapi.xfyun.cn/api/ppt/v2/createOutline"
|
||
headers, err := getSparkRequestHeaders(model.APPID, model.APISecret)
|
||
if err != nil {
|
||
log.Println("Spark PPT Theme Error getting headers:", err)
|
||
return proto.SparkCreateOutlineResponse{}, err
|
||
}
|
||
log.Println("Spark create outline by doc headers:", headers)
|
||
|
||
//自定义内容
|
||
req := proto.SparkCreateOutlineByDocRequest{
|
||
Query: userReq.Query,
|
||
}
|
||
reqStr, err := json.Marshal(req)
|
||
if err != nil {
|
||
log.Println("Spark create PPT Error encoding request:", err)
|
||
}
|
||
log.Println("Spark create PPT req:", string(reqStr))
|
||
|
||
err, resp := worker.DoPostRequestForm(url, reqStr, headers)
|
||
log.Println("Spark create PPT Response:", string(resp))
|
||
var response proto.SparkCreateOutlineResponse
|
||
if err != nil {
|
||
log.Println("Spark create PPT Error:", err)
|
||
} else {
|
||
err = json.Unmarshal(resp, &response)
|
||
if err != nil {
|
||
log.Println("Spark create PPT Error decoding response:", err)
|
||
}
|
||
}
|
||
log.Println("Spark create PPT Response struct:", response)
|
||
return response, err
|
||
}
|
||
|
||
func SparkCreatePPTByOutline(model *proto.ModelParam, userReq *proto.SparkCreatePPTByOutlineUserRequest) (proto.SparkCreatePPTResponse, error) {
|
||
url := "https://zwapi.xfyun.cn/api/ppt/v2/createPptByOutline"
|
||
headers, err := getSparkRequestHeaders(model.APPID, model.APISecret)
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error encoding request:", err)
|
||
return proto.SparkCreatePPTResponse{}, err
|
||
}
|
||
log.Println("Spark create ppt by outline headers:", headers)
|
||
|
||
req := proto.SparkCreatePptByOutlineRequest{
|
||
Query: userReq.Query,
|
||
FileUrl: userReq.FileUrl,
|
||
FileName: userReq.FileName,
|
||
Outline: userReq.Outline,
|
||
}
|
||
|
||
reqStr, err := json.Marshal(req)
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error encoding request:", err)
|
||
}
|
||
log.Println("Spark create ppt by outline req:", string(reqStr))
|
||
|
||
err, resp := worker.DoPostRequestJSON(url, reqStr, headers)
|
||
log.Println("Spark create ppt by outline Response:", string(resp))
|
||
var response proto.SparkCreatePPTResponse
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error:", err)
|
||
} else {
|
||
err = json.Unmarshal(resp, &response)
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error decoding response:", err)
|
||
}
|
||
}
|
||
log.Println("Spark create ppt by outline Response struct:", response)
|
||
return response, err
|
||
}
|
||
|
||
func SparkGetPPTInfoBySID(model *proto.ModelParam, userReq *proto.SparkCreatePPTByOutlineUserRequest) (proto.SparkCreatePPTByOutlineResponse, error) {
|
||
url := "https://zwapi.xfyun.cn/api/ppt/v2/progress?sid=" + userReq.Sid
|
||
headers, err := getSparkRequestHeaders(model.APPID, model.APISecret)
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error encoding request:", err)
|
||
return proto.SparkCreatePPTByOutlineResponse{}, err
|
||
}
|
||
log.Println("Spark create ppt by outline headers:", headers)
|
||
|
||
err, resp := worker.DoGetRequest(url, headers)
|
||
log.Println("Spark create ppt by outline Response:", string(resp))
|
||
var response proto.SparkCreatePPTByOutlineResponse
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error:", err)
|
||
} else {
|
||
err = json.Unmarshal(resp, &response)
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error decoding response:", err)
|
||
}
|
||
}
|
||
log.Println("Spark create ppt by outline Response struct:", response)
|
||
return response, err
|
||
}
|
||
|
||
// 处理制作大纲请求
|
||
func SparkDoCreateOutline(userReq *proto.SparkCreateOutlineRequest, base *proto.SparkCreatePPTBaseInfo) (proto.SparkCreateOutlineResponse, error) {
|
||
var modelParam proto.ModelParam
|
||
var err error
|
||
//需要保存会话
|
||
sessionName := userReq.Query[:10]
|
||
err2, sessionID := service.CreateSession(base.UserID, proto.SessionTypeUserCreatePPT, sessionName)
|
||
if err2 != nil {
|
||
log.Println("Create session error:", err2)
|
||
return proto.SparkCreateOutlineResponse{}, err2
|
||
}
|
||
base.SessionID = sessionID
|
||
//获取模型
|
||
models, funcs, mferr := service.FindFuncModelListByFunctionV2(userReq.Function)
|
||
if mferr != nil {
|
||
return proto.SparkCreateOutlineResponse{}, mferr
|
||
}
|
||
model := models[0]
|
||
base.ModelID = int(model.ID)
|
||
err = json.Unmarshal([]byte(model.Parameter), &modelParam)
|
||
if err != nil {
|
||
return proto.SparkCreateOutlineResponse{}, err
|
||
}
|
||
modelParam.Url = model.Url
|
||
modelParam.System = funcs[0].Info //系统功能
|
||
//保存消息
|
||
userReqStr, err := json.Marshal(userReq)
|
||
if err != nil {
|
||
log.Println("Error parsing JSON:", err)
|
||
return proto.SparkCreateOutlineResponse{}, err
|
||
}
|
||
err3, userMsgID := service.CreateMessage(proto.UserToModelMsgType, int(sessionID), base.UserID, int(model.ID), string(userReqStr), proto.UserToModelPPTMsgType, int(funcs[0].ID))
|
||
if err3 != nil {
|
||
log.Println("Create message error:", err3, userMsgID)
|
||
return proto.SparkCreateOutlineResponse{}, err3
|
||
}
|
||
|
||
var outline proto.SparkCreateOutlineResponse
|
||
//获取大纲
|
||
if userReq.FileUrl == "" {
|
||
outline, err = SparkCreateOutline(&modelParam, userReq) //根据query生成大纲
|
||
} else {
|
||
outline, err = SparkCreateOutlineByDoc(&modelParam, userReq) //根据文档生成大纲
|
||
}
|
||
//进行保存
|
||
if err != nil {
|
||
log.Println("Spark create outline Error:", err)
|
||
return outline, err
|
||
}
|
||
outlineStr, _ := json.Marshal(userReq)
|
||
//返回的数据保存
|
||
service.CreateAIStreamMsg(base.UserID, base.ModelID, int(base.SessionID), outline, modelParam.Model)
|
||
//保存大纲
|
||
err3, userMsgID = service.CreateMessage(proto.ModelToUserMsgType, int(sessionID), base.UserID, int(model.ID), string(outlineStr), proto.UserToModelPPTMsgType, int(funcs[0].ID))
|
||
if err3 != nil {
|
||
log.Println("Create message error:", err3, userMsgID)
|
||
return proto.SparkCreateOutlineResponse{}, err3
|
||
}
|
||
return outline, err
|
||
}
|
||
|
||
// 处理制作PPT请求
|
||
func SparkDoCreatePPTByOutline(userReq *proto.SparkCreatePPTByOutlineUserRequest, base *proto.SparkCreatePPTBaseInfo) (proto.SparkCreatePPTResponse, error) {
|
||
var modelParam proto.ModelParam
|
||
var err error
|
||
//需要保存会话
|
||
if base.SessionID == 0 {
|
||
log.Println("session id is 0, create ppt")
|
||
return proto.SparkCreatePPTResponse{}, errors.New("create ppt,session id is 0, please create session first")
|
||
}
|
||
//获取模型
|
||
models, funcs, mferr := service.FindFuncModelListByFunctionV2(userReq.Function)
|
||
if mferr != nil {
|
||
return proto.SparkCreatePPTResponse{}, mferr
|
||
}
|
||
model := models[0]
|
||
base.ModelID = int(model.ID)
|
||
err = json.Unmarshal([]byte(model.Parameter), &modelParam)
|
||
if err != nil {
|
||
return proto.SparkCreatePPTResponse{}, err
|
||
}
|
||
modelParam.Url = model.Url
|
||
modelParam.System = funcs[0].Info //系统功能
|
||
//保存消息
|
||
userReqStr, err := json.Marshal(userReq)
|
||
if err != nil {
|
||
log.Println("Error parsing JSON:", err)
|
||
return proto.SparkCreatePPTResponse{}, err
|
||
}
|
||
err3, userMsgID := service.CreateMessage(proto.UserToModelMsgType, int(base.SessionID), base.UserID, int(model.ID), string(userReqStr), proto.UserToModelPPTMsgType, int(funcs[0].ID))
|
||
if err3 != nil {
|
||
log.Println("Create message error:", err3, userMsgID)
|
||
return proto.SparkCreatePPTResponse{}, err3
|
||
}
|
||
|
||
resp, err4 := SparkCreatePPTByOutline(&modelParam, userReq) //根据大纲生成ppt
|
||
if err4 != nil {
|
||
log.Println("Spark create ppt by outline Error:", err4)
|
||
return proto.SparkCreatePPTResponse{}, err4
|
||
}
|
||
//进行保存
|
||
service.CreateAIStreamMsg(base.UserID, base.ModelID, int(base.SessionID), resp, modelParam.Model) //保存消息
|
||
//新协程获取ppt制作进度
|
||
userReq.Sid = resp.Data.SID
|
||
go SparkDoGetPPTInfoBySID(&modelParam, userReq, base)
|
||
return resp, err4
|
||
}
|
||
|
||
// 每3秒查询一次ppt制作情况
|
||
func SparkDoGetPPTInfoBySID(modelParam *proto.ModelParam, userReq *proto.SparkCreatePPTByOutlineUserRequest, base *proto.SparkCreatePPTBaseInfo) {
|
||
//定时
|
||
ticker := time.NewTicker(3 * time.Second)
|
||
defer ticker.Stop()
|
||
i := 0
|
||
for {
|
||
select {
|
||
case <-ticker.C:
|
||
i += 1
|
||
if i > 20 {
|
||
log.Println("SparkDoGetPPTInfoBySID timeout more 1 min, stop ticker")
|
||
return
|
||
}
|
||
log.Println("SparkDoGetPPTInfoBySID ticker")
|
||
resp, err := SparkGetPPTInfoBySID(modelParam, userReq) //根据大纲生成ppt
|
||
if err != nil {
|
||
log.Println("Spark create ppt by outline Error:", err)
|
||
return
|
||
}
|
||
respStr, _ := json.Marshal(resp)
|
||
service.CreateAIStreamMsg(base.UserID, base.ModelID, int(base.SessionID), resp, modelParam.Model) //保存消息
|
||
//添加到消息队列,用于前端查询
|
||
key := fmt.Sprintf("user-%d-session-%d-create-pptx", base.UserID, base.SessionID)
|
||
worker.SetRedisWithExpire(key, string(respStr), time.Second*60*5) //设置5分钟过期
|
||
if resp.Data.PptStatus == "done" {
|
||
log.Println("ppt done")
|
||
//下载文件，保存到指定用户的文件
|
||
fileName, filePath, err2 := SparkSavePPTFileToUser(resp.Data.PptUrl)
|
||
if err2 != nil {
|
||
log.Println("Spark save ppt file to user Error:", err2)
|
||
return
|
||
}
|
||
//保存文件路径到数据库
|
||
fileRecord := dao.CreateFile(fileName, userReq.Outline.Title+".pptx", "", filePath, "", 0, base.UserID, false)
|
||
fileAuth := service.CreateUserFile(base.UserID, fileRecord.FileName, int(fileRecord.ID), proto.UserFileTypeFile)
|
||
if fileRecord.ID == 0 || fileAuth.ID == 0 {
|
||
log.Println("save file info failed", fileRecord.ID, fileAuth.ID)
|
||
}
|
||
return
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 将下载的文件保存到对应指定用户的文件，并对该会话添加ppt完成消息
|
||
func SparkSavePPTFileToUser(fileUrl string) (string, string, error) {
|
||
log.Println("Spark save ppt file to user")
|
||
//https://bjcdn.openstorage.cn/xinghuo-privatedata/zhiwen/2025-04-02/6ebfc0b2-824a-4acc-a030-01acce938dab/3209de1bca814ac48884e015277360fd.pptx
|
||
fileName := fileUrl[strings.LastIndex(fileUrl, "/")+1:]
|
||
//文件保存路径
|
||
path := service.GetFilePath(proto.FILE_BASE_DIR)
|
||
saveFile := path + "/" + fileName
|
||
//下载文件
|
||
err := DownloadFile(fileUrl, saveFile)
|
||
if err != nil {
|
||
log.Println("Download file error:", err)
|
||
return "", "", err
|
||
}
|
||
log.Println("Download file success:", fileName)
|
||
return fileName, path, nil
|
||
}
|
||
|
||
func DownloadFile(url, filePath string) error {
|
||
// 创建 HTTP GET 请求
|
||
resp, err := http.Get(url)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
// 确保在函数结束时关闭响应体
|
||
defer resp.Body.Close()
|
||
|
||
// 检查 HTTP 响应状态码
|
||
if resp.StatusCode != http.StatusOK {
|
||
return fmt.Errorf("unexpected status code: %d", resp.StatusCode)
|
||
}
|
||
|
||
// 创建本地文件
|
||
out, err := os.Create(filePath)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
// 确保在函数结束时关闭文件
|
||
defer out.Close()
|
||
|
||
// 将响应体内容复制到本地文件
|
||
_, err = io.Copy(out, resp.Body)
|
||
return err
|
||
}
|