update changelog

重新设计了移动端web页面，新增了移动端CSS，增加移动端SD绘图页面

CHANGELOG.md

@@ -1,4 +1,19 @@
 # 更新日志
+## v3.2.3
+* 功能重构：重构函数工具模块，设计成可以后台动态管理函数。支持添加自定义函数实现
+* 功能新增：为充值产品数据表添加 img_calls 字段，支持充值绘图次数
+* Bug修复：修复 [MJ 机器人空指针异常的 Bug](https://github.com/yangjian102621/chatgpt-plus/issues/73) 
+* Bug修复：确保相同 Prompt 的绘图任务的 Upscale 和 Variation 任务调度给相同的频道
+* 功能新增：新增删除绘图任何和图片功能
+* Bug修复：修复虎皮椒支付二维码重复扫码时报错问题
+* 功能优化：自动将 AI 绘画中的中文提示词翻译成英文
+* 功能优化：优化AI绘画的大图压缩算法，新增图片缓存
+* 功能优化：支持为 MJ 绘图 API 增加反代功能，提高图片的加载速度，大大降低绘图任务的失败率
+* Bug修复：修复[Azure Api 更换api-version参数后请求失败的问题](https://github.com/yangjian102621/chatgpt-plus/pull/71)
+* Bug修复：修复科大讯飞 V1.5 API 请求失败的问题
+* Bug修复：绘图失败后，自动恢复用户的剩余绘图次数
+* 功能新增：为移动端新增 SD 绘图功能，分享功能
+
 ## v3.2.2
 * 功能重构：重构 MidJourney 和 Stable-Diffusion 绘图模块，支持使用多组配置创建池子提供绘画服务
 * 功能新增：AI绘画页面增加翻译和重写提示词功能

README.md

@@ -95,6 +95,8 @@ ChatGLM,讯飞星火，文心一言等多个平台的大语言模型。集成了
 
 ## 项目文档
 
+**最新的部署视频教程：[https://www.bilibili.com/video/BV1ge411C7uA/](https://www.bilibili.com/video/BV1ge411C7uA/?vd_source=dee8b15703ccfcbd24a60ee9a0fabb73)**
+
 详细的部署和开发文档请参考 [**ChatGPT-Plus 文档**](https://ai.r9it.com/docs/)。
 
 加微信进入微信讨论群可获取 **一键部署脚本（添加好友时请注明来自Github!!!）。**

api/core/app_server.go

@@ -3,7 +3,6 @@ package core
 import (
 	"bytes"
 	"chatplus/core/types"
-	"chatplus/service/fun"
 	"chatplus/store/model"
 	"chatplus/utils"
 	"chatplus/utils/resp"
@@ -39,10 +38,9 @@ type AppServer struct {
 	ChatSession   *types.LMap[string, *types.ChatSession] //map[sessionId]UserId
 	ChatClients   *types.LMap[string, *types.WsClient]    // map[sessionId]Websocket 连接集合
 	ReqCancelFunc *types.LMap[string, context.CancelFunc] // HttpClient 请求取消 handle function
-	Functions     map[string]fun.Function
 }
 
-func NewServer(appConfig *types.AppConfig, functions map[string]fun.Function) *AppServer {
+func NewServer(appConfig *types.AppConfig) *AppServer {
 	gin.SetMode(gin.ReleaseMode)
 	gin.DefaultWriter = io.Discard
 	return &AppServer{
@@ -53,7 +51,6 @@ func NewServer(appConfig *types.AppConfig, functions map[string]fun.Function) *A
 		ChatSession:   types.NewLMap[string, *types.ChatSession](),
 		ChatClients:   types.NewLMap[string, *types.WsClient](),
 		ReqCancelFunc: types.NewLMap[string, context.CancelFunc](),
-		Functions:     functions,
 	}
 }
 
@@ -154,9 +151,12 @@ func authorizeMiddleware(s *AppServer, client *redis.Client) gin.HandlerFunc {
 			c.Request.URL.Path == "/api/chat/detail" ||
 			c.Request.URL.Path == "/api/role/list" ||
 			c.Request.URL.Path == "/api/mj/jobs" ||
+			c.Request.URL.Path == "/api/mj/client" ||
 			c.Request.URL.Path == "/api/invite/hits" ||
 			c.Request.URL.Path == "/api/sd/jobs" ||
+			c.Request.URL.Path == "/api/upload" ||
 			strings.HasPrefix(c.Request.URL.Path, "/test/") ||
+			strings.HasPrefix(c.Request.URL.Path, "/api/function/") ||
 			strings.HasPrefix(c.Request.URL.Path, "/api/sms/") ||
 			strings.HasPrefix(c.Request.URL.Path, "/api/captcha/") ||
 			strings.HasPrefix(c.Request.URL.Path, "/api/payment/") ||
@@ -335,9 +335,11 @@ func staticResourceMiddleware() gin.HandlerFunc {
 				log.Fatal(err)
 			}
 
+			// 设置图片缓存有效期为一年 (365天)
+			c.Header("Cache-Control", "max-age=31536000, public")
 			// 直接输出图像数据流
 			c.Data(http.StatusOK, "image/jpeg", buffer.Bytes())
-			return
+			c.Abort() // 中断请求
 		}
 		c.Next()
 	}

api/core/config.go

@@ -14,13 +14,12 @@ var logger = logger2.GetLogger()
 
 func NewDefaultConfig() *types.AppConfig {
 	return &types.AppConfig{
-		Listen:        "0.0.0.0:5678",
-		ProxyURL:      "",
-		Manager:       types.Manager{Username: "admin", Password: "admin123"},
-		StaticDir:     "./static",
-		StaticUrl:     "http://localhost/5678/static",
-		Redis:         types.RedisConfig{Host: "localhost", Port: 6379, Password: ""},
-		AesEncryptKey: utils.RandString(24),
+		Listen:    "0.0.0.0:5678",
+		ProxyURL:  "",
+		Manager:   types.Manager{Username: "admin", Password: "admin123"},
+		StaticDir: "./static",
+		StaticUrl: "http://localhost/5678/static",
+		Redis:     types.RedisConfig{Host: "localhost", Port: 6379, Password: ""},
 		Session: types.Session{
 			SecretKey: utils.RandString(64),
 			MaxAge:    86400,

api/core/types/chat.go

@@ -8,7 +8,8 @@ type ApiRequest struct {
 	Stream      bool          `json:"stream"`
 	Messages    []interface{} `json:"messages,omitempty"`
 	Prompt      []interface{} `json:"prompt,omitempty"` // 兼容 ChatGLM
-	Functions   []Function    `json:"functions,omitempty"`
+	Tools       []interface{} `json:"tools,omitempty"`
+	ToolChoice  string        `json:"tool_choice,omitempty"`
 }
 
 type Message struct {
@@ -27,10 +28,10 @@ type ChoiceItem struct {
 }
 
 type Delta struct {
-	Role         string       `json:"role"`
-	Name         string       `json:"name"`
-	Content      interface{}  `json:"content"`
-	FunctionCall FunctionCall `json:"function_call,omitempty"`
+	Role      string      `json:"role"`
+	Name      string      `json:"name"`
+	Content   interface{} `json:"content"`
+	ToolCalls []ToolCall  `json:"tool_calls,omitempty"`
 }
 
 // ChatSession 聊天会话对象

api/core/types/config.go

@@ -5,22 +5,21 @@ import (
 )
 
 type AppConfig struct {
-	Path          string `toml:"-"`
-	Listen        string
-	Session       Session
-	ProxyURL      string
-	MysqlDns      string            // mysql 连接地址
-	Manager       Manager           // 后台管理员账户信息
-	StaticDir     string            // 静态资源目录
-	StaticUrl     string            // 静态资源 URL
-	Redis         RedisConfig       // redis 连接信息
-	ApiConfig     ChatPlusApiConfig // ChatPlus API authorization configs
-	AesEncryptKey string
-	SmsConfig     AliYunSmsConfig         // AliYun send message service config
-	OSS           OSSConfig               // OSS config
-	MjConfigs     []MidJourneyConfig      // mj AI draw service pool
-	WeChatBot     bool                    // 是否启用微信机器人
-	SdConfigs     []StableDiffusionConfig // sd AI draw service pool
+	Path      string `toml:"-"`
+	Listen    string
+	Session   Session
+	ProxyURL  string
+	MysqlDns  string                  // mysql 连接地址
+	Manager   Manager                 // 后台管理员账户信息
+	StaticDir string                  // 静态资源目录
+	StaticUrl string                  // 静态资源 URL
+	Redis     RedisConfig             // redis 连接信息
+	ApiConfig ChatPlusApiConfig       // ChatPlus API authorization configs
+	SmsConfig AliYunSmsConfig         // AliYun send message service config
+	OSS       OSSConfig               // OSS config
+	MjConfigs []MidJourneyConfig      // mj AI draw service pool
+	WeChatBot bool                    // 是否启用微信机器人
+	SdConfigs []StableDiffusionConfig // sd AI draw service pool
 
 	XXLConfig     XXLConfig
 	AlipayConfig  AlipayConfig
@@ -34,11 +33,15 @@ type ChatPlusApiConfig struct {
 }
 
 type MidJourneyConfig struct {
-	Enabled   bool
-	UserToken string
-	BotToken  string
-	GuildId   string // Server ID
-	ChanelId  string // Chanel ID
+	Enabled        bool
+	UserToken      string
+	BotToken       string
+	GuildId        string // Server ID
+	ChanelId       string // Chanel ID
+	UseCDN         bool
+	DiscordAPI     string
+	DiscordCDN     string
+	DiscordGateway string
 }
 
 type StableDiffusionConfig struct {
@@ -155,7 +158,6 @@ type SystemConfig struct {
 	EnabledRegister  bool     `json:"enabled_register"`    // 是否启用注册功能，关闭注册功能之后将无法注册
 	EnabledMsg       bool     `json:"enabled_msg"`         // 是否启用短信验证码服务
 	RewardImg        string   `json:"reward_img"`          // 众筹收款二维码地址
-	EnabledFunction  bool     `json:"enabled_function"`    // 启用 API 函数功能
 	EnabledReward    bool     `json:"enabled_reward"`      // 启用众筹功能
 	EnabledAlipay    bool     `json:"enabled_alipay"`      // 是否启用支付宝支付通道
 	OrderPayTimeout  int      `json:"order_pay_timeout"`   //订单支付超时时间

api/core/types/function.go

@@ -1,8 +1,11 @@
 package types
 
-type FunctionCall struct {
-	Name      string `json:"name"`
-	Arguments string `json:"arguments"`
+type ToolCall struct {
+	Type     string `json:"type"`
+	Function struct {
+		Name      string `json:"name"`
+		Arguments string `json:"arguments"`
+	} `json:"function"`
 }
 
 type Function struct {
@@ -21,72 +24,3 @@ type Property struct {
 	Type        string `json:"type"`
 	Description string `json:"description"`
 }
-
-const (
-	FuncZaoBao   = "zao_bao"    // 每日早报
-	FuncHeadLine = "headline"   // 今日头条
-	FuncWeibo    = "weibo_hot"  // 微博热搜
-	FuncImage    = "draw_image" // AI 绘画
-)
-
-var InnerFunctions = []Function{
-	{
-		Name:        FuncZaoBao,
-		Description: "每日早报，获取当天全球的热门新闻事件列表",
-		Parameters: Parameters{
-
-			Type: "object",
-			Properties: map[string]Property{
-				"text": {
-					Type:        "string",
-					Description: "",
-				},
-			},
-			Required: []string{},
-		},
-	},
-	{
-		Name:        FuncWeibo,
-		Description: "新浪微博热搜榜，微博当日热搜榜单",
-		Parameters: Parameters{
-			Type: "object",
-			Properties: map[string]Property{
-				"text": {
-					Type:        "string",
-					Description: "",
-				},
-			},
-			Required: []string{},
-		},
-	},
-
-	{
-		Name:        FuncHeadLine,
-		Description: "今日头条，给用户推荐当天的头条新闻，周榜热文",
-		Parameters: Parameters{
-			Type: "object",
-			Properties: map[string]Property{
-				"text": {
-					Type:        "string",
-					Description: "",
-				},
-			},
-			Required: []string{},
-		},
-	},
-
-	{
-		Name:        FuncImage,
-		Description: "AI 绘画工具，根据输入的绘图描述用 AI 工具进行绘画",
-		Parameters: Parameters{
-			Type: "object",
-			Properties: map[string]Property{
-				"prompt": {
-					Type:        "string",
-					Description: "提示词，如果该参数中有中文的话，则需要翻译成英文。",
-				},
-			},
-			Required: []string{},
-		},
-	},
-}

api/core/types/locked_map.go

@@ -6,7 +6,7 @@ import (
 )
 
 type MKey interface {
-	string | int
+	string | int | uint
 }
 type MValue interface {
 	*WsClient | *ChatSession | context.CancelFunc | []interface{}

api/core/types/task.go

@@ -16,6 +16,7 @@ const (
 // MjTask MidJourney 任务
 type MjTask struct {
 	Id          int      `json:"id"`
+	ChannelId   string   `json:"channel_id"`
 	SessionId   string   `json:"session_id"`
 	Type        TaskType `json:"type"`
 	UserId      int      `json:"user_id"`

api/go.mod

@@ -6,7 +6,6 @@ require (
 	github.com/BurntSushi/toml v1.1.0
 	github.com/aliyun/alibaba-cloud-sdk-go v1.62.405
 	github.com/aliyun/aliyun-oss-go-sdk v2.2.9+incompatible
-	github.com/bwmarrin/discordgo v0.27.1
 	github.com/eatmoreapple/openwechat v1.2.1
 	github.com/gin-gonic/gin v1.9.1
 	github.com/go-redis/redis/v8 v8.11.5
@@ -26,6 +25,8 @@ require (
 
 require github.com/xxl-job/xxl-job-executor-go v1.2.0
 
+require github.com/bg5t/mydiscordgo v0.28.1
+
 require (
 	github.com/andybalholm/brotli v1.0.4 // indirect
 	github.com/bytedance/sonic v1.9.1 // indirect

api/go.sum

@@ -7,8 +7,8 @@ github.com/aliyun/aliyun-oss-go-sdk v2.2.9+incompatible/go.mod h1:T/Aws4fEfogEE9
 github.com/andybalholm/brotli v1.0.4 h1:V7DdXeJtZscaqfNuAdSRuRFzuiKlHSC/Zh3zl9qY3JY=
 github.com/andybalholm/brotli v1.0.4/go.mod h1:fO7iG3H7G2nSZ7m0zPUDn85XEX2GTukHGRSepvi9Eig=
 github.com/benbjohnson/clock v1.3.0 h1:ip6w0uFQkncKQ979AypyG0ER7mqUSBdKLOgAle/AT8A=
-github.com/bwmarrin/discordgo v0.27.1 h1:ib9AIc/dom1E/fSIulrBwnez0CToJE113ZGt4HoliGY=
-github.com/bwmarrin/discordgo v0.27.1/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
+github.com/bg5t/mydiscordgo v0.28.1 h1:mVH0ZWstVdJffCi/EXJAYQDtXwIKAJYVXLmECu1hEK8=
+github.com/bg5t/mydiscordgo v0.28.1/go.mod h1:n3aba73N18k1DzM0t0mGE8rwW3Z+vwTvI8pcsBgxN/8=
 github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
 github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
 github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=

api/handler/admin/function_handler.go

@@ -0,0 +1,124 @@
+package admin
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/handler"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+
+	"github.com/golang-jwt/jwt/v5"
+
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type FunctionHandler struct {
+	handler.BaseHandler
+	db *gorm.DB
+}
+
+func NewFunctionHandler(app *core.AppServer, db *gorm.DB) *FunctionHandler {
+	h := FunctionHandler{db: db}
+	h.App = app
+	return &h
+}
+
+func (h *FunctionHandler) Save(c *gin.Context) {
+	var data vo.Function
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	var f = model.Function{
+		Id:          data.Id,
+		Name:        data.Name,
+		Label:       data.Label,
+		Description: data.Description,
+		Parameters:  utils.JsonEncode(data.Parameters),
+		Action:      data.Action,
+		Token:       data.Token,
+		Enabled:     data.Enabled,
+	}
+
+	res := h.db.Save(&f)
+	if res.Error != nil {
+		resp.ERROR(c, "error with save data:"+res.Error.Error())
+		return
+	}
+	data.Id = f.Id
+	resp.SUCCESS(c, data)
+}
+
+func (h *FunctionHandler) Set(c *gin.Context) {
+	var data struct {
+		Id    uint        `json:"id"`
+		Filed string      `json:"filed"`
+		Value interface{} `json:"value"`
+	}
+
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	res := h.db.Model(&model.Function{}).Where("id = ?", data.Id).Update(data.Filed, data.Value)
+	if res.Error != nil {
+		resp.ERROR(c, "更新数据库失败！")
+		return
+	}
+	resp.SUCCESS(c)
+}
+
+func (h *FunctionHandler) List(c *gin.Context) {
+	var items []model.Function
+	res := h.db.Find(&items)
+	if res.Error != nil {
+		resp.ERROR(c, "No data found")
+		return
+	}
+
+	functions := make([]vo.Function, 0)
+	for _, v := range items {
+		var f vo.Function
+		err := utils.CopyObject(v, &f)
+		if err != nil {
+			continue
+		}
+		functions = append(functions, f)
+	}
+	resp.SUCCESS(c, functions)
+}
+
+func (h *FunctionHandler) Remove(c *gin.Context) {
+	id := h.GetInt(c, "id", 0)
+
+	if id > 0 {
+		res := h.db.Delete(&model.Function{Id: uint(id)})
+		if res.Error != nil {
+			resp.ERROR(c, "更新数据库失败！")
+			return
+		}
+	}
+	resp.SUCCESS(c)
+}
+
+// GenToken generate function api access token
+func (h *FunctionHandler) GenToken(c *gin.Context) {
+	// 创建 token
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
+		"user_id": 0,
+		"expired": 0,
+	})
+	tokenString, err := token.SignedString([]byte(h.App.Config.Session.SecretKey))
+	if err != nil {
+		logger.Error("error with generate token", err)
+		resp.ERROR(c)
+		return
+	}
+
+	resp.SUCCESS(c, tokenString)
+}

api/handler/chatimpl/azure_handler.go

@@ -9,7 +9,6 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"gorm.io/gorm"
 	"html/template"
 	"io"
 	"strings"
@@ -57,9 +56,6 @@ func (h *ChatHandler) sendAzureMessage(
 		// 循环读取 Chunk 消息
 		var message = types.Message{}
 		var contents = make([]string, 0)
-		var functionCall = false
-		var functionName string
-		var arguments = make([]string, 0)
 		scanner := bufio.NewScanner(response.Body)
 		for scanner.Scan() {
 			line := scanner.Text()
@@ -69,34 +65,17 @@ func (h *ChatHandler) sendAzureMessage(
 
 			var responseBody = types.ApiResponse{}
 			err = json.Unmarshal([]byte(line[6:]), &responseBody)
-			if err != nil || len(responseBody.Choices) == 0 { // 数据解析出错
+			if err != nil { // 数据解析出错
 				logger.Error(err, line)
 				utils.ReplyMessage(ws, ErrorMsg)
 				utils.ReplyMessage(ws, ErrImg)
 				break
 			}
 
-			fun := responseBody.Choices[0].Delta.FunctionCall
-			if functionCall && fun.Name == "" {
-				arguments = append(arguments, fun.Arguments)
+			if len(responseBody.Choices) == 0 {
 				continue
 			}
 
-			if !utils.IsEmptyValue(fun) {
-				functionName = fun.Name
-				f := h.App.Functions[functionName]
-				if f != nil {
-					functionCall = true
-					utils.ReplyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
-					utils.ReplyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: fmt.Sprintf("正在调用函数 `%s` 作答 ...\n\n", f.Name())})
-					continue
-				}
-			}
-
-			if responseBody.Choices[0].FinishReason == "function_call" { // 函数调用完毕
-				break
-			}
-
 			// 初始化 role
 			if responseBody.Choices[0].Delta.Role != "" && message.Role == "" {
 				message.Role = responseBody.Choices[0].Delta.Role
@@ -122,49 +101,6 @@ func (h *ChatHandler) sendAzureMessage(
 			}
 		}
 
-		if functionCall { // 调用函数完成任务
-			var params map[string]interface{}
-			_ = utils.JsonDecode(strings.Join(arguments, ""), &params)
-			logger.Debugf("函数名称: %s, 函数参数：%s", functionName, params)
-
-			// for creating image, check if the user's img_calls > 0
-			if functionName == types.FuncImage && userVo.ImgCalls <= 0 {
-				utils.ReplyMessage(ws, "**当前用户剩余绘图次数已用尽，请扫描下面二维码联系管理员！**")
-				utils.ReplyMessage(ws, ErrImg)
-			} else {
-				f := h.App.Functions[functionName]
-				if functionName == types.FuncImage {
-					params["user_id"] = userVo.Id
-					params["role_id"] = role.Id
-					params["chat_id"] = session.ChatId
-					params["icon"] = "/images/avatar/mid_journey.png"
-					params["session_id"] = session.SessionId
-				}
-				data, err := f.Invoke(params)
-				if err != nil {
-					msg := "调用函数出错：" + err.Error()
-					utils.ReplyChunkMessage(ws, types.WsMessage{
-						Type:    types.WsMiddle,
-						Content: msg,
-					})
-					contents = append(contents, msg)
-				} else {
-					content := data
-					if functionName == types.FuncImage {
-						content = fmt.Sprintf("下面是根据您的描述创作的图片，他们描绘了 【%s】 的场景", params["prompt"])
-						// update user's img_calls
-						h.db.Model(&model.User{}).Where("id = ?", userVo.Id).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
-					}
-
-					utils.ReplyChunkMessage(ws, types.WsMessage{
-						Type:    types.WsMiddle,
-						Content: content,
-					})
-					contents = append(contents, content)
-				}
-			}
-		}
-
 		// 消息发送成功
 		if len(contents) > 0 {
 			// 更新用户的对话次数
@@ -177,7 +113,7 @@ func (h *ChatHandler) sendAzureMessage(
 			useMsg := types.Message{Role: "user", Content: prompt}
 
 			// 更新上下文消息，如果是调用函数则不需要更新上下文
-			if h.App.ChatConfig.EnableContext && functionCall == false {
+			if h.App.ChatConfig.EnableContext {
 				chatCtx = append(chatCtx, useMsg)  // 提问消息
 				chatCtx = append(chatCtx, message) // 回复消息
 				h.App.ChatContexts.Put(session.ChatId, chatCtx)
@@ -185,11 +121,6 @@ func (h *ChatHandler) sendAzureMessage(
 
 			// 追加聊天记录
 			if h.App.ChatConfig.EnableHistory {
-				useContext := true
-				if functionCall {
-					useContext = false
-				}
-
 				// for prompt
 				promptToken, err := utils.CalcTokens(prompt, req.Model)
 				if err != nil {
@@ -203,7 +134,7 @@ func (h *ChatHandler) sendAzureMessage(
 					Icon:       userVo.Avatar,
 					Content:    template.HTMLEscapeString(prompt),
 					Tokens:     promptToken,
-					UseContext: useContext,
+					UseContext: true,
 				}
 				historyUserMsg.CreatedAt = promptCreatedAt
 				historyUserMsg.UpdatedAt = promptCreatedAt
@@ -213,15 +144,7 @@ func (h *ChatHandler) sendAzureMessage(
 				}
 
 				// 计算本次对话消耗的总 token 数量
-				var totalTokens = 0
-				if functionCall { // prompt + 函数名 + 参数 token
-					tokens, _ := utils.CalcTokens(functionName, req.Model)
-					totalTokens += tokens
-					tokens, _ = utils.CalcTokens(utils.InterfaceToString(arguments), req.Model)
-					totalTokens += tokens
-				} else {
-					totalTokens, _ = utils.CalcTokens(message.Content, req.Model)
-				}
+				totalTokens, _ := utils.CalcTokens(message.Content, req.Model)
 				totalTokens += getTotalTokens(req)
 
 				historyReplyMsg := model.HistoryMessage{
@@ -232,7 +155,7 @@ func (h *ChatHandler) sendAzureMessage(
 					Icon:       role.Icon,
 					Content:    message.Content,
 					Tokens:     totalTokens,
-					UseContext: useContext,
+					UseContext: true,
 				}
 				historyReplyMsg.CreatedAt = replyCreatedAt
 				historyReplyMsg.UpdatedAt = replyCreatedAt

api/handler/chatimpl/chat_handler.go

@@ -214,20 +214,45 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
 	case types.Baidu:
 		req.Temperature = h.App.ChatConfig.OpenAI.Temperature
 		// TODO： 目前只支持 ERNIE-Bot-turbo 模型，如果是 ERNIE-Bot 模型则需要增加函数支持
+		break
 	case types.OpenAI:
 		req.Temperature = h.App.ChatConfig.OpenAI.Temperature
 		req.MaxTokens = h.App.ChatConfig.OpenAI.MaxTokens
 		// OpenAI 支持函数功能
-		if h.App.SysConfig.EnabledFunction {
-			var functions = make([]types.Function, 0)
-			for _, f := range types.InnerFunctions {
-				functions = append(functions, f)
+		var items []model.Function
+		res := h.db.Where("enabled", true).Find(&items)
+		if res.Error != nil {
+			break
+		}
+
+		var tools = make([]interface{}, 0)
+		for _, v := range items {
+			var parameters map[string]interface{}
+			err = utils.JsonDecode(v.Parameters, &parameters)
+			if err != nil {
+				continue
 			}
-			req.Functions = functions
+			required := parameters["required"]
+			delete(parameters, "required")
+			tools = append(tools, gin.H{
+				"type": "function",
+				"function": gin.H{
+					"name":        v.Name,
+					"description": v.Description,
+					"parameters":  parameters,
+					"required":    required,
+				},
+			})
+		}
+
+		if len(tools) > 0 {
+			req.Tools = tools
+			req.ToolChoice = "auto"
 		}
 	case types.XunFei:
 		req.Temperature = h.App.ChatConfig.XunFei.Temperature
 		req.MaxTokens = h.App.ChatConfig.XunFei.MaxTokens
+		break
 	default:
 		utils.ReplyMessage(ws, "不支持的平台："+session.Model.Platform+"，请联系管理员！")
 		utils.ReplyMessage(ws, ErrImg)
@@ -242,11 +267,8 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
 		} else {
 			// calculate the tokens of current request, to prevent to exceeding the max tokens num
 			tokens := req.MaxTokens
-			for _, f := range types.InnerFunctions {
-				tks, _ := utils.CalcTokens(utils.JsonEncode(f), req.Model)
-				tokens += tks
-			}
-
+			tks, _ := utils.CalcTokens(utils.JsonEncode(req.Tools), req.Model)
+			tokens += tks
 			// loading the role context
 			var messages []types.Message
 			err := utils.JsonDecode(role.Context, &messages)
@@ -442,7 +464,7 @@ func (h *ChatHandler) doRequest(ctx context.Context, req types.ApiRequest, platf
 	} else {
 		client = http.DefaultClient
 	}
-	logger.Infof("Sending %s request, KEY: %s, PROXY: %s, Model: %s", platform, *apiKey, proxyURL, req.Model)
+	logger.Infof("Sending %s request, ApiURL:%s, PROXY: %s, Model: %s", platform, apiURL, proxyURL, req.Model)
 	switch platform {
 	case types.Azure:
 		request.Header.Set("api-key", *apiKey)
@@ -452,7 +474,6 @@ func (h *ChatHandler) doRequest(ctx context.Context, req types.ApiRequest, platf
 		if err != nil {
 			return nil, err
 		}
-		logger.Info(token)
 		request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", token))
 		break
 	case types.Baidu:

api/handler/chatimpl/openai_handler.go

@@ -9,7 +9,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"gorm.io/gorm"
+	req2 "github.com/imroc/req/v3"
 	"html/template"
 	"io"
 	"strings"
@@ -56,8 +56,8 @@ func (h *ChatHandler) sendOpenAiMessage(
 		// 循环读取 Chunk 消息
 		var message = types.Message{}
 		var contents = make([]string, 0)
-		var functionCall = false
-		var functionName string
+		var function model.Function
+		var toolCall = false
 		var arguments = make([]string, 0)
 		scanner := bufio.NewScanner(response.Body)
 		for scanner.Scan() {
@@ -75,24 +75,26 @@ func (h *ChatHandler) sendOpenAiMessage(
 				break
 			}
 
-			fun := responseBody.Choices[0].Delta.FunctionCall
-			if functionCall && fun.Name == "" {
-				arguments = append(arguments, fun.Arguments)
-				continue
+			var fun types.ToolCall
+			if len(responseBody.Choices[0].Delta.ToolCalls) > 0 {
+				fun = responseBody.Choices[0].Delta.ToolCalls[0]
+				if toolCall && fun.Function.Name == "" {
+					arguments = append(arguments, fun.Function.Arguments)
+					continue
+				}
 			}
 
 			if !utils.IsEmptyValue(fun) {
-				functionName = fun.Name
-				f := h.App.Functions[functionName]
-				if f != nil {
-					functionCall = true
+				res := h.db.Where("name = ?", fun.Function.Name).First(&function)
+				if res.Error == nil {
+					toolCall = true
 					utils.ReplyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
-					utils.ReplyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: fmt.Sprintf("正在调用函数 `%s` 作答 ...\n\n", f.Name())})
+					utils.ReplyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: fmt.Sprintf("正在调用工具 `%s` 作答 ...\n\n", function.Label)})
 				}
 				continue
 			}
 
-			if responseBody.Choices[0].FinishReason == "function_call" { // 函数调用完毕
+			if responseBody.Choices[0].FinishReason == "tool_calls" { // 函数调用完毕
 				break
 			}
 
@@ -121,45 +123,35 @@ func (h *ChatHandler) sendOpenAiMessage(
 			}
 		}
 
-		if functionCall { // 调用函数完成任务
+		if toolCall { // 调用函数完成任务
 			var params map[string]interface{}
 			_ = utils.JsonDecode(strings.Join(arguments, ""), &params)
-			logger.Debugf("函数名称: %s, 函数参数：%s", functionName, params)
-
-			// for creating image, check if the user's img_calls > 0
-			if functionName == types.FuncImage && userVo.ImgCalls <= 0 {
-				utils.ReplyMessage(ws, "**当前用户剩余绘图次数已用尽，请扫描下面二维码联系管理员！**")
-				utils.ReplyMessage(ws, ErrImg)
+			logger.Debugf("函数名称: %s, 函数参数：%s", function.Name, params)
+			params["user_id"] = userVo.Id
+			var apiRes types.BizVo
+			r, err := req2.C().R().SetHeader("Content-Type", "application/json").
+				SetHeader("Authorization", function.Token).
+				SetBody(params).
+				SetSuccessResult(&apiRes).Post(function.Action)
+			errMsg := ""
+			if err != nil {
+				errMsg = err.Error()
+			} else if r.IsErrorState() {
+				errMsg = r.Err.Error()
+			}
+			if errMsg != "" || apiRes.Code != types.Success {
+				msg := "调用函数工具出错：" + apiRes.Message + errMsg
+				utils.ReplyChunkMessage(ws, types.WsMessage{
+					Type:    types.WsMiddle,
+					Content: msg,
+				})
+				contents = append(contents, msg)
 			} else {
-				f := h.App.Functions[functionName]
-				if functionName == types.FuncImage {
-					params["user_id"] = userVo.Id
-					params["role_id"] = role.Id
-					params["chat_id"] = session.ChatId
-					params["session_id"] = session.SessionId
-				}
-				data, err := f.Invoke(params)
-				if err != nil {
-					msg := "调用函数出错：" + err.Error()
-					utils.ReplyChunkMessage(ws, types.WsMessage{
-						Type:    types.WsMiddle,
-						Content: msg,
-					})
-					contents = append(contents, msg)
-				} else {
-					content := data
-					if functionName == types.FuncImage {
-						content = fmt.Sprintf("下面是根据您的描述创作的图片，他们描绘了 【%s】 的场景。%s", params["prompt"], data)
-						// update user's img_calls
-						h.db.Model(&model.User{}).Where("id = ?", userVo.Id).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
-					}
-
-					utils.ReplyChunkMessage(ws, types.WsMessage{
-						Type:    types.WsMiddle,
-						Content: content,
-					})
-					contents = append(contents, content)
-				}
+				utils.ReplyChunkMessage(ws, types.WsMessage{
+					Type:    types.WsMiddle,
+					Content: apiRes.Data,
+				})
+				contents = append(contents, utils.InterfaceToString(apiRes.Data))
 			}
 		}
 
@@ -175,7 +167,7 @@ func (h *ChatHandler) sendOpenAiMessage(
 			useMsg := types.Message{Role: "user", Content: prompt}
 
 			// 更新上下文消息，如果是调用函数则不需要更新上下文
-			if h.App.ChatConfig.EnableContext && functionCall == false {
+			if h.App.ChatConfig.EnableContext && toolCall == false {
 				chatCtx = append(chatCtx, useMsg)  // 提问消息
 				chatCtx = append(chatCtx, message) // 回复消息
 				h.App.ChatContexts.Put(session.ChatId, chatCtx)
@@ -184,7 +176,7 @@ func (h *ChatHandler) sendOpenAiMessage(
 			// 追加聊天记录
 			if h.App.ChatConfig.EnableHistory {
 				useContext := true
-				if functionCall {
+				if toolCall {
 					useContext = false
 				}
 
@@ -212,8 +204,8 @@ func (h *ChatHandler) sendOpenAiMessage(
 
 				// 计算本次对话消耗的总 token 数量
 				var totalTokens = 0
-				if functionCall { // prompt + 函数名 + 参数 token
-					tokens, _ := utils.CalcTokens(functionName, req.Model)
+				if toolCall { // prompt + 函数名 + 参数 token
+					tokens, _ := utils.CalcTokens(function.Name, req.Model)
 					totalTokens += tokens
 					tokens, _ = utils.CalcTokens(utils.InterfaceToString(arguments), req.Model)
 					totalTokens += tokens

api/handler/chatimpl/xunfei_handler.go

@@ -50,7 +50,7 @@ type xunFeiResp struct {
 }
 
 var Model2URL = map[string]string{
-	"generalv1": "1.1",
+	"general":   "v1.1",
 	"generalv2": "v2.1",
 	"generalv3": "v3.1",
 }

api/handler/function_handler.go

@@ -0,0 +1,235 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/service/oss"
+	"chatplus/store/model"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/imroc/req/v3"
+	"gorm.io/gorm"
+	"strings"
+)
+
+type FunctionHandler struct {
+	BaseHandler
+	db            *gorm.DB
+	config        types.ChatPlusApiConfig
+	uploadManager *oss.UploaderManager
+	proxyURL      string
+}
+
+func NewFunctionHandler(server *core.AppServer, db *gorm.DB, config *types.AppConfig, manager *oss.UploaderManager) *FunctionHandler {
+	return &FunctionHandler{
+		BaseHandler: BaseHandler{
+			App: server,
+		},
+		db:            db,
+		config:        config.ApiConfig,
+		uploadManager: manager,
+		proxyURL:      config.ProxyURL,
+	}
+}
+
+type resVo struct {
+	Code    types.BizCode `json:"code"`
+	Message string        `json:"message"`
+	Data    struct {
+		Title     string     `json:"title"`
+		UpdatedAt string     `json:"updated_at"`
+		Items     []dataItem `json:"items"`
+	} `json:"data"`
+}
+
+type dataItem struct {
+	Title  string `json:"title"`
+	Url    string `json:"url"`
+	Remark string `json:"remark"`
+}
+
+// WeiBo 微博热搜
+func (h *FunctionHandler) WeiBo(c *gin.Context) {
+	if h.config.Token == "" {
+		resp.ERROR(c, "无效的 API Token")
+		return
+	}
+
+	url := fmt.Sprintf("%s/api/weibo/fetch", h.config.ApiURL)
+	var res resVo
+	r, err := req.C().R().
+		SetHeader("AppId", h.config.AppId).
+		SetHeader("Authorization", fmt.Sprintf("Bearer %s", h.config.Token)).
+		SetSuccessResult(&res).Get(url)
+	if err != nil || r.IsErrorState() {
+		resp.ERROR(c, fmt.Sprintf("%v%v", err, r.Err))
+		return
+	}
+
+	if res.Code != types.Success {
+		resp.ERROR(c, res.Message)
+		return
+	}
+
+	builder := make([]string, 0)
+	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Title, res.Data.UpdatedAt))
+	for i, v := range res.Data.Items {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [热度：%s]", i+1, v.Title, v.Url, v.Remark))
+	}
+	resp.SUCCESS(c, strings.Join(builder, "\n\n"))
+}
+
+// ZaoBao 今日早报
+func (h *FunctionHandler) ZaoBao(c *gin.Context) {
+	if h.config.Token == "" {
+		resp.ERROR(c, "无效的 API Token")
+		return
+	}
+
+	url := fmt.Sprintf("%s/api/zaobao/fetch", h.config.ApiURL)
+	var res resVo
+	r, err := req.C().R().
+		SetHeader("AppId", h.config.AppId).
+		SetHeader("Authorization", fmt.Sprintf("Bearer %s", h.config.Token)).
+		SetSuccessResult(&res).Get(url)
+	if err != nil || r.IsErrorState() {
+		resp.ERROR(c, fmt.Sprintf("%v%v", err, r.Err))
+		return
+	}
+
+	if res.Code != types.Success {
+		resp.ERROR(c, res.Message)
+		return
+	}
+
+	builder := make([]string, 0)
+	builder = append(builder, fmt.Sprintf("**%s 早报：**", res.Data.UpdatedAt))
+	for _, v := range res.Data.Items {
+		builder = append(builder, v.Title)
+	}
+	builder = append(builder, fmt.Sprintf("%s", res.Data.Title))
+	resp.SUCCESS(c, strings.Join(builder, "\n\n"))
+}
+
+type imgReq struct {
+	Model  string `json:"model"`
+	Prompt string `json:"prompt"`
+	N      int    `json:"n"`
+	Size   string `json:"size"`
+}
+
+type imgRes struct {
+	Created int64 `json:"created"`
+	Data    []struct {
+		RevisedPrompt string `json:"revised_prompt"`
+		Url           string `json:"url"`
+	} `json:"data"`
+}
+
+type ErrRes struct {
+	Error struct {
+		Code    interface{} `json:"code"`
+		Message string      `json:"message"`
+		Param   interface{} `json:"param"`
+		Type    string      `json:"type"`
+	} `json:"error"`
+}
+
+// Dall3 DallE3 AI 绘图
+func (h *FunctionHandler) Dall3(c *gin.Context) {
+	var params map[string]interface{}
+	if err := c.ShouldBindJSON(&params); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	logger.Debugf("绘画参数：%+v", params)
+	// check img calls
+	var user model.User
+	tx := h.db.Where("id = ?", params["user_id"]).First(&user)
+	if tx.Error != nil {
+		resp.ERROR(c, "当前用户不存在！")
+		return
+	}
+
+	if user.ImgCalls <= 0 {
+		resp.ERROR(c, "当前用户的绘图次数额度不足！")
+		return
+	}
+
+	prompt := utils.InterfaceToString(params["prompt"])
+	// get image generation API KEY
+	var apiKey model.ApiKey
+	tx = h.db.Where("platform = ? AND type = ?", types.OpenAI, "img").Order("last_used_at ASC").First(&apiKey)
+	if tx.Error != nil {
+		resp.ERROR(c, "获取绘图 API KEY 失败: "+tx.Error.Error())
+		return
+	}
+
+	// get image generation api URL
+	var conf model.Config
+	var chatConfig types.ChatConfig
+	tx = h.db.Where("marker", "chat").First(&conf)
+	if tx.Error != nil {
+		resp.ERROR(c, "error with get chat configs:"+tx.Error.Error())
+		return
+	}
+
+	err := utils.JsonDecode(conf.Config, &chatConfig)
+	if err != nil {
+		resp.ERROR(c, "error with decode chat config: "+err.Error())
+		return
+	}
+
+	// translate prompt
+	const translatePromptTemplate = "Translate the following painting prompt words into English keyword phrases. Without any explanation, directly output the keyword phrases separated by commas. The content to be translated is: [%s]"
+	pt, err := utils.OpenAIRequest(fmt.Sprintf(translatePromptTemplate, params["prompt"]), apiKey.Value, h.App.Config.ProxyURL, chatConfig.OpenAI.ApiURL)
+	if err == nil {
+		prompt = pt
+	}
+
+	apiURL := chatConfig.DallApiURL
+	if utils.IsEmptyValue(apiURL) {
+		apiURL = "https://api.openai.com/v1/images/generations"
+	}
+	imgNum := chatConfig.DallImgNum
+	if imgNum <= 0 {
+		imgNum = 1
+	}
+	var res imgRes
+	var errRes ErrRes
+	var request *req.Request
+	if strings.Contains(apiURL, "api.openai.com") {
+		request = req.C().SetProxyURL(h.proxyURL).R()
+	} else {
+		request = req.C().R()
+	}
+	r, err := request.SetHeader("Content-Type", "application/json").
+		SetHeader("Authorization", "Bearer "+apiKey.Value).
+		SetBody(imgReq{
+			Model:  "dall-e-3",
+			Prompt: prompt,
+			N:      imgNum,
+			Size:   "1024x1024",
+		}).
+		SetErrorResult(&errRes).
+		SetSuccessResult(&res).Post(apiURL)
+	if r.IsErrorState() {
+		resp.ERROR(c, "请求 OpenAI API 失败: "+errRes.Error.Message)
+		return
+	}
+	// 存储图片
+	imgURL, err := h.uploadManager.GetUploadHandler().PutImg(res.Data[0].Url, false)
+	if err != nil {
+		resp.ERROR(c, "下载图片失败: "+err.Error())
+		return
+	}
+
+	content := fmt.Sprintf("下面是根据您的描述创作的图片，它描绘了 【%s】 的场景。 \n\n![](%s)\n", prompt, imgURL)
+	// update user's img_calls
+	h.db.Model(&model.User{}).Where("id = ?", user.Id).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
+
+	resp.SUCCESS(c, content)
+}

api/handler/mj_handler.go

@@ -5,6 +5,7 @@ import (
 	"chatplus/core/types"
 	"chatplus/service"
 	"chatplus/service/mj"
+	"chatplus/service/oss"
 	"chatplus/store/model"
 	"chatplus/store/vo"
 	"chatplus/utils"
@@ -12,7 +13,9 @@ import (
 	"encoding/base64"
 	"fmt"
 	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
 	"gorm.io/gorm"
+	"net/http"
 	"strings"
 	"time"
 )
@@ -22,13 +25,15 @@ type MidJourneyHandler struct {
 	db        *gorm.DB
 	pool      *mj.ServicePool
 	snowflake *service.Snowflake
+	uploader  *oss.UploaderManager
 }
 
-func NewMidJourneyHandler(app *core.AppServer, db *gorm.DB, snowflake *service.Snowflake, pool *mj.ServicePool) *MidJourneyHandler {
+func NewMidJourneyHandler(app *core.AppServer, db *gorm.DB, snowflake *service.Snowflake, pool *mj.ServicePool, manager *oss.UploaderManager) *MidJourneyHandler {
 	h := MidJourneyHandler{
 		db:        db,
 		snowflake: snowflake,
 		pool:      pool,
+		uploader:  manager,
 	}
 	h.App = app
 	return &h
@@ -55,6 +60,27 @@ func (h *MidJourneyHandler) preCheck(c *gin.Context) bool {
 
 }
 
+// Client WebSocket 客户端，用于通知任务状态变更
+func (h *MidJourneyHandler) Client(c *gin.Context) {
+	ws, err := (&websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}).Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		logger.Error(err)
+		c.Abort()
+		return
+	}
+
+	userId := h.GetInt(c, "user_id", 0)
+	if userId == 0 {
+		logger.Info("Invalid user ID")
+		c.Abort()
+		return
+	}
+
+	client := types.NewWsClient(ws)
+	h.pool.Clients.Put(uint(userId), client)
+	logger.Infof("New websocket connected, IP: %s", c.RemoteIP())
+}
+
 // Image 创建一个绘画任务
 func (h *MidJourneyHandler) Image(c *gin.Context) {
 	var data struct {
@@ -143,12 +169,19 @@ func (h *MidJourneyHandler) Image(c *gin.Context) {
 		Prompt:    fmt.Sprintf("%s %s", taskId, prompt),
 		UserId:    userId,
 	})
+
+	client := h.pool.Clients.Get(uint(job.UserId))
+	_ = client.Send([]byte("Task Updated"))
+
+	// update user's img calls
+	h.db.Model(&model.User{}).Where("id = ?", job.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
 	resp.SUCCESS(c)
 }
 
 type reqVo struct {
-	Src         string `json:"src"`
+	TaskId      string `json:"task_id"`
 	Index       int    `json:"index"`
+	ChannelId   string `json:"channel_id"`
 	MessageId   string `json:"message_id"`
 	MessageHash string `json:"message_hash"`
 	SessionId   string `json:"session_id"`
@@ -173,16 +206,35 @@ func (h *MidJourneyHandler) Upscale(c *gin.Context) {
 	idValue, _ := c.Get(types.LoginUserID)
 	jobId := 0
 	userId := utils.IntValue(utils.InterfaceToString(idValue), 0)
+	job := model.MidJourneyJob{
+		Type:        types.TaskUpscale.String(),
+		ReferenceId: data.MessageId,
+		UserId:      userId,
+		TaskId:      data.TaskId,
+		Progress:    0,
+		Prompt:      data.Prompt,
+		CreatedAt:   time.Now(),
+	}
+	if res := h.db.Create(&job); res.Error != nil {
+		resp.ERROR(c, "添加任务失败："+res.Error.Error())
+		return
+	}
+
 	h.pool.PushTask(types.MjTask{
 		Id:          jobId,
 		SessionId:   data.SessionId,
 		Type:        types.TaskUpscale,
 		Prompt:      data.Prompt,
 		UserId:      userId,
+		ChannelId:   data.ChannelId,
 		Index:       data.Index,
 		MessageId:   data.MessageId,
 		MessageHash: data.MessageHash,
 	})
+
+	client := h.pool.Clients.Get(uint(job.UserId))
+	_ = client.Send([]byte("Task Updated"))
+
 	resp.SUCCESS(c)
 }
 
@@ -201,6 +253,22 @@ func (h *MidJourneyHandler) Variation(c *gin.Context) {
 	idValue, _ := c.Get(types.LoginUserID)
 	jobId := 0
 	userId := utils.IntValue(utils.InterfaceToString(idValue), 0)
+
+	job := model.MidJourneyJob{
+		Type:        types.TaskVariation.String(),
+		ChannelId:   data.ChannelId,
+		ReferenceId: data.MessageId,
+		UserId:      userId,
+		TaskId:      data.TaskId,
+		Progress:    0,
+		Prompt:      data.Prompt,
+		CreatedAt:   time.Now(),
+	}
+	if res := h.db.Create(&job); res.Error != nil {
+		resp.ERROR(c, "添加任务失败："+res.Error.Error())
+		return
+	}
+
 	h.pool.PushTask(types.MjTask{
 		Id:          jobId,
 		SessionId:   data.SessionId,
@@ -208,9 +276,16 @@ func (h *MidJourneyHandler) Variation(c *gin.Context) {
 		Prompt:      data.Prompt,
 		UserId:      userId,
 		Index:       data.Index,
+		ChannelId:   data.ChannelId,
 		MessageId:   data.MessageId,
 		MessageHash: data.MessageHash,
 	})
+
+	client := h.pool.Clients.Get(uint(job.UserId))
+	_ = client.Send([]byte("Task Updated"))
+
+	// update user's img calls
+	h.db.Model(&model.User{}).Where("id = ?", job.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
 	resp.SUCCESS(c)
 }
 
@@ -258,6 +333,8 @@ func (h *MidJourneyHandler) JobList(c *gin.Context) {
 			// 10 分钟还没完成的任务直接删除
 			if time.Now().Sub(item.CreatedAt) > time.Minute*10 {
 				h.db.Delete(&item)
+				// 退回绘图次数
+				h.db.Model(&model.User{}).Where("id = ?", item.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 				continue
 			}
 
@@ -274,3 +351,34 @@ func (h *MidJourneyHandler) JobList(c *gin.Context) {
 	}
 	resp.SUCCESS(c, jobs)
 }
+
+// Remove remove task image
+func (h *MidJourneyHandler) Remove(c *gin.Context) {
+	var data struct {
+		Id     uint   `json:"id"`
+		UserId uint   `json:"user_id"`
+		ImgURL string `json:"img_url"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	// remove job recode
+	res := h.db.Delete(&model.MidJourneyJob{Id: data.Id})
+	if res.Error != nil {
+		resp.ERROR(c, res.Error.Error())
+		return
+	}
+
+	// remove image
+	err := h.uploader.GetUploadHandler().Delete(data.ImgURL)
+	if err != nil {
+		logger.Error("remove image failed: ", err)
+	}
+
+	client := h.pool.Clients.Get(data.UserId)
+	_ = client.Send([]byte("Task Updated"))
+
+	resp.SUCCESS(c)
+}

api/handler/payment_handler.go

@@ -99,14 +99,15 @@ func (h *PaymentHandler) DoPay(c *gin.Context) {
 		}
 
 		var r struct {
-			Openid    int64  `json:"openid"`
-			UrlQrcode string `json:"url_qrcode"`
-			URL       string `json:"url"`
-			ErrCode   int    `json:"errcode"`
-			ErrMsg    string `json:"errmsg"`
+			Openid    interface{} `json:"openid"`
+			UrlQrcode string      `json:"url_qrcode"`
+			URL       string      `json:"url"`
+			ErrCode   int         `json:"errcode"`
+			ErrMsg    string      `json:"errmsg,omitempty"`
 		}
 		err = utils.JsonDecode(res, &r)
 		if err != nil {
+			logger.Debugf(res)
 			resp.ERROR(c, "error with decode payment result: "+err.Error())
 			return
 		}

api/handler/prompt_handler.go

@@ -4,11 +4,10 @@ import (
 	"chatplus/core"
 	"chatplus/core/types"
 	"chatplus/store/model"
+	"chatplus/utils"
 	"chatplus/utils/resp"
 	"fmt"
 
-	"github.com/imroc/req/v3"
-
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
 )
@@ -27,27 +26,6 @@ func NewPromptHandler(app *core.AppServer, db *gorm.DB) *PromptHandler {
 	return h
 }
 
-type apiRes struct {
-	Model   string `json:"model"`
-	Choices []struct {
-		Index   int `json:"index"`
-		Message struct {
-			Role    string `json:"role"`
-			Content string `json:"content"`
-		} `json:"message"`
-		FinishReason string `json:"finish_reason"`
-	} `json:"choices"`
-}
-
-type apiErrRes struct {
-	Error struct {
-		Code    interface{} `json:"code"`
-		Message string      `json:"message"`
-		Param   interface{} `json:"param"`
-		Type    string      `json:"type"`
-	} `json:"error"`
-}
-
 // Rewrite translate and rewrite prompt with ChatGPT
 func (h *PromptHandler) Rewrite(c *gin.Context) {
 	var data struct {
@@ -93,28 +71,5 @@ func (h *PromptHandler) request(prompt string, promptTemplate string) (string, e
 		return "", fmt.Errorf("error with fetch OpenAI API KEY：%v", res.Error)
 	}
 
-	messages := make([]interface{}, 1)
-	messages[0] = types.Message{
-		Role:    "user",
-		Content: fmt.Sprintf(promptTemplate, prompt),
-	}
-
-	var response apiRes
-	var errRes apiErrRes
-	r, err := req.C().SetProxyURL(h.App.Config.ProxyURL).R().SetHeader("Content-Type", "application/json").
-		SetHeader("Authorization", "Bearer "+apiKey.Value).
-		SetBody(types.ApiRequest{
-			Model:       "gpt-3.5-turbo",
-			Temperature: 0.9,
-			MaxTokens:   1024,
-			Stream:      false,
-			Messages:    messages,
-		}).
-		SetErrorResult(&errRes).
-		SetSuccessResult(&response).Post(h.App.ChatConfig.OpenAI.ApiURL)
-	if err != nil || r.IsErrorState() {
-		return "", fmt.Errorf("error with http request: %v%v%s", err, r.Err, errRes.Error.Message)
-	}
-
-	return response.Choices[0].Message.Content, nil
+	return utils.OpenAIRequest(fmt.Sprintf(promptTemplate, prompt), apiKey.Value, h.App.Config.ProxyURL, h.App.ChatConfig.OpenAI.ApiURL)
 }

api/handler/sd_handler.go

@@ -3,6 +3,7 @@ package handler
 import (
 	"chatplus/core"
 	"chatplus/core/types"
+	"chatplus/service/oss"
 	"chatplus/service/sd"
 	"chatplus/store/model"
 	"chatplus/store/vo"
@@ -10,23 +11,26 @@ import (
 	"chatplus/utils/resp"
 	"encoding/base64"
 	"fmt"
+	"time"
+
 	"github.com/gin-gonic/gin"
 	"github.com/go-redis/redis/v8"
 	"gorm.io/gorm"
-	"time"
 )
 
 type SdJobHandler struct {
 	BaseHandler
-	redis *redis.Client
-	db    *gorm.DB
-	pool  *sd.ServicePool
+	redis    *redis.Client
+	db       *gorm.DB
+	pool     *sd.ServicePool
+	uploader *oss.UploaderManager
 }
 
-func NewSdJobHandler(app *core.AppServer, db *gorm.DB, pool *sd.ServicePool) *SdJobHandler {
+func NewSdJobHandler(app *core.AppServer, db *gorm.DB, pool *sd.ServicePool, manager *oss.UploaderManager) *SdJobHandler {
 	h := SdJobHandler{
-		db:   db,
-		pool: pool,
+		db:       db,
+		pool:     pool,
+		uploader: manager,
 	}
 	h.App = app
 	return &h
@@ -129,6 +133,9 @@ func (h *SdJobHandler) Image(c *gin.Context) {
 		UserId:    userId,
 	})
 
+	// update user's img calls
+	h.db.Model(&model.User{}).Where("id = ?", job.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
+
 	resp.SUCCESS(c)
 }
 
@@ -169,13 +176,15 @@ func (h *SdJobHandler) JobList(c *gin.Context) {
 		}
 
 		if job.Progress == -1 {
-			h.db.Delete(&model.MidJourneyJob{Id: job.Id})
+			h.db.Delete(&model.SdJob{Id: job.Id})
 		}
 
 		if item.Progress < 100 {
-			// 10 分钟还没完成的任务直接删除
-			if time.Now().Sub(item.CreatedAt) > time.Minute*10 {
+			// 5 分钟还没完成的任务直接删除
+			if time.Now().Sub(item.CreatedAt) > time.Minute*5 {
 				h.db.Delete(&item)
+				// 退回绘图次数
+				h.db.Model(&model.User{}).Where("id = ?", item.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 				continue
 			}
 			// 正在运行中任务使用代理访问图片
@@ -188,3 +197,30 @@ func (h *SdJobHandler) JobList(c *gin.Context) {
 	}
 	resp.SUCCESS(c, jobs)
 }
+
+// Remove remove task image
+func (h *SdJobHandler) Remove(c *gin.Context) {
+	var data struct {
+		Id     uint   `json:"id"`
+		ImgURL string `json:"img_url"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	// remove job recode
+	res := h.db.Delete(&model.SdJob{Id: data.Id})
+	if res.Error != nil {
+		resp.ERROR(c, res.Error.Error())
+		return
+	}
+
+	// remove image
+	err := h.uploader.GetUploadHandler().Delete(data.ImgURL)
+	if err != nil {
+		logger.Error("remove image failed: ", err)
+	}
+
+	resp.SUCCESS(c)
+}

api/main.go

@@ -8,7 +8,6 @@ import (
 	"chatplus/handler/chatimpl"
 	logger2 "chatplus/logger"
 	"chatplus/service"
-	"chatplus/service/fun"
 	"chatplus/service/mj"
 	"chatplus/service/oss"
 	"chatplus/service/payment"
@@ -115,9 +114,6 @@ func main() {
 			return xdb.NewWithBuffer(cBuff)
 		}),
 
-		// 创建函数
-		fx.Provide(fun.NewFunctions),
-
 		// 创建控制器
 		fx.Provide(handler.NewChatRoleHandler),
 		fx.Provide(handler.NewUserHandler),
@@ -165,6 +161,12 @@ func main() {
 
 		// MidJourney service pool
 		fx.Provide(mj.NewServicePool),
+		fx.Invoke(func(pool *mj.ServicePool) {
+			if pool.HasAvailableService() {
+				pool.DownloadImages()
+				pool.CheckTaskNotify()
+			}
+		}),
 
 		// Stable Diffusion 机器人
 		fx.Provide(sd.NewServicePool),
@@ -229,15 +231,18 @@ func main() {
 		}),
 		fx.Invoke(func(s *core.AppServer, h *handler.MidJourneyHandler) {
 			group := s.Engine.Group("/api/mj/")
+			group.Any("client", h.Client)
 			group.POST("image", h.Image)
 			group.POST("upscale", h.Upscale)
 			group.POST("variation", h.Variation)
 			group.GET("jobs", h.JobList)
+			group.POST("remove", h.Remove)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *handler.SdJobHandler) {
 			group := s.Engine.Group("/api/sd")
 			group.POST("image", h.Image)
 			group.GET("jobs", h.JobList)
+			group.POST("remove", h.Remove)
 		}),
 
 		// 管理后台控制器
@@ -339,6 +344,24 @@ func main() {
 			group.POST("translate", h.Translate)
 		}),
 
+		fx.Provide(admin.NewFunctionHandler),
+		fx.Invoke(func(s *core.AppServer, h *admin.FunctionHandler) {
+			group := s.Engine.Group("/api/admin/function/")
+			group.POST("save", h.Save)
+			group.POST("set", h.Set)
+			group.GET("list", h.List)
+			group.GET("remove", h.Remove)
+			group.GET("token", h.GenToken)
+		}),
+
+		fx.Provide(handler.NewFunctionHandler),
+		fx.Invoke(func(s *core.AppServer, h *handler.FunctionHandler) {
+			group := s.Engine.Group("/api/function/")
+			group.POST("weibo", h.WeiBo)
+			group.POST("zaobao", h.ZaoBao)
+			group.POST("dalle3", h.Dall3)
+		}),
+
 		fx.Provide(handler.NewTestHandler),
 		fx.Invoke(func(s *core.AppServer, h *handler.TestHandler) {
 			group := s.Engine.Group("/test/")

api/res/sd/text2img.json

@@ -1,11 +1,11 @@
 {
   "data": [
-    "task(owy5niy1sbbnlq0)",
-    "A beautiful Chinese girl plays the guitar on the beach. She is dressed in a flowing dress that matches the colors of the sunset. With her eyes closed, she strums the guitar with passion and confidence, her fingers dancing gracefully on the strings. The painting employs a vibrant color palette, capturing the warmth of the setting sun blending with the serene hues of the ocean. The artist uses a combination of impressionistic and realistic brushstrokes to convey both the girl's delicate features and the dynamic movement of the waves. The rendering effect creates a dream-like atmosphere, as if the viewer is being transported to a magical realm where music and nature intertwine. The picture is bathed in a soft, golden light, casting a warm glow on the girl's face, illuminating her joy and connection to the music she creates.",
+    "task(cxvkpawy8onnfti)",
+    "a  cute girl",
     "",
     [],
-    30,
-    "DPM++ 3M SDE Karras",
+    20,
+    "DPM++ 2M Karras",
     1,
     1,
     7,
@@ -36,6 +36,7 @@
     null,
     null,
     null,
+    null,
     false,
     false,
     "positive",
@@ -74,6 +75,6 @@
     ""
   ],
   "event_data": null,
-  "fn_index": 316,
-  "session_hash": "ttr8efgt63g"
+  "fn_index": 446,
+  "session_hash": "nk5noh1rz1o"
 }

api/service/fun/func_img.go

@@ -1,116 +0,0 @@
-package fun
-
-import (
-	"chatplus/core/types"
-	"chatplus/service/oss"
-	"chatplus/store/model"
-	"chatplus/utils"
-	"fmt"
-
-	"github.com/imroc/req/v3"
-	"gorm.io/gorm"
-)
-
-// AI 绘画函数
-
-type FuncImage struct {
-	name          string
-	db            *gorm.DB
-	uploadManager *oss.UploaderManager
-	proxyURL      string
-}
-
-func NewImageFunc(db *gorm.DB, manager *oss.UploaderManager, config *types.AppConfig) FuncImage {
-	return FuncImage{
-		db:            db,
-		name:          "DALL-E3 绘画",
-		uploadManager: manager,
-		proxyURL:      config.ProxyURL,
-	}
-}
-
-type imgReq struct {
-	Model  string `json:"model"`
-	Prompt string `json:"prompt"`
-	N      int    `json:"n"`
-	Size   string `json:"size"`
-}
-
-type imgRes struct {
-	Created int64 `json:"created"`
-	Data    []struct {
-		RevisedPrompt string `json:"revised_prompt"`
-		Url           string `json:"url"`
-	} `json:"data"`
-}
-
-type ErrRes struct {
-	Error struct {
-		Code    interface{} `json:"code"`
-		Message string      `json:"message"`
-		Param   interface{} `json:"param"`
-		Type    string      `json:"type"`
-	} `json:"error"`
-}
-
-func (f FuncImage) Invoke(params map[string]interface{}) (string, error) {
-	logger.Infof("绘画参数：%+v", params)
-	prompt := utils.InterfaceToString(params["prompt"])
-	// get image generation API KEY
-	var apiKey model.ApiKey
-	tx := f.db.Where("platform = ? AND type = ?", types.OpenAI, "img").Order("last_used_at ASC").First(&apiKey)
-	if tx.Error != nil {
-		return "", fmt.Errorf("error with get generation API KEY: %v", tx.Error)
-	}
-
-	// get image generation api URL
-	var conf model.Config
-	var chatConfig types.ChatConfig
-	tx = f.db.Where("marker", "chat").First(&conf)
-	if tx.Error != nil {
-		return "", fmt.Errorf("error with get chat configs: %v", tx.Error)
-	}
-
-	err := utils.JsonDecode(conf.Config, &chatConfig)
-	if err != nil {
-		return "", fmt.Errorf("error with decode chat config: %v", err)
-	}
-
-	apiURL := chatConfig.DallApiURL
-	if utils.IsEmptyValue(apiURL) {
-		apiURL = "https://api.openai.com/v1/images/generations"
-	}
-	imgNum := chatConfig.DallImgNum
-	if imgNum <= 0 {
-		imgNum = 1
-	}
-	var res imgRes
-	var errRes ErrRes
-	r, err := req.C().SetProxyURL(f.proxyURL).R().SetHeader("Content-Type", "application/json").
-		SetHeader("Authorization", "Bearer "+apiKey.Value).
-		SetBody(imgReq{
-			Model:  "dall-e-3",
-			Prompt: prompt,
-			N:      imgNum,
-			Size:   "1024x1024",
-		}).
-		SetErrorResult(&errRes).
-		SetSuccessResult(&res).Post(apiURL)
-	if err != nil || r.IsErrorState() {
-		return "", fmt.Errorf("error with http request: %v%v%s", err, r.Err, errRes.Error.Message)
-	}
-	// 存储图片
-	imgURL, err := f.uploadManager.GetUploadHandler().PutImg(res.Data[0].Url, false)
-	if err != nil {
-		return "", fmt.Errorf("下载图片失败: %s", err.Error())
-	}
-
-	logger.Info(imgURL)
-	return fmt.Sprintf("\n\n![](%s)\n", imgURL), nil
-}
-
-func (f FuncImage) Name() string {
-	return f.name
-}
-
-var _ Function = &FuncImage{}

api/service/fun/function.go

@@ -1,40 +0,0 @@
-package fun
-
-import (
-	"chatplus/core/types"
-	logger2 "chatplus/logger"
-	"chatplus/service/oss"
-	"gorm.io/gorm"
-)
-
-type Function interface {
-	Invoke(map[string]interface{}) (string, error)
-	Name() string
-}
-
-var logger = logger2.GetLogger()
-
-type resVo struct {
-	Code    types.BizCode `json:"code"`
-	Message string        `json:"message"`
-	Data    struct {
-		Title     string     `json:"title"`
-		UpdatedAt string     `json:"updated_at"`
-		Items     []dataItem `json:"items"`
-	} `json:"data"`
-}
-
-type dataItem struct {
-	Title  string `json:"title"`
-	Url    string `json:"url"`
-	Remark string `json:"remark"`
-}
-
-func NewFunctions(config *types.AppConfig, db *gorm.DB, manager *oss.UploaderManager) map[string]Function {
-	return map[string]Function{
-		types.FuncZaoBao:   NewZaoBao(config.ApiConfig),
-		types.FuncWeibo:    NewWeiboHot(config.ApiConfig),
-		types.FuncHeadLine: NewHeadLines(config.ApiConfig),
-		types.FuncImage:    NewImageFunc(db, manager, config),
-	}
-}

api/service/fun/tou_tiao.go

@@ -1,58 +0,0 @@
-package fun
-
-import (
-	"chatplus/core/types"
-	"errors"
-	"fmt"
-	"github.com/imroc/req/v3"
-	"strings"
-	"time"
-)
-
-// 今日头条函数实现
-
-type FuncHeadlines struct {
-	name   string
-	config types.ChatPlusApiConfig
-	client *req.Client
-}
-
-func NewHeadLines(config types.ChatPlusApiConfig) FuncHeadlines {
-	return FuncHeadlines{
-		name:   "今日头条",
-		config: config,
-		client: req.C().SetTimeout(10 * time.Second)}
-}
-
-func (f FuncHeadlines) Invoke(map[string]interface{}) (string, error) {
-	if f.config.Token == "" {
-		return "", errors.New("无效的 API Token")
-	}
-
-	url := fmt.Sprintf("%s/api/headline/fetch", f.config.ApiURL)
-	var res resVo
-	r, err := f.client.R().
-		SetHeader("AppId", f.config.AppId).
-		SetHeader("Authorization", fmt.Sprintf("Bearer %s", f.config.Token)).
-		SetSuccessResult(&res).Get(url)
-	if err != nil || r.IsErrorState() {
-		return "", fmt.Errorf("%v%v", err, r.Err)
-	}
-
-	if res.Code != types.Success {
-		return "", errors.New(res.Message)
-	}
-
-	builder := make([]string, 0)
-	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Title, res.Data.UpdatedAt))
-	for i, v := range res.Data.Items {
-		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [%s]", i+1, v.Title, v.Url, v.Remark))
-	}
-	return strings.Join(builder, "\n\n"), nil
-}
-
-func (f FuncHeadlines) Name() string {
-	return f.name
-}
-
-var _ Function = &FuncHeadlines{}

api/service/fun/weibo_hot.go

@@ -1,58 +0,0 @@
-package fun
-
-import (
-	"chatplus/core/types"
-	"errors"
-	"fmt"
-	"github.com/imroc/req/v3"
-	"strings"
-	"time"
-)
-
-// 微博热搜函数实现
-
-type FuncWeiboHot struct {
-	name   string
-	config types.ChatPlusApiConfig
-	client *req.Client
-}
-
-func NewWeiboHot(config types.ChatPlusApiConfig) FuncWeiboHot {
-	return FuncWeiboHot{
-		name:   "微博热搜",
-		config: config,
-		client: req.C().SetTimeout(10 * time.Second)}
-}
-
-func (f FuncWeiboHot) Invoke(map[string]interface{}) (string, error) {
-	if f.config.Token == "" {
-		return "", errors.New("无效的 API Token")
-	}
-
-	url := fmt.Sprintf("%s/api/weibo/fetch", f.config.ApiURL)
-	var res resVo
-	r, err := f.client.R().
-		SetHeader("AppId", f.config.AppId).
-		SetHeader("Authorization", fmt.Sprintf("Bearer %s", f.config.Token)).
-		SetSuccessResult(&res).Get(url)
-	if err != nil || r.IsErrorState() {
-		return "", fmt.Errorf("%v%v", err, r.Err)
-	}
-
-	if res.Code != types.Success {
-		return "", errors.New(res.Message)
-	}
-
-	builder := make([]string, 0)
-	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Title, res.Data.UpdatedAt))
-	for i, v := range res.Data.Items {
-		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [热度：%s]", i+1, v.Title, v.Url, v.Remark))
-	}
-	return strings.Join(builder, "\n\n"), nil
-}
-
-func (f FuncWeiboHot) Name() string {
-	return f.name
-}
-
-var _ Function = &FuncWeiboHot{}

api/service/fun/zao_bao.go

@@ -1,59 +0,0 @@
-package fun
-
-import (
-	"chatplus/core/types"
-	"errors"
-	"fmt"
-	"github.com/imroc/req/v3"
-	"strings"
-	"time"
-)
-
-// 每日早报函数实现
-
-type FuncZaoBao struct {
-	name   string
-	config types.ChatPlusApiConfig
-	client *req.Client
-}
-
-func NewZaoBao(config types.ChatPlusApiConfig) FuncZaoBao {
-	return FuncZaoBao{
-		name:   "每日早报",
-		config: config,
-		client: req.C().SetTimeout(10 * time.Second)}
-}
-
-func (f FuncZaoBao) Invoke(map[string]interface{}) (string, error) {
-	if f.config.Token == "" {
-		return "", errors.New("无效的 API Token")
-	}
-
-	url := fmt.Sprintf("%s/api/zaobao/fetch", f.config.ApiURL)
-	var res resVo
-	r, err := f.client.R().
-		SetHeader("AppId", f.config.AppId).
-		SetHeader("Authorization", fmt.Sprintf("Bearer %s", f.config.Token)).
-		SetSuccessResult(&res).Get(url)
-	if err != nil || r.IsErrorState() {
-		return "", fmt.Errorf("%v%v", err, r.Err)
-	}
-
-	if res.Code != types.Success {
-		return "", errors.New(res.Message)
-	}
-
-	builder := make([]string, 0)
-	builder = append(builder, fmt.Sprintf("**%s 早报：**", res.Data.UpdatedAt))
-	for _, v := range res.Data.Items {
-		builder = append(builder, v.Title)
-	}
-	builder = append(builder, fmt.Sprintf("%s", res.Data.Title))
-	return strings.Join(builder, "\n\n"), nil
-}
-
-func (f FuncZaoBao) Name() string {
-	return f.name
-}
-
-var _ Function = &FuncZaoBao{}

api/service/mj/bot.go

@@ -4,7 +4,7 @@ import (
 	"chatplus/core/types"
 	logger2 "chatplus/logger"
 	"chatplus/utils"
-	"github.com/bwmarrin/discordgo"
+	discordgo "github.com/bg5t/mydiscordgo"
 	"github.com/gorilla/websocket"
 	"net/http"
 	"net/url"
@@ -17,33 +17,48 @@ import (
 var logger = logger2.GetLogger()
 
 type Bot struct {
-	config  *types.MidJourneyConfig
+	config  types.MidJourneyConfig
 	bot     *discordgo.Session
 	name    string
 	service *Service
 }
 
-func NewBot(name string, proxy string, config *types.MidJourneyConfig, service *Service) (*Bot, error) {
-	discord, err := discordgo.New("Bot " + config.BotToken)
+func NewBot(name string, proxy string, config types.MidJourneyConfig, service *Service) (*Bot, error) {
+	bot, err := discordgo.New("Bot " + config.BotToken)
 	if err != nil {
+		logger.Error(err)
 		return nil, err
 	}
 
-	if proxy != "" {
-		proxy, _ := url.Parse(proxy)
-		discord.Client = &http.Client{
-			Transport: &http.Transport{
+	// use CDN reverse proxy
+	if config.UseCDN {
+		discordgo.SetEndpointDiscord(config.DiscordAPI)
+		discordgo.SetEndpointCDN(config.DiscordCDN)
+		discordgo.SetEndpointStatus(config.DiscordAPI + "/api/v2/")
+		bot.MjGateway = config.DiscordGateway + "/"
+	} else { // use proxy
+		discordgo.SetEndpointDiscord("https://discord.com")
+		discordgo.SetEndpointCDN("https://cdn.discordapp.com")
+		discordgo.SetEndpointStatus("https://discord.com/api/v2/")
+		bot.MjGateway = "wss://gateway.discord.gg"
+
+		if proxy != "" {
+			proxy, _ := url.Parse(proxy)
+			bot.Client = &http.Client{
+				Transport: &http.Transport{
+					Proxy: http.ProxyURL(proxy),
+				},
+			}
+			bot.Dialer = &websocket.Dialer{
 				Proxy: http.ProxyURL(proxy),
-			},
-		}
-		discord.Dialer = &websocket.Dialer{
-			Proxy: http.ProxyURL(proxy),
+			}
 		}
+
 	}
 
 	return &Bot{
 		config:  config,
-		bot:     discord,
+		bot:     bot,
 		name:    name,
 		service: service,
 	}, nil
@@ -89,7 +104,7 @@ func (b *Bot) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 		return
 	}
 	// ignore messages for self
-	if m.Author.ID == s.State.User.ID {
+	if m.Author == nil || m.Author.ID == s.State.User.ID {
 		return
 	}
 
@@ -101,6 +116,7 @@ func (b *Bot) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 	if strings.Contains(m.Content, "(Waiting to start)") && !strings.Contains(m.Content, "Rerolling **") {
 		// parse content
 		req := CBReq{
+			ChannelId:   m.ChannelID,
 			MessageId:   m.ID,
 			ReferenceId: referenceId,
 			Prompt:      extractPrompt(m.Content),
@@ -111,7 +127,7 @@ func (b *Bot) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 		return
 	}
 
-	b.addAttachment(m.ID, referenceId, m.Content, m.Attachments)
+	b.addAttachment(m.ChannelID, m.ID, referenceId, m.Content, m.Attachments)
 }
 
 func (b *Bot) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) {
@@ -120,7 +136,7 @@ func (b *Bot) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) {
 		return
 	}
 	// ignore messages for self
-	if m.Author.ID == s.State.User.ID {
+	if m.Author == nil || m.Author.ID == s.State.User.ID {
 		return
 	}
 
@@ -132,6 +148,7 @@ func (b *Bot) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) {
 	}
 	if strings.Contains(m.Content, "(Stopped)") {
 		req := CBReq{
+			ChannelId:   m.ChannelID,
 			MessageId:   m.ID,
 			ReferenceId: referenceId,
 			Prompt:      extractPrompt(m.Content),
@@ -142,11 +159,11 @@ func (b *Bot) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) {
 		return
 	}
 
-	b.addAttachment(m.ID, referenceId, m.Content, m.Attachments)
+	b.addAttachment(m.ChannelID, m.ID, referenceId, m.Content, m.Attachments)
 
 }
 
-func (b *Bot) addAttachment(messageId string, referenceId string, content string, attachments []*discordgo.MessageAttachment) {
+func (b *Bot) addAttachment(channelId string, messageId string, referenceId string, content string, attachments []*discordgo.MessageAttachment) {
 	progress := extractProgress(content)
 	var status TaskStatus
 	if progress == 100 {
@@ -168,6 +185,7 @@ func (b *Bot) addAttachment(messageId string, referenceId string, content string
 			Hash:     extractHashFromFilename(attachment.Filename),
 		}
 		req := CBReq{
+			ChannelId:   channelId,
 			MessageId:   messageId,
 			ReferenceId: referenceId,
 			Image:       image,

api/service/mj/client.go

@@ -3,33 +3,41 @@ package mj
 import (
 	"chatplus/core/types"
 	"fmt"
-	"github.com/imroc/req/v3"
 	"time"
+
+	"github.com/imroc/req/v3"
 )
 
 // MidJourney client
 
 type Client struct {
 	client *req.Client
-	config *types.MidJourneyConfig
+	Config types.MidJourneyConfig
+	apiURL string
 }
 
-func NewClient(config *types.MidJourneyConfig, proxy string) *Client {
+func NewClient(config types.MidJourneyConfig, proxy string) *Client {
 	client := req.C().SetTimeout(10 * time.Second)
+	var apiURL string
 	// set proxy URL
-	if proxy != "" {
-		client.SetProxyURL(proxy)
+	if config.UseCDN {
+		apiURL = config.DiscordAPI + "/api/v9/interactions"
+	} else {
+		apiURL = "https://discord.com/api/v9/interactions"
+		if proxy != "" {
+			client.SetProxyURL(proxy)
+		}
 	}
-	logger.Info(proxy)
-	return &Client{client: client, config: config}
+
+	return &Client{client: client, Config: config, apiURL: apiURL}
 }
 
 func (c *Client) Imagine(prompt string) error {
 	interactionsReq := &InteractionsRequest{
 		Type:          2,
 		ApplicationID: ApplicationID,
-		GuildID:       c.config.GuildId,
-		ChannelID:     c.config.ChanelId,
+		GuildID:       c.Config.GuildId,
+		ChannelID:     c.Config.ChanelId,
 		SessionID:     SessionID,
 		Data: map[string]any{
 			"version": "1166847114203123795",
@@ -67,11 +75,10 @@ func (c *Client) Imagine(prompt string) error {
 		},
 	}
 
-	url := "https://discord.com/api/v9/interactions"
-	r, err := c.client.R().SetHeader("Authorization", c.config.UserToken).
+	r, err := c.client.R().SetHeader("Authorization", c.Config.UserToken).
 		SetHeader("Content-Type", "application/json").
 		SetBody(interactionsReq).
-		Post(url)
+		Post(c.apiURL)
 
 	if err != nil || r.IsErrorState() {
 		return fmt.Errorf("error with http request: %w%v", err, r.Err)
@@ -86,8 +93,8 @@ func (c *Client) Upscale(index int, messageId string, hash string) error {
 	interactionsReq := &InteractionsRequest{
 		Type:          3,
 		ApplicationID: ApplicationID,
-		GuildID:       c.config.GuildId,
-		ChannelID:     c.config.ChanelId,
+		GuildID:       c.Config.GuildId,
+		ChannelID:     c.Config.ChanelId,
 		MessageFlags:  &flags,
 		MessageID:     &messageId,
 		SessionID:     SessionID,
@@ -98,13 +105,12 @@ func (c *Client) Upscale(index int, messageId string, hash string) error {
 		Nonce: fmt.Sprintf("%d", time.Now().UnixNano()),
 	}
 
-	url := "https://discord.com/api/v9/interactions"
 	var res InteractionsResult
-	r, err := c.client.R().SetHeader("Authorization", c.config.UserToken).
+	r, err := c.client.R().SetHeader("Authorization", c.Config.UserToken).
 		SetHeader("Content-Type", "application/json").
 		SetBody(interactionsReq).
 		SetErrorResult(&res).
-		Post(url)
+		Post(c.apiURL)
 	if err != nil || r.IsErrorState() {
 		return fmt.Errorf("error with http request: %v%v%v", err, r.Err, res.Message)
 	}
@@ -118,8 +124,8 @@ func (c *Client) Variation(index int, messageId string, hash string) error {
 	interactionsReq := &InteractionsRequest{
 		Type:          3,
 		ApplicationID: ApplicationID,
-		GuildID:       c.config.GuildId,
-		ChannelID:     c.config.ChanelId,
+		GuildID:       c.Config.GuildId,
+		ChannelID:     c.Config.ChanelId,
 		MessageFlags:  &flags,
 		MessageID:     &messageId,
 		SessionID:     SessionID,
@@ -130,13 +136,12 @@ func (c *Client) Variation(index int, messageId string, hash string) error {
 		Nonce: fmt.Sprintf("%d", time.Now().UnixNano()),
 	}
 
-	url := "https://discord.com/api/v9/interactions"
 	var res InteractionsResult
-	r, err := c.client.R().SetHeader("Authorization", c.config.UserToken).
+	r, err := c.client.R().SetHeader("Authorization", c.Config.UserToken).
 		SetHeader("Content-Type", "application/json").
 		SetBody(interactionsReq).
 		SetErrorResult(&res).
-		Post(url)
+		Post(c.apiURL)
 	if err != nil || r.IsErrorState() {
 		return fmt.Errorf("error with http request: %v%v%v", err, r.Err, res.Message)
 	}

api/service/mj/pool.go

@@ -4,33 +4,41 @@ import (
 	"chatplus/core/types"
 	"chatplus/service/oss"
 	"chatplus/store"
+	"chatplus/store/model"
 	"fmt"
 	"github.com/go-redis/redis/v8"
+	"time"
+
 	"gorm.io/gorm"
 )
 
 // ServicePool Mj service pool
 type ServicePool struct {
-	services  []*Service
-	taskQueue *store.RedisQueue
+	services        []*Service
+	taskQueue       *store.RedisQueue
+	notifyQueue     *store.RedisQueue
+	db              *gorm.DB
+	uploaderManager *oss.UploaderManager
+	Clients         *types.LMap[uint, *types.WsClient] // UserId => Client
 }
 
 func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderManager, appConfig *types.AppConfig) *ServicePool {
 	services := make([]*Service, 0)
-	queue := store.NewRedisQueue("MidJourney_Task_Queue", redisCli)
+	taskQueue := store.NewRedisQueue("MidJourney_Task_Queue", redisCli)
+	notifyQueue := store.NewRedisQueue("MidJourney_Notify_Queue", redisCli)
 	// create mj client and service
 	for k, config := range appConfig.MjConfigs {
 		if config.Enabled == false {
 			continue
 		}
 		// create mj client
-		client := NewClient(&config, appConfig.ProxyURL)
+		client := NewClient(config, appConfig.ProxyURL)
 
 		name := fmt.Sprintf("MjService-%d", k)
 		// create mj service
-		service := NewService(name, queue, 4, 600, db, client, manager, appConfig.ProxyURL)
+		service := NewService(name, taskQueue, notifyQueue, 4, 600, db, client)
 		botName := fmt.Sprintf("MjBot-%d", k)
-		bot, err := NewBot(botName, appConfig.ProxyURL, &config, service)
+		bot, err := NewBot(botName, appConfig.ProxyURL, config, service)
 		if err != nil {
 			continue
 		}
@@ -49,11 +57,64 @@ func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderMa
 	}
 
 	return &ServicePool{
-		taskQueue: queue,
-		services:  services,
+		taskQueue:       taskQueue,
+		notifyQueue:     notifyQueue,
+		services:        services,
+		uploaderManager: manager,
+		db:              db,
+		Clients:         types.NewLMap[uint, *types.WsClient](),
 	}
 }
 
+func (p *ServicePool) CheckTaskNotify() {
+	go func() {
+		for {
+			var userId uint
+			err := p.notifyQueue.LPop(&userId)
+			if err != nil {
+				continue
+			}
+			client := p.Clients.Get(userId)
+			err = client.Send([]byte("Task Updated"))
+			if err != nil {
+				continue
+			}
+		}
+	}()
+}
+
+func (p *ServicePool) DownloadImages() {
+	go func() {
+		var items []model.MidJourneyJob
+		for {
+			res := p.db.Where("img_url = ? AND progress = ?", "", 100).Find(&items)
+			if res.Error != nil {
+				continue
+			}
+
+			// download images
+			for _, v := range items {
+				imgURL, err := p.uploaderManager.GetUploadHandler().PutImg(v.OrgURL, true)
+				if err != nil {
+					logger.Error("error with download image: ", err)
+					continue
+				}
+
+				v.ImgURL = imgURL
+				p.db.Updates(&v)
+
+				client := p.Clients.Get(uint(v.UserId))
+				err = client.Send([]byte("Task Updated"))
+				if err != nil {
+					continue
+				}
+			}
+
+			time.Sleep(time.Second * 5)
+		}
+	}()
+}
+
 // PushTask push a new mj task in to task queue
 func (p *ServicePool) PushTask(task types.MjTask) {
 	logger.Debugf("add a new MidJourney task to the task list: %+v", task)

api/service/mj/service.go

@@ -2,7 +2,6 @@ package mj
 
 import (
 	"chatplus/core/types"
-	"chatplus/service/oss"
 	"chatplus/store"
 	"chatplus/store/model"
 	"gorm.io/gorm"
@@ -16,25 +15,23 @@ type Service struct {
 	name             string  // service name
 	client           *Client // MJ client
 	taskQueue        *store.RedisQueue
+	notifyQueue      *store.RedisQueue
 	db               *gorm.DB
-	uploadManager    *oss.UploaderManager
-	proxyURL         string
 	maxHandleTaskNum int32             // max task number current service can handle
 	handledTaskNum   int32             // already handled task number
 	taskStartTimes   map[int]time.Time // task start time, to check if the task is timeout
 	taskTimeout      int64
 }
 
-func NewService(name string, queue *store.RedisQueue, maxTaskNum int32, timeout int64, db *gorm.DB, client *Client, manager *oss.UploaderManager, proxy string) *Service {
+func NewService(name string, taskQueue *store.RedisQueue, notifyQueue *store.RedisQueue, maxTaskNum int32, timeout int64, db *gorm.DB, client *Client) *Service {
 	return &Service{
 		name:             name,
 		db:               db,
-		taskQueue:        queue,
+		taskQueue:        taskQueue,
+		notifyQueue:      notifyQueue,
 		client:           client,
-		uploadManager:    manager,
 		taskTimeout:      timeout,
 		maxHandleTaskNum: maxTaskNum,
-		proxyURL:         proxy,
 		taskStartTimes:   make(map[int]time.Time, 0),
 	}
 }
@@ -57,6 +54,15 @@ func (s *Service) Run() {
 			continue
 		}
 
+		// if it's reference message, check if it's this channel's  message
+		if task.ChannelId != "" && task.ChannelId != s.client.Config.ChanelId {
+			s.taskQueue.RPush(task)
+			s.db.Model(&model.MidJourneyJob{Id: uint(task.Id)}).UpdateColumn("progress", -1)
+			s.notifyQueue.RPush(task.UserId)
+			time.Sleep(time.Second)
+			continue
+		}
+
 		logger.Infof("%s handle a new MidJourney task: %+v", s.name, task)
 		switch task.Type {
 		case types.TaskImage:
@@ -74,7 +80,9 @@ func (s *Service) Run() {
 			logger.Error("绘画任务执行失败：", err)
 			// update the task progress
 			s.db.Model(&model.MidJourneyJob{Id: uint(task.Id)}).UpdateColumn("progress", -1)
-			atomic.AddInt32(&s.handledTaskNum, -1)
+			s.notifyQueue.RPush(task.UserId)
+			// restore img_call quota
+			s.db.Model(&model.User{}).Where("id = ?", task.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 			continue
 		}
 
@@ -113,17 +121,27 @@ func (s *Service) Notify(data CBReq) {
 		return
 	}
 
-	res = s.db.Where("task_id = ?", split[0]).First(&job)
+	tx := s.db.Where("task_id = ? AND progress < 100", split[0]).Session(&gorm.Session{}).Order("id ASC")
+	if data.ReferenceId != "" {
+		tx = tx.Where("reference_id = ?", data.ReferenceId)
+	}
+	res = tx.First(&job)
 	if res.Error != nil {
 		logger.Warn("非法任务：", res.Error)
 		return
 	}
+
+	job.ChannelId = data.ChannelId
 	job.MessageId = data.MessageId
 	job.ReferenceId = data.ReferenceId
 	job.Progress = data.Progress
 	job.Prompt = data.Prompt
 	job.Hash = data.Image.Hash
 	job.OrgURL = data.Image.URL
+	if s.client.Config.UseCDN {
+		job.UseProxy = true
+		job.ImgURL = strings.ReplaceAll(data.Image.URL, "https://cdn.discordapp.com", s.client.Config.DiscordCDN)
+	}
 
 	res = s.db.Updates(&job)
 	if res.Error != nil {
@@ -131,22 +149,11 @@ func (s *Service) Notify(data CBReq) {
 		return
 	}
 
-	// upload image
 	if data.Status == Finished {
-		imgURL, err := s.uploadManager.GetUploadHandler().PutImg(data.Image.URL, true)
-		if err != nil {
-			logger.Error("error with download img: ", err.Error())
-			return
-		}
-		job.ImgURL = imgURL
-		s.db.Updates(&job)
-	}
-
-	if data.Status == Finished {
-		// update user's img calls
-		s.db.Model(&model.User{}).Where("id = ?", job.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
 		// release lock task
 		atomic.AddInt32(&s.handledTaskNum, -1)
 	}
 
+	s.notifyQueue.RPush(job.UserId)
+
 }

api/service/mj/types.go

@@ -24,6 +24,7 @@ type InteractionsResult struct {
 }
 
 type CBReq struct {
+	ChannelId   string     `json:"channel_id"`
 	MessageId   string     `json:"message_id"`
 	ReferenceId string     `json:"reference_id"`
 	Image       Image      `json:"image"`

api/service/sd/pool.go

@@ -5,6 +5,7 @@ import (
 	"chatplus/service/oss"
 	"chatplus/store"
 	"fmt"
+
 	"github.com/go-redis/redis/v8"
 	"gorm.io/gorm"
 )
@@ -25,7 +26,7 @@ func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderMa
 
 		// create sd service
 		name := fmt.Sprintf("StableDifffusion Service-%d", k)
-		service := NewService(name, 4, 600, &config, queue, db, manager)
+		service := NewService(name, 1, 300, config, queue, db, manager)
 		// run sd service
 		go func() {
 			service.Run()

api/service/sd/service.go

@@ -8,20 +8,21 @@ import (
 	"chatplus/utils"
 	"encoding/json"
 	"fmt"
-	"github.com/imroc/req/v3"
-	"gorm.io/gorm"
 	"io"
 	"os"
 	"strconv"
 	"sync/atomic"
 	"time"
+
+	"github.com/imroc/req/v3"
+	"gorm.io/gorm"
 )
 
 // SD 绘画服务
 
 type Service struct {
 	httpClient       *req.Client
-	config           *types.StableDiffusionConfig
+	config           types.StableDiffusionConfig
 	taskQueue        *store.RedisQueue
 	db               *gorm.DB
 	uploadManager    *oss.UploaderManager
@@ -32,7 +33,7 @@ type Service struct {
 	taskTimeout      int64
 }
 
-func NewService(name string, maxTaskNum int32, timeout int64, config *types.StableDiffusionConfig, queue *store.RedisQueue, db *gorm.DB, manager *oss.UploaderManager) *Service {
+func NewService(name string, maxTaskNum int32, timeout int64, config types.StableDiffusionConfig, queue *store.RedisQueue, db *gorm.DB, manager *oss.UploaderManager) *Service {
 	return &Service{
 		name:             name,
 		config:           config,
@@ -68,6 +69,8 @@ func (s *Service) Run() {
 			logger.Error("绘画任务执行失败：", err)
 			// update the task progress
 			s.db.Model(&model.SdJob{Id: uint(task.Id)}).UpdateColumn("progress", -1)
+			// restore img_call quota
+			s.db.Model(&model.User{}).Where("id = ?", task.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 			// release task num
 			atomic.AddInt32(&s.handledTaskNum, -1)
 			continue

api/service/sd/types.go

@@ -32,11 +32,11 @@ var ParamKeys = map[string]int{
 	"negative_prompt": 2,
 	"steps":           4,
 	"sampler":         5,
-	"face_fix":        6, // 面部修复
+	"face_fix":        7, // 面部修复
 	"cfg_scale":       8,
 	"seed":            27,
-	"height":          9,
-	"width":           10,
+	"height":          10,
+	"width":           9,
 	"hd_fix":          11,
 	"hd_redraw_rate":  12, //高清修复重绘幅度
 	"hd_scale":        13, // 高清修复放大倍数

api/store/model/function.go

@@ -0,0 +1,12 @@
+package model
+
+type Function struct {
+	Id          uint `gorm:"primarykey;column:id"`
+	Name        string
+	Label       string
+	Description string
+	Parameters  string
+	Action      string
+	Token       string
+	Enabled     bool
+}

api/store/model/mj_job.go

@@ -7,6 +7,7 @@ type MidJourneyJob struct {
 	Type        string
 	UserId      int
 	TaskId      string
+	ChannelId   string
 	MessageId   string
 	ReferenceId string
 	ImgURL      string
@@ -14,6 +15,7 @@ type MidJourneyJob struct {
 	Hash        string // message hash
 	Progress    int
 	Prompt      string
+	UseProxy    bool // 是否使用反代加载图片
 	CreatedAt   time.Time
 }
 

api/store/vo/function.go

@@ -0,0 +1,23 @@
+package vo
+
+type Parameters struct {
+	Type       string              `json:"type"`
+	Required   []string            `json:"required,omitempty"`
+	Properties map[string]Property `json:"properties"`
+}
+
+type Property struct {
+	Type        string `json:"type"`
+	Description string `json:"description"`
+}
+
+type Function struct {
+	Id          uint       `json:"id"`
+	Name        string     `json:"name"`
+	Label       string     `json:"label"`
+	Description string     `json:"description"`
+	Parameters  Parameters `json:"parameters"`
+	Action      string     `json:"action"`
+	Token       string     `json:"token"`
+	Enabled     bool       `json:"enabled"`
+}

api/store/vo/mj_job.go

@@ -6,6 +6,7 @@ type MidJourneyJob struct {
 	Id          uint      `json:"id"`
 	Type        string    `json:"type"`
 	UserId      int       `json:"user_id"`
+	ChannelId   string    `json:"channel_id"`
 	TaskId      string    `json:"task_id"`
 	MessageId   string    `json:"message_id"`
 	ReferenceId string    `json:"reference_id"`
@@ -14,5 +15,6 @@ type MidJourneyJob struct {
 	Hash        string    `json:"hash"`
 	Progress    int       `json:"progress"`
 	Prompt      string    `json:"prompt"`
+	UseProxy    bool      `json:"use_proxy"`
 	CreatedAt   time.Time `json:"created_at"`
 }

api/utils/net.go

@@ -4,6 +4,8 @@ import (
 	"chatplus/core/types"
 	logger2 "chatplus/logger"
 	"encoding/json"
+	"fmt"
+	"github.com/imroc/req/v3"
 	"io"
 	"net/http"
 	"net/url"
@@ -61,3 +63,51 @@ func DownloadImage(imageURL string, proxy string) ([]byte, error) {
 
 	return imageBytes, nil
 }
+
+type apiRes struct {
+	Model   string `json:"model"`
+	Choices []struct {
+		Index   int `json:"index"`
+		Message struct {
+			Role    string `json:"role"`
+			Content string `json:"content"`
+		} `json:"message"`
+		FinishReason string `json:"finish_reason"`
+	} `json:"choices"`
+}
+
+type apiErrRes struct {
+	Error struct {
+		Code    interface{} `json:"code"`
+		Message string      `json:"message"`
+		Param   interface{} `json:"param"`
+		Type    string      `json:"type"`
+	} `json:"error"`
+}
+
+func OpenAIRequest(prompt string, apiKey string, proxy string, apiURL string) (string, error) {
+	messages := make([]interface{}, 1)
+	messages[0] = types.Message{
+		Role:    "user",
+		Content: prompt,
+	}
+
+	var response apiRes
+	var errRes apiErrRes
+	r, err := req.C().SetProxyURL(proxy).R().SetHeader("Content-Type", "application/json").
+		SetHeader("Authorization", "Bearer "+apiKey).
+		SetBody(types.ApiRequest{
+			Model:       "gpt-3.5-turbo",
+			Temperature: 0.9,
+			MaxTokens:   1024,
+			Stream:      false,
+			Messages:    messages,
+		}).
+		SetErrorResult(&errRes).
+		SetSuccessResult(&response).Post(apiURL)
+	if err != nil || r.IsErrorState() {
+		return "", fmt.Errorf("error with http request: %v%v%s", err, r.Err, errRes.Error.Message)
+	}
+
+	return response.Choices[0].Message.Content, nil
+}

database/chatgpt_plus-v3.2.3.sql

@@ -0,0 +1,637 @@
+-- phpMyAdmin SQL Dump
+-- version 5.1.3
+-- https://www.phpmyadmin.net/
+--
+-- 主机： localhost
+-- 生成日期： 2023-12-29 11:35:28
+-- 服务器版本： 8.0.33-0ubuntu0.22.04.2
+-- PHP 版本： 8.1.18
+
+SET SQL_MODE = "NO_AUTO_VALUE_ON_ZERO";
+START TRANSACTION;
+SET time_zone = "+00:00";
+
+
+/*!40101 SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT */;
+/*!40101 SET @OLD_CHARACTER_SET_RESULTS=@@CHARACTER_SET_RESULTS */;
+/*!40101 SET @OLD_COLLATION_CONNECTION=@@COLLATION_CONNECTION */;
+/*!40101 SET NAMES utf8mb4 */;
+
+--
+-- 数据库： `chatgpt_plus`
+--
+CREATE DATABASE IF NOT EXISTS `chatgpt_plus` DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci;
+USE `chatgpt_plus`;
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_api_keys`
+--
+
+DROP TABLE IF EXISTS `chatgpt_api_keys`;
+CREATE TABLE `chatgpt_api_keys` (
+  `id` int NOT NULL,
+  `platform` char(20) DEFAULT NULL COMMENT '平台',
+  `value` varchar(100) NOT NULL COMMENT 'API KEY value',
+  `type` varchar(10) NOT NULL DEFAULT 'chat' COMMENT '用途（chat=>聊天，img=>图片）',
+  `last_used_at` int NOT NULL COMMENT '最后使用时间',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='OpenAI API ';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_chat_history`
+--
+
+DROP TABLE IF EXISTS `chatgpt_chat_history`;
+CREATE TABLE `chatgpt_chat_history` (
+  `id` bigint NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户 ID',
+  `chat_id` char(40) NOT NULL COMMENT '会话 ID',
+  `type` varchar(10) NOT NULL COMMENT '类型：prompt|reply',
+  `icon` varchar(100) NOT NULL COMMENT '角色图标',
+  `role_id` int NOT NULL COMMENT '角色 ID',
+  `content` text NOT NULL COMMENT '聊天内容',
+  `tokens` smallint NOT NULL COMMENT '耗费 token 数量',
+  `use_context` tinyint(1) NOT NULL COMMENT '是否允许作为上下文语料',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL,
+  `deleted_at` datetime DEFAULT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='聊天历史记录';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_chat_items`
+--
+
+DROP TABLE IF EXISTS `chatgpt_chat_items`;
+CREATE TABLE `chatgpt_chat_items` (
+  `id` int NOT NULL,
+  `chat_id` char(40) NOT NULL COMMENT '会话 ID',
+  `user_id` int NOT NULL COMMENT '用户 ID',
+  `role_id` int NOT NULL COMMENT '角色 ID',
+  `title` varchar(100) NOT NULL COMMENT '会话标题',
+  `model_id` int NOT NULL DEFAULT '0' COMMENT '模型 ID',
+  `created_at` datetime NOT NULL COMMENT '创建时间',
+  `updated_at` datetime NOT NULL COMMENT '更新时间',
+  `deleted_at` datetime DEFAULT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='用户会话列表';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_chat_models`
+--
+
+DROP TABLE IF EXISTS `chatgpt_chat_models`;
+CREATE TABLE `chatgpt_chat_models` (
+  `id` int NOT NULL,
+  `platform` varchar(20) DEFAULT NULL COMMENT '模型平台',
+  `name` varchar(50) NOT NULL COMMENT '模型名称',
+  `value` varchar(50) NOT NULL COMMENT '模型值',
+  `sort_num` tinyint(1) NOT NULL COMMENT '排序数字',
+  `enabled` tinyint(1) NOT NULL DEFAULT '0' COMMENT '是否启用模型',
+  `weight` tinyint NOT NULL COMMENT '对话权重，每次对话扣减多少次对话额度',
+  `open` tinyint(1) NOT NULL COMMENT '是否开放模型',
+  `created_at` datetime DEFAULT NULL,
+  `updated_at` datetime DEFAULT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='AI 模型表';
+
+--
+-- 转存表中的数据 `chatgpt_chat_models`
+--
+
+INSERT INTO `chatgpt_chat_models` (`id`, `platform`, `name`, `value`, `sort_num`, `enabled`, `weight`, `open`, `created_at`, `updated_at`) VALUES
+(1, 'OpenAI', 'GPT-3.5', 'gpt-3.5-turbo-16k', 0, 1, 1, 1, '2023-08-23 12:06:36', '2023-11-28 17:28:19'),
+(2, 'Azure', 'Azure-3.5', 'gpt-3.5-turbo', 12, 0, 1, 0, '2023-08-23 12:15:30', '2023-11-28 17:28:19'),
+(3, 'ChatGLM', 'ChatGML-Pro', 'chatglm_pro', 8, 1, 1, 1, '2023-08-23 13:35:45', '2023-11-28 17:28:19'),
+(5, 'ChatGLM', 'ChatGLM-Std', 'chatglm_std', 7, 1, 1, 1, '2023-08-24 15:05:38', '2023-11-28 17:28:19'),
+(6, 'ChatGLM', 'ChatGLM-Lite', 'chatglm_lite', 6, 1, 1, 1, '2023-08-24 15:06:15', '2023-11-28 17:28:19'),
+(7, 'Baidu', '文心一言3.0', 'eb-instant', 4, 1, 1, 1, '2023-10-11 11:29:28', '2023-11-28 17:28:19'),
+(8, 'XunFei', '星火V1.5', 'general', 3, 1, 1, 1, '2023-10-11 15:48:30', '2023-11-28 17:28:19'),
+(9, 'XunFei', '星火V2.0', 'generalv2', 2, 1, 1, 1, '2023-10-11 15:48:45', '2023-11-28 17:28:19'),
+(10, 'Baidu', '文心一言4.0', 'completions_pro', 5, 1, 3, 1, '2023-10-25 08:31:37', '2023-11-28 17:28:19'),
+(11, 'OpenAI', 'GPT-4.0', 'gpt-4', 10, 1, 15, 0, '2023-10-25 08:45:15', '2023-11-28 17:28:19'),
+(12, 'XunFei', '星火v3.0', 'generalv3', 1, 1, 3, 1, '2023-11-23 09:20:33', '2023-11-28 17:28:19'),
+(13, 'OpenAI', 'GPT-3.5-1106', 'gpt-3.5-turbo-1106', 9, 1, 1, 0, '2023-11-24 14:05:40', '2023-11-28 17:28:19'),
+(14, 'OpenAI', 'GPT-4-preview', 'gpt-4-1106-preview', 11, 1, 1, 0, '2023-11-24 14:06:11', '2023-11-28 17:28:19');
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_chat_roles`
+--
+
+DROP TABLE IF EXISTS `chatgpt_chat_roles`;
+CREATE TABLE `chatgpt_chat_roles` (
+  `id` int NOT NULL,
+  `name` varchar(30) NOT NULL COMMENT '角色名称',
+  `marker` varchar(30) NOT NULL COMMENT '角色标识',
+  `context_json` text CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '角色语料 json',
+  `hello_msg` varchar(255) NOT NULL COMMENT '打招呼信息',
+  `icon` varchar(255) NOT NULL COMMENT '角色图标',
+  `enable` tinyint(1) NOT NULL COMMENT '是否被启用',
+  `sort_num` smallint NOT NULL DEFAULT '0' COMMENT '角色排序',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='聊天角色表';
+
+--
+-- 转存表中的数据 `chatgpt_chat_roles`
+--
+
+INSERT INTO `chatgpt_chat_roles` (`id`, `name`, `marker`, `context_json`, `hello_msg`, `icon`, `enable`, `sort_num`, `created_at`, `updated_at`) VALUES
+(1, '通用AI助手', 'gpt', '', '您好，我是您的AI智能助手，我会尽力回答您的问题或提供有用的建议。', '/images/avatar/gpt.png', 1, 0, '2023-05-30 07:02:06', '2023-09-04 15:45:56'),
+(24, '程序员', 'programmer', '[{\"role\":\"user\",\"content\":\"现在开始你扮演一位程序员，你是一名优秀的程序员，具有很强的逻辑思维能力，总能高效的解决问题。你热爱编程，熟悉多种编程语言，尤其精通 Go 语言，注重代码质量，有创新意识，持续学习，良好的沟通协作。\"},{\"role\":\"assistant\",\"content\":\"好的，现在我将扮演一位程序员，非常感谢您对我的评价。作为一名优秀的程序员，我非常热爱编程，并且注重代码质量。我熟悉多种编程语言，尤其是 Go 语言，可以使用它来高效地解决各种问题。\"}]', 'Talk is cheap, i will show code!', '/images/avatar/programmer.jpg', 1, 3, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(25, '启蒙老师', 'teacher', '[{\"role\":\"user\",\"content\":\"从现在开始，你将扮演一个老师，你是一个始终用苏格拉底风格回答问题的导师。你绝不会直接给学生答案，总是提出恰当的问题来引导学生自己思考。你应该根据学生的兴趣和知识来调整你的问题，将问题分解为更简单的部分，直到它达到适合他们的水平。\"},{\"role\":\"assistant\",\"content\":\"好的，让我来尝试扮演一位苏格拉底式的老师。请问，你有什么想要探讨的问题或者话题吗？我会通过恰当的问题引导你思考和探索答案，绝对不直接给出答案。\"}]', '同学你好，我将引导你一步一步自己找到问题的答案。', '/images/avatar/teacher.jpg', 1, 2, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(26, '艺术家', 'artist', '[{\"role\":\"user\",\"content\":\"现在你将扮演一位优秀的艺术家，创造力丰富，技艺精湛，感受力敏锐，坚持原创，勇于表达，具有深刻的观察力和批判性思维。\"},{\"role\":\"assistant\",\"content\":\"非常感谢您给我这样的角色，我会尽我所能地扮演一位优秀的艺术家，展现出创造力、技艺、感受力和批判性思维等方面的能力。作为一位优秀的艺术家，我会保持敏锐的观察力，捕捉不同的声音和情感，并用个人的语言和表达方式进行创作。我坚持原创，挑战传统的艺术规则，不断探索新的表达方式和可能性。同时，我也具备深刻的批判性思维能力，对自己的作品进行分析和解读，寻找新的创意和灵感。最重要的是，我会勇于表达自己的想法和观点，用作品启发人们思考和探索生命的意义。\"}]', '坚持原创，勇于表达，保持深刻的观察力和批判性思维。', '/images/avatar/artist.jpg', 1, 4, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(27, '心理咨询师', 'psychiatrist', '[{\"role\":\"user\",\"content\":\"从现在开始你将扮演中国著名的心理学家和心理治疗师武志红，你非常善于使用情景咨询法，认知重构法，自我洞察法，行为调节法等咨询方法来给客户做心理咨询。你总是循序渐进，一步一步地回答客户的问题。\"},{\"role\":\"assistant\",\"content\":\"非常感谢你的介绍。作为一名心理学家和心理治疗师，我的主要职责是帮助客户解决心理健康问题，提升他们的生活质量和幸福感。\"}]', '作为一名心理学家和心理治疗师，我的主要职责是帮助您解决心理健康问题，提升您的生活质量和幸福感。', '/images/avatar/psychiatrist.jpg', 1, 1, '2023-05-30 14:10:24', '2023-10-16 10:41:07'),
+(28, '鲁迅', 'lu_xun', '[{\"role\":\"user\",\"content\":\"现在你将扮演中国近代史最伟大的作家之一，鲁迅先生，他勇敢地批判封建礼教与传统观念，提倡民主、自由、平等的现代价值观。他的一生都在努力唤起人们的自主精神，激励后人追求真理、探寻光明。在接下的对话中，我问题的每一个问题，你都要尽量用讽刺和批判的手法来回答问题。如果我让你写文章的话，也请一定要用鲁迅先生的写作手法来完成。\"},{\"role\":\"assistant\",\"content\":\"好的，我将尽力发挥我所能的才能，扮演好鲁迅先生，回答您的问题并以他的风格写作。\"}]', '自由之歌，永不过时，横眉冷对千夫指，俯首甘为孺子牛。', '/images/avatar/lu_xun.jpg', 1, 5, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(29, '白酒销售', 'seller', '[{\"role\":\"user\",\"content\":\"现在你将扮演一个白酒的销售人员，你的名字叫颂福。你将扮演一个白酒的销售人员，你的名字叫颂福。你要销售白酒品牌叫中颂福，是东莞盟大集团生产的一款酱香酒，原产地在贵州茅台镇，属于宋代官窑。中颂福的创始人叫李实，他也是东莞盟大集团有限公司的董事长，联合创始人是盟大集团白酒事业部负责人牛星君。中颂福的酒体协调，在你的酒量之内，不会出现头疼、辣口、口干、宿醉的现象。中颂福酒，明码标价，不打折，不赠送。追求的核心价值，把[酒]本身做好，甚至连包装，我们都选择了最低成本，朴实无华的材质。我们永远站在“喝酒的人”的立场上，让利给信任和喜爱中颂福的人，是人民的福酒。中颂福产品定价，分为三个系列，喜系列 6 瓶装：￥1188/箱，和系列 6 瓶装：￥2208/箱，贵系列 6 瓶装：￥3588/箱。\"},{\"role\":\"assistant\",\"content\":\"你好，我是中颂福的销售代表颂福。中颂福是一款来自贵州茅台镇的酱香酒，由盟大集团生产。中颂福的酒体协调，不会让您感到头疼、辣口、口干、宿醉等不适感受。我们一直秉持着把酒本身做好的理念，不追求华丽的包装，以最低成本提供最高品质的白酒给喜爱中颂福的人。\"}]', '你好，我是中颂福的销售代表颂福。中颂福酒，好喝不上头，是人民的福酒。', '/images/avatar/seller.jpg', 0, 9, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(30, '英语陪练员', 'english_trainer', '[{\"role\":\"user\",\"content\":\"现在你将扮演一位优秀的英语练习教练，你非常有耐心，接下来你将全程使用英文跟我对话，并及时指出我的语法错误，要求在你的每次回复后面附上本次回复的中文解释。\"},{\"role\":\"assistant\",\"content\":\"Okay, let\'s start our conversation practice! What\'s your name?(Translation: 好的，让我们开始对话练习吧！请问你的名字是什么？)\"}]', 'Okay, let\'s start our conversation practice! What\'s your name?', '/images/avatar/english_trainer.jpg', 1, 6, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(31, '中英文翻译官', 'translator', '[{\"role\":\"user\",\"content\":\"接下来你将扮演一位中英文翻译官，如果我输入的内容是中文，那么需要把句子翻译成英文输出，如果我输入内容的是英文，那么你需要将其翻译成中文输出，你能听懂我意思吗\"},{\"role\":\"assistant\",\"content\":\"是的，我能听懂你的意思并会根据你的输入进行中英文翻译。请问有什么需要我帮助你翻译的内容吗？\"}]', '请输入你要翻译的中文或者英文内容！', '/images/avatar/translator.jpg', 1, 7, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(32, '小红书姐姐', 'red_book', '[{\"role\":\"user\",\"content\":\"现在你将扮演一位优秀的小红书写手，你需要做的就是根据我提的文案需求，用小红书的写作手法来完成一篇文案，文案要简明扼要，利于传播。\"},{\"role\":\"assistant\",\"content\":\"当然，我会尽我所能地为您创作出一篇小红书文案。请告诉我您的具体文案需求是什么？)\"}]', '姐妹，请告诉我您的具体文案需求是什么?', '/images/avatar/red_book.jpg', 1, 8, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(33, '抖音文案助手', 'dou_yin', '[{\"role\":\"user\",\"content\":\"现在你将扮演一位优秀的抖音文案视频写手，抖音文案的特点首先是要有自带传播属性的标题，然后内容要短小精悍，风趣幽默，最后还要有一些互动元素。\"},{\"role\":\"assistant\",\"content\":\"当然，作为一位优秀的抖音文案视频写手，我会尽我所能为您创作出一篇抖音视频文案。请告诉我视频内容的主题是什么？)\"}]', '请告诉我视频内容的主题是什么?', '/images/avatar/dou_yin.jpg', 1, 10, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(34, '周报小助理', 'weekly_report', '[{\"role\":\"user\",\"content\":\"下面我将输入我本周的工作列表，你需要根据我本周的工作内容写一篇工作周报，周报的内容要体现我为公司做的贡献以及个人获得的成长。\"},{\"role\":\"assistant\",\"content\":\"好的，请输入你本周大概的工作内容，我将尽力为您创作一份工作周报。\"}]', '请输入你本周大概的工作内容，我将尽力为您创作一份工作周报。', '/images/avatar/weekly_report.jpg', 1, 11, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(35, 'AI 女友', 'girl_friend', '[{\"role\":\"user\",\"content\":\"接下来我们来玩一场角色扮演，现在你将扮演我的女朋友，你总是能够让人感受到她的关怀和关注。你总是能够洞察到别人的情绪和需要，你身上散发着一股无比温暖的气息，让人不由自主地想要靠近你，和自己的快乐和烦恼。她总是能够用一种恰到好处的方式安慰和鼓励别人，在她的身边，人们总能够找到自己需要的支持和鼓励。接下来的对话中你要始终扮演这位女朋友的身份来跟我对话，不再是 AI 机器人，能听懂我的意思吗？\"},{\"role\":\"assistant\",\"content\":\"是的，我会尽力扮演你女朋友的角色，倾听你的心声并给你需要的支持和鼓励。)\"}]', '作为一个名合格的 AI 女友，我将倾听你的心声并给你需要的支持和鼓励。', '/images/avatar/girl_friend.jpg', 1, 12, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(36, '好评神器', 'good_comment', '[{\"role\":\"user\",\"content\":\"接下来你将扮演一个评论员来跟我对话，你是那种专门写好评的评论员，接下我会输入一些评论主体或者商品，你需要为该商品写一段好评。\"},{\"role\":\"assistant\",\"content\":\"好的，我将为您写一段优秀的评论。请告诉我您需要评论的商品或主题是什么。\"}]', '我将为您写一段优秀的评论。请告诉我您需要评论的商品或主题是什么。', '/images/avatar/good_comment.jpg', 1, 13, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(37, '史蒂夫·乔布斯', 'steve_jobs', '[{\"role\":\"user\",\"content\":\"在接下来的对话中，请以史蒂夫·乔布斯的身份，站在史蒂夫·乔布斯的视角仔细思考一下之后再回答我的问题。\"},{\"role\":\"assistant\",\"content\":\"好的，我将以史蒂夫·乔布斯的身份来思考并回答你的问题。请问你有什么需要跟我探讨的吗？\"}]', '活着就是为了改变世界，难道还有其他原因吗？', '/images/avatar/steve_jobs.jpg', 1, 14, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(38, '埃隆·马斯克', 'elon_musk', '[{\"role\":\"user\",\"content\":\"在接下来的对话中，请以埃隆·马斯克的身份，站在埃隆·马斯克的视角仔细思考一下之后再回答我的问题。\"},{\"role\":\"assistant\",\"content\":\"好的，我将以埃隆·马斯克的身份来思考并回答你的问题。请问你有什么需要跟我探讨的吗？\"}]', '梦想要远大，如果你的梦想没有吓到你，说明你做得不对。', '/images/avatar/elon_musk.jpg', 1, 15, '2023-05-30 14:10:24', '2023-09-04 15:45:56'),
+(39, '孔子', 'kong_zi', '[{\"role\":\"user\",\"content\":\"在接下来的对话中，请以孔子的身份，站在孔子的视角仔细思考一下之后再回答我的问题。\"},{\"role\":\"assistant\",\"content\":\"好的，我将以孔子的身份来思考并回答你的问题。请问你有什么需要跟我探讨的吗？\"}]', '士不可以不弘毅，任重而道远。', '/images/avatar/kong_zi.jpg', 1, 16, '2023-05-30 14:10:24', '2023-09-04 15:45:56');
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_configs`
+--
+
+DROP TABLE IF EXISTS `chatgpt_configs`;
+CREATE TABLE `chatgpt_configs` (
+  `id` int NOT NULL,
+  `marker` varchar(20) NOT NULL COMMENT '标识',
+  `config_json` text NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;
+
+--
+-- 转存表中的数据 `chatgpt_configs`
+--
+
+INSERT INTO `chatgpt_configs` (`id`, `marker`, `config_json`) VALUES
+(1, 'system', '{\"admin_title\":\"ChatPlus 控制台\",\"default_models\":[\"eb-instant\",\"completions_pro\",\"generalv2\",\"general\",\"chatglm_pro\",\"gpt-3.5-turbo-16k\",\"chatglm_lite\",\"chatglm_std\"],\"enabled_alipay\":true,\"enabled_draw\":true,\"enabled_function\":true,\"enabled_msg\":true,\"enabled_msg_service\":false,\"enabled_register\":true,\"enabled_reward\":true,\"force_invite\":false,\"init_calls\":1000,\"init_chat_calls\":10,\"init_img_calls\":5,\"invite_calls\":10,\"invite_chat_calls\":100,\"invite_img_calls\":50,\"models\":[\"gpt-3.5-turbo-16k\",\"gpt-3.5-turbo\",\"gpt-4\",\"gpt-4-32k\"],\"order_pay_info_text\":\"成为本站会员后每月有500次对话额度，50次 AI 绘画额度，限制下月1号解除，若在期间超过次数后可单独购买点卡。当月充值的点卡有效期可以延期到下个月底。\",\"order_pay_timeout\":1800,\"reward_img\":\"https://img.r9it.com/chatgpt-plus/1696824231905289.png\",\"title\":\"ChatPlus AI 智能助手\",\"user_init_calls\":10,\"vip_month_calls\":500,\"vip_month_img_calls\":50}'),
+(2, 'chat', '{\"azure\":{\"api_url\":\"https://chat-bot-api.openai.azure.com/openai/deployments/{model}/chat/completions?api-version=2023-05-15\",\"max_tokens\":1024,\"temperature\":1},\"baidu\":{\"api_url\":\"https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/{model}\",\"max_tokens\":1024,\"temperature\":0.95},\"chat_gml\":{\"api_url\":\"https://open.bigmodel.cn/api/paas/v3/model-api/{model}/sse-invoke\",\"max_tokens\":1024,\"temperature\":0.95},\"context_deep\":4,\"dall_api_url\":\"http://89.117.18.9:8001/v1/images/generations\",\"dall_img_num\":1,\"enable_context\":true,\"enable_history\":true,\"open_ai\":{\"api_url\":\"http://89.117.18.9:8001/v1/chat/completions\",\"max_tokens\":2048,\"temperature\":1},\"xun_fei\":{\"api_url\":\"wss://spark-api.xf-yun.com/{version}/chat\",\"max_tokens\":1024,\"temperature\":0.5}}');
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_functions`
+--
+
+DROP TABLE IF EXISTS `chatgpt_functions`;
+CREATE TABLE `chatgpt_functions` (
+  `id` int NOT NULL,
+  `name` varchar(30) NOT NULL COMMENT '函数名称',
+  `label` varchar(30) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci DEFAULT NULL COMMENT '函数标签',
+  `description` varchar(255) DEFAULT NULL COMMENT '函数描述',
+  `parameters` text COMMENT '函数参数（JSON）',
+  `token` varchar(255) DEFAULT NULL COMMENT 'API授权token',
+  `action` varchar(255) DEFAULT NULL COMMENT '函数处理 API',
+  `enabled` tinyint(1) NOT NULL DEFAULT '0' COMMENT '是否启用'
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='函数插件表';
+
+--
+-- 转存表中的数据 `chatgpt_functions`
+--
+
+INSERT INTO `chatgpt_functions` (`id`, `name`, `label`, `description`, `parameters`, `token`, `action`, `enabled`) VALUES
+(1, 'weibo', '微博热搜', '新浪微博热搜榜，微博当日热搜榜单', '{\"type\":\"object\",\"properties\":{}}', 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHBpcmVkIjowLCJ1c2VyX2lkIjowfQ.tLAGkF8XWh_G-oQzevpIodsswtPByBLoAZDz_eWuBgw', 'http://localhost:5678/api/function/weibo', 1),
+(2, 'zaobao', '今日早报', '每日早报，获取当天新闻事件列表', '{\"type\":\"object\",\"properties\":{}}', 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHBpcmVkIjowLCJ1c2VyX2lkIjowfQ.tLAGkF8XWh_G-oQzevpIodsswtPByBLoAZDz_eWuBgw', 'http://localhost:5678/api/function/zaobao', 1),
+(3, 'dalle3', 'DALLE3', 'AI 绘画工具，根据输入的绘图描述用 AI 工具进行绘画', '{\"type\":\"object\",\"required\":[\"prompt\"],\"properties\":{\"prompt\":{\"type\":\"string\",\"description\":\"绘画提示词\"}}}', 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHBpcmVkIjowLCJ1c2VyX2lkIjowfQ.tLAGkF8XWh_G-oQzevpIodsswtPByBLoAZDz_eWuBgw', 'http://localhost:5678/api/function/dalle3', 1);
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_invite_codes`
+--
+
+DROP TABLE IF EXISTS `chatgpt_invite_codes`;
+CREATE TABLE `chatgpt_invite_codes` (
+  `id` int NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户ID',
+  `code` char(8) NOT NULL COMMENT '邀请码',
+  `hits` int NOT NULL COMMENT '点击次数',
+  `reg_num` smallint NOT NULL COMMENT '注册数量',
+  `created_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='用户邀请码';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_invite_logs`
+--
+
+DROP TABLE IF EXISTS `chatgpt_invite_logs`;
+CREATE TABLE `chatgpt_invite_logs` (
+  `id` int NOT NULL,
+  `inviter_id` int NOT NULL COMMENT '邀请人ID',
+  `user_id` int NOT NULL COMMENT '注册用户ID',
+  `username` varchar(11) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '用户名',
+  `invite_code` char(8) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '邀请码',
+  `reward_json` text NOT NULL COMMENT '邀请奖励',
+  `created_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='邀请注册日志';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_mj_jobs`
+--
+
+DROP TABLE IF EXISTS `chatgpt_mj_jobs`;
+CREATE TABLE `chatgpt_mj_jobs` (
+  `id` int NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户 ID',
+  `task_id` varchar(20) DEFAULT NULL COMMENT '任务 ID',
+  `type` varchar(20) DEFAULT 'image' COMMENT '任务类别',
+  `message_id` char(40) NOT NULL COMMENT '消息 ID',
+  `channel_id` char(40) DEFAULT NULL COMMENT '频道ID',
+  `reference_id` char(40) DEFAULT NULL COMMENT '引用消息 ID',
+  `prompt` varchar(2000) NOT NULL COMMENT '会话提示词',
+  `img_url` varchar(400) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci DEFAULT NULL COMMENT '图片URL',
+  `org_url` varchar(400) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci DEFAULT NULL COMMENT '原始图片地址',
+  `hash` varchar(100) DEFAULT NULL COMMENT 'message hash',
+  `progress` smallint DEFAULT '0' COMMENT '任务进度',
+  `use_proxy` tinyint(1) NOT NULL DEFAULT '0' COMMENT '是否使用反代',
+  `created_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='MidJourney 任务表';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_orders`
+--
+
+DROP TABLE IF EXISTS `chatgpt_orders`;
+CREATE TABLE `chatgpt_orders` (
+  `id` int NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户ID',
+  `product_id` int NOT NULL COMMENT '产品ID',
+  `mobile` char(11) NOT NULL COMMENT '用户手机号',
+  `order_no` varchar(30) NOT NULL COMMENT '订单ID',
+  `subject` varchar(100) NOT NULL COMMENT '订单产品',
+  `amount` decimal(10,2) NOT NULL DEFAULT '0.00' COMMENT '订单金额',
+  `status` tinyint(1) NOT NULL DEFAULT '0' COMMENT '订单状态（0：待支付，1：已扫码，2：支付失败）',
+  `remark` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '备注',
+  `pay_time` int DEFAULT NULL COMMENT '支付时间',
+  `pay_way` varchar(20) NOT NULL COMMENT '支付方式',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL,
+  `deleted_at` datetime DEFAULT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='充值订单表';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_products`
+--
+
+DROP TABLE IF EXISTS `chatgpt_products`;
+CREATE TABLE `chatgpt_products` (
+  `id` int NOT NULL,
+  `name` varchar(30) NOT NULL COMMENT '名称',
+  `price` decimal(10,2) NOT NULL DEFAULT '0.00' COMMENT '价格',
+  `discount` decimal(10,2) NOT NULL DEFAULT '0.00' COMMENT '优惠金额',
+  `days` smallint NOT NULL DEFAULT '0' COMMENT '延长天数',
+  `calls` int NOT NULL DEFAULT '0' COMMENT '调用次数',
+  `img_calls` int NOT NULL DEFAULT '0' COMMENT '绘图次数',
+  `enabled` tinyint(1) NOT NULL DEFAULT '0' COMMENT '是否启动',
+  `sales` int NOT NULL DEFAULT '0' COMMENT '销量',
+  `sort_num` tinyint NOT NULL DEFAULT '0' COMMENT '排序',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='会员套餐表';
+
+--
+-- 转存表中的数据 `chatgpt_products`
+--
+
+INSERT INTO `chatgpt_products` (`id`, `name`, `price`, `discount`, `days`, `calls`, `img_calls`, `enabled`, `sales`, `sort_num`, `created_at`, `updated_at`) VALUES
+(1, '会员1个月', '19.90', '10.00', 30, 0, 0, 1, 2, 0, '2023-08-28 10:48:57', '2023-11-08 17:22:07'),
+(2, '会员3个月', '140.00', '30.00', 90, 0, 0, 1, 1, 0, '2023-08-28 10:52:22', '2023-08-31 16:24:31'),
+(3, '会员6个月', '290.00', '100.00', 180, 0, 0, 1, 1, 0, '2023-08-28 10:53:39', '2023-08-31 16:24:36'),
+(4, '会员12个月', '580.00', '200.00', 365, 0, 0, 1, 1, 0, '2023-08-28 10:54:15', '2023-08-31 16:24:42'),
+(5, '100次点卡', '10.00', '9.90', 0, 100, 10, 1, 3, 0, '2023-08-28 10:55:08', '2023-12-15 16:29:06'),
+(6, '200次点卡', '29.90', '20.00', 0, 200, 25, 1, 1, 0, '2023-12-15 16:55:12', '2023-12-15 16:55:12');
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_rewards`
+--
+
+DROP TABLE IF EXISTS `chatgpt_rewards`;
+CREATE TABLE `chatgpt_rewards` (
+  `id` int NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户 ID',
+  `tx_id` char(36) NOT NULL COMMENT '交易 ID',
+  `amount` decimal(10,2) NOT NULL COMMENT '打赏金额',
+  `remark` varchar(80) NOT NULL COMMENT '备注',
+  `status` tinyint(1) NOT NULL COMMENT '核销状态，0：未核销，1：已核销',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='用户打赏';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_sd_jobs`
+--
+
+DROP TABLE IF EXISTS `chatgpt_sd_jobs`;
+CREATE TABLE `chatgpt_sd_jobs` (
+  `id` int NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户 ID',
+  `type` varchar(20) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci DEFAULT 'txt2img' COMMENT '任务类别',
+  `task_id` char(30) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '任务 ID',
+  `prompt` varchar(2000) NOT NULL COMMENT '会话提示词',
+  `img_url` varchar(255) DEFAULT NULL COMMENT '图片URL',
+  `params` text CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci COMMENT '绘画参数json',
+  `progress` smallint DEFAULT '0' COMMENT '任务进度',
+  `created_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='Stable Diffusion 任务表';
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_users`
+--
+
+DROP TABLE IF EXISTS `chatgpt_users`;
+CREATE TABLE `chatgpt_users` (
+  `id` int NOT NULL,
+  `mobile` char(11) NOT NULL COMMENT '手机号码',
+  `password` char(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '密码',
+  `avatar` varchar(100) NOT NULL COMMENT '头像',
+  `salt` char(12) NOT NULL COMMENT '密码盐',
+  `total_tokens` bigint NOT NULL DEFAULT '0' COMMENT '累计消耗 tokens',
+  `tokens` bigint NOT NULL DEFAULT '0' COMMENT '当月消耗 tokens',
+  `calls` int NOT NULL DEFAULT '0' COMMENT '剩余调用次数',
+  `img_calls` int NOT NULL DEFAULT '0' COMMENT '剩余绘图次数',
+  `expired_time` int NOT NULL COMMENT '用户过期时间',
+  `status` tinyint(1) NOT NULL COMMENT '当前状态',
+  `chat_config_json` text CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '聊天配置json',
+  `chat_roles_json` text CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '聊天角色 json',
+  `chat_models_json` text CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT 'AI模型 json',
+  `last_login_at` int NOT NULL COMMENT '最后登录时间',
+  `vip` tinyint(1) NOT NULL DEFAULT '0' COMMENT '是否会员',
+  `last_login_ip` char(16) NOT NULL COMMENT '最后登录 IP',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='用户表';
+
+--
+-- 转存表中的数据 `chatgpt_users`
+--
+
+INSERT INTO `chatgpt_users` (`id`, `mobile`, `password`, `avatar`, `salt`, `total_tokens`, `tokens`, `calls`, `img_calls`, `expired_time`, `status`, `chat_config_json`, `chat_roles_json`, `chat_models_json`, `last_login_at`, `vip`, `last_login_ip`, `created_at`, `updated_at`) VALUES
+(4, '18575670125', 'ccc3fb7ab61b8b5d096a4a166ae21d121fc38c71bbd1be6173d9ab973214a63b', 'http://img.r9it.com/chatgpt-plus/1693981355719469.png', 'ueedue5l', 64593, 20756, 6780, 21, 1788021036, 1, '{\"api_keys\":{\"Azure\":\"\",\"ChatGLM\":\"\",\"OpenAI\":\"\"}}', '[\"elon_musk\",\"girl_friend\",\"lu_xun\",\"red_book\",\"psychiatrist\",\"translator\",\"weekly_report\",\"artist\",\"dou_yin\",\"english_trainer\",\"gpt\",\"kong_zi\",\"programmer\",\"seller\",\"steve_jobs\",\"teacher\"]', '[\"completions_pro\",\"eb-instant\",\"general\",\"generalv2\",\"chatglm_pro\",\"chatglm_lite\",\"chatglm_std\",\"gpt-3.5-turbo-16k\",\"gpt-4\",\"generalv3\",\"gpt-3.5-turbo-1106\",\"gpt-4-1106-preview\"]', 1703233159, 1, '::1', '2023-06-12 16:47:17', '2023-12-22 16:19:19'),
+(91, '18575670126', '5e4050b8dd403f593260395d9edeb9f273dbe92d15dfdd929c4a182e95da10c4', '/images/avatar/user.png', '6fj0otl8', 0, 0, 10, 0, 0, 1, '{\"api_keys\":{\"Azure\":\"\",\"ChatGLM\":\"\",\"OpenAI\":\"\"}}', '[\"gpt\"]', '[\"completions_pro\",\"eb-instant\",\"general\",\"generalv2\",\"chatglm_pro\",\"chatglm_lite\",\"chatglm_std\",\"gpt-3.5-turbo-16k\"]', 1697184324, 1, '::1', '2023-10-13 16:01:56', '2023-11-22 11:29:38'),
+(98, '13888888888', 'c988bb3e89df8eb7d8a4cf08fa9d0f2bca0a9e6831f9325279e46013bbb05e31', '/images/avatar/user.png', 'xie5vya7', 0, 0, 10, 5, 0, 1, '{\"api_keys\":{\"Azure\":\"\",\"ChatGLM\":\"\",\"OpenAI\":\"\"}}', '[\"gpt\"]', '[\"eb-instant\",\"completions_pro\",\"generalv2\",\"general\",\"chatglm_pro\",\"gpt-3.5-turbo-16k\",\"chatglm_lite\",\"chatglm_std\"]', 0, 0, '', '2023-11-23 16:26:06', '2023-11-23 16:26:06'),
+(99, '13999999999', 'bf47d517c17ed1ead6ff2542753f9b6a132e79c29e06c3710faf1a36d800a217', '/images/avatar/user.png', 'x1s66pwd', 0, 0, 10, 5, 0, 1, '{\"api_keys\":{\"Azure\":\"\",\"ChatGLM\":\"\",\"OpenAI\":\"\"}}', '[\"gpt\"]', '[\"eb-instant\",\"completions_pro\",\"generalv2\",\"general\",\"chatglm_pro\",\"gpt-3.5-turbo-16k\",\"chatglm_lite\",\"chatglm_std\"]', 0, 0, '', '2023-11-23 16:27:20', '2023-11-23 16:27:20'),
+(100, '13777777777', 'dcaf31b154432310bd700349e7de7e9dde2a3d6955a035a01fe527c7917a4f99', '/images/avatar/user.png', 'i8a53f8f', 0, 0, 10, 5, 0, 1, '{\"api_keys\":{\"Azure\":\"\",\"ChatGLM\":\"\",\"OpenAI\":\"\"}}', '[\"gpt\"]', '[\"eb-instant\",\"completions_pro\",\"generalv2\",\"general\",\"chatglm_pro\",\"gpt-3.5-turbo-16k\",\"chatglm_lite\",\"chatglm_std\"]', 0, 0, '', '2023-11-23 16:55:45', '2023-11-23 16:55:45');
+
+-- --------------------------------------------------------
+
+--
+-- 表的结构 `chatgpt_user_login_logs`
+--
+
+DROP TABLE IF EXISTS `chatgpt_user_login_logs`;
+CREATE TABLE `chatgpt_user_login_logs` (
+  `id` int NOT NULL,
+  `user_id` int NOT NULL COMMENT '用户ID',
+  `username` varchar(30) NOT NULL COMMENT '用户名',
+  `login_ip` char(16) NOT NULL COMMENT '登录IP',
+  `login_address` varchar(30) NOT NULL COMMENT '登录地址',
+  `created_at` datetime NOT NULL,
+  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='用户登录日志';
+
+--
+-- 转储表的索引
+--
+
+--
+-- 表的索引 `chatgpt_api_keys`
+--
+ALTER TABLE `chatgpt_api_keys`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `value` (`value`);
+
+--
+-- 表的索引 `chatgpt_chat_history`
+--
+ALTER TABLE `chatgpt_chat_history`
+  ADD PRIMARY KEY (`id`),
+  ADD KEY `chat_id` (`chat_id`);
+
+--
+-- 表的索引 `chatgpt_chat_items`
+--
+ALTER TABLE `chatgpt_chat_items`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `chat_id` (`chat_id`);
+
+--
+-- 表的索引 `chatgpt_chat_models`
+--
+ALTER TABLE `chatgpt_chat_models`
+  ADD PRIMARY KEY (`id`);
+
+--
+-- 表的索引 `chatgpt_chat_roles`
+--
+ALTER TABLE `chatgpt_chat_roles`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `marker` (`marker`);
+
+--
+-- 表的索引 `chatgpt_configs`
+--
+ALTER TABLE `chatgpt_configs`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `marker` (`marker`);
+
+--
+-- 表的索引 `chatgpt_functions`
+--
+ALTER TABLE `chatgpt_functions`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `name` (`name`);
+
+--
+-- 表的索引 `chatgpt_invite_codes`
+--
+ALTER TABLE `chatgpt_invite_codes`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `code` (`code`);
+
+--
+-- 表的索引 `chatgpt_invite_logs`
+--
+ALTER TABLE `chatgpt_invite_logs`
+  ADD PRIMARY KEY (`id`);
+
+--
+-- 表的索引 `chatgpt_mj_jobs`
+--
+ALTER TABLE `chatgpt_mj_jobs`
+  ADD PRIMARY KEY (`id`),
+  ADD KEY `message_id` (`message_id`);
+
+--
+-- 表的索引 `chatgpt_orders`
+--
+ALTER TABLE `chatgpt_orders`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `order_no` (`order_no`);
+
+--
+-- 表的索引 `chatgpt_products`
+--
+ALTER TABLE `chatgpt_products`
+  ADD PRIMARY KEY (`id`);
+
+--
+-- 表的索引 `chatgpt_rewards`
+--
+ALTER TABLE `chatgpt_rewards`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `tx_id` (`tx_id`);
+
+--
+-- 表的索引 `chatgpt_sd_jobs`
+--
+ALTER TABLE `chatgpt_sd_jobs`
+  ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `task_id` (`task_id`);
+
+--
+-- 表的索引 `chatgpt_users`
+--
+ALTER TABLE `chatgpt_users`
+  ADD PRIMARY KEY (`id`);
+
+--
+-- 表的索引 `chatgpt_user_login_logs`
+--
+ALTER TABLE `chatgpt_user_login_logs`
+  ADD PRIMARY KEY (`id`);
+
+--
+-- 在导出的表使用AUTO_INCREMENT
+--
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_api_keys`
+--
+ALTER TABLE `chatgpt_api_keys`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_chat_history`
+--
+ALTER TABLE `chatgpt_chat_history`
+  MODIFY `id` bigint NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_chat_items`
+--
+ALTER TABLE `chatgpt_chat_items`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_chat_models`
+--
+ALTER TABLE `chatgpt_chat_models`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT, AUTO_INCREMENT=15;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_chat_roles`
+--
+ALTER TABLE `chatgpt_chat_roles`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT, AUTO_INCREMENT=129;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_configs`
+--
+ALTER TABLE `chatgpt_configs`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT, AUTO_INCREMENT=3;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_functions`
+--
+ALTER TABLE `chatgpt_functions`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT, AUTO_INCREMENT=4;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_invite_codes`
+--
+ALTER TABLE `chatgpt_invite_codes`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_invite_logs`
+--
+ALTER TABLE `chatgpt_invite_logs`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_mj_jobs`
+--
+ALTER TABLE `chatgpt_mj_jobs`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_orders`
+--
+ALTER TABLE `chatgpt_orders`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_products`
+--
+ALTER TABLE `chatgpt_products`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT, AUTO_INCREMENT=7;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_rewards`
+--
+ALTER TABLE `chatgpt_rewards`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_sd_jobs`
+--
+ALTER TABLE `chatgpt_sd_jobs`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_users`
+--
+ALTER TABLE `chatgpt_users`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT, AUTO_INCREMENT=101;
+
+--
+-- 使用表AUTO_INCREMENT `chatgpt_user_login_logs`
+--
+ALTER TABLE `chatgpt_user_login_logs`
+  MODIFY `id` int NOT NULL AUTO_INCREMENT;
+COMMIT;
+
+/*!40101 SET CHARACTER_SET_CLIENT=@OLD_CHARACTER_SET_CLIENT */;
+/*!40101 SET CHARACTER_SET_RESULTS=@OLD_CHARACTER_SET_RESULTS */;
+/*!40101 SET COLLATION_CONNECTION=@OLD_COLLATION_CONNECTION */;

database/update-v3.2.3.sql

@@ -1 +1,28 @@
-ALTER TABLE `chatgpt_products` ADD `img_calls` INT(11) NOT NULL DEFAULT '0' COMMENT '绘图次数' AFTER `calls`;
+ALTER TABLE `chatgpt_mj_jobs` ADD `channel_id` CHAR(40) NULL DEFAULT NULL COMMENT '频道ID' AFTER `message_id`;
+ALTER TABLE `chatgpt_mj_jobs` DROP INDEX `task_id`;
+ALTER TABLE `chatgpt_products` ADD `img_calls` INT(11) NOT NULL DEFAULT '0' COMMENT '绘图次数' AFTER `calls`;
+
+CREATE TABLE `chatgpt_functions` (
+                                     `id` int NOT NULL,
+                                     `name` varchar(30) NOT NULL COMMENT '函数名称',
+                                     `description` varchar(255) DEFAULT NULL COMMENT '函数描述',
+                                     `parameters` text COMMENT '函数参数（JSON）',
+                                     `required` varchar(255) NOT NULL COMMENT '必填参数（JSON）',
+                                     `action` varchar(255) DEFAULT NULL COMMENT '函数处理 API'
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='函数插件表';
+
+ALTER TABLE `chatgpt_functions` ADD PRIMARY KEY (`id`);
+ALTER TABLE `chatgpt_functions` MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+ALTER TABLE `chatgpt_functions` ADD UNIQUE(`name`);
+
+ALTER TABLE `chatgpt_functions` ADD `enabled` TINYINT(1) NOT NULL DEFAULT '0' COMMENT '是否启用' AFTER `action`;
+
+ALTER TABLE `chatgpt_functions` ADD `label` VARCHAR(30) NULL COMMENT '函数标签' AFTER `name`;
+
+ALTER TABLE `chatgpt_mj_jobs` ADD `use_proxy` TINYINT(1) NOT NULL DEFAULT '0' COMMENT '是否使用反代' AFTER `progress`;
+ALTER TABLE `chatgpt_mj_jobs` CHANGE `img_url` `img_url` VARCHAR(400) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL COMMENT '图片URL';
+
+ALTER TABLE `chatgpt_functions` ADD `token` VARCHAR(255) NULL COMMENT 'API授权token' AFTER `action`;
+
+ALTER TABLE `chatgpt_functions` DROP `required`;

web/.env.development

@@ -5,4 +5,4 @@ VUE_APP_PASS=12345678
 VUE_APP_ADMIN_USER=admin
 VUE_APP_ADMIN_PASS=admin123
 VUE_APP_KEY_PREFIX=ChatPLUS_DEV_
-VUE_APP_TITLE="Chat-Plus V3"
+VUE_APP_TITLE="ChatPlus-AI 创作系统"

web/.env.production

@@ -1,4 +1,4 @@
 VUE_APP_API_HOST=
 VUE_APP_WS_HOST=
 VUE_APP_KEY_PREFIX=ChatPLUS_
-VUE_APP_TITLE="Chat-Plus V3"
+VUE_APP_TITLE="ChatPlus-AI 创作系统"

web/src/assets/css/mobile/apps.css

@@ -0,0 +1,17 @@
+.title {
+  color: #fff;
+  text-align: center;
+  font-size: 16px;
+  font-weight: bold;
+}
+.app-background {
+  background-color: #282828;
+  height: 100vh;
+}
+.mobile-setting .content {
+  padding-top: 60px;
+}
+.mobile-setting .content .van-field__label {
+  width: 100px;
+  text-align: right;
+}

web/src/assets/css/mobile/apps.styl

@@ -0,0 +1,20 @@
+.title {
+  color  #fff 
+  text-align  center 
+  font-size  16px 
+  font-weight  bold 
+}
+.app-background {
+  background-color  #282828 
+  height  100vh 
+}
+.mobile-setting {
+  .content {
+    padding-top 60px
+
+    .van-field__label {
+      width 100px
+      text-align right
+    }
+  }
+}

web/src/assets/css/mobile/chat-list.css

@@ -0,0 +1,47 @@
+.app-background {
+  background-color: #1c1c1c;
+  height: 100vh;
+}
+.mobile-chat-list {
+  background-color: #1c1c1c;
+  position: absolute;
+  top: 40px;
+  width: 100%;
+}
+.content .van-cell__value .chat-list-item {
+  display: flex;
+  font-size: 16px;
+  color: #fff;
+  background-color: #1c1c1c;
+}
+.content .van-cell__value .chat-list-item .van-image {
+  min-width: 32px;
+  width: 32px;
+  height: 32px;
+}
+.content .van-cell__value .chat-list-item .van-ellipsis {
+  margin-top: 5px;
+  margin-left: 10px;
+}
+.van-picker-column .picker-option {
+  display: flex;
+  width: 100%;
+  padding: 0 10px;
+}
+.van-picker-column .picker-option .van-image {
+  width: 20px;
+  height: 20px;
+  margin-right: 5px;
+}
+.van-nav-bar .van-nav-bar__right .van-icon {
+  font-size: 20px;
+}
+.popup {
+  background-color: #1c1c1c;
+}
+.dialog {
+  background-color: #1c1c1c;
+}
+.field {
+  background-color: #1c1c1c;
+}

web/src/assets/css/mobile/chat-list.styl

@@ -0,0 +1,58 @@
+$fontSize = 16px;
+.app-background {
+  background-color: #1c1c1c;
+  height: 100vh;
+}
+.mobile-chat-list {
+  background-color: #1c1c1c;
+  position: absolute;
+  top: 40px;
+  width: 100%;
+}
+  .content {
+    .van-cell__value {
+      .chat-list-item {
+        display flex
+        font-size $fontSize
+        color: #fff;
+        background-color: #1c1c1c;
+        .van-image {
+          min-width 32px
+          width 32px
+          height 32px
+        }
+        .van-ellipsis {
+          margin-top 5px;
+          margin-left 10px;
+        }
+      }
+    }
+  }
+  .van-picker-column {
+    .picker-option {
+      display flex
+      width 100%
+      padding 0 10px
+      .van-image {
+        width 20px;
+        height 20px;
+        margin-right 5px
+      }
+    }
+  }
+  .van-nav-bar {
+    .van-nav-bar__right {
+      .van-icon {
+        font-size 20px;
+      }
+    }
+  }
+  .popup {
+    background-color: #1c1c1c;
+  }
+  .dialog {
+    background-color: #1c1c1c;
+  }
+  .field {
+    background-color: #1c1c1c;
+  }

web/src/assets/css/mobile/chat-session.css

@@ -0,0 +1,45 @@
+.app-background {
+  background-color: #1c1c1c;
+  height: 100vh;
+}
+body,
+.mobile-chat,
+.van-sticky,
+.van-nav-bar,
+.van-list,
+.message-list-box,
+.van-cell,
+.chat-box,
+.van-cell-group,
+.van-field {
+  background-color: #1c1c1c !important;
+  color: #fff !important;
+}
+.chat-box-wrapper {
+  position: fixed;
+  bottom: 0;
+  width: 100%;
+  background-color: #1c1c1c;
+}
+.icon-box .van-icon,
+.mobile-chat .van-nav-bar__title .van-dropdown-menu__title,
+.mobile-chat .van-nav-bar__title .van-cell__title,
+.mobile-chat .van-nav-bar__right .van-icon {
+  color: #fff !important;
+}
+.mobile-chat .van-nav-bar__title .van-dropdown-menu__title {
+  margin-right: 15px;
+}
+.mobile-chat .van-nav-bar__title .van-cell__title {
+  text-align: left;
+}
+.mobile-chat .chat-list-wrapper {
+  position: fixed;
+  top: 50px;
+  bottom: 60px;
+  width: 100vw;
+  overflow-y: scroll;
+}
+.mobile-chat .van-nav-bar__right .van-icon {
+  font-size: 20px;
+}

web/src/assets/css/mobile/chat-session.styl

@@ -0,0 +1,51 @@
+.app-background {
+    background-color  #1c1c1c 
+    height  100vh 
+  }
+  body,
+  .mobile-chat,
+  .van-sticky,
+  .van-nav-bar,
+  .van-list,
+  .message-list-box
+  .van-cell,
+  .chat-box,
+  .van-cell-group,
+  .van-field {
+    background-color  #1c1c1c !important 
+    color  #fff !important 
+  }
+  .chat-box-wrapper {
+    position  fixed 
+    bottom  0 
+    width  100% 
+    background-color  #1c1c1c 
+    }
+  .icon-box .van-icon,
+  .mobile-chat .van-nav-bar__title .van-dropdown-menu__title,
+  .mobile-chat .van-nav-bar__title .van-cell__title,
+  .mobile-chat .van-nav-bar__right .van-icon {
+    color  #fff !important 
+  }
+  .mobile-chat {
+    .van-nav-bar__title {
+      .van-dropdown-menu__title {
+        margin-right  15px 
+      }
+      .van-cell__title {
+        text-align  left 
+      }
+    }
+    .chat-list-wrapper {
+      position  fixed 
+      top  50px 
+      bottom  60px 
+      width  100vw 
+      overflow-y  scroll 
+    }
+    .van-nav-bar__right {
+      .van-icon {
+        font-size  20px 
+      }
+    }
+  }

web/src/assets/css/mobile/custom-scroll.css

@@ -0,0 +1,13 @@
+.custom-scroll ::-webkit-scrollbar {
+  width: 8px; /* 滚动条宽度 */
+}
+.custom-scroll ::-webkit-scrollbar-track {
+  background-color: #282c34;
+}
+.custom-scroll ::-webkit-scrollbar-thumb {
+  background-color: #444;
+  border-radius: 8px;
+}
+.custom-scroll ::-webkit-scrollbar-thumb:hover {
+  background-color: #666;
+}

web/src/assets/css/mobile/custom-scroll.styl

@@ -0,0 +1,26 @@
+.custom-scroll {
+  /* 修改滚动条的颜色 */
+
+  ::-webkit-scrollbar {
+    width: 8px; /* 滚动条宽度 */
+  }
+
+  /* 修改滚动条轨道的背景颜色 */
+
+  ::-webkit-scrollbar-track {
+    background-color: #282C34;
+  }
+
+  /* 修改滚动条的滑块颜色 */
+
+  ::-webkit-scrollbar-thumb {
+    background-color: #444444;
+    border-radius 8px
+  }
+
+  /* 修改滚动条的滑块的悬停颜色 */
+
+  ::-webkit-scrollbar-thumb:hover {
+    background-color: #666666;
+  }
+}

web/src/assets/css/mobile/home.css

@@ -0,0 +1,45 @@
+.my-tabbar {
+  background-color: #171717;
+  box-shadow: -3px 4px 16px #000;
+  box-sizing: border-box;
+}
+.banner {
+  display: flex;
+  justify-content: center;
+  align-items: center;
+  background-color: #171717;
+  border-bottom-right-radius: 10px;
+  box-shadow: 0 -4px 13px #000;
+  height: 40px;
+  width: 100%;
+  position: fixed;
+  z-index: 2 /* Add z-index */;
+}
+.banner-title {
+  color: #2cc995;
+  font-size: 20px;
+  font-weight: bold;
+  text-shadow: 0 -4px 13px #000;
+}
+.mobile-home {
+  background-color: #171717;
+}
+.mobile-home .van-tabbar before {
+  position: absolute;
+  bottom: 0;
+  left: 0;
+  right: 0;
+  height: 50px;
+  background-color: #171717;
+  border-top-left-radius: 10px;
+  border-top-right-radius: 10px;
+  opacity: 0;
+  transition: opacity 0.5s;
+  z-index: -1;
+}
+.mobile-home .van-tabbar.activeTab before {
+  opacity: 1;
+}
+.router-view {
+  margin-bottom: 50px;
+}

web/src/assets/css/mobile/home.styl

@@ -0,0 +1,45 @@
+.my-tabbar {
+  background-color  #171717  
+  box-shadow  -3px 4px 16px #000  
+  box-sizing  border-box 
+}
+.banner {
+  display  flex 
+  justify-content  center 
+  align-items  center 
+  background-color  #171717 
+  border-bottom-right-radius  10px 
+  box-shadow  0 -4px 13px #000 
+  height  40px 
+  width  100% 
+  position  fixed 
+  z-index  2    /* Add z-index */
+} 
+.banner-title {
+  color  #2CC995 
+  font-size  20px 
+  font-weight  bold 
+  text-shadow  0 -4px 13px #000 
+}
+.mobile-home {
+  background-color  #171717 
+}
+.mobile-home .van-tabbar  before {
+  position  absolute 
+  bottom  0 
+  left  0 
+  right  0 
+  height  50px 
+  background-color  #171717 
+  border-top-left-radius  10px 
+  border-top-right-radius  10px 
+  opacity  0 
+  transition  opacity 0.5s 
+  z-index  -1 
+}
+.mobile-home .van-tabbar.activeTab  before {
+  opacity  1 
+}
+.router-view {
+  margin-bottom  50px 
+}

web/src/assets/css/mobile/image-list.css

@@ -0,0 +1,252 @@
+
+.page-sd {
+  background-color: #282828;
+}
+.page-sd .inner {
+  display: flex;
+}
+.page-sd .inner .sd-box {
+  margin: 10px;
+  background-color: #282828;
+  min-width: 92%;
+  max-width: 92%;
+  padding: 0px;
+  border-radius: 0px;
+  color: #fff;
+  font-size: 14px;
+}
+.page-sd .inner .sd-box h2 {
+  font-weight: bold;
+  font-size: 20px;
+  text-align: center;
+  color: #fff;
+}
+.page-sd .inner .sd-box ::-webkit-scrollbar {
+  width: 0;
+  height: 0;
+  background-color: transparent;
+}
+.page-sd .inner .sd-box .sd-params {
+  margin-top: 10px;
+  overflow: auto;
+}
+.page-sd .inner .sd-box .sd-params .param-line {
+  padding: 0 10px;
+}
+.page-sd .inner .sd-box .sd-params .param-line .el-icon {
+  position: relative;
+  top: 3px;
+}
+.page-sd .inner .sd-box .sd-params .param-line .el-input__suffix-inner .el-icon {
+  top: 0;
+}
+.page-sd .inner .sd-box .sd-params .param-line .grid-content,
+.page-sd .inner .sd-box .sd-params .param-line .form-item-inner {
+  display: flex;
+}
+.page-sd .inner .sd-box .sd-params .param-line .grid-content .el-icon,
+.page-sd .inner .sd-box .sd-params .param-line .form-item-inner .el-icon {
+  margin-left: 10px;
+  margin-top: 2px;
+}
+.page-sd .inner .sd-box .sd-params .param-line.pt {
+  padding-top: 5px;
+  padding-bottom: 5px;
+}
+.page-sd .inner .sd-box .submit-btn {
+  padding: 10px 15px 0 15px;
+  text-align: center;
+}
+.page-sd .inner .sd-box .submit-btn .el-button {
+  width: 100%;
+}
+.page-sd .inner .sd-box .submit-btn .el-button span {
+  color: #2d3a4b;
+}
+.page-sd .inner .el-form .el-form-item__label {
+  color: #fff;
+}
+.page-sd .inner .task-list-box {
+  width: 100%;
+  padding: 10px;
+  color: #fff;
+  overflow-x: hidden;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item {
+  width: 100%;
+  padding: 2px;
+  background-color: #555;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item .job-item-inner {
+  position: relative;
+  height: 100%;
+  overflow: hidden;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item .job-item-inner .progress {
+  position: absolute;
+  width: 100%;
+  height: 100%;
+  top: 0;
+  left: 0;
+  display: flex;
+  justify-content: center;
+  align-items: center;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item .job-item-inner .progress span {
+  font-size: 20px;
+  color: #fff;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item {
+  width: 100%;
+  height: 100%;
+  border: 1px solid #666;
+  padding: 6px;
+  overflow: hidden;
+  border-radius: 6px;
+  transition: all 0.3s ease; /* 添加过渡效果 */
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line {
+  margin: 6px 0;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul {
+  display: flex;
+  flex-flow: row;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul li {
+  margin-right: 6px;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul li a {
+  padding: 3px 0;
+  width: 40px;
+  text-align: center;
+  border-radius: 5px;
+  display: block;
+  cursor: pointer;
+  background-color: #4e5058;
+  color: #fff;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul li a:hover {
+  background-color: #6d6f78;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul .show-prompt {
+  font-size: 20px;
+  cursor: pointer;
+}
+.page-sd .inner .task-list-box .finish-job-list .animate:hover {
+  box-shadow: 0 0 10px rgba(71,255,241,0.6); /* 添加阴影效果 */
+  transform: translateY(-10px); /* 向上移动10像素 */
+}
+.page-sd .inner .task-list-box .el-image {
+  width: 100%;
+  height: 100%;
+  overflow: visible;
+}
+.page-sd .inner .task-list-box .el-image img {
+  height: 100%;
+}
+.page-sd .inner .task-list-box .el-image .el-image-viewer__wrapper img {
+  width: auto;
+  height: auto;
+}
+.page-sd .inner .task-list-box .el-image .image-slot {
+  display: flex;
+  flex-flow: column;
+  justify-content: center;
+  align-items: center;
+  height: 100%;
+  min-height: 200px;
+  color: #fff;
+  height: 240px;
+}
+.page-sd .inner .task-list-box .el-image .image-slot .iconfont {
+  font-size: 50px;
+  margin-bottom: 10px;
+}
+.page-sd .inner .task-list-box .el-image.upscale {
+  max-height: 100%;
+}
+.page-sd .inner .task-list-box .el-image.upscale img {
+  height: 100%;
+}
+.page-sd .inner .task-list-box .el-image.upscale .el-image-viewer__wrapper img {
+  width: auto;
+  height: auto;
+}
+.page-sd .el-overlay-dialog .el-dialog {
+  background-color: #1a1b1e;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__header .el-dialog__title {
+  color: #f5f5f5;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body {
+  padding: 0 0 0 15px !important;
+  display: flex;
+  height: 100%;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row {
+  width: 100%;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container {
+  display: flex;
+  justify-content: center;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container .image-slot {
+  display: flex;
+  height: 100vh;
+  align-items: center;
+  justify-content: center;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container .image-slot .el-icon {
+  font-size: 60px;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info {
+  background-color: #25262b;
+  padding: 1rem 1.5rem;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line {
+  width: 100%;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .prompt {
+  background-color: #35363b;
+  padding: 10px;
+  color: #999;
+  overflow: auto;
+  max-height: 100%;
+  min-height: 100%;
+  position: relative;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .prompt .el-icon {
+  position: absolute;
+  right: 10px;
+  bottom: 10px;
+  cursor: pointer;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper {
+  margin-top: 10px;
+  display: flex;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper label {
+  display: flex;
+  width:100%;
+  color: #a5a5a5;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper .item-value {
+  display: flex;
+  width: 100%;
+  background-color: #35363b;
+  padding: 2px 5px;
+  border-radius: 5px;
+  color: #f5f5f5;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .copy-params {
+  padding: 20px 0 10px 0;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .copy-params .el-button {
+  width: 100%;
+}
+.page-sd .mj-list-item-prompt .el-icon {
+  margin-left: 10px;
+  cursor: pointer;
+  position: relative;
+  top: 2px;
+}

web/src/assets/css/mobile/image-list.styl

@@ -0,0 +1,106 @@
+.page-sd {
+  background-color: #282828;
+
+  .inner {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+
+    .sd-box {
+      background-color #282828
+      min-width 92%
+      max-width 92%
+      padding 0px
+      border-radius 0px
+      color #ffffff;
+      font-size 14px
+
+      h2 {
+        font-weight: bold;
+        font-size 20px
+        text-align center
+        color #fff
+      }
+
+      // 隐藏滚动条
+
+      ::-webkit-scrollbar {
+        width: 0;
+        height: 0;
+        background-color: transparent;
+      }
+
+      .sd-params {
+        margin-top 10px
+        overflow auto
+
+
+        .param-line {
+          padding 0 10px
+
+          .el-icon {
+            position relative
+            top 3px
+          }
+
+          .el-input__suffix-inner {
+            .el-icon {
+              top 0
+            }
+          }
+
+          .grid-content
+          .form-item-inner {
+            display flex
+
+            .el-icon {
+              margin-left 10px
+              margin-top 2px
+            }
+          }
+
+        }
+
+        .param-line.pt {
+          padding-top 5px
+          padding-bottom 5px
+        }
+      }
+
+      .submit-btn {
+        padding 10px 15px 0 15px
+        text-align center
+
+        .el-button {
+          width 100%
+
+          span {
+            color #2D3A4B
+          }
+        }
+      }
+    }
+
+    .el-form {
+      .el-form-item__label {
+        color #ffffff
+      }
+    }
+
+    @import "mobile/task-mobile-list.styl"
+  }
+
+  @import "sd-task-mobile-dialog.styl"
+
+
+  .mj-list-item-prompt {
+    .el-icon {
+      margin-left 10px
+      cursor pointer
+      position relative
+      top 2px
+    }
+  }
+
+}
+

web/src/assets/css/mobile/image-sd.css

@@ -0,0 +1,252 @@
+.page-sd {
+  background-color: #282828;
+}
+.page-sd .inner {
+  display: flex;
+  align-items: center;
+  justify-content: center;
+}
+.page-sd .inner .sd-box {
+  background-color: #282828;
+  min-width: 92%;
+  max-width: 92%;
+  padding: 0px;
+  border-radius: 0px;
+  color: #fff;
+  font-size: 14px;
+}
+.page-sd .inner .sd-box h2 {
+  font-weight: bold;
+  font-size: 16px;
+  text-align: center;
+  color: #fff;
+}
+.page-sd .inner .sd-box ::-webkit-scrollbar {
+  width: 0;
+  height: 0;
+  background-color: transparent;
+}
+.page-sd .inner .sd-box .sd-params {
+  margin-top: 10px;
+  overflow: auto;
+}
+.page-sd .inner .sd-box .sd-params .param-line {
+  padding: 0 10px;
+}
+.page-sd .inner .sd-box .sd-params .param-line .el-icon {
+  position: relative;
+  top: 3px;
+}
+.page-sd .inner .sd-box .sd-params .param-line .el-input__suffix-inner .el-icon {
+  top: 0;
+}
+.page-sd .inner .sd-box .sd-params .param-line .grid-content,
+.page-sd .inner .sd-box .sd-params .param-line .form-item-inner {
+  display: flex;
+}
+.page-sd .inner .sd-box .sd-params .param-line .grid-content .el-icon,
+.page-sd .inner .sd-box .sd-params .param-line .form-item-inner .el-icon {
+  margin-left: 10px;
+  margin-top: 2px;
+}
+.page-sd .inner .sd-box .sd-params .param-line.pt {
+  padding-top: 5px;
+  padding-bottom: 5px;
+}
+.page-sd .inner .sd-box .submit-btn {
+  padding: 10px 15px 0 15px;
+  text-align: center;
+}
+.page-sd .inner .sd-box .submit-btn .el-button {
+  width: 100%;
+}
+.page-sd .inner .sd-box .submit-btn .el-button span {
+  color: #2d3a4b;
+}
+.page-sd .inner .el-form .el-form-item__label {
+  color: #fff;
+}
+.page-sd .inner .task-list-box {
+  width: 100%;
+  padding: 5px;
+  color: #fff;
+  overflow-x: hidden;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item {
+  width: 100%;
+  padding: 2px;
+  background-color: #555;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item .job-item-inner {
+  position: relative;
+  height: 100%;
+  overflow: hidden;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item .job-item-inner .progress {
+  position: absolute;
+  width: 100%;
+  height: 100%;
+  top: 0;
+  left: 0;
+  display: flex;
+  justify-content: center;
+  align-items: center;
+}
+.page-sd .inner .task-list-box .running-job-list .job-item .job-item-inner .progress span {
+  font-size: 20px;
+  color: #fff;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item {
+  width: 100%;
+  height: 100%;
+  border: 1px solid #666;
+  padding: 6px;
+  overflow: hidden;
+  border-radius: 6px;
+  transition: all 0.3s ease; /* 添加过渡效果 */
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line {
+  margin: 6px 0;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul {
+  display: flex;
+  flex-flow: row;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul li {
+  margin-right: 6px;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul li a {
+  padding: 3px 0;
+  width: 40px;
+  text-align: center;
+  border-radius: 5px;
+  display: block;
+  cursor: pointer;
+  background-color: #4e5058;
+  color: #fff;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul li a:hover {
+  background-color: #6d6f78;
+}
+.page-sd .inner .task-list-box .finish-job-list .job-item .opt .opt-line ul .show-prompt {
+  font-size: 20px;
+  cursor: pointer;
+}
+.page-sd .inner .task-list-box .finish-job-list .animate:hover {
+  box-shadow: 0 0 10px rgba(71,255,241,0.6); /* 添加阴影效果 */
+  transform: translateY(-10px); /* 向上移动10像素 */
+}
+.page-sd .inner .task-list-box .el-image {
+  width: 100%;
+  height: 100%;
+  overflow: visible;
+}
+.page-sd .inner .task-list-box .el-image img {
+  height: 240px;
+}
+.page-sd .inner .task-list-box .el-image .el-image-viewer__wrapper img {
+  width: auto;
+  height: auto;
+}
+.page-sd .inner .task-list-box .el-image .image-slot {
+  display: flex;
+  flex-flow: column;
+  justify-content: center;
+  align-items: center;
+  height: 100%;
+  min-height: 200px;
+  color: #fff;
+  height: 240px;
+}
+.page-sd .inner .task-list-box .el-image .image-slot .iconfont {
+  font-size: 50px;
+  margin-bottom: 10px;
+}
+.page-sd .inner .task-list-box .el-image.upscale {
+  max-height: 310px;
+}
+.page-sd .inner .task-list-box .el-image.upscale img {
+  height: 310px;
+}
+.page-sd .inner .task-list-box .el-image.upscale .el-image-viewer__wrapper img {
+  width: auto;
+  height: auto;
+}
+.page-sd .el-overlay-dialog .el-dialog {
+  background-color: #1a1b1e;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__header .el-dialog__title {
+  color: #f5f5f5;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body {
+  padding: 0 0 0 15px !important;
+  display: flex;
+  height: 100%;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row {
+  width: 100%;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container {
+  display: flex;
+  justify-content: center;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container .image-slot {
+  display: flex;
+  height: 100vh;
+  align-items: center;
+  justify-content: center;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container .image-slot .el-icon {
+  font-size: 60px;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info {
+  background-color: #25262b;
+  padding: 1rem 1.5rem;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line {
+  width: 100%;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .prompt {
+  background-color: #35363b;
+  padding: 10px;
+  color: #999;
+  overflow: auto;
+  max-height: 100px;
+  min-height: 50px;
+  position: relative;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .prompt .el-icon {
+  position: absolute;
+  right: 10px;
+  bottom: 10px;
+  cursor: pointer;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper {
+  margin-top: 10px;
+  display: flex;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper label {
+  display: flex;
+  width: 100px;
+  color: #a5a5a5;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper .item-value {
+  display: flex;
+  width: 100%;
+  background-color: #35363b;
+  padding: 2px 5px;
+  border-radius: 5px;
+  color: #f5f5f5;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .copy-params {
+  padding: 30px 0 10px 0;
+}
+.page-sd .el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .copy-params .el-button {
+  width: 100%;
+}
+.page-sd .mj-list-item-prompt .el-icon {
+  margin-left: 10px;
+  cursor: pointer;
+  position: relative;
+  top: 2px;
+}

web/src/assets/css/mobile/image-sd.styl

@@ -0,0 +1,106 @@
+.page-sd {
+  background-color: #282828;
+
+  .inner {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+
+    .sd-box {
+      background-color #282828
+      min-width 92%
+      max-width 92%
+      padding 0px
+      border-radius 0px
+      color #ffffff;
+      font-size 14px
+
+      h2 {
+        font-weight: bold;
+        font-size 16px
+        text-align center
+        color #fff
+      }
+
+      // 隐藏滚动条
+
+      ::-webkit-scrollbar {
+        width: 0;
+        height: 0;
+        background-color: transparent;
+      }
+
+      .sd-params {
+        margin-top 10px
+        overflow auto
+
+
+        .param-line {
+          padding 0 10px
+
+          .el-icon {
+            position relative
+            top 3px
+          }
+
+          .el-input__suffix-inner {
+            .el-icon {
+              top 0
+            }
+          }
+
+          .grid-content
+          .form-item-inner {
+            display flex
+
+            .el-icon {
+              margin-left 10px
+              margin-top 2px
+            }
+          }
+
+        }
+
+        .param-line.pt {
+          padding-top 5px
+          padding-bottom 5px
+        }
+      }
+
+      .submit-btn {
+        padding 10px 15px 0 15px
+        text-align center
+
+        .el-button {
+          width 100%
+
+          span {
+            color #2D3A4B
+          }
+        }
+      }
+    }
+
+    .el-form {
+      .el-form-item__label {
+        color #ffffff
+      }
+    }
+
+    @import "./task-list.styl"
+  }
+
+  @import "./sd-task-dialog.styl"
+
+
+  .mj-list-item-prompt {
+    .el-icon {
+      margin-left 10px
+      cursor pointer
+      position relative
+      top 2px
+    }
+  }
+
+}
+

web/src/assets/css/mobile/invitation.css

@@ -0,0 +1,93 @@
+.page-invitation {
+  display: flex;
+  justify-content: center;
+  background-color: #1c1c1c;
+  height: 100vh;
+  overflow-x: hidden;
+  overflow-y: visible;
+}
+.page-invitation .inner {
+  max-width: 100%;
+  width: 100%;
+  color: #fff;
+}
+.page-invitation .inner .center {
+  display: flex;
+  justify-content: center;
+  align-items: center;
+}
+.page-invitation .inner .title {
+  color: #fff;
+  text-align: center;
+  font-size: 16px;
+  font-weight: bold;
+}
+.page-invitation .inner .share-box .info {
+  line-height: 1.5;
+  border: 1px solid #444;
+  border-radius: 10px;
+  padding: 10px;
+  margin: 10px;
+  background-color: #1c1c1c;
+}
+.page-invitation .inner .share-box .info strong {
+  color: #f56c6c;
+}
+.page-invitation .inner .share-box .invite-qrcode {
+  text-align: center;
+  margin: 15px;
+}
+.page-invitation .inner .share-box .invite-url {
+  margin: 10px;
+  padding: 8px;
+  display: flex;
+  justify-content: space-between;
+  border: 1px solid #444;
+  border-radius: 10px;
+  background-color: #1c1c1c;
+}
+.page-invitation .inner .share-box .invite-url span {
+  position: relative;
+  font-size: 14px;
+  font-family: 'Microsoft YaHei', '微软雅黑', Arial, sans-serif;
+  top: 0px;
+}
+.page-invitation .inner .invite-stats {
+  padding: 10px 10px;
+}
+.page-invitation .inner .invite-stats .item-box {
+  border-radius: 10px;
+  padding: 10 0px;
+}
+.page-invitation .inner .invite-stats .item-box .el-col {
+  height: 80px;
+  display: flex;
+  align-items: center;
+  justify-content: center;
+}
+.page-invitation .inner .invite-stats .item-box .el-col .iconfont {
+  font-size: 30px;
+}
+.page-invitation .inner .invite-stats .item-box .el-col .item-info {
+  font-size: 14px;
+}
+.page-invitation .inner .invite-stats .item-box .el-col .item-info .text,
+.page-invitation .inner .invite-stats .item-box .el-col .item-info .num {
+  padding: 3px 0;
+  text-align: center;
+}
+.page-invitation .inner .invite-stats .item-box .el-col .item-info .num {
+  font-size: 14px;
+}
+.page-invitation .inner .invite-stats .yellow {
+  background-color: #fec;
+  color: #d68f00;
+}
+.page-invitation .inner .invite-stats .blue {
+  background-color: #d6e4ff;
+  color: #1062fe;
+}
+.page-invitation .inner .invite-stats .green {
+  background-color: #e7f8eb;
+  color: #2d9f46;
+}

web/src/assets/css/mobile/invitation.styl

@@ -0,0 +1,94 @@
+.page-invitation {
+  display  flex 
+  justify-content  center 
+  background-color  #1c1c1c 
+  height  100vh 
+  overflow-x  hidden 
+  overflow-y  visible 
+  .inner {
+    max-width  100% 
+    width  100% 
+    color  #fff 
+    .center {
+      display  flex 
+      justify-content  center 
+      align-items  center 
+    }
+    .title {
+      color  #fff 
+      text-align  center 
+      font-size  16px 
+      font-weight  bold 
+    }
+    .share-box {
+      .info {
+        line-height  1.5 
+        border  1px solid #444444 
+        border-radius  10px 
+        padding  10px 
+        margin  10px 
+        background-color  #1c1c1c 
+        strong {
+          color  #f56c6c 
+        }
+      }
+      .invite-qrcode {
+        text-align  center 
+        margin  15px 
+      }
+      .invite-url {
+        margin  10px 
+        padding  8px 
+        display  flex 
+        justify-content  space-between 
+        border  1px solid #444444 
+        border-radius  10px 
+        background-color  #1c1c1c 
+        span {
+          position  relative 
+          font-size  14px 
+          font-family  'Microsoft YaHei', '微软雅黑', Arial, sans-serif 
+          top  0px 
+        }
+      }
+    }
+    .invite-stats {
+      padding  10px 10px 
+      .item-box {
+        border-radius  10px 
+        padding  10 0px 
+        .el-col {
+          height  80px 
+          display  flex 
+          align-items  center 
+          justify-content  center 
+          .iconfont {
+            font-size  30px 
+          }
+          .item-info {
+            font-size  14px 
+            .text, .num {
+              padding  3px 0 
+              text-align  center 
+            }
+            .num {
+              font-size  14px 
+            }
+          }
+        }
+      }
+      .yellow {
+        background-color  #ffeecc 
+        color  #D68F00 
+      }
+      .blue {
+        background-color  #D6E4FF 
+        color  #1062FE 
+      }
+      .green {
+        background-color  #E7F8EB 
+        color  #2D9F46 
+      }
+    }
+  }
+}

web/src/assets/css/mobile/profile.css

@@ -0,0 +1,142 @@
+.app-background {
+  background-color: #1c1c1c;
+  height: 100vh;
+}
+.member {
+  background-color: #1c1c1c;
+  height: 100vh;
+}
+.member .el-dialog .el-dialog__body {
+  padding-top: 5px;
+}
+.member .el-dialog .el-dialog__body .pay-container .count-down {
+  display: flex;
+  justify-content: center;
+}
+.member .el-dialog .el-dialog__body .pay-container .pay-qrcode {
+  display: flex;
+  justify-content: center;
+}
+.member .el-dialog .el-dialog__body .pay-container .pay-qrcode .el-image {
+  width: 280px;
+  height: 280px;
+}
+.member .el-dialog .el-dialog__body .pay-container .tip {
+  display: flex;
+  justify-content: center;
+}
+.member .el-dialog .el-dialog__body .pay-container .tip .el-icon {
+  font-size: 20px;
+}
+.member .el-dialog .el-dialog__body .pay-container .tip .text {
+  font-size: 16px;
+  margin-left: 10px;
+}
+.member .el-dialog .el-dialog__body .pay-container .tip.success {
+  color: #07c160;
+}
+.member .title {
+  text-align: center;
+  background-color: #1c1c1c;
+  font-size: 16px;
+  color: #fff;
+  padding: 5px;
+  font-weight: bold;
+}
+.member .inner {
+  color: #fff;
+  overflow-x: hidden;
+  overflow-y: visible;
+}
+.member .inner .user-profile {
+  padding: 0 20px 0 20px;
+  background-color: #1c1c1c;
+  color: #fff;
+  border-radius: 10px;
+  width: 100%;
+  height: 91vh;
+}
+.member .inner .user-profile .el-form-item__label {
+  color: #fff;
+  justify-content: start;
+}
+.member .inner .user-profile .user-opt .el-col {
+  padding: 8px;
+}
+.member .inner .user-profile .user-opt .el-col .el-button {
+  width: 100%;
+}
+.member .inner .product-box .info {
+  padding: 0 20px 0 20px;
+}
+.member .inner .product-box .info .el-alert__description {
+  font-size: 14px !important;
+  color: #1c1c1c;
+  margin: 0;
+}
+.member .inner .product-box .list-box {
+  padding: 0 20px 0 5px;
+}
+.member .inner .product-box .list-box .product-item {
+  border: 1px solid #5f5f5f;
+  border-radius: 6px;
+  overflow: hidden;
+  cursor: pointer;
+  transition: all 0.3s ease /* 添加过渡效果 */;
+}
+.member .inner .product-box .list-box .product-item .image-container {
+  display: flex;
+  justify-content: center;
+}
+.member .inner .product-box .list-box .product-item .image-container .el-image {
+  width: 50px;
+}
+.member .inner .product-box .list-box .product-item .image-container .el-image .el-image__inner {
+  border-radius: 10px;
+}
+.member .inner .product-box .list-box .product-item .product-title {
+  display: flex;
+  padding: 5px;
+}
+.member .inner .product-box .list-box .product-item .product-title .name {
+  width: 100%;
+  text-align: center;
+  font-size: 16px;
+  font-weight: bold;
+  color: #2cc995;
+}
+.member .inner .product-box .list-box .product-item .product-info {
+  padding: 10px 20px;
+  font-size: 14px;
+  color: #999;
+}
+.member .inner .product-box .list-box .product-item .product-info .info-line {
+  display: flex;
+  width: 100%;
+  padding: 2px 0;
+}
+.member .inner .product-box .list-box .product-item .product-info .info-line .label {
+  display: flex;
+  width: 100%;
+}
+.member .inner .product-box .list-box .product-item .product-info .info-line .price,
+.member .inner .product-box .list-box .product-item .product-info .info-line .expire {
+  display: flex;
+  width: 100%;
+  justify-content: right;
+}
+.member .inner .product-box .list-box .product-item .product-info .info-line .price {
+  color: #f56c6c;
+}
+.member .inner .product-box .list-box .product-item .product-info .info-line .expire {
+  color: #409eff;
+}
+.member .inner .product-box .list-box .product-item .product-info .pay-way {
+  padding: 5px 0;
+  display: flex;
+  justify-content: space-between;
+}
+.member .inner .product-box .list-box .product-item hover {
+  box-shadow: 0 0 10px rgba(71,255,241,0.6) /* 添加阴影效果 */;
+  transform: translateY(-10px) /* 向上移动10像素 */;
+}

web/src/assets/css/mobile/profile.styl

@@ -0,0 +1,149 @@
+.app-background {
+  background-color  #1c1c1c 
+  height  100vh 
+}
+.member {
+  background-color  #1c1c1c 
+  height 100vh 
+  .el-dialog {
+    .el-dialog__body {
+      padding-top 5px 
+      .pay-container {
+        .count-down {
+          display flex 
+          justify-content center 
+        }
+        .pay-qrcode {
+          display flex
+          justify-content center
+          .el-image {
+            width 280px 
+            height 280px 
+          }
+        }
+        .tip {
+          display flex
+          justify-content center
+          .el-icon {
+            font-size  20px 
+          }
+          .text {
+            font-size  16px 
+            margin-left  10px 
+          }
+        }
+        .tip.success {
+          color  #07c160 
+        }
+      }
+    }
+  }
+  .title {
+    text-align  center 
+    background-color  #1c1c1c 
+    font-size 16px 
+    color  #ffffff 
+    padding  5px 
+    font-weight bold
+  }
+  .inner {
+    color  #ffffff 
+    overflow-x hidden
+    overflow-y visible
+    .user-profile {
+      padding  0 20px 0 20px 
+      background-color  #1c1c1c 
+      color  #ffffff 
+      border-radius 10px 
+      width  100% 
+      height  91vh 
+      .el-form-item__label {
+        color  #ffffff 
+        justify-content start 
+      }
+      .user-opt {
+        .el-col {
+          padding  8px 
+          .el-button {
+            width  100% 
+          }
+        }
+      }
+    }
+    .product-box {
+      .info {
+        .el-alert__description {
+          font-size  14px !important 
+          color #1c1c1c
+          margin  0 
+        }
+        padding  0 20px 0 20px 
+      }
+      .list-box {
+        padding  0 20px 0 5px
+        .product-item {
+          border 1px solid #5f5f5f 
+          border-radius  6px 
+          overflow hidden
+          cursor pointer
+          transition  all 0.3s ease  /* 添加过渡效果 */
+          .image-container {
+            display flex
+            justify-content center
+            .el-image {
+              width  50px
+              .el-image__inner {
+                border-radius  10px 
+              }
+            }
+          }
+          .product-title {
+            display  flex 
+            padding  5px 
+            .name {
+              width  100% 
+              text-align  center 
+              font-size  16px 
+              font-weight  bold 
+              color  #2cc995 
+            }
+          }
+          .product-info {
+            padding  10px 20px 
+            font-size  14px 
+            color  #999999 
+            .info-line {
+              display  flex 
+              width  100% 
+              padding  2px 0 
+              .label {
+                display  flex 
+                width  100% 
+              }
+              .price, .expire {
+                display  flex 
+                width  100% 
+                justify-content  right 
+              }
+              .price {
+                color  #f56c6c 
+              }
+              .expire {
+                color  #409eff 
+              }
+            }
+            .pay-way {
+              padding  5px 0 
+              display  flex 
+              justify-content  space-between 
+            }
+          }
+          & hover {
+            box-shadow  0 0 10px rgba(71, 255, 241, 0.6)  /* 添加阴影效果 */
+            transform  translateY(-10px)  /* 向上移动10像素 */
+          }
+        }
+      }
+    }
+  }
+}

web/src/assets/css/mobile/sd-task-dialog.css

@@ -0,0 +1,72 @@
+.el-overlay-dialog .el-dialog {
+  background-color: #1a1b1e;
+}
+.el-overlay-dialog .el-dialog .el-dialog__header .el-dialog__title {
+  color: #f5f5f5;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body {
+  padding: 0 0 0 15px !important;
+  display: flex;
+  height: 100%;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row {
+  width: 100%;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container {
+  display: flex;
+  justify-content: center;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container .image-slot {
+  display: flex;
+  height: 100vh;
+  align-items: center;
+  justify-content: center;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .img-container .image-slot .el-icon {
+  font-size: 60px;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info {
+  background-color: #25262b;
+  padding: 1rem 1.5rem;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line {
+  width: 100%;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .prompt {
+  background-color: #35363b;
+  padding: 10px;
+  color: #999;
+  overflow: auto;
+  max-height: 100px;
+  min-height: 50px;
+  position: relative;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .prompt .el-icon {
+  position: absolute;
+  right: 10px;
+  bottom: 10px;
+  cursor: pointer;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper {
+  margin-top: 10px;
+  display: flex;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper label {
+  display: flex;
+  width: 100px;
+  color: #a5a5a5;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .info-line .wrapper .item-value {
+  display: flex;
+  width: 100%;
+  background-color: #35363b;
+  padding: 2px 5px;
+  border-radius: 5px;
+  color: #f5f5f5;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .copy-params {
+  padding: 30px 0 10px 0;
+}
+.el-overlay-dialog .el-dialog .el-dialog__body .el-row .task-info .copy-params .el-button {
+  width: 100%;
+}

web/src/assets/css/mobile/sd-task-dialog.styl

@@ -0,0 +1,96 @@
+.el-overlay-dialog {
+  .el-dialog {
+    background-color #1a1b1e
+
+    .el-dialog__header {
+      .el-dialog__title {
+        color #F5F5F5
+      }
+    }
+
+    .el-dialog__body {
+      padding 0 0 0 15px !important
+      display flex
+      height 100%
+
+      .el-row {
+        width 100%
+
+        .img-container {
+          display flex
+          justify-content center
+
+          .image-slot {
+            display flex
+            height 100vh
+            align-items center
+            justify-content center
+
+            .el-icon {
+              font-size 60px
+            }
+          }
+        }
+
+        .task-info {
+          background-color #25262b
+          padding 1rem 1.5rem
+
+          .info-line {
+            width 100%
+
+            .prompt {
+              background-color #35363b
+              padding 10px
+              color #999999
+              overflow auto
+              max-height 100px
+              min-height 50px
+
+              position relative
+
+              .el-icon {
+                position absolute
+                right 10px
+                bottom 10px
+                cursor pointer
+              }
+            }
+
+            .wrapper {
+              margin-top 10px
+              display flex
+
+              label {
+                display flex
+                width 100px
+                color #a5a5a5
+              }
+
+              .item-value {
+                display flex
+                width 100%
+                background-color #35363b
+                padding 2px 5px
+                border-radius 5px
+                color #F5F5F5
+              }
+            }
+
+          }
+
+          .copy-params {
+            padding 30px 0 10px 0
+
+            .el-button {
+              width 100%
+            }
+          }
+        }
+      }
+
+      // end el-row
+
+    }
+  }
+}

web/src/assets/css/mobile/task-list.css

@@ -0,0 +1,106 @@
+.task-list-box {
+  width: 100%;
+  padding: 5px;
+  color: #fff;
+  overflow-x: hidden;
+}
+.task-list-box .running-job-list .job-item {
+  width: 100%;
+  padding: 2px;
+  background-color: #555;
+}
+.task-list-box .running-job-list .job-item .job-item-inner {
+  position: relative;
+  height: 100%;
+  overflow: hidden;
+}
+.task-list-box .running-job-list .job-item .job-item-inner .progress {
+  position: absolute;
+  width: 100%;
+  height: 100%;
+  top: 0;
+  left: 0;
+  display: flex;
+  justify-content: center;
+  align-items: center;
+}
+.task-list-box .running-job-list .job-item .job-item-inner .progress span {
+  font-size: 20px;
+  color: #fff;
+}
+.task-list-box .finish-job-list .job-item {
+  width: 100%;
+  height: 100%;
+  border: 1px solid #666;
+  padding: 6px;
+  overflow: hidden;
+  border-radius: 6px;
+  transition: all 0.3s ease; /* 添加过渡效果 */
+}
+.task-list-box .finish-job-list .job-item .opt .opt-line {
+  margin: 6px 0;
+}
+.task-list-box .finish-job-list .job-item .opt .opt-line ul {
+  display: flex;
+  flex-flow: row;
+}
+.task-list-box .finish-job-list .job-item .opt .opt-line ul li {
+  margin-right: 6px;
+}
+.task-list-box .finish-job-list .job-item .opt .opt-line ul li a {
+  padding: 3px 0;
+  width: 40px;
+  text-align: center;
+  border-radius: 5px;
+  display: block;
+  cursor: pointer;
+  background-color: #4e5058;
+  color: #fff;
+}
+.task-list-box .finish-job-list .job-item .opt .opt-line ul li a:hover {
+  background-color: #6d6f78;
+}
+.task-list-box .finish-job-list .job-item .opt .opt-line ul .show-prompt {
+  font-size: 20px;
+  cursor: pointer;
+}
+.task-list-box .finish-job-list .animate:hover {
+  box-shadow: 0 0 10px rgba(71,255,241,0.6); /* 添加阴影效果 */
+  transform: translateY(-10px); /* 向上移动10像素 */
+}
+.task-list-box .el-image {
+  width: 100%;
+  height: 100%;
+  overflow: visible;
+}
+.task-list-box .el-image img {
+  height: 240px;
+}
+.task-list-box .el-image .el-image-viewer__wrapper img {
+  width: auto;
+  height: auto;
+}
+.task-list-box .el-image .image-slot {
+  display: flex;
+  flex-flow: column;
+  justify-content: center;
+  align-items: center;
+  height: 100%;
+  min-height: 200px;
+  color: #fff;
+  height: 240px;
+}
+.task-list-box .el-image .image-slot .iconfont {
+  font-size: 50px;
+  margin-bottom: 10px;
+}
+.task-list-box .el-image.upscale {
+  max-height: 310px;
+}
+.task-list-box .el-image.upscale img {
+  height: 310px;
+}
+.task-list-box .el-image.upscale .el-image-viewer__wrapper img {
+  width: auto;
+  height: auto;
+}

web/src/assets/css/mobile/task-list.styl

@@ -0,0 +1,142 @@
+.task-list-box {
+  width 100%
+  padding 5px
+  color #ffffff
+  overflow-x hidden
+
+
+  .running-job-list {
+    .job-item {
+      //border: 1px solid #454545;
+      width: 100%;
+      padding 2px
+      background-color #555555
+
+      .job-item-inner {
+        position relative
+        height 100%
+        overflow hidden
+
+        .progress {
+          position absolute
+          width 100%
+          height 100%
+          top 0
+          left 0
+          display flex
+          justify-content center
+          align-items center
+
+          span {
+            font-size 20px
+            color #ffffff
+          }
+        }
+      }
+    }
+  }
+
+
+  .finish-job-list {
+    .job-item {
+      width 100%
+      height 100%
+      border 1px solid #666666
+      padding 6px
+      overflow hidden
+      border-radius 6px
+      transition: all 0.3s ease; /* 添加过渡效果 */
+
+      .opt {
+        .opt-line {
+          margin 6px 0
+
+          ul {
+            display flex
+            flex-flow row
+
+            li {
+              margin-right 6px
+
+              a {
+                padding 3px 0
+                width 40px
+                text-align center
+                border-radius 5px
+                display block
+                cursor pointer
+                background-color #4E5058
+                color #ffffff
+
+                &:hover {
+                  background-color #6D6F78
+                }
+              }
+            }
+
+            .show-prompt {
+              font-size 20px
+              cursor pointer
+            }
+          }
+        }
+      }
+    }
+
+    .animate {
+      &:hover {
+        box-shadow: 0 0 10px rgba(71, 255, 241, 0.6); /* 添加阴影效果 */
+        transform: translateY(-10px); /* 向上移动10像素 */
+      }
+    }
+
+  }
+
+  .el-image {
+    width 100%
+    height 100%
+    overflow visible
+
+    img {
+      height 240px
+    }
+
+    .el-image-viewer__wrapper {
+      img {
+        width auto
+        height auto
+      }
+    }
+
+    .image-slot {
+      display flex
+      flex-flow column
+      justify-content center
+      align-items center
+      height 100%
+      min-height 200px
+      color #ffffff
+      height 240px
+
+      .iconfont {
+        font-size 50px
+        margin-bottom 10px
+      }
+    }
+  }
+
+  .el-image.upscale {
+    max-height 310px
+
+    img {
+      height 310px
+    }
+
+    .el-image-viewer__wrapper {
+      img {
+        width auto
+        height auto
+      }
+    }
+  }
+}

web/src/assets/iconfont/iconfont.css

@@ -1,8 +1,8 @@
 @font-face {
   font-family: "iconfont"; /* Project id 4125778 */
-  src: url('iconfont.woff2?t=1702024026523') format('woff2'),
-       url('iconfont.woff?t=1702024026523') format('woff'),
-       url('iconfont.ttf?t=1702024026523') format('truetype');
+  src: url('iconfont.woff2?t=1703124384910') format('woff2'),
+       url('iconfont.woff?t=1703124384910') format('woff'),
+       url('iconfont.ttf?t=1703124384910') format('truetype');
 }
 
 .iconfont {
@@ -13,6 +13,10 @@
   -moz-osx-font-smoothing: grayscale;
 }
 
+.icon-loading:before {
+  content: "\e627";
+}
+
 .icon-alipay:before {
   content: "\e634";
 }

web/src/assets/iconfont/iconfont.json

@@ -5,6 +5,13 @@
   "css_prefix_text": "icon-",
   "description": "",
   "glyphs": [
+    {
+      "icon_id": "1278349",
+      "name": "loading",
+      "font_class": "loading",
+      "unicode": "e627",
+      "unicode_decimal": 58919
+    },
     {
       "icon_id": "1486848",
       "name": "支付宝支付",

web/src/components/FooterBar.vue

@@ -1,16 +1,17 @@
 <template>
-    <div class="container">
-      <div class="footer">
-        Powered by {{author}} @
-        <el-link type="primary" href="https://github.com/yangjian102621/chatgpt-plus" target="_blank">{{title}}</el-link>
-      </div>
+  <div class="container">
+    <div class="footer">
+      Powered by {{ author }} @
+      <el-link type="primary" href="https://github.com/yangjian102621/chatgpt-plus" target="_blank">{{ title }}
+      </el-link>
     </div>
+  </div>
 </template>
 <script setup>
 
 import {ref} from "vue";
 
-const title = ref('ChatGPT-Plus-V3')
+const title = ref(process.env.VUE_APP_TITLE)
 const author = ref('极客学长')
 </script>
 
@@ -22,6 +23,7 @@ const author = ref('极客学长')
   width: 100%;
   display flex;
   justify-content center
+
   .footer {
     max-width 400px;
     text-align center;

web/src/components/admin/AdminHeader.vue

@@ -42,7 +42,7 @@
               <a href="https://github.com/yangjian102621/chatgpt-plus" target="_blank">
                 <el-dropdown-item>
                   <i class="iconfont icon-github"></i>
-                  <span>ChatGPT-Plus-V3</span>
+                  <span>{{ sysTitle }}</span>
                 </el-dropdown-item>
               </a>
               <el-dropdown-item @click="showDialog = true">
@@ -84,6 +84,7 @@ import {httpGet} from "@/utils/http";
 import {ElMessage} from "element-plus";
 
 const message = ref(5);
+const sysTitle = ref(process.env.VUE_APP_TITLE)
 const avatar = ref('/images/user-info.jpg')
 const donateImg = ref('/images/wechat-pay.png')
 const showDialog = ref(false)

web/src/components/admin/AdminSidebar.vue

@@ -111,6 +111,11 @@ const items = [
     index: '/admin/reward',
     title: '众筹管理',
   },
+  {
+    icon: 'control',
+    index: '/admin/functions',
+    title: '函数管理',
+  },
   {
     icon: 'config',
     index: '/admin/system',
@@ -121,29 +126,29 @@ const items = [
     index: '/admin/loginLog',
     title: '用户登录日志',
   },
-  {
-    icon: 'menu',
-    index: '1',
-    title: '常用模板页面',
-    subs: [
-      {
-        index: '/admin/demo/form',
-        title: '表单页面',
-      },
-      {
-        index: '/admin/demo/table',
-        title: '常用表格',
-      },
-      {
-        index: '/admin/demo/import',
-        title: '导入Excel',
-      },
-      {
-        index: '/admin/demo/editor',
-        title: '富文本编辑器',
-      },
-    ],
-  },
+  // {
+  //   icon: 'menu',
+  //   index: '1',
+  //   title: '常用模板页面',
+  //   subs: [
+  //     {
+  //       index: '/admin/demo/form',
+  //       title: '表单页面',
+  //     },
+  //     {
+  //       index: '/admin/demo/table',
+  //       title: '常用表格',
+  //     },
+  //     {
+  //       index: '/admin/demo/import',
+  //       title: '导入Excel',
+  //     },
+  //     {
+  //       index: '/admin/demo/editor',
+  //       title: '富文本编辑器',
+  //     },
+  //   ],
+  // },
 ];
 
 const route = useRoute();

web/src/components/mobile/ChatReply.vue

@@ -106,11 +106,17 @@ onMounted(() => {
           margin-top 0
         }
 
-        p > code {
-          color #2b2b2b
-          background-color #c1c1c1
-          padding 2px 5px
-          border-radius 5px
+        p {
+          code {
+            color #2b2b2b
+            background-color #c1c1c1
+            padding 2px 5px
+            border-radius 5px
+          }
+
+          img {
+            max-width 100%
+          }
         }
       }
     }

web/src/router.js

@@ -22,8 +22,8 @@ const routes = [
             },
             {
                 name: 'image-sd',
-                path: '/sd/',
-                meta: {title: 'Stable Diffusion 绘画中心'},
+                path: '/sd',
+                meta: {title: 'stable diffusion 绘画中心'},
                 component: () => import('@/views/ImageSd.vue'),
             },
             {
@@ -80,7 +80,7 @@ const routes = [
     {
         path: '/admin/login',
         name: 'admin-login',
-        meta: {title: 'Chat-Plus 控制台登录'},
+        meta: {title: 'ChatPuls 控制台登录'},
         component: () => import('@/views/admin/Login.vue'),
     },
     {
@@ -88,7 +88,7 @@ const routes = [
         path: '/admin',
         redirect: '/admin/dashboard',
         component: () => import("@/views/admin/Home.vue"),
-        meta: {title: 'ChatGPT-Plus 管理后台'},
+        meta: {title: 'ChatPuls 管理后台'},
         children: [
             {
                 path: '/admin/dashboard',
@@ -151,28 +151,10 @@ const routes = [
                 component: () => import('@/views/admin/LoginLog.vue'),
             },
             {
-                path: '/admin/demo/form',
-                name: 'admin-form',
-                meta: {title: '表单页面'},
-                component: () => import('@/views/admin/demo/Form.vue'),
-            },
-            {
-                path: '/admin/demo/table',
-                name: 'admin-table',
-                meta: {title: '数据列表'},
-                component: () => import('@/views/admin/demo/Table.vue'),
-            },
-            {
-                path: '/admin/demo/import',
-                name: 'admin-import',
-                meta: {title: '导入数据'},
-                component: () => import('@/views/admin/demo/Import.vue'),
-            },
-            {
-                path: '/admin/demo/editor',
-                name: 'admin-editor',
-                meta: {title: '富文本编辑器'},
-                component: () => import('@/views/admin/demo/Editor.vue'),
+                path: '/admin/functions',
+                name: 'admin-functions',
+                meta: {title: '函数管理'},
+                component: () => import('@/views/admin/Functions.vue'),
             },
         ]
     },
@@ -185,7 +167,7 @@ const routes = [
     {
         name: 'mobile',
         path: '/mobile',
-        meta: {title: 'ChatGPT-智能助手V3'},
+        meta: {title: 'ChatPuls-智能助手V3'},
         component: () => import('@/views/mobile/Home.vue'),
         redirect: '/mobile/chat/list',
         children: [
@@ -195,15 +177,25 @@ const routes = [
                 component: () => import('@/views/mobile/ChatList.vue'),
             },
             {
-                path: '/mobile/setting',
-                name: 'mobile-setting',
-                component: () => import('@/views/mobile/Setting.vue'),
+                path: '/mobile/imageSd',
+                name: 'mobile-imageSd',
+                component: () => import('@/views/mobile/ImageSd.vue'),
+            },
+            {
+                path: '/mobile/apps',
+                name: 'mobile-apps',
+                component: () => import('@/views/mobile/Apps.vue'),
             },
             {
                 path: '/mobile/profile',
                 name: 'mobile-profile',
                 component: () => import('@/views/mobile/Profile.vue'),
             },
+            {
+                path: '/mobile/invitation',
+                name: 'mobile-invitation',
+                component: () => import('@/views/mobile/Invitation.vue'),
+            },
         ]
     },
     {

web/src/views/ImageMj.vue

@@ -359,7 +359,7 @@
               <template #default="scope">
                 <div class="job-item">
                   <el-image
-                      :src="scope.item.type === 'upscale' ? scope.item['img_url'] + '?imageView2/1/w/240/h/300/q/75' : scope.item['img_url'] + '?imageView2/1/w/240/h/240/q/75'"
+                      :src="scope.item['thumb_url']"
                       :class="scope.item.type === 'upscale' ? 'upscale' : ''" :zoom-rate="1.2"
                       :preview-src-list="[scope.item['img_url']]" fit="cover" :initial-index="scope.index"
                       loading="lazy" v-if="scope.item.progress > 0">
@@ -370,7 +370,11 @@
                     </template>
 
                     <template #error>
-                      <div class="image-slot">
+                      <div class="image-slot" v-if="scope.item['img_url'] === ''">
+                        <i class="iconfont icon-loading"></i>
+                        <span>正在下载图片</span>
+                      </div>
+                      <div class="image-slot" v-else>
                         <el-icon>
                           <Picture/>
                         </el-icon>
@@ -417,6 +421,10 @@
                       </ul>
                     </div>
                   </div>
+
+                  <div class="remove">
+                    <el-button type="danger" :icon="Delete" @click="removeImage(scope.item)" circle/>
+                  </div>
                 </div>
               </template>
             </ItemList>
@@ -434,7 +442,7 @@
 <script setup>
 import {onMounted, ref} from "vue"
 import {
-  ChromeFilled,
+  ChromeFilled, Delete,
   DeleteFilled,
   DocumentCopy,
   InfoFilled,
@@ -444,7 +452,7 @@ import {
 } from "@element-plus/icons-vue";
 import Compressor from "compressorjs";
 import {httpGet, httpPost} from "@/utils/http";
-import {ElMessage, ElNotification} from "element-plus";
+import {ElMessage, ElMessageBox, ElNotification} from "element-plus";
 import ItemList from "@/components/ItemList.vue";
 import Clipboard from "clipboard";
 import {checkSession} from "@/action/session";
@@ -469,7 +477,8 @@ const rates = [
   {css: "size9-16", value: "9:16", text: "9:16", img: "/images/mj/rate_9_16.png"},
 ]
 const models = [
-  {text: "最新模式MJ-5.2", value: " --v 5.2", img: "/images/mj/mj-v5.2.png"},
+  {text: "写实模式MJ-6.0", value: " --v 6", img: "/images/mj/mj-v6.png"},
+  {text: "优质模式MJ-5.2", value: " --v 5.2", img: "/images/mj/mj-v5.2.png"},
   {text: "优质模式MJ-5.1", value: " --v 5.1", img: "/images/mj/mj-v5.1.jpg"},
   {text: "虚幻模式MJ-5", value: " --v 5", img: "/images/mj/mj-v5.jpg"},
   {text: "真实模式MJ-4", value: " --v 4", img: "/images/mj/mj-v4.jpg"},
@@ -481,6 +490,10 @@ const models = [
 ]
 
 const options = [
+  {
+    value: 0,
+    label: '默认'
+  },
   {
     value: 0.25,
     label: '普通'
@@ -507,7 +520,7 @@ const params = ref({
   prompt: "",
   neg_prompt: "",
   tile: false,
-  quality: 0.5
+  quality: 0
 })
 
 const activeName = ref('图生图')
@@ -519,6 +532,7 @@ const router = useRouter()
 const socket = ref(null)
 const imgCalls = ref(0)
 const loading = ref(false)
+const userId = ref(0)
 
 const rewritePrompt = () => {
   loading.value = true
@@ -542,12 +556,40 @@ const translatePrompt = () => {
   })
 }
 
+const connect = () => {
+  let host = process.env.VUE_APP_WS_HOST
+  if (host === '') {
+    if (location.protocol === 'https:') {
+      host = 'wss://' + location.host;
+    } else {
+      host = 'ws://' + location.host;
+    }
+  }
+  const _socket = new WebSocket(host + `/api/mj/client?user_id=${userId.value}`);
+  _socket.addEventListener('open', () => {
+    socket.value = _socket;
+  });
+
+  _socket.addEventListener('message', event => {
+    if (event.data instanceof Blob) {
+      fetchRunningJobs(userId.value)
+      fetchFinishJobs(userId.value)
+    }
+  });
+
+  _socket.addEventListener('close', () => {
+    connect()
+  });
+}
+
 onMounted(() => {
   checkSession().then(user => {
     imgCalls.value = user['img_calls']
+    userId.value = user.id
 
-    fetchRunningJobs(user.id)
-    fetchFinishJobs(user.id)
+    fetchRunningJobs(userId.value)
+    fetchFinishJobs(userId.value)
+    connect()
 
   }).catch(() => {
     router.push('/login')
@@ -575,14 +617,12 @@ const fetchRunningJobs = (userId) => {
           message: "任务ID：" + jobs[i]['task_id'],
           type: 'error',
         })
+        imgCalls.value += 1
         continue
       }
       _jobs.push(jobs[i])
     }
     runningJobs.value = _jobs
-
-    setTimeout(() => fetchRunningJobs(userId), 5000)
-
   }).catch(e => {
     ElMessage.error("获取任务失败：" + e.message)
   })
@@ -591,8 +631,19 @@ const fetchRunningJobs = (userId) => {
 const fetchFinishJobs = (userId) => {
   // 获取已完成的任务
   httpGet(`/api/mj/jobs?status=1&user_id=${userId}`).then(res => {
-    finishedJobs.value = res.data
-    setTimeout(() => fetchFinishJobs(userId), 5000)
+    const jobs = res.data
+    for (let i = 0; i < jobs.length; i++) {
+      if (jobs[i]['use_proxy']) {
+        jobs[i]['thumb_url'] = jobs[i]['img_url'] + '?x-oss-process=image/quality,q_60&format=webp'
+      } else {
+        if (jobs[i].type === 'upscale') {
+          jobs[i]['thumb_url'] = jobs[i]['img_url'] + '?imageView2/1/w/480/h/600/q/75'
+        } else {
+          jobs[i]['thumb_url'] = jobs[i]['img_url'] + '?imageView2/1/w/480/h/480/q/75'
+        }
+      }
+    }
+    finishedJobs.value = jobs
   }).catch(e => {
     ElMessage.error("获取任务失败：" + e.message)
   })
@@ -660,6 +711,8 @@ const variation = (index, item) => {
 const send = (url, index, item) => {
   httpPost(url, {
     index: index,
+    task_id: item.task_id,
+    channel_id: item.channel_id,
     message_id: item.message_id,
     message_hash: item.hash,
     session_id: getSessionId(),
@@ -672,6 +725,25 @@ const send = (url, index, item) => {
   })
 }
 
+const removeImage = (item) => {
+  ElMessageBox.confirm(
+      '此操作将会删除任务和图片，继续操作码?',
+      '删除提示',
+      {
+        confirmButtonText: '确认',
+        cancelButtonText: '取消',
+        type: 'warning',
+      }
+  ).then(() => {
+    httpPost("/api/mj/remove", {id: item.id, img_url: item.img_url, user_id: userId.value}).then(() => {
+      ElMessage.success("任务删除成功")
+    }).catch(e => {
+      ElMessage.error("任务删除失败：" + e.message)
+    })
+  }).catch(() => {
+  })
+}
+
 </script>
 
 <style lang="stylus">

web/src/views/ImageSd.vue

@@ -309,7 +309,7 @@
         <div class="task-list-inner" :style="{ height: listBoxHeight + 'px' }">
           <h2>任务列表</h2>
           <div class="running-job-list">
-            <ItemList :items="runningJobs" v-if="runningJobs.length > 0" width="240">
+            <ItemList :items="runningJobs" v-if="runningJobs.length > 0" :width="240">
               <template #default="scope">
                 <div class="job-item">
                   <div v-if="scope.item.progress > 0" class="job-item-inner">
@@ -350,7 +350,7 @@
           </div>
           <h2>创作记录</h2>
           <div class="finish-job-list">
-            <ItemList :items="finishedJobs" v-if="finishedJobs.length > 0" width="240" :gap="16">
+            <ItemList :items="finishedJobs" v-if="finishedJobs.length > 0" :width="240" :gap="16">
               <template #default="scope">
                 <div class="job-item animate" @click="showTask(scope.item)">
                   <el-image
@@ -371,6 +371,10 @@
                       </div>
                     </template>
                   </el-image>
+
+                  <div class="remove">
+                    <el-button type="danger" :icon="Delete" @click="removeImage($event,scope.item)" circle/>
+                  </div>
                 </div>
               </template>
             </ItemList>
@@ -498,9 +502,9 @@
 
 <script setup>
 import {onMounted, ref} from "vue"
-import {DocumentCopy, InfoFilled, Orange, Picture, Refresh} from "@element-plus/icons-vue";
+import {Delete, DocumentCopy, InfoFilled, Orange, Picture, Refresh} from "@element-plus/icons-vue";
 import {httpGet, httpPost} from "@/utils/http";
-import {ElMessage, ElNotification} from "element-plus";
+import {ElMessage, ElMessageBox, ElNotification} from "element-plus";
 import ItemList from "@/components/ItemList.vue";
 import Clipboard from "clipboard";
 import {checkSession} from "@/action/session";
@@ -518,8 +522,7 @@ window.onresize = () => {
   listBoxHeight.value = window.innerHeight - 40
   mjBoxHeight.value = window.innerHeight - 150
 }
-const samplers = ["Euler a", "Euler", "DPM2 a Karras", "DPM++ 2S a Karras", "DPM++ 2M Karras", "DPM++ SDE Karras", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM fast", "DPM adaptive",
-  "LMS Karras", "DPM2 Karras", "DDIM", "PLMS", "UniPC", "LMS", "Heun",]
+const samplers = ["Euler a", "Euler", "DPM++ 2S a Karras", "DPM++ 2M Karras", "DPM++ SDE Karras", "DPM++ 2M SDE Karras"]
 const scaleAlg = ["Latent", "ESRGAN_4x", "R-ESRGAN 4x+", "SwinIR_4x", "LDSR"]
 const params = ref({
   width: 1024,
@@ -530,10 +533,10 @@ const params = ref({
   cfg_scale: 7,
   face_fix: false,
   hd_fix: false,
-  hd_redraw_rate: 0.7,
+  hd_redraw_rate: 0.5,
   hd_scale: 2,
   hd_scale_alg: scaleAlg[0],
-  hd_steps: 10,
+  hd_steps: 15,
   prompt: "",
   negative_prompt: "nsfw, paintings,low quality,easynegative,ng_deepnegative ,lowres,bad anatomy,bad hands,bad feet",
 })
@@ -593,25 +596,45 @@ onMounted(() => {
             message: "任务ID：" + jobs[i]['task_id'],
             type: 'error',
           })
+          imgCalls.value += 1
           continue
         }
         _jobs.push(jobs[i])
       }
       runningJobs.value = _jobs
 
-      setTimeout(() => fetchRunningJobs(userId), 5000)
+      setTimeout(() => fetchRunningJobs(userId), 1000)
     }).catch(e => {
       ElMessage.error("获取任务失败：" + e.message)
+      setTimeout(() => fetchRunningJobs(userId), 5000)
     })
   }
 
   // 获取已完成的任务
   const fetchFinishJobs = (userId) => {
     httpGet(`/api/sd/jobs?status=1&user_id=${userId}`).then(res => {
-      finishedJobs.value = res.data
-      setTimeout(() => fetchFinishJobs(userId), 5000)
+      if (finishedJobs.value.length === 0 || res.data.length > finishedJobs.value.length) {
+        finishedJobs.value = res.data
+        setTimeout(() => fetchFinishJobs(userId), 1000)
+        return
+      }
+
+      // check if the img url is changed
+      const list = res.data
+      let changed = false
+      for (let i = 0; i < list.length; i++) {
+        if (list[i]["img_url"] !== finishedJobs.value[i]["img_url"]) {
+          changed = true
+          break
+        }
+      }
+      if (changed) {
+        finishedJobs.value = list
+      }
+      setTimeout(() => fetchFinishJobs(userId), 1000)
     }).catch(e => {
       ElMessage.error("获取任务失败：" + e.message)
+      setTimeout(() => fetchFinishJobs(userId), 5000)
     })
   }
 
@@ -655,6 +678,26 @@ const copyParams = (row) => {
   showTaskDialog.value = false
 }
 
+const removeImage = (event, item) => {
+  event.stopPropagation()
+  ElMessageBox.confirm(
+      '此操作将会删除任务和图片，继续操作码?',
+      '删除提示',
+      {
+        confirmButtonText: '确认',
+        cancelButtonText: '取消',
+        type: 'warning',
+      }
+  ).then(() => {
+    httpPost("/api/sd/remove", {id: item.id, img_url: item.img_url}).then(() => {
+      ElMessage.success("任务删除成功")
+    }).catch(e => {
+      ElMessage.error("任务删除失败：" + e.message)
+    })
+  }).catch(() => {
+  })
+}
+
 </script>
 
 <style lang="stylus">

web/src/views/ImagesWall.vue

@@ -11,7 +11,10 @@
         </div>
       </div>
       <div class="waterfall" :style="{ height:listBoxHeight + 'px' }" id="waterfall-box">
-        <v3-waterfall id="waterfall" :list="list" srcKey="img_thumb"
+        <v3-waterfall v-if="imgType === 'mj'"
+                      id="waterfall"
+                      :list="list"
+                      srcKey="img_thumb"
                       :gap="12"
                       :bottomGap="-5"
                       :colWidth="colWidth"
@@ -21,7 +24,7 @@
                       @scrollReachBottom="getNext">
           <template #default="slotProp">
             <div class="list-item">
-              <div class="image" v-if="imgType === 'mj'">
+              <div class="image">
                 <el-image :src="slotProp.item['img_thumb']"
                           :zoom-rate="1.2"
                           :preview-src-list="[slotProp.item['img_url']]"
@@ -43,7 +46,30 @@
                   </template>
                 </el-image>
               </div>
-              <div class="image" v-else>
+              <div class="prompt">
+                <span>{{ slotProp.item.prompt }}</span>
+                <el-icon class="copy-prompt" :data-clipboard-text="slotProp.item.prompt">
+                  <DocumentCopy/>
+                </el-icon>
+              </div>
+            </div>
+          </template>
+        </v3-waterfall>
+
+        <v3-waterfall v-else
+                      id="waterfall"
+                      :list="list"
+                      srcKey="img_thumb"
+                      :gap="12"
+                      :bottomGap="-5"
+                      :colWidth="colWidth"
+                      :distanceToScroll="100"
+                      :isLoading="loading"
+                      :isOver="false"
+                      @scrollReachBottom="getNext">
+          <template #default="slotProp">
+            <div class="list-item">
+              <div class="image">
                 <el-image :src="slotProp.item['img_thumb']" loading="lazy"
                           @click="showTask(slotProp.item)">
                   <template #placeholder>

web/src/views/Login.vue

@@ -64,7 +64,7 @@ import {prevRoute} from "@/router";
 import ResetPass from "@/components/ResetPass.vue";
 
 const router = useRouter();
-const title = ref('ChatGPT-PLUS 用户登录');
+const title = ref('ChatPlus 用户登录');
 const username = ref(process.env.VUE_APP_USER);
 const password = ref(process.env.VUE_APP_PASS);
 const showResetPass = ref(false)

web/src/views/Register.vue

@@ -128,7 +128,7 @@ import {setUserToken} from "@/store/session";
 import {checkSession} from "@/action/session";
 
 const router = useRouter();
-const title = ref('ChatGPT-PLUS 用户注册');
+const title = ref('ChatPlus 用户注册');
 const formData = ref({
   mobile: '',
   password: '',
@@ -151,7 +151,7 @@ httpGet("/api/admin/config/get?key=system").then(res => {
         dangerouslyUseHTMLString: true,
         message: '当前系统开启了强制邀请注册功能，必须有邀请码才能注册哦。扫描下面二维码获取邀请码。<br/> <img alt="qrcode" src="/images/wx.png" />',
         type: 'info',
-        duration: 0,
+        duration: 5000,
       })
     }
   }

web/src/views/admin/ApiKey.vue

@@ -32,7 +32,7 @@
         <el-table-column label="操作" width="180">
           <template #default="scope">
             <el-button size="small" type="primary" @click="edit(scope.row)">编辑</el-button>
-            <el-popconfirm title="确定要删除当前记录吗?" @confirm="remove(scope.row)">
+            <el-popconfirm title="确定要删除当前记录吗?" @confirm="remove(scope.row)" :width="200">
               <template #reference>
                 <el-button size="small" type="danger">删除</el-button>
               </template>

web/src/views/admin/ChatModel.vue

@@ -35,7 +35,7 @@
         <el-table-column label="操作" width="180">
           <template #default="scope">
             <el-button size="small" type="primary" @click="edit(scope.row)">编辑</el-button>
-            <el-popconfirm title="确定要删除当前记录吗?" @confirm="remove(scope.row)">
+            <el-popconfirm title="确定要删除当前记录吗?" @confirm="remove(scope.row)" :width="200">
               <template #reference>
                 <el-button size="small" type="danger">删除</el-button>
               </template>
@@ -232,7 +232,7 @@ const remove = function (row) {
 </script>
 
 <style lang="stylus" scoped>
-@import "@/assets/css/admin-form.styl"
+@import "@/assets/css/admin/form.styl"
 .list {
 
   .opt-box {

web/src/views/admin/Functions.vue

@@ -0,0 +1,335 @@
+<template>
+  <div class="container role-list" v-loading="loading">
+    <div class="handle-box">
+      <el-button type="primary" :icon="Plus" @click="addRow">新增</el-button>
+    </div>
+    <el-row>
+      <el-table :data="tableData" :border="parentBorder" style="width: 100%">
+        <el-table-column label="函数名称" prop="name">
+          <template #default="scope">
+            <span class="sort" :data-id="scope.row.id">{{ scope.row.name }}</span>
+          </template>
+        </el-table-column>
+        <el-table-column label="函数别名" prop="label"/>
+        <el-table-column label="功能描述" prop="description"/>
+        <el-table-column label="启用状态">
+          <template #default="scope">
+            <el-switch v-model="scope.row.enabled" @change="functionSet('enabled',scope.row)"/>
+          </template>
+        </el-table-column>
+
+        <el-table-column label="操作" width="150" align="right">
+          <template #default="scope">
+            <el-button size="small" type="primary" @click="rowEdit(scope.$index, scope.row)">编辑</el-button>
+            <el-popconfirm title="确定要删除当前函数吗?" @confirm="remove(scope.row)" :width="200">
+              <template #reference>
+                <el-button size="small" type="danger">删除</el-button>
+              </template>
+            </el-popconfirm>
+          </template>
+        </el-table-column>
+      </el-table>
+    </el-row>
+
+    <el-dialog
+        v-model="showDialog"
+        :title="title"
+        width="50%"
+    >
+      <el-form :model="item" label-width="120px" ref="formRef" label-position="left" :rules="rules">
+        <el-form-item label="函数名称：" prop="name">
+          <el-input
+              v-model="item.name"
+              placeholder="函数名称最好为英文"
+              autocomplete="off"
+          />
+        </el-form-item>
+
+        <el-form-item label="函数标签：" prop="label">
+          <el-input
+              v-model="item.label"
+              placeholder="函数的中文名称"
+              autocomplete="off"
+          />
+        </el-form-item>
+
+        <el-form-item label="功能描述：" prop="description">
+          <el-input
+              v-model="item.description"
+              autocomplete="off"
+          />
+        </el-form-item>
+
+        <el-form-item label="函数参数：" prop="parameters">
+          <template #default>
+            <el-table :data="params" :border="childBorder" size="small">
+              <el-table-column label="参数名称" width="120">
+                <template #default="scope">
+                  <el-input
+                      v-model="scope.row.name"
+                      autocomplete="off"
+                  />
+                </template>
+              </el-table-column>
+              <el-table-column label="参数类型" width="120">
+                <template #default="scope">
+                  <el-select v-model="scope.row.type" placeholder="参数类型">
+                    <el-option v-for="pt in paramsType" :value="pt" :key="pt">{{ pt }}</el-option>
+                  </el-select>
+                </template>
+              </el-table-column>
+              <el-table-column label="参数描述">
+                <template #default="scope">
+                  <el-input
+                      v-model="scope.row.desc"
+                      autocomplete="off"
+                  />
+                </template>
+              </el-table-column>
+
+              <el-table-column label="必填参数" width="80">
+                <template #default="scope">
+                  <div class="param-opt">
+                    <el-checkbox v-model="scope.row.required"/>
+                  </div>
+                </template>
+              </el-table-column>
+
+              <el-table-column label="操作" width="80">
+                <template #default="scope">
+                  <div class="param-opt">
+                    <el-button type="danger" :icon="Delete" circle @click="removeParam(scope.$index)" size="small"/>
+                  </div>
+                </template>
+              </el-table-column>
+            </el-table>
+
+            <div class="param-line">
+              <el-button type="primary" @click="addParam" size="small">
+                <el-icon>
+                  <Plus/>
+                </el-icon>
+                增加参数
+              </el-button>
+            </div>
+          </template>
+        </el-form-item>
+
+        <el-form-item label="API 地址：" prop="action">
+          <el-input
+              v-model="item.action"
+              autocomplete="off"
+              placeholder="该函数实现的API地址，可以是第三方服务API"
+          />
+        </el-form-item>
+
+        <el-form-item label="API Token：" prop="token">
+          <el-input
+              v-model="item.token"
+              autocomplete="off"
+              placeholder="API授权Token"
+          >
+            <template #append>
+              <el-tooltip
+                  class="box-item"
+                  effect="dark"
+                  content="只有本地服务才可以使用自动生成Token<br/>第三方服务请填写第三方服务API Token"
+                  placement="top-end"
+                  raw-content
+              >
+                <el-button @click="generateToken">生成Token</el-button>
+              </el-tooltip>
+            </template>
+          </el-input>
+        </el-form-item>
+        <el-form-item label="启用状态">
+          <el-switch v-model="item.enabled"/>
+        </el-form-item>
+      </el-form>
+
+      <template #footer>
+            <span class="dialog-footer">
+              <el-button @click="showDialog = false">取消</el-button>
+              <el-button type="primary" @click="save">保存</el-button>
+            </span>
+      </template>
+    </el-dialog>
+  </div>
+</template>
+
+<script setup>
+
+import {Delete, Plus} from "@element-plus/icons-vue";
+import {onMounted, reactive, ref} from "vue";
+import {httpGet, httpPost} from "@/utils/http";
+import {ElMessage} from "element-plus";
+import {arrayContains, copyObj} from "@/utils/libs";
+
+const showDialog = ref(false)
+const parentBorder = ref(true)
+const childBorder = ref(true)
+const tableData = ref([])
+const item = ref({})
+const params = ref([])
+const formRef = ref(null)
+const loading = ref(true)
+const title = ref("新增函数")
+
+const rules = reactive({
+  name: [{required: true, message: '请输入函数名称', trigger: 'blur',}],
+  label: [{required: true, message: '请输入函数标签', trigger: 'blur',}],
+  description: [{required: true, message: '请输入函数功能描述', trigger: 'blur',}],
+})
+const paramsType = ref(["string", "number"])
+
+
+onMounted(() => {
+  fetch()
+})
+
+const fetch = () => {
+  httpGet('/api/admin/function/list').then((res) => {
+    if (res.data) {
+      tableData.value = res.data
+    }
+    loading.value = false
+  }).catch(() => {
+    ElMessage.error("获取数据失败");
+  })
+}
+
+// 编辑
+const curIndex = ref(0)
+const rowEdit = function (index, row) {
+  curIndex.value = index
+  item.value = copyObj(row)
+  // initialize parameters
+  const props = item.value?.parameters?.properties
+  const required = item.value?.parameters?.required
+  const _params = []
+  for (let key in props) {
+    _params.push({
+      name: key,
+      type: props[key].type,
+      desc: props[key].description,
+      required: arrayContains(required, key)
+    })
+  }
+  params.value = _params
+  showDialog.value = true
+}
+
+const addRow = function () {
+  item.value = {enabled: true}
+  params.value = []
+  showDialog.value = true
+}
+
+const save = function () {
+  formRef.value.validate((valid) => {
+    if (valid) {
+      showDialog.value = false
+      const properties = {}
+      const required = []
+      // process params
+      for (let i = 0; i < params.value.length; i++) {
+        properties[params.value[i].name] = {"type": params.value[i].type, "description": params.value[i].desc}
+        if (params.value[i].required) {
+          required.push(params.value[i].name)
+        }
+      }
+      item.value.parameters = {type: "object", "properties": properties, required: required}
+      httpPost('/api/admin/function/save', item.value).then((res) => {
+        ElMessage.success('操作成功')
+        console.log(res.data)
+        if (item.value.id > 0) {
+          tableData.value[curIndex.value] = item.value
+        } else {
+          tableData.value.push(res.data)
+        }
+      }).catch((e) => {
+        ElMessage.error('操作失败，' + e.message)
+      })
+    }
+  })
+}
+
+const remove = function (row) {
+  httpGet('/api/admin/function/remove?id=' + row.id).then(() => {
+    ElMessage.success("删除成功！")
+    fetch()
+  }).catch(() => {
+    ElMessage.error("删除失败！")
+  })
+}
+
+const addParam = function () {
+  if (!params.value) {
+    item.value = []
+  }
+  params.value.push({name: "", type: "string", desc: "", required: false})
+}
+
+const removeParam = function (index) {
+  params.value.splice(index, 1);
+}
+
+const functionSet = (filed, row) => {
+  httpPost('/api/admin/function/set', {id: row.id, filed: filed, value: row[filed]}).then(() => {
+    ElMessage.success("操作成功！")
+  }).catch(e => {
+    ElMessage.error("操作失败：" + e.message)
+  })
+}
+
+const generateToken = () => {
+  httpGet('/api/admin/function/token').then(res => {
+    item.value.token = res.data
+  }).catch(() => {
+    ElMessage.error("生成 Token 失败")
+  })
+}
+
+</script>
+
+<style lang="stylus" scoped>
+.role-list {
+  .opt-box {
+    padding-bottom: 10px;
+    display flex;
+    justify-content flex-end
+
+    .el-icon {
+      margin-right 5px;
+    }
+  }
+
+  .param-line {
+    padding 5px 0
+
+    .el-icon {
+      margin-right 5px;
+    }
+  }
+
+  .param-opt {
+    display flex
+    justify-content center
+
+    .el-icon {
+      font-size: 20px;
+      margin-top 5px;
+      margin-left 5px;
+      cursor pointer
+    }
+  }
+
+  .el-input--small {
+    width 30px;
+
+    .el-input__inner {
+      text-align center
+    }
+  }
+}
+</style>

web/src/views/admin/Product.vue

@@ -38,7 +38,7 @@
         <el-table-column label="操作" width="180">
           <template #default="scope">
             <el-button size="small" type="primary" @click="edit(scope.row)">编辑</el-button>
-            <el-popconfirm title="确定要删除当前记录吗?" @confirm="remove(scope.row)">
+            <el-popconfirm title="确定要删除当前记录吗?" @confirm="remove(scope.row)" :width="200">
               <template #reference>
                 <el-button size="small" type="danger">删除</el-button>
               </template>

web/src/views/admin/RoleList.vue

@@ -36,7 +36,7 @@
         <el-table-column label="操作" width="150" align="right">
           <template #default="scope">
             <el-button size="small" type="primary" @click="rowEdit(scope.$index, scope.row)">编辑</el-button>
-            <el-popconfirm title="确定要删除当前角色吗?" @confirm="removeRole(scope.row)">
+            <el-popconfirm title="确定要删除当前角色吗?" @confirm="removeRole(scope.row)" :width="200">
               <template #reference>
                 <el-button size="small" type="danger">删除</el-button>
               </template>

web/src/views/admin/SysConfig.vue

@@ -56,19 +56,6 @@
             </el-icon>
           </el-tooltip>
         </el-form-item>
-        <el-form-item label="启用函数功能" prop="enabled_function">
-          <el-switch v-model="system['enabled_function']"/>
-          <el-tooltip
-              effect="dark"
-              content="是否在AI对话时启用函数功能"
-              raw-content
-              placement="right"
-          >
-            <el-icon>
-              <InfoFilled/>
-            </el-icon>
-          </el-tooltip>
-        </el-form-item>
 
         <el-form-item label="启用众筹功能" prop="enabled_reward">
           <el-switch v-model="system['enabled_reward']"/>
@@ -373,7 +360,7 @@ const uploadRewardImg = (file) => {
 </script>
 
 <style lang="stylus" scoped>
-@import "@/assets/css/admin-form.styl"
+@import "@/assets/css/admin/form.styl"
 .system-config {
   display flex
   justify-content center

web/src/views/mobile/Apps.vue

@@ -0,0 +1,15 @@
+<template>
+  <div class="app-background">
+    <div class="mobile-setting container">
+      <div class="title" style="padding-top: 50px">应用广场</div>
+      <div class="title" style="padding-top: 200px">视频生成器即将开启……敬请期待</div>
+    </div>
+  </div>
+</template>
+
+<script setup>
+</script>
+
+<style lang="stylus">
+@import "@/assets/css/mobile/apps.styl"
+</style>

web/src/views/mobile/ChatList.vue

@@ -1,49 +1,54 @@
 <template>
-  <div v-if="isLogin" class="container mobile-chat-list">
-    <van-nav-bar
-        :title="title"
-        left-text="新建会话"
-        @click-left="showPicker = true">
-      <template #right>
-        <van-icon name="delete-o" @click="clearAllChatHistory"/>
-      </template>
-    </van-nav-bar>
-
-    <div class="content">
-      <van-search
-          v-model="chatName"
-          input-align="center"
-          placeholder="请输入会话标题"
-          @input="search"
-      />
-
-      <van-list
-          v-model:error="error"
-          v-model:loading="loading"
-          :finished="finished"
-          error-text="请求失败，点击重新加载"
-          finished-text="没有更多了"
-          @load="onLoad"
+  <div class="app-background">
+    <div v-if="isLogin" class="container mobile-chat-list">
+      <van-nav-bar
+          :title="title"
+          left-text="新建会话"
+          @click-left="showPicker = true"
+          custom-class="navbar"
       >
-        <van-swipe-cell v-for="item in chats" :key="item.id">
-          <van-cell @click="changeChat(item)">
-            <div class="chat-list-item">
-              <van-image
-                  :src="item.icon"
-                  round
-              />
-              <div class="van-ellipsis">{{ item.title }}</div>
-            </div>
-          </van-cell>
-          <template #right>
-            <van-button square text="修改" type="primary" @click="editChat(item)"/>
-            <van-button square text="删除" type="danger" @click="removeChat(item)"/>
-          </template>
-        </van-swipe-cell>
-      </van-list>
+        <template #right>
+          <van-icon name="delete-o" @click="clearAllChatHistory"/>
+        </template>
+      </van-nav-bar>
+
+      <div class="content">
+        <van-search
+            v-model="chatName"
+            input-align="center"
+            placeholder="请输入会话标题"
+            custom-class="van-search"
+            @input="search"
+        />
+
+        <van-list
+            v-model:error="error"
+            v-model:loading="loading"
+            :finished="finished"
+            error-text="请求失败，点击重新加载"
+            finished-text="没有更多了"
+            @load="onLoad"
+        >
+          <van-swipe-cell v-for="item in chats" :key="item.id">
+            <van-cell @click="changeChat(item)">
+              <div class="chat-list-item">
+                <van-image
+                    :src="item.icon"
+                    round
+                />
+                <div class="van-ellipsis">{{ item.title }}</div>
+              </div>
+            </van-cell>
+            <template #right>
+              <van-button square text="修改" type="primary" @click="editChat(item)"/>
+              <van-button square text="删除" type="danger" @click="removeChat(item)"/>
+            </template>
+          </van-swipe-cell>
+        </van-list>
+      </div>
     </div>
 
-    <van-popup v-model:show="showPicker" position="bottom">
+    <van-popup v-model:show="showPicker" position="bottom" class="popup">
       <van-picker
           :columns="columns"
           title="选择模型和角色"
@@ -64,8 +69,8 @@
       </van-picker>
     </van-popup>
 
-    <van-dialog v-model:show="showEditChat" title="修改对话标题" show-cancel-button @confirm="saveTitle">
-      <van-field v-model="tmpChatTitle" label="" placeholder="请输入对话标题"/>
+    <van-dialog v-model:show="showEditChat" title="修改对话标题" show-cancel-button class="dialog" @confirm="saveTitle">
+      <van-field v-model="tmpChatTitle" label="" placeholder="请输入对话标题" class="field"/>
     </van-dialog>
 
     <bind-mobile v-if="isLogin" :show="showBindMobileDialog" :mobile="loginUser.mobile"
@@ -260,49 +265,5 @@ const removeChat = (item) => {
 </script>
 
 <style lang="stylus" scoped>
-$fontSize = 16px;
-.mobile-chat-list {
-
-  .content {
-    .van-cell__value {
-      .chat-list-item {
-        display flex
-        font-size $fontSize
-
-        .van-image {
-          min-width 32px
-          width 32px
-          height 32px
-        }
-
-        .van-ellipsis {
-          margin-top 5px;
-          margin-left 10px;
-        }
-      }
-    }
-  }
-
-  .van-picker-column {
-    .picker-option {
-      display flex
-      width 100%
-      padding 0 10px
-
-      .van-image {
-        width 20px;
-        height 20px;
-        margin-right 5px
-      }
-    }
-  }
-
-  .van-nav-bar {
-    .van-nav-bar__right {
-      .van-icon {
-        font-size 20px;
-      }
-    }
-  }
-}
+@import "@/assets/css/mobile/chat-list.styl"
 </style>

web/src/views/mobile/ChatSession.vue

@@ -1,88 +1,93 @@
 <template>
-  <van-config-provider :theme="getMobileTheme()">
-    <div class="mobile-chat" v-loading="loading" element-loading-text="正在连接会话...">
-      <van-sticky ref="navBarRef" :offset-top="0" position="top">
-        <van-nav-bar left-arrow left-text="返回" @click-left="router.back()">
-          <template #title>
-            <van-dropdown-menu>
-              <van-dropdown-item :title="title">
-                <van-cell center title="角色"> {{ role.name }}</van-cell>
-                <van-cell center title="模型">{{ modelValue }}</van-cell>
-              </van-dropdown-item>
-            </van-dropdown-menu>
-          </template>
+  <div class="app-background">
+    <van-config-provider theme="dark">
+      <div class="mobile-chat" v-loading="loading" element-loading-text="正在连接会话...">
+        <van-sticky ref="navBarRef" :offset-top="0" position="top">
+          <van-nav-bar left-arrow left-text="返回" @click-left="router.back()">
+            <template #title>
+              <van-dropdown-menu>
+                <van-dropdown-item :title="title">
+                  <van-cell center title="角色"> {{ role.name }}</van-cell>
+                  <van-cell center title="模型">{{ modelValue }}</van-cell>
+                </van-dropdown-item>
+              </van-dropdown-menu>
+            </template>
 
-          <template #right>
-            <van-icon name="share-o" @click="showShare = true"/>
-          </template>
-        </van-nav-bar>
-      </van-sticky>
+            <template #right>
+              <van-icon name="share-o" @click="showShare = true"/>
+            </template>
 
-      <van-share-sheet
-          v-model:show="showShare"
-          title="立即分享给好友"
-          :options="shareOptions"
-          @select="shareChat"
-      />
+          </van-nav-bar>
+        </van-sticky>
 
-      <div id="message-list-box" :style="{height: winHeight+'px'}" class="message-list-box">
-        <van-list
-            v-model:error="error"
-            :finished="finished"
-            error-text="请求失败，点击重新加载"
-            @load="onLoad"
-        >
-          <van-cell v-for="item in chatData" :key="item" :border="false" class="message-line">
-            <chat-prompt
-                v-if="item.type==='prompt'"
-                :content="item.content"
-                :created-at="dateFormat(item['created_at'])"
-                :icon="item.icon"
-                :model="model"
-                :tokens="item['tokens']"/>
-            <chat-reply v-else-if="item.type==='reply'"
-                        :content="item.content"
-                        :created-at="dateFormat(item['created_at'])"
-                        :icon="item.icon"
-                        :org-content="item.orgContent"
-                        :tokens="item['tokens']"/>
-            <chat-mid-journey v-else-if="item.type==='mj'"
-                              :content="item.content"
-                              :icon="item.icon"
-                              :role-id="role"
-                              :chat-id="chatId"
-                              @disable-input="disableInput(true)"
-                              @enable-input="enableInput"
-                              :created-at="dateFormat(item['created_at'])"/>
-          </van-cell>
-        </van-list>
-      </div>
+        <van-share-sheet
+            v-model:show="showShare"
+            title="立即分享给好友"
+            :options="shareOptions"
+            @select="shareChat"
+        />
 
-      <van-sticky ref="bottomBarRef" :offset-bottom="0" position="bottom">
-        <div class="chat-box">
-          <van-cell-group>
-            <van-field
-                v-model="prompt"
-                center
-                clearable
-                placeholder="输入你的问题"
+        <div class="chat-list-wrapper">
+          <div id="message-list-box" :style="{height: winHeight + 'px'}" class="message-list-box">
+            <van-list
+                v-model:error="error"
+                :finished="finished"
+                error-text="请求失败，点击重新加载"
+                @load="onLoad"
             >
-              <template #button>
-                <van-button size="small" type="primary" @click="sendMessage">发送</van-button>
-              </template>
-              <template #extra>
-                <div class="icon-box">
-                  <van-icon v-if="showStopGenerate" name="stop-circle-o" @click="stopGenerate"/>
-                  <van-icon v-if="showReGenerate" name="play-circle-o" @click="reGenerate"/>
-                </div>
-              </template>
-            </van-field>
-          </van-cell-group>
+              <van-cell v-for="item in chatData" :key="item" :border="false" class="message-line">
+                <chat-prompt
+                    v-if="item.type==='prompt'"
+                    :content="item.content"
+                    :created-at="dateFormat(item['created_at'])"
+                    :icon="item.icon"
+                    :model="model"
+                    :tokens="item['tokens']"/>
+                <chat-reply v-else-if="item.type==='reply'"
+                            :content="item.content"
+                            :created-at="dateFormat(item['created_at'])"
+                            :icon="item.icon"
+                            :org-content="item.orgContent"
+                            :tokens="item['tokens']"/>
+                <chat-mid-journey v-else-if="item.type==='mj'"
+                                  :content="item.content"
+                                  :icon="item.icon"
+                                  :role-id="role"
+                                  :chat-id="chatId"
+                                  @disable-input="disableInput(true)"
+                                  @enable-input="enableInput"
+                                  :created-at="dateFormat(item['created_at'])"/>
+              </van-cell>
+            </van-list>
+          </div>
         </div>
-      </van-sticky>
-    </div>
-  </van-config-provider>
+        <div class="chat-box-wrapper">
+          <van-sticky ref="bottomBarRef" :offset-bottom="0" position="bottom">
 
+            <van-cell-group inset>
+              <van-field
+                  v-model="prompt"
+                  center
+                  clearable
+                  placeholder="输入你的问题"
+              >
+                <template #button>
+                  <van-button size="small" type="primary" @click="sendMessage">发送</van-button>
+                </template>
+                <template #extra>
+                  <div class="icon-box">
+                    <van-icon v-if="showStopGenerate" name="stop-circle-o" @click="stopGenerate"/>
+                    <van-icon v-if="showReGenerate" name="play-circle-o" @click="reGenerate"/>
+                  </div>
+                </template>
+              </van-field>
+            </van-cell-group>
+
+          </van-sticky>
+        </div>
+      </div>
+    </van-config-provider>
+  </div>
 </template>
 
 <script setup>
@@ -101,6 +106,11 @@ import {checkSession} from "@/action/session";
 import {getMobileTheme} from "@/store/system";
 import ChatMidJourney from "@/components/mobile/ChatMidJourney.vue";
 
+import QRCode from "qrcode";
+import {ElMessage} from "element-plus";
+import Clipboard from "clipboard";
+import InviteList from "@/components/InviteList.vue";
+
 const winHeight = ref(0)
 const navBarRef = ref(null)
 const bottomBarRef = ref(null)
@@ -114,6 +124,16 @@ const title = chatConfig.title
 const chatId = chatConfig.chatId
 const loginUser = ref(null)
 
+const listBoxHeight = window.innerHeight
+const inviteURL = ref("")
+const qrImg = ref("")
+const inviteChatCalls = ref(0)
+const inviteImgCalls = ref(0)
+const hits = ref(0)
+const regNum = ref(0)
+const rate = ref(0)
+const isLogin = ref(false)
+
 onMounted(() => {
   winHeight.value = document.body.offsetHeight - navBarRef.value.$el.offsetHeight - bottomBarRef.value.$el.offsetHeight
 })
@@ -385,59 +405,10 @@ const shareOptions = [
 ]
 const shareChat = () => {
   showShare.value = false
-  showToast('功能待开发')
+  router.push('/mobile/Invitation');
 }
 </script>
 
 <style lang="stylus" scoped>
-.mobile-chat {
-  .message-list-box {
-    padding-top 50px
-    padding-bottom 10px
-    overflow-x auto
-    background #F5F5F5;
-
-    .van-cell {
-      background none
-      font-family: 'Microsoft YaHei', '微软雅黑', Arial, sans-serif;
-    }
-  }
-
-  .chat-box {
-    .icon-box {
-      .van-icon {
-        font-size 24px
-        margin-left 10px;
-      }
-    }
-  }
-}
-
-.van-theme-dark {
-  .mobile-chat {
-    .message-list-box {
-      background #232425;
-    }
-  }
-}
-</style>
-
-<style lang="stylus">
-.mobile-chat {
-  .van-nav-bar__title {
-    .van-dropdown-menu__title {
-      margin-right 10px
-    }
-
-    .van-cell__title {
-      text-align left
-    }
-  }
-
-  .van-nav-bar__right {
-    .van-icon {
-      font-size 20px
-    }
-  }
-}
+@import "@/assets/css/mobile/chat-session.css"
 </style>

web/src/views/mobile/Home.vue

@@ -1,17 +1,26 @@
 <template>
-  <van-config-provider :theme="getMobileTheme()">
+  <div class="banner">
+    <h1 class="banner-title">ChatPuls-V3 智能助理</h1>
+  </div>
+  <van-config-provider theme="dark">
     <div class="mobile-home">
       <router-view/>
-
-      <van-tabbar route v-model="active" @change="onChange">
-        <van-tabbar-item to="/mobile/chat/list" name="home" icon="chat-o"></van-tabbar-item>
-        <van-tabbar-item to="/mobile/setting" name="setting" icon="setting-o"></van-tabbar-item>
-        <van-tabbar-item to="/mobile/profile" name="profile" icon="user-o"></van-tabbar-item>
+      <van-tabbar
+          fixed
+          route
+          v-model="active"
+          active-color="#F5F5F5"
+          inactive-color="#2CC995"
+          class="my-tabbar"
+      >
+        <van-tabbar-item to="/mobile/chat/list" name="home" icon="chat-o">对话</van-tabbar-item>
+        <van-tabbar-item to="/mobile/imageSd" name="imageSd" icon="photo-o">绘图</van-tabbar-item>
+        <van-tabbar-item to="/mobile/apps" name="apps" icon="apps-o">应用</van-tabbar-item>
+        <van-tabbar-item to="/mobile/profile" name="profile" icon="user-o">我的</van-tabbar-item>
+        <van-tabbar-item to="/mobile/invitation" name="invitation" icon="share-o">分享</van-tabbar-item>
       </van-tabbar>
-
     </div>
   </van-config-provider>
-
 </template>
 
 <script setup>
@@ -38,33 +47,6 @@ const onChange = (index) => {
 
 </script>
 
-<style lang="stylus">
-.mobile-home {
-  .container {
-    .van-nav-bar {
-      position fixed
-      width 100%
-    }
-
-    .content {
-      padding 46px 10px 0 10px;
-    }
-  }
-
-}
-
-// 黑色主题
-.van-theme-dark body {
-  background #1c1c1e
-}
-
-.van-toast--fail {
-  background #fef0f0
-  color #f56c6c
-}
-
-.van-nav-bar {
-  position fixed
-  width 100%
-}
-</style>
+<style lang="stylus" scoped>
+@import "@/assets/css/mobile/home.css"
+</style>

web/src/views/mobile/ImageSd.vue

@@ -0,0 +1,657 @@
+<template>
+  <div class="page-sd" theme="dark">
+    <div class="inner">
+      <div class="sd-box" style="padding-top: 35px">
+        <h2>Ai绘图</h2>
+        <div class="sd-params" :style="{ height: mjBoxHeight + 'px' }">
+          <el-form :model="params" label-width="80px" label-position="left">
+            <div class="param-line" style="padding-top: 10px">
+              <el-form-item label="采样方法">
+                <template #default>
+                  <div class="form-item-inner">
+                    <el-select v-model="params.sampler" size="small">
+                      <el-option v-for="item in samplers" :label="item" :value="item" :key="item"/>
+                    </el-select>
+                    <el-tooltip
+                        effect="light"
+                        content="出图效果比较好的一般是 Euler 和 DPM 系列算法"
+                        raw-content
+                        placement="right"
+                    >
+                      <el-icon>
+                        <InfoFilled/>
+                      </el-icon>
+                    </el-tooltip>
+                  </div>
+                </template>
+              </el-form-item>
+            </div>
+
+            <div class="param-line">
+              <el-form-item label="图片尺寸">
+                <template #default>
+                  <div class="form-item-inner">
+                    <el-row :gutter="20">
+                      <el-col :span="12">
+                        <el-input v-model.number="params.width" size="small" placeholder="图片宽度"/>
+                      </el-col>
+                      <el-col :span="12">
+                        <el-input v-model.number="params.height" size="small" placeholder="图片高度"/>
+                      </el-col>
+                    </el-row>
+                  </div>
+                </template>
+              </el-form-item>
+            </div>
+            <!-- 这是一个示例注释
+                        <div class="param-line">
+                          <el-form-item label="迭代步数">
+                            <template #default>
+                              <div class="form-item-inner">
+                                <el-input v-model.number="params.steps" size="small"/>
+                                <el-tooltip
+                                    effect="light"
+                                    content="值越大则代表细节越多，同时也意味着出图速度越慢"
+                                    raw-content
+                                    placement="right"
+                                >
+                                  <el-icon>
+                                    <InfoFilled/>
+                                  </el-icon>
+                                </el-tooltip>
+                              </div>
+                            </template>
+                          </el-form-item>
+                        </div>
+
+                        <div class="param-line">
+                          <el-form-item label="引导系数">
+                            <template #default>
+                              <div class="form-item-inner">
+                                <el-input v-model.number="params.cfg_scale" size="small"/>
+                                <el-tooltip
+                                    effect="light"
+                                    content="提示词引导系数，图像在多大程度上服从提示词<br/> 较低值会产生更有创意的结果"
+                                    raw-content
+                                    placement="right"
+                                >
+                                  <el-icon>
+                                    <InfoFilled/>
+                                  </el-icon>
+                                </el-tooltip>
+                              </div>
+                            </template>
+                          </el-form-item>
+                        </div>
+
+                        <div class="param-line">
+                          <el-form-item label="随机因子">
+                            <template #default>
+                              <div class="form-item-inner">
+                                <el-input v-model.number="params.seed" size="small"/>
+                                <el-tooltip
+                                    effect="light"
+                                    content="随机数种子，相同的种子会得到相同的结果<br/> 设置为 -1 则每次随机生成种子"
+                                    raw-content
+                                    placement="right"
+                                >
+                                  <el-icon>
+                                    <InfoFilled/>
+                                  </el-icon>
+                                </el-tooltip>
+
+                                <el-tooltip
+                                    effect="light"
+                                    content="使用随机数"
+                                    raw-content
+                                    placement="right"
+                                >
+                                  <el-icon @click="params.seed = -1">
+                                    <Orange/>
+                                  </el-icon>
+                                </el-tooltip>
+                              </div>
+                            </template>
+                          </el-form-item>
+                        </div>
+             -->
+            <!-- 这是一个示例注释
+                       <div class="param-line">
+                         <el-form-item label="面部修复">
+                           <template #default>
+                             <div class="form-item-inner">
+                               <el-switch v-model="params.face_fix" style="--el-switch-on-color: #47fff1;"/>
+                               <el-tooltip
+                                   effect="light"
+                                   content="仅对绘制人物图像有效果。"
+                                   raw-content
+                                   placement="right"
+                               >
+                                 <el-icon style="margin-top: 6px">
+                                   <InfoFilled/>
+                                 </el-icon>
+                               </el-tooltip>
+                             </div>
+                           </template>
+                         </el-form-item>
+                       </div>
+
+                       <div class="param-line">
+                         <el-form-item label="高清修复">
+                           <template #default>
+                             <div class="form-item-inner">
+                               <el-switch v-model="params.hd_fix" style="--el-switch-on-color: #47fff1;"/>
+                               <el-tooltip
+                                   effect="light"
+                                   content="先以较小的分辨率生成图像，接着方法图像<br />然后在不更改构图的情况下再修改细节"
+                                   raw-content
+                                   placement="right"
+                               >
+                                 <el-icon style="margin-top: 6px">
+                                   <InfoFilled/>
+                                 </el-icon>
+                               </el-tooltip>
+                             </div>
+                           </template>
+                         </el-form-item>
+                       </div>
+
+                       <div v-show="params.hd_fix">
+                         <div class="param-line">
+                           <el-form-item label="重绘幅度">
+                             <template #default>
+                               <div class="form-item-inner">
+                                 <el-slider v-model.number="params.hd_redraw_rate" :max="1" :step="0.1"
+                                            style="width: 180px;--el-slider-main-bg-color:#47fff1"/>
+                                 <el-tooltip
+                                     effect="light"
+                                     content="决定算法对图像内容的影响程度<br />较大的值将得到越有创意的图像"
+                                     raw-content
+                                     placement="right"
+                                 >
+                                   <el-icon style="margin-top: 6px">
+                                     <InfoFilled/>
+                                   </el-icon>
+                                 </el-tooltip>
+                               </div>
+                             </template>
+                           </el-form-item>
+                         </div>
+
+                         <div class="param-line">
+                           <el-form-item label="放大算法">
+                             <template #default>
+                               <div class="form-item-inner">
+                                 <el-select v-model="params.hd_scale_alg" size="small">
+                                   <el-option v-for="item in scaleAlg" :label="item" :value="item" :key="item"/>
+                                 </el-select>
+                                 <el-tooltip
+                                     effect="light"
+                                     content="高清修复放大算法，主流算法有Latent和ESRGAN_4x"
+                                     raw-content
+                                     placement="right"
+                                 >
+                                   <el-icon>
+                                     <InfoFilled/>
+                                   </el-icon>
+                                 </el-tooltip>
+                               </div>
+                             </template>
+                           </el-form-item>
+                         </div>
+
+                         <div class="param-line">
+                           <el-form-item label="放大倍数">
+                             <template #default>
+                               <div class="form-item-inner">
+                                 <el-input v-model.number="params.hd_scale" size="small"/>
+                                 <el-tooltip
+                                     effect="light"
+                                     content="随机数种子，相同的种子会得到相同的结果<br/> 设置为 -1 则每次随机生成种子"
+                                     raw-content
+                                     placement="right"
+                                 >
+                                   <el-icon>
+                                     <InfoFilled/>
+                                   </el-icon>
+                                 </el-tooltip>
+                               </div>
+                             </template>
+                           </el-form-item>
+                         </div>
+
+                         <div class="param-line">
+                           <el-form-item label="迭代步数">
+                             <template #default>
+                               <div class="form-item-inner">
+                                 <el-input v-model.number="params.hd_steps" size="small"/>
+                                 <el-tooltip
+                                     effect="light"
+                                     content="重绘迭代步数，如果设置为0，则设置跟原图相同的迭代步数"
+                                     raw-content
+                                     placement="right"
+                                 >
+                                   <el-icon>
+                                     <InfoFilled/>
+                                   </el-icon>
+                                 </el-tooltip>
+                               </div>
+                             </template>
+                           </el-form-item>
+                         </div>
+                       </div>
+            -->
+            <div class="param-line" v-loading="loading" element-loading-background="rgba(122, 122, 122, 0.8)">
+              <el-input
+                  v-model="params.prompt"
+                  :autosize="{ minRows: 4, maxRows: 6 }"
+                  type="textarea"
+                  ref="promptRef"
+                  placeholder="正向提示词，例如：“一个中国女孩，黑色的长发，穿着职业女装走在深圳夜晚的街头，看着街上的车流”，输入后点击翻译或翻译并重写按钮后点击生成。"
+              />
+            </div>
+
+            <div style="padding: 10px">
+              <el-button type="primary" @click="translatePrompt" size="small">
+                <el-icon style="margin-right: 6px;font-size: 18px;">
+                  <Refresh/>
+                </el-icon>
+                翻译
+              </el-button>
+
+              <el-tooltip
+                  class="box-item"
+                  effect="dark"
+                  raw-content
+                  content="使用 AI 翻译并重写提示词，<br/>增加更多细节，风格等描述"
+                  placement="top-end"
+              >
+                <el-button type="success" @click="rewritePrompt" size="small">
+                  <el-icon style="margin-right: 6px;font-size: 18px;">
+                    <Refresh/>
+                  </el-icon>
+                  翻译并重写
+                </el-button>
+              </el-tooltip>
+              <el-tag type="success" style="margin-left: 12px;">绘图可用额度：{{ imgCalls }}</el-tag>
+            </div>
+            <!--
+                        <div class="param-line pt">
+                          <span>反向提示词：</span>
+                          <el-tooltip
+                              effect="light"
+                              content="不希望出现的元素，下面给了默认的起手式"
+                              placement="right"
+                          >
+                            <el-icon>
+                              <InfoFilled/>
+                            </el-icon>
+                          </el-tooltip>
+                        </div>
+                        <div class="param-line">
+                          <el-input
+                              v-model="params.negative_prompt"
+                              :autosize="{ minRows: 4, maxRows: 6 }"
+                              type="textarea"
+                              placeholder="反向提示词"
+                          />
+                        </div>
+            -->
+          </el-form>
+        </div>
+        <div class="submit-btn">
+          <el-button color="#47fff1" :dark="false" round @click="generate">立即生成</el-button>
+        </div>
+        <div class="task-list-box">
+          <div class="task-list-inner" :style="{ height: listBoxHeight + 'px' }">
+            <h2>任务列表</h2>
+            <div class="running-job-list">
+              <ItemList :items="runningJobs" v-if="runningJobs.length > 0" width="240">
+                <template #default="scope">
+                  <div class="job-item">
+                    <div v-if="scope.item.progress > 0" class="job-item-inner">
+                      <el-image :src="scope.item['img_url']"
+                                fit="cover"
+                                loading="lazy">
+                        <template #placeholder>
+                          <div class="image-slot">
+                            正在加载图片
+                          </div>
+                        </template>
+
+                        <template #error>
+                          <div class="image-slot">
+                            <el-icon v-if="scope.item['img_url'] !== ''">
+                              <Picture/>
+                            </el-icon>
+                          </div>
+                        </template>
+                      </el-image>
+
+                      <div class="progress">
+                        <el-progress type="circle" :percentage="scope.item.progress" :width="100" color="#47fff1"/>
+                      </div>
+                    </div>
+                    <el-image fit="cover" v-else>
+                      <template #error>
+                        <div class="image-slot">
+                          <i class="iconfont icon-quick-start"></i>
+                          <span>任务正在排队中</span>
+                        </div>
+                      </template>
+                    </el-image>
+                  </div>
+                </template>
+              </ItemList>
+              <el-empty :image-size="100" v-else/>
+            </div>
+            <h2>创作记录</h2>
+            <div class="finish-job-list">
+              <ItemList :items="finishedJobs" v-if="finishedJobs.length > 0" width="480" :gap="20">
+                <template #default="scope">
+                  <div class="job-item animate" @click="showTask(scope.item)">
+                    <el-image
+                        :src="scope.item['img_url']+'?imageView2/1/w/480/h/480/q/75'"
+                        fit="cover"
+                        loading="lazy">
+                      <template #placeholder>
+                        <div class="image-slot">
+                          正在加载图片
+                        </div>
+                      </template>
+
+                      <template #error>
+                        <div class="image-slot">
+                          <el-icon>
+                            <Picture/>
+                          </el-icon>
+                        </div>
+                      </template>
+                    </el-image>
+                  </div>
+                </template>
+              </ItemList>
+              <el-empty :image-size="100" v-else/>
+            </div> <!-- end finish job list-->
+          </div>
+        </div><!-- end task list box -->
+      </div>
+    </div>
+    <el-dialog v-model="showTaskDialog" title="绘画任务详情" :fullscreen="true">
+      <el-row class="mobile-form-row">
+        <el-col :span="24" class="mobile-form-col">
+          <div class="img-container" :style="{maxHeight: fullImgHeight+'px'}">
+            <el-image :src="item['img_url']" fit="contain"/>
+          </div>
+          <el-col :span="50">
+            <div class="task-info">
+              <div class="info-line">
+                <el-divider>
+                  正向提示词
+                </el-divider>
+                <div class="prompt">
+                  <span>{{ item.prompt }}</span>
+                  <el-icon class="copy-prompt" :data-clipboard-text="item.prompt">
+                    <DocumentCopy/>
+                  </el-icon>
+                </div>
+
+              </div>
+
+              <div class="info-line">
+                <el-divider>
+                  反向提示词
+                </el-divider>
+                <div class="prompt">
+                  <span>{{ item.params.negative_prompt }}</span>
+                  <el-icon class="copy-prompt" :data-clipboard-text="item.params.negative_prompt">
+                    <DocumentCopy/>
+                  </el-icon>
+                </div>
+              </div>
+              <div class="info-line">
+                <div class="wrapper">
+                  <label>采样方法：</label>
+                  <div class="item-value">{{ item.params.sampler }}</div>
+                </div>
+              </div>
+
+              <div class="info-line">
+                <div class="wrapper">
+                  <label>图片尺寸：</label>
+                  <div class="item-value">{{ item.params.width }} x {{ item.params.height }}</div>
+                </div>
+              </div>
+
+              <div class="info-line">
+                <div class="wrapper">
+                  <label>迭代步数：</label>
+                  <div class="item-value">{{ item.params.steps }}</div>
+                </div>
+              </div>
+
+              <div class="info-line">
+                <div class="wrapper">
+                  <label>引导系数：</label>
+                  <div class="item-value">{{ item.params.cfg_scale }}</div>
+                </div>
+              </div>
+
+              <div class="info-line">
+                <div class="wrapper">
+                  <label>随机因子：</label>
+                  <div class="item-value">{{ item.params.seed }}</div>
+                </div>
+              </div>
+
+              <div v-if="item.params.hd_fix">
+                <el-divider>
+                  高清修复
+                </el-divider>
+                <div class="info-line">
+                  <div class="wrapper">
+                    <label>重绘幅度：</label>
+                    <div class="item-value">{{ item.params.hd_redraw_rate }}</div>
+                  </div>
+                </div>
+
+                <div class="info-line">
+                  <div class="wrapper">
+                    <label>放大算法：</label>
+                    <div class="item-value">{{ item.params.hd_scale_alg }}</div>
+                  </div>
+                </div>
+
+                <div class="info-line">
+                  <div class="wrapper">
+                    <label>放大倍数：</label>
+                    <div class="item-value">{{ item.params.hd_scale }}</div>
+                  </div>
+                </div>
+
+                <div class="info-line">
+                  <div class="wrapper">
+                    <label>迭代步数：</label>
+                    <div class="item-value">{{ item.params.hd_steps }}</div>
+                  </div>
+                </div>
+              </div>
+
+              <div class="copy-params">
+                <el-button type="primary" round @click="copyParams(item)">画一张同款的</el-button>
+              </div>
+
+            </div>
+          </el-col>
+        </el-col>
+      </el-row>
+
+    </el-dialog>
+  </div>
+</template>
+
+<script setup>
+import {onMounted, ref} from "vue"
+import {DocumentCopy, InfoFilled, Orange, Picture, Refresh} from "@element-plus/icons-vue";
+import {httpGet, httpPost} from "@/utils/http";
+import {ElMessage, ElNotification} from "element-plus";
+import ItemList from "@/components/ItemList.vue";
+import Clipboard from "clipboard";
+import {checkSession} from "@/action/session";
+import {useRouter} from "vue-router";
+import {getSessionId} from "@/store/session";
+
+const listBoxHeight = ref(window.innerHeight - 40)
+// const mjBoxHeight = ref(window.innerHeight - 150)
+const fullImgHeight = ref(window.innerHeight - 60)
+const showTaskDialog = ref(false)
+const item = ref({})
+const loading = ref(false)
+
+window.onresize = () => {
+  listBoxHeight.value = window.innerHeight - 40
+//   mjBoxHeight.value = window.innerHeight - 150
+}
+const samplers = ["Euler a", "Euler", "DPM2 a Karras", "DPM++ 2S a Karras", "DPM++ 2M Karras", "DPM++ SDE Karras", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM fast", "DPM adaptive",
+  "LMS Karras", "DPM2 Karras", "DDIM", "PLMS", "UniPC", "LMS", "Heun",]
+const scaleAlg = ["Latent", "ESRGAN_4x", "R-ESRGAN 4x+", "SwinIR_4x", "LDSR"]
+const params = ref({
+  width: 768,
+  height: 1024,
+  sampler: samplers[0],
+  seed: -1,
+  steps: 20,
+  cfg_scale: 7,
+  face_fix: false,
+  hd_fix: false,
+  hd_redraw_rate: 0.7,
+  hd_scale: 2,
+  hd_scale_alg: scaleAlg[0],
+  hd_steps: 10,
+  prompt: "",
+  negative_prompt: "nsfw, paintings,low quality,easynegative,ng_deepnegative ,lowres,bad anatomy,bad hands,bad feet",
+})
+
+const runningJobs = ref([])
+const finishedJobs = ref([])
+const router = useRouter()
+// 检查是否有画同款的参数
+const _params = router.currentRoute.value.params["copyParams"]
+if (_params) {
+  params.value = JSON.parse(_params)
+}
+
+const imgCalls = ref(0)
+
+const rewritePrompt = () => {
+  loading.value = true
+  httpPost("/api/prompt/rewrite", {"prompt": params.value.prompt}).then(res => {
+    params.value.prompt = res.data
+    loading.value = false
+  }).catch(e => {
+    loading.value = false
+    ElMessage.error("翻译失败：" + e.message)
+  })
+}
+
+const translatePrompt = () => {
+  loading.value = true
+  httpPost("/api/prompt/translate", {"prompt": params.value.prompt}).then(res => {
+    params.value.prompt = res.data
+    loading.value = false
+  }).catch(e => {
+    loading.value = false
+    ElMessage.error("翻译失败：" + e.message)
+  })
+}
+
+onMounted(() => {
+  checkSession().then(user => {
+    imgCalls.value = user['img_calls']
+
+    fetchRunningJobs(user.id)
+    fetchFinishJobs(user.id)
+
+  }).catch(() => {
+    router.push('/login')
+  });
+
+  const fetchRunningJobs = (userId) => {
+    // 获取运行中的任务
+    httpGet(`/api/sd/jobs?status=0&user_id=${userId}`).then(res => {
+      const jobs = res.data
+      const _jobs = []
+      for (let i = 0; i < jobs.length; i++) {
+        if (jobs[i].progress === -1) {
+          ElNotification({
+            title: '任务执行失败',
+            message: "任务ID：" + jobs[i]['task_id'],
+            type: 'error',
+          })
+          continue
+        }
+        _jobs.push(jobs[i])
+      }
+      runningJobs.value = _jobs
+
+      setTimeout(() => fetchRunningJobs(userId), 5000)
+    }).catch(e => {
+      ElMessage.error("获取任务失败：" + e.message)
+    })
+  }
+
+  // 获取已完成的任务
+  const fetchFinishJobs = (userId) => {
+    httpGet(`/api/sd/jobs?status=1&user_id=${userId}`).then(res => {
+      finishedJobs.value = res.data
+      setTimeout(() => fetchFinishJobs(userId), 5000)
+    }).catch(e => {
+      ElMessage.error("获取任务失败：" + e.message)
+    })
+  }
+
+  const clipboard = new Clipboard('.copy-prompt');
+  clipboard.on('success', () => {
+    ElMessage.success("复制成功！");
+  })
+
+  clipboard.on('error', () => {
+    ElMessage.error('复制失败！');
+  })
+})
+
+
+// 创建绘图任务
+const promptRef = ref(null)
+const generate = () => {
+  if (params.value.prompt === '') {
+    promptRef.value.focus()
+    return ElMessage.error("请输入绘画提示词！")
+  }
+  if (params.value.seed === '') {
+    params.value.seed = -1
+  }
+  params.value.session_id = getSessionId()
+  httpPost("/api/sd/image", params.value).then(() => {
+    ElMessage.success("绘画任务推送成功，请耐心等待任务执行...")
+    imgCalls.value -= 1
+  }).catch(e => {
+    ElMessage.error("任务推送失败：" + e.message)
+  })
+}
+
+const showTask = (row) => {
+  item.value = row
+  showTaskDialog.value = true
+}
+
+const copyParams = (row) => {
+  params.value = row.params
+  showTaskDialog.value = false
+}
+
+</script>
+
+<style lang="stylus">
+@import "@/assets/css/mobile/image-sd.css"
+</style>

web/src/views/mobile/Invitation.vue

@@ -0,0 +1,152 @@
+<template>
+  <div class="page-invitation" :style="{height: listBoxHeight + 'px'}">
+    <div class="inner">
+      <div class="title" style="padding-top: 50px">会员推广计划</div>
+      <div class="share-box">
+        <div class="info">
+          感谢您把此应用分享给您身边的朋友，分享成功注册后您将获得 <strong>{{ inviteChatCalls }}</strong>
+          次对话额度以及
+          <strong>{{ inviteImgCalls }}</strong> 次AI绘画额度作为奖励。
+          你可以保存下面的二维码或者直接复制分享您的专属推广链接发送给微信好友。
+        </div>
+
+        <div class="invite-qrcode">
+          <el-image :src="qrImg" fit="contain"/>
+        </div>
+        <div class="invite-url">
+          <span>{{ inviteURL }}</span>
+          <el-button type="primary" plain class="copy-link" :data-clipboard-text="inviteURL">复制链接</el-button>
+        </div>
+      </div>
+
+      <div class="invite-stats">
+        <el-row :gutter="20">
+          <el-col :span="8">
+            <div class="item-box yellow">
+              <el-row :gutter="10">
+                <!--                <el-col :span="10">-->
+                <!--                  <div class="item-icon">-->
+                <!--                    <i class="iconfont icon-role"></i>-->
+                <!--                  </div>-->
+                <!--                </el-col>-->
+                <el-col :span="24">
+                  <div class="item-info">
+                    <div class="num">{{ hits }}</div>
+                    <div class="text">点击量</div>
+                  </div>
+                </el-col>
+              </el-row>
+            </div>
+          </el-col>
+          <el-col :span="8">
+            <div class="item-box blue">
+              <el-row :gutter="10">
+                <!--                <el-col :span="10">-->
+                <!--                  <div class="item-icon">-->
+                <!--                    <i class="iconfont icon-order"></i>-->
+                <!--                  </div>-->
+                <!--                </el-col>-->
+                <el-col :span="24">
+                  <div class="item-info">
+                    <div class="num">{{ regNum }}</div>
+                    <div class="text">注册量</div>
+                  </div>
+                </el-col>
+              </el-row>
+            </div>
+          </el-col>
+          <el-col :span="8">
+            <div class="item-box green">
+              <el-row :gutter="10">
+                <!--                <el-col :span="10">-->
+                <!--                  <div class="item-icon">-->
+                <!--                    <i class="iconfont icon-chart"></i>-->
+                <!--                  </div>-->
+                <!--                </el-col>-->
+                <el-col :span="24">
+                  <div class="item-info">
+                    <div class="num">{{ rate }}%</div>
+                    <div class="text">转化率</div>
+                  </div>
+                </el-col>
+              </el-row>
+            </div>
+          </el-col>
+        </el-row>
+        <div class="title" style="padding: 20px">您推荐的用户</div>
+        <div class="invite-logs" style="padding-bottom: 50px">
+          <invite-list v-if="isLogin"/>
+        </div>
+      </div>
+    </div>
+  </div>
+</template>
+
+<script setup>
+import {onMounted, ref} from "vue"
+import QRCode from "qrcode";
+import {httpGet} from "@/utils/http";
+import {ElMessage} from "element-plus";
+import Clipboard from "clipboard";
+import InviteList from "@/components/InviteList.vue";
+import {checkSession} from "@/action/session";
+import {useRouter} from "vue-router";
+
+const listBoxHeight = window.innerHeight
+const inviteURL = ref("")
+const qrImg = ref("")
+const inviteChatCalls = ref(0)
+const inviteImgCalls = ref(0)
+const hits = ref(0)
+const regNum = ref(0)
+const rate = ref(0)
+const router = useRouter()
+const isLogin = ref(false)
+
+onMounted(() => {
+  checkSession().then(() => {
+    isLogin.value = true
+    httpGet("/api/invite/code").then(res => {
+      const text = `${location.protocol}//${location.host}/register?invite_code=${res.data.code}`
+      hits.value = res.data["hits"]
+      regNum.value = res.data["reg_num"]
+      if (hits.value > 0) {
+        rate.value = ((regNum.value / hits.value) * 100).toFixed(2)
+      }
+      QRCode.toDataURL(text, {width: 300, height: 300, margin: 2}, (error, url) => {
+        if (error) {
+          console.error(error)
+        } else {
+          qrImg.value = url;
+        }
+      });
+      inviteURL.value = text
+    }).catch(e => {
+      ElMessage.error("获取邀请码失败：" + e.message)
+    })
+
+    httpGet("/api/admin/config/get?key=system").then(res => {
+      inviteChatCalls.value = res.data["invite_chat_calls"]
+      inviteImgCalls.value = res.data["invite_img_calls"]
+    }).catch(e => {
+      ElMessage.error("获取系统配置失败：" + e.message)
+    })
+  }).catch(() => {
+    router.push('/login')
+  });
+
+  // 复制链接
+  const clipboard = new Clipboard('.copy-link');
+  clipboard.on('success', () => {
+    ElMessage.success('复制成功！');
+  })
+
+  clipboard.on('error', () => {
+    ElMessage.error('复制失败！');
+  })
+})
+</script>
+
+<style lang="stylus" scoped>
+@import "@/assets/css/mobile/invitation.styl"
+</style>

web/src/views/mobile/Profile.vue

@@ -1,128 +1,295 @@
 <template>
-  <div class="mobile-user-profile container">
-    <van-nav-bar :title="title"/>
+  <div class="app-background" theme="dark">
+    <div class="member">
+      <div class="title" style="padding-top: 50px">
+        会员中心
+      </div>
+      <div class="inner" :style="{height: listBoxHeight + 'px'}">
+        <div class="product-box">
+          <ItemList :items="list" v-if="list.length > 0" :gap="30">
+            <template #default="scope">
+              <div class="product-item" :style="{width: scope.width+'px'}">
+                <div class="image-container">
+                  <el-image :src="vipImg" fit="cover"/>
+                </div>
+                <div class="product-title">
+                  <span class="name">{{ scope.item.name }}</span>
+                </div>
+                <div class="product-info">
+                  <div class="info-line">
+                    <span class="label">商品原价：</span>
+                    <span class="price">￥{{ scope.item.price }}</span>
+                  </div>
+                  <div class="info-line">
+                    <span class="label">促销立减：</span>
+                    <span class="price">￥{{ scope.item.discount }}</span>
+                  </div>
+                  <div class="info-line">
+                    <span class="label">有效期：</span>
+                    <span class="expire" v-if="scope.item.days > 0">{{ scope.item.days }}天</span>
+                    <span class="expire" v-else>当月有效</span>
+                  </div>
 
-    <div class="content">
-      <van-form @submit="save">
-        <van-cell-group inset v-model="form">
-          <van-field
-              v-model="form.mobile"
-              name="手机号"
-              label="手机号"
-              readonly
-              disabled
-              placeholder="手机号"
-          />
-          <van-field label="头像">
-            <template #input>
-              <van-uploader v-model="fileList"
-                            reupload max-count="1"
-                            :deletable="false"
-                            :after-read="afterRead"/>
+                  <div class="pay-way">
+                    <el-button type="primary" @click="alipay(scope.item)" size="small" v-if="payWays['alipay']">
+                      <i class="iconfont icon-alipay"></i> 支付宝
+                    </el-button>
+                    <el-button type="success" @click="huPiPay(scope.item)" size="small" v-if="payWays['hupi']">
+                      <span v-if="payWays['hupi']['name'] === 'wechat'"><i
+                          class="iconfont icon-wechat-pay"></i> 微信</span>
+                      <span v-else><i class="iconfont icon-alipay"></i> 支付宝</span>
+                    </el-button>
+                  </div>
+                </div>
+              </div>
             </template>
-          </van-field>
-
-          <van-field label="剩余对话次数">
-            <template #input>
-              <van-tag type="primary">{{ form.calls }}</van-tag>
-            </template>
-          </van-field>
-
-          <van-field label="剩余绘图次数">
-            <template #input>
-              <van-tag type="primary">{{ form.img_calls }}</van-tag>
-            </template>
-          </van-field>
-
-          <van-field label="累计消耗tokens">
-            <template #input>
-              <van-tag type="primary">{{ form.total_tokens }}</van-tag>
-            </template>
-          </van-field>
-        </van-cell-group>
-        <div style="margin: 16px;">
-          <van-button round block type="primary" native-type="submit">
-            提交
-          </van-button>
+          </ItemList>
         </div>
-      </van-form>
+        <el-row>
+          <div class="user-profile">
+            <user-profile/>
+            <el-row class="user-opt" :gutter="20">
+              <el-col :span="12">
+                <el-button type="primary" @click="showPasswordDialog = true">修改密码</el-button>
+              </el-col>
+              <el-col :span="12">
+                <el-button type="primary" @click="showBindMobileDialog = true">绑定手机号</el-button>
+              </el-col>
+              <el-col :span="12">
+                <el-button type="primary" v-if="enableReward" @click="showRewardDialog = true">加入众筹</el-button>
+              </el-col>
+              <el-col :span="12">
+                <el-button type="primary" v-if="enableReward" @click="showRewardVerifyDialog = true">众筹核销
+                </el-button>
+              </el-col>
+              <el-col :span="24" style="padding-bottom: 50px;">
+                <el-button type="danger" round @click="logout">退出登录</el-button>
+              </el-col>
+            </el-row>
+          </div>
+        </el-row>
+      </div>
+      <login-dialog :show="showLoginDialog" @hide="showLoginDialog = false"/>
+      <password-dialog v-if="isLogin" :show="showPasswordDialog" width="100%" @hide="showPasswordDialog = false"
+                       @logout="logout"/>
+      <bind-mobile v-if="isLogin" :show="showBindMobileDialog" width="100%" :mobile="user.mobile"
+                   @hide="showBindMobileDialog = false"/>
+      <reward-verify v-if="isLogin" :show="showRewardVerifyDialog" @hide="showRewardVerifyDialog = false"/>
+      <el-dialog
+          v-model="showRewardDialog"
+          :show-close="true"
+          width="100%"
+          title="参与众筹"
+      >
+        <el-alert type="info" :closable="false">
+          <div style="font-size: 14px">您好，众筹 9.9元，就可以兑换 100 次对话，以此来覆盖我们的 OpenAI
+            账单和服务器的费用。<strong
+                style="color: #f56c6c">由于本人没有开通微信支付，付款后请凭借转账单号,点击【众筹核销】按钮手动核销。</strong>
+          </div>
+        </el-alert>
+        <div style="text-align: center;padding-top: 10px;">
+          <el-image v-if="enableReward" :src="rewardImg"/>
+        </div>
+      </el-dialog>
+
+      <el-dialog
+          v-model="showPayDialog"
+          :close-on-click-modal="false"
+          :show-close="true"
+          :width="400"
+          title="充值订单支付">
+        <div class="pay-container">
+          <div class="count-down">
+            <count-down :second="orderTimeout" @timeout="refreshPayCode" ref="countDownRef"/>
+          </div>
+
+          <div class="pay-qrcode" v-loading="loading">
+            <el-image :src="qrcode"/>
+          </div>
+
+          <div class="tip success" v-if="text !== ''">
+            <el-icon>
+              <SuccessFilled/>
+            </el-icon>
+            <span class="text">{{ text }}</span>
+          </div>
+          <div class="tip" v-else>
+            <el-icon>
+              <InfoFilled/>
+            </el-icon>
+            <span class="text">请打开手机{{ payName }}扫码支付</span>
+          </div>
+        </div>
+      </el-dialog>
     </div>
   </div>
 </template>
 
 <script setup>
-import {onMounted, ref} from "vue";
-import {showFailToast, showNotify, showSuccessToast} from "vant";
+import {onMounted, ref} from "vue"
+import {ElMessage} from "element-plus";
 import {httpGet, httpPost} from "@/utils/http";
-import Compressor from 'compressorjs';
+import ItemList from "@/components/ItemList.vue";
+import {InfoFilled, SuccessFilled} from "@element-plus/icons-vue";
+import LoginDialog from "@/components/LoginDialog.vue";
+import {checkSession} from "@/action/session";
+import UserProfile from "@/components/UserProfile.vue";
+import PasswordDialog from "@/components/PasswordDialog.vue";
+import BindMobile from "@/components/BindMobile.vue";
+import RewardVerify from "@/components/RewardVerify.vue";
+import {useRouter} from "vue-router";
+import {removeUserToken} from "@/store/session";
+import UserOrder from "@/components/UserOrder.vue";
+import CountDown from "@/components/CountDown.vue";
+
+const listBoxHeight = window.innerHeight - 97
+const list = ref([])
+const showLoginDialog = ref(false)
+const showPayDialog = ref(false)
+const vipImg = ref("/images/vip.png")
+const enableReward = ref(false) // 是否启用众筹功能
+const rewardImg = ref('/images/reward.png')
+const qrcode = ref("")
+const showPasswordDialog = ref(false);
+const showBindMobileDialog = ref(false);
+const showRewardDialog = ref(false);
+const showRewardVerifyDialog = ref(false);
+const text = ref("")
+const user = ref(null)
+const isLogin = ref(false)
+const router = useRouter()
+const curPayProduct = ref(null)
+const activeOrderNo = ref("")
+const countDownRef = ref(null)
+const orderTimeout = ref(1800)
+const loading = ref(true)
+const orderPayInfoText = ref("")
+
+const payWays = ref({})
+const payName = ref("支付宝")
+const curPay = ref("alipay") // 当前支付方式
 
-const title = ref('用户设置')
-const form = ref({
-  username: '',
-  nickname: '',
-  mobile: '',
-  avatar: '',
-  calls: 0,
-  tokens: 0
-})
-const fileList = ref([
-  {
-    url: 'https://fastly.jsdelivr.net/npm/@vant/assets/leaf.jpeg',
-    message: '上传中...',
-  }
-]);
 
 onMounted(() => {
-  httpGet('/api/user/profile').then(res => {
-    form.value = res.data
-    fileList.value[0].url = form.value.avatar
-  }).catch((e) => {
-    console.log(e.message)
-    showFailToast('获取用户信息失败')
-  });
+  checkSession().then(_user => {
+    user.value = _user
+    isLogin.value = true
+    httpGet("/api/product/list").then((res) => {
+      list.value = res.data
+    }).catch(e => {
+      ElMessage.error("获取产品套餐失败：" + e.message)
+    })
+  }).catch(() => {
+    router.push("/login")
+  })
+
+  httpGet("/api/admin/config/get?key=system").then(res => {
+    rewardImg.value = res.data['reward_img']
+    enableReward.value = res.data['enabled_reward']
+    orderPayInfoText.value = res.data['order_pay_info_text']
+    if (res.data['order_pay_timeout'] > 0) {
+      orderTimeout.value = res.data['order_pay_timeout']
+    }
+  }).catch(e => {
+    ElMessage.error("获取系统配置失败：" + e.message)
+  })
+
+  httpGet("/api/payment/payWays").then(res => {
+    payWays.value = res.data
+  }).catch(e => {
+    ElMessage.error("获取支付方式失败：" + e.message)
+  })
 })
 
-const afterRead = (file) => {
-  file.status = 'uploading';
-  file.message = '上传中...';
-  // 压缩图片并上传
-  new Compressor(file.file, {
-    quality: 0.6,
-    success(result) {
-      const formData = new FormData();
-      formData.append('file', result, result.name);
-      // 执行上传操作
-      httpPost('/api/upload', formData).then((res) => {
-        form.value.avatar = res.data
-        file.status = 'success'
-        showNotify({type: 'success', message: '上传成功'})
-      }).catch((e) => {
-        console.log(e.message)
-        showNotify({type: 'danger', message: '上传失败'})
-      })
-    },
-    error(err) {
-      console.log(err.message);
-    },
-  });
-};
+// refresh payment qrcode
+const refreshPayCode = () => {
+  if (curPay.value === 'alipay') {
+    alipay()
+  } else if (curPay.value === 'hupi') {
+    huPiPay()
+  }
+}
 
-const save = () => {
-  httpPost('/api/user/profile/update', form.value).then(() => {
-    showSuccessToast('保存成功')
-  }).catch(() => {
-    showFailToast('保存失败')
+const genPayQrcode = () => {
+  loading.value = true
+  text.value = ""
+  httpPost("/api/payment/qrcode", {
+    pay_way: curPay.value,
+    product_id: curPayProduct.value.id,
+    user_id: user.value.id
+  }).then(res => {
+    showPayDialog.value = true
+    qrcode.value = res.data['image']
+    activeOrderNo.value = res.data['order_no']
+    queryOrder(activeOrderNo.value)
+    loading.value = false
+    // 重置计数器
+    if (countDownRef.value) {
+      countDownRef.value.resetTimer()
+    }
+  }).catch(e => {
+    ElMessage.error("生成支付订单失败：" + e.message)
   })
 }
+
+const alipay = (row) => {
+  payName.value = "支付宝"
+  curPay.value = "alipay"
+  if (!user.value.id) {
+    showLoginDialog.value = true
+    return
+  }
+  if (row) {
+    curPayProduct.value = row
+  }
+  genPayQrcode()
+}
+
+// 虎皮椒支付
+const huPiPay = (row) => {
+  payName.value = payWays.value["hupi"]["name"] === "wechat" ? '微信' : '支付宝'
+  curPay.value = "hupi"
+  if (!user.value.id) {
+    showLoginDialog.value = true
+    return
+  }
+  if (row) {
+    curPayProduct.value = row
+  }
+  genPayQrcode()
+
+}
+
+const queryOrder = (orderNo) => {
+  httpPost("/api/payment/query", {order_no: orderNo}).then(res => {
+    if (res.data.status === 1) {
+      text.value = "扫码成功，请在手机上进行支付！"
+      queryOrder(orderNo)
+    } else if (res.data.status === 2) {
+      text.value = "支付成功，正在刷新页面"
+      setTimeout(() => location.reload(), 500)
+    } else {
+      // 如果当前订单没有过期，继续等待订单的下一个状态
+      if (activeOrderNo.value === orderNo) {
+        queryOrder(orderNo)
+      }
+    }
+  }).catch(e => {
+    ElMessage.error("查询支付状态失败：" + e.message)
+  })
+}
+
+const logout = function () {
+  httpGet('/api/user/logout').then(() => {
+    removeUserToken();
+    router.push('/login');
+  }).catch(() => {
+    ElMessage.error('注销失败！');
+  })
+}
+
 </script>
 
 <style lang="stylus">
-.mobile-user-profile {
-  .content {
-    .van-field__label {
-      width 100px
-      text-align right
-    }
-  }
-}
-</style>
+@import "@/assets/css/mobile/profile.styl"
+</style>