公众号助手新增图片

This commit is contained in:
Jackzhou
2026-08-26 17:34:21 +08:00
parent b00e55f0e5
commit f864cc2c87
11 changed files with 1348 additions and 30 deletions
+2 -1
View File
@@ -28,4 +28,5 @@ docs/09_Research/工业AI观察员/raw/
# ===== 操作系统 =====
.DS_Store
Thumbs.db
Thumbs.db
.ollama_models/
@@ -17,6 +17,7 @@ import (
const (
CapabilityAIChat = "ai_chat" // PathCoach 对话(每轮扣 1 点)
CapabilityTextGen = "text_gen" // 快捷动作(情景演练/查佣金/产品对比,扣 1 点)
CapabilityImageGen = "image_gen" // 文生图(按次扣点)
CapabilityEmbed = "embed" // 知识检索内部 embedding(不扣点,仅审计)
CapabilityEssayGrade = "essay_grade" // 简答题 LLM 评分(系统自动,不扣点,仅审计)
)
@@ -25,6 +26,7 @@ const (
var CapabilityCredits = map[string]int{
CapabilityAIChat: 1,
CapabilityTextGen: 1,
CapabilityImageGen: 1,
CapabilityEmbed: 0,
CapabilityEssayGrade: 0,
}
@@ -73,6 +73,28 @@ func syncWorkflowFromOfficialAccountArticle(workflow *officialAccountWorkflowSta
workflow.Shared.SelectedTitle = article.SelectedTitle
workflow.Shared.Outline = article.Outline
workflow.Shared.Content = article.Content
workflow.Shared.ContentWithPrompts = article.ContentWithPrompts
if strings.TrimSpace(article.ImagePromptsJSON) != "" {
var items []officialAccountImagePrompt
if json.Unmarshal([]byte(article.ImagePromptsJSON), &items) == nil {
workflow.Shared.ImagePrompts = items
}
}
if strings.TrimSpace(article.ImageRefsJSON) != "" {
var items []officialAccountImageRef
if json.Unmarshal([]byte(article.ImageRefsJSON), &items) == nil {
workflow.Shared.ImageRefs = items
}
}
workflow.Shared.PreviewMarkdown = article.PreviewMarkdown
workflow.Shared.PreviewHTML = article.PreviewHTML
workflow.Shared.PageMarkdown = article.PageMarkdown
if strings.TrimSpace(article.PagesJSON) != "" {
var items []officialAccountPageChunk
if json.Unmarshal([]byte(article.PagesJSON), &items) == nil {
workflow.Shared.Pages = items
}
}
}
func syncOfficialAccountArticleFromWorkflow(article *model.OfficialAccountArticle, workflow officialAccountWorkflowState) {
@@ -92,14 +114,35 @@ func syncOfficialAccountArticleFromWorkflow(article *model.OfficialAccountArticl
article.SelectedTitle = workflow.Shared.SelectedTitle
article.Outline = workflow.Shared.Outline
article.Content = workflow.Shared.Content
article.ContentWithPrompts = workflow.Shared.ContentWithPrompts
if data, err := json.Marshal(workflow.Shared.TopicCandidates); err == nil {
article.TopicCandidatesJSON = string(data)
}
if data, err := json.Marshal(workflow.Shared.TitleCandidates); err == nil {
article.TitleCandidatesJSON = string(data)
}
if workflow.Shared.Content != "" {
article.Status = "content_ready"
if data, err := json.Marshal(workflow.Shared.ImagePrompts); err == nil {
article.ImagePromptsJSON = string(data)
}
if data, err := json.Marshal(workflow.Shared.ImageRefs); err == nil {
article.ImageRefsJSON = string(data)
}
article.PreviewMarkdown = workflow.Shared.PreviewMarkdown
article.PreviewHTML = workflow.Shared.PreviewHTML
article.PageMarkdown = workflow.Shared.PageMarkdown
if data, err := json.Marshal(workflow.Shared.Pages); err == nil {
article.PagesJSON = string(data)
}
if workflow.Shared.PageMarkdown != "" {
article.Status = "page_md_ready"
} else if workflow.Shared.PreviewHTML != "" {
article.Status = "preview_ready"
} else if len(workflow.Shared.ImageRefs) > 0 {
article.Status = "images_ready"
} else if len(workflow.Shared.ImagePrompts) > 0 || workflow.Shared.ContentWithPrompts != "" {
article.Status = "image_prompts_ready"
} else if workflow.Shared.Content != "" {
article.Status = "content_ready"
} else if workflow.Shared.Outline != "" {
article.Status = "outline_ready"
} else if workflow.Shared.SelectedTitle != "" {
@@ -132,6 +175,13 @@ func resetOfficialAccountArticle(article *model.OfficialAccountArticle, req offi
article.Outline = ""
article.ContentTargetWords = normalizeOfficialAccountContentTargetWords(req.ContentTargetWords)
article.Content = ""
article.ContentWithPrompts = ""
article.ImagePromptsJSON = "[]"
article.ImageRefsJSON = "[]"
article.PreviewMarkdown = ""
article.PreviewHTML = ""
article.PageMarkdown = ""
article.PagesJSON = "[]"
article.Status = "draft"
}
@@ -0,0 +1,593 @@
package api
import (
"bytes"
"encoding/json"
"fmt"
"html"
"io"
"net/http"
"net/url"
"strings"
"time"
"github.com/gin-gonic/gin"
"eaisalestrain/backend/internal/ai"
"eaisalestrain/backend/internal/config"
"eaisalestrain/backend/internal/middleware"
"eaisalestrain/backend/internal/model"
"eaisalestrain/backend/internal/web"
)
type officialAccountSection struct {
Title string
Content []string
}
func executeOfficialAccountImagePromptStep(workflow *officialAccountWorkflowState, article *model.OfficialAccountArticle, user *model.User, now time.Time) officialAccountStepExecution {
prompts := buildOfficialAccountImagePrompts(workflow)
contentWithPrompts := buildOfficialAccountContentWithPrompts(workflow.Shared.Content, prompts)
workflow.Shared.ContentWithPrompts = contentWithPrompts
workflow.Shared.ImagePrompts = prompts
output := map[string]any{
"content_with_prompts": contentWithPrompts,
"image_prompts": prompts,
"count": len(prompts),
}
return officialAccountStepExecution{
Summary: fmt.Sprintf("已生成 %d 条配图提示词。", len(prompts)),
TaskStatus: workerTaskStatusDraft,
ArtifactTitle: "公众号配图提示词",
ArtifactType: "image_prompts",
ArtifactStatus: workerArtifactStatusDraft,
ArtifactContent: contentWithPrompts,
Output: output,
Logs: []string{
fmt.Sprintf("%s 已基于正文提炼配图提示词。", now.Format("15:04:05")),
fmt.Sprintf("%s 已写回带提示词标记的正文稿。", now.Format("15:04:05")),
},
}
}
func executeOfficialAccountImageGenerationStep(workflow *officialAccountWorkflowState, article *model.OfficialAccountArticle, user *model.User, now time.Time) (officialAccountStepExecution, *web.AppError) {
prompts := workflow.Shared.ImagePrompts
if len(prompts) == 0 {
prompts = buildOfficialAccountImagePrompts(workflow)
workflow.Shared.ImagePrompts = prompts
workflow.Shared.ContentWithPrompts = buildOfficialAccountContentWithPrompts(workflow.Shared.Content, prompts)
}
route, refs, appErr := generateOfficialAccountImages(article.TaskID, workflow, prompts, user)
if appErr != nil {
return officialAccountStepExecution{}, appErr
}
workflow.Shared.ImageRefs = refs
output := map[string]any{
"image_refs": refs,
"count": len(refs),
"route_id": route.RouteID,
"route_model": route.Model,
"route_provider": route.Provider,
}
return officialAccountStepExecution{
Summary: fmt.Sprintf("已生成 %d 张图片结果。", len(refs)),
TaskStatus: workerTaskStatusDraft,
ArtifactTitle: "公众号配图结果",
ArtifactType: "images",
ArtifactStatus: workerArtifactStatusDraft,
ArtifactContent: buildOfficialAccountImageArtifact(refs),
Output: output,
Logs: []string{
fmt.Sprintf("%s 已通过 %s 生成图片结果。", now.Format("15:04:05"), route.Model),
fmt.Sprintf("%s 图片已可直接用于预览与导出。", now.Format("15:04:05")),
},
}, nil
}
func executeOfficialAccountPreviewStep(workflow *officialAccountWorkflowState, article *model.OfficialAccountArticle, user *model.User, now time.Time) officialAccountStepExecution {
previewMarkdown := buildOfficialAccountPreviewMarkdown(workflow)
previewHTML := renderOfficialAccountHTML(previewMarkdown)
workflow.Shared.PreviewMarkdown = previewMarkdown
workflow.Shared.PreviewHTML = previewHTML
output := map[string]any{
"preview_markdown": previewMarkdown,
"preview_html": previewHTML,
"image_count": len(workflow.Shared.ImageRefs),
}
return officialAccountStepExecution{
Summary: "预览稿已生成,可直接检查排版并导出。",
TaskStatus: workerTaskStatusDraft,
ArtifactTitle: "公众号预览稿",
ArtifactType: "preview",
ArtifactStatus: workerArtifactStatusDraft,
ArtifactContent: previewMarkdown,
Output: output,
Logs: []string{
fmt.Sprintf("%s 已整合正文与配图生成预览稿。", now.Format("15:04:05")),
fmt.Sprintf("%s 已同步产出预览 HTML。", now.Format("15:04:05")),
},
}
}
func executeOfficialAccountPageMarkdownStep(workflow *officialAccountWorkflowState, article *model.OfficialAccountArticle, user *model.User, now time.Time) officialAccountStepExecution {
pages := buildOfficialAccountPages(workflow)
pageMarkdown := buildOfficialAccountPageMarkdown(pages)
workflow.Shared.Pages = pages
workflow.Shared.PageMarkdown = pageMarkdown
if workflow.Shared.PreviewMarkdown == "" {
workflow.Shared.PreviewMarkdown = buildOfficialAccountPreviewMarkdown(workflow)
}
if workflow.Shared.PreviewHTML == "" {
workflow.Shared.PreviewHTML = renderOfficialAccountHTML(workflow.Shared.PreviewMarkdown)
}
output := map[string]any{
"page_markdown": pageMarkdown,
"pages": pages,
"count": len(pages),
}
return officialAccountStepExecution{
Summary: fmt.Sprintf("分页 MD 已生成,共 %d 页。", len(pages)),
TaskStatus: workerTaskStatusDraft,
ArtifactTitle: "公众号分页 MD",
ArtifactType: "page_markdown",
ArtifactStatus: workerArtifactStatusDraft,
ArtifactContent: pageMarkdown,
Output: output,
Logs: []string{
fmt.Sprintf("%s 已按段落结构生成分页 MD。", now.Format("15:04:05")),
fmt.Sprintf("%s 分页结果可直接导出。", now.Format("15:04:05")),
},
}
}
func buildOfficialAccountImagePrompts(workflow *officialAccountWorkflowState) []officialAccountImagePrompt {
title := firstNonEmpty(workflow.Shared.SelectedTitle, workflow.Form.Keyword, "公众号文章")
sections := extractOfficialAccountSections(workflow.Shared.Content)
prompts := []officialAccountImagePrompt{
{
Key: "cover",
Section: "封面配图",
Prompt: buildOfficialAccountVisualPrompt(workflow, title, "公众号封面主视觉"),
},
}
for idx, item := range sections {
if idx >= 3 {
break
}
prompts = append(prompts, officialAccountImagePrompt{
Key: fmt.Sprintf("section_%02d", idx+1),
Section: item.Title,
Prompt: buildOfficialAccountVisualPrompt(workflow, item.Title, strings.Join(item.Content, " ")),
})
}
return prompts
}
func buildOfficialAccountContentWithPrompts(content string, prompts []officialAccountImagePrompt) string {
content = strings.TrimSpace(content)
if content == "" {
return ""
}
blocks := []string{content, "", "## 配图提示词"}
for idx, item := range prompts {
blocks = append(blocks, fmt.Sprintf("- IMAGE_PROMPT_%02d [%s] %s", idx+1, firstNonEmpty(item.Section, "段落配图"), item.Prompt))
}
return strings.Join(blocks, "\n")
}
func buildOfficialAccountVisualPrompt(workflow *officialAccountWorkflowState, sectionTitle, context string) string {
keyword := firstNonEmpty(workflow.Form.Keyword, "业务主题")
tone := firstNonEmpty(workflow.Form.Tone, "专业但好懂")
domain := officialAccountBusinessDomainLabel(workflow.Form.BusinessDomain)
return fmt.Sprintf("realistic editorial illustration, Chinese business article cover, %s, focus on %s, %s, clean composition, natural lighting, brand-safe, no text, no watermark", domain, firstNonEmpty(sectionTitle, keyword), firstNonEmpty(context, tone))
}
func buildOfficialAccountImageArtifact(refs []officialAccountImageRef) string {
lines := []string{"# 图片结果", ""}
for idx, item := range refs {
lines = append(lines,
fmt.Sprintf("%d. %s", idx+1, firstNonEmpty(item.Section, item.Key)),
fmt.Sprintf("- 提示词:%s", item.Prompt),
fmt.Sprintf("- 图片地址:%s", item.URL),
fmt.Sprintf("- 生成模型:%s / %s", firstNonEmpty(item.RouteProvider, "unknown"), firstNonEmpty(item.RouteModel, "unknown")),
"",
)
}
return strings.Join(lines, "\n")
}
func generateOfficialAccountImages(taskID uint, workflow *officialAccountWorkflowState, prompts []officialAccountImagePrompt, user *model.User) (*config.RouteConfig, []officialAccountImageRef, *web.AppError) {
route, err := config.GetRoute("image_gen")
if err != nil {
return nil, nil, web.NewLLMNotConfigured("未找到可用的图片生成路由,请先检查 ai_config.json 中的 image_gen 配置")
}
if route.Category != "image" || strings.TrimSpace(route.FullURL) == "" || strings.TrimSpace(route.Model) == "" {
return nil, nil, web.NewLLMNotConfigured("图片生成路由配置不完整")
}
client := &http.Client{Timeout: 240 * time.Second}
refs := make([]officialAccountImageRef, 0, len(prompts))
for idx, item := range prompts {
imageSize := officialAccountImageSlotSize(idx)
requestSize := officialAccountImageRequestSize(imageSize)
started := time.Now()
imageURL, reqErr := requestOfficialAccountImage(client, route, item.Prompt, requestSize)
latencyMs := int(time.Since(started) / time.Millisecond)
ai.LogCall(ai.LogEntry{
UserID: user.ID,
Capability: ai.CapabilityImageGen,
Provider: route.Provider,
RouteID: route.RouteID,
Model: route.Model,
Success: reqErr == nil,
ErrorMessage: errorMessage(reqErr),
LatencyMs: latencyMs,
})
if reqErr != nil {
return nil, nil, web.NewLLMError(fmt.Sprintf("图片生成失败:%s", reqErr.Error()))
}
if strings.HasPrefix(strings.TrimSpace(imageURL), "data:image/") {
imageURL, reqErr = persistOfficialAccountDataURL(taskID, item.Key, imageURL)
if reqErr != nil {
return nil, nil, web.NewLLMError(fmt.Sprintf("图片保存失败:%s", reqErr.Error()))
}
}
refs = append(refs, officialAccountImageRef{
Key: item.Key,
Section: item.Section,
Prompt: item.Prompt,
URL: imageURL,
ImageSize: imageSize,
RouteID: route.RouteID,
RouteModel: route.Model,
RouteProvider: route.Provider,
})
}
return route, refs, nil
}
func requestOfficialAccountImage(client *http.Client, route *config.RouteConfig, prompt, size string) (string, error) {
payload := map[string]any{
"model": route.Model,
"prompt": strings.TrimSpace(prompt),
"n": 1,
"size": size,
}
imageURL, err := postOfficialAccountImageRequest(client, route, payload)
if err == nil {
return imageURL, nil
}
payload["response_format"] = "b64_json"
return postOfficialAccountImageRequest(client, route, payload)
}
func postOfficialAccountImageRequest(client *http.Client, route *config.RouteConfig, payload map[string]any) (string, error) {
body, err := json.Marshal(payload)
if err != nil {
return "", err
}
req, err := http.NewRequest(http.MethodPost, route.FullURL, bytes.NewReader(body))
if err != nil {
return "", err
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Accept", "application/json")
if strings.TrimSpace(route.APIKey) != "" {
req.Header.Set("Authorization", "Bearer "+strings.TrimSpace(route.APIKey))
}
resp, err := client.Do(req)
if err != nil {
return "", err
}
defer resp.Body.Close()
raw, err := io.ReadAll(resp.Body)
if err != nil {
return "", err
}
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
return "", fmt.Errorf("图片接口返回 %d:%s", resp.StatusCode, truncateOfficialAccountErrorBody(string(raw)))
}
var out struct {
Data []struct {
URL string `json:"url"`
B64JSON string `json:"b64_json"`
} `json:"data"`
}
if err := json.Unmarshal(raw, &out); err != nil {
return "", fmt.Errorf("图片结果解析失败:%w", err)
}
if len(out.Data) == 0 {
return "", fmt.Errorf("图片接口未返回 data")
}
first := out.Data[0]
if strings.TrimSpace(first.URL) != "" {
return strings.TrimSpace(first.URL), nil
}
if strings.TrimSpace(first.B64JSON) != "" {
return "data:image/png;base64," + strings.TrimSpace(first.B64JSON), nil
}
return "", fmt.Errorf("图片接口未返回 url 或 b64_json")
}
func officialAccountImageSlotSize(index int) string {
if index <= 0 {
return "landscape_16_9"
}
return "landscape_4_3"
}
func officialAccountImageRequestSize(size string) string {
switch strings.TrimSpace(size) {
case "landscape_16_9":
return "1536x864"
case "landscape_4_3":
return "1536x1152"
case "portrait_16_9":
return "864x1536"
case "portrait_4_3":
return "1152x1536"
case "square_hd":
return "1536x1536"
default:
return "1024x1024"
}
}
func truncateOfficialAccountErrorBody(raw string) string {
raw = strings.TrimSpace(raw)
if len(raw) <= 240 {
return raw
}
return raw[:240] + "..."
}
func errorMessage(err error) string {
if err == nil {
return ""
}
return err.Error()
}
func buildOfficialAccountPreviewMarkdown(workflow *officialAccountWorkflowState) string {
title := firstNonEmpty(workflow.Shared.SelectedTitle, workflow.Form.Keyword, "公众号文章")
sections := extractOfficialAccountSections(workflow.Shared.Content)
refs := workflow.Shared.ImageRefs
lines := []string{fmt.Sprintf("# %s", title), ""}
if len(refs) > 0 {
lines = append(lines, fmt.Sprintf("![封面配图](%s)", refs[0].URL), "")
}
for idx, item := range sections {
lines = append(lines, fmt.Sprintf("## %s", item.Title))
lines = append(lines, item.Content...)
lines = append(lines, "")
if idx+1 < len(refs) {
lines = append(lines, fmt.Sprintf("![段落配图](%s)", refs[idx+1].URL), "")
}
}
if len(sections) == 0 {
lines = append(lines, workflow.Shared.Content)
}
return strings.TrimSpace(strings.Join(lines, "\n"))
}
func buildOfficialAccountPages(workflow *officialAccountWorkflowState) []officialAccountPageChunk {
title := firstNonEmpty(workflow.Shared.SelectedTitle, workflow.Form.Keyword, "公众号文章")
sections := extractOfficialAccountSections(workflow.Shared.Content)
refs := workflow.Shared.ImageRefs
pages := make([]officialAccountPageChunk, 0, len(sections)+1)
intro := []string{fmt.Sprintf("# %s", title)}
if len(refs) > 0 {
intro = append(intro, "", fmt.Sprintf("![封面配图](%s)", refs[0].URL))
}
pages = append(pages, officialAccountPageChunk{
Index: 1,
Title: title,
Content: strings.Join(intro, "\n"),
})
for idx, item := range sections {
lines := []string{fmt.Sprintf("## %s", item.Title)}
lines = append(lines, item.Content...)
if idx+1 < len(refs) {
lines = append(lines, "", fmt.Sprintf("![段落配图](%s)", refs[idx+1].URL))
}
pages = append(pages, officialAccountPageChunk{
Index: idx + 2,
Title: item.Title,
Content: strings.TrimSpace(strings.Join(lines, "\n")),
})
}
return pages
}
func buildOfficialAccountPageMarkdown(pages []officialAccountPageChunk) string {
chunks := make([]string, 0, len(pages))
for _, item := range pages {
chunks = append(chunks, fmt.Sprintf("<!-- PAGE %d: %s -->\n%s", item.Index, item.Title, item.Content))
}
return strings.Join(chunks, "\n\n---\n\n")
}
func extractOfficialAccountSections(content string) []officialAccountSection {
lines := strings.Split(strings.ReplaceAll(content, "\r\n", "\n"), "\n")
sections := make([]officialAccountSection, 0, 6)
current := officialAccountSection{}
for _, raw := range lines {
line := strings.TrimSpace(raw)
if strings.HasPrefix(line, "# ") {
continue
}
if strings.HasPrefix(line, "## ") {
if current.Title != "" || len(current.Content) > 0 {
sections = append(sections, current)
}
current = officialAccountSection{Title: strings.TrimSpace(strings.TrimPrefix(line, "## "))}
continue
}
if current.Title == "" && line == "" {
continue
}
current.Content = append(current.Content, raw)
}
if current.Title != "" || len(current.Content) > 0 {
sections = append(sections, current)
}
for idx := range sections {
if sections[idx].Title == "" {
sections[idx].Title = fmt.Sprintf("内容页 %d", idx+1)
}
sections[idx].Content = trimEmptyLines(sections[idx].Content)
}
return sections
}
func trimEmptyLines(lines []string) []string {
start := 0
for start < len(lines) && strings.TrimSpace(lines[start]) == "" {
start++
}
end := len(lines)
for end > start && strings.TrimSpace(lines[end-1]) == "" {
end--
}
return append([]string(nil), lines[start:end]...)
}
func renderOfficialAccountHTML(markdown string) string {
var builder strings.Builder
builder.WriteString(`<article class="oa-preview">`)
listOpen := false
lines := strings.Split(strings.ReplaceAll(markdown, "\r\n", "\n"), "\n")
for _, raw := range lines {
line := strings.TrimSpace(raw)
if line == "" {
if listOpen {
builder.WriteString(`</ul>`)
listOpen = false
}
continue
}
if strings.HasPrefix(line, "![") && strings.Contains(line, "](") && strings.HasSuffix(line, ")") {
if listOpen {
builder.WriteString(`</ul>`)
listOpen = false
}
alt, src := parseOfficialAccountImageMarkdown(line)
builder.WriteString(fmt.Sprintf(`<figure class="oa-preview-figure"><img src="%s" alt="%s" /><figcaption>%s</figcaption></figure>`, html.EscapeString(src), html.EscapeString(alt), html.EscapeString(alt)))
continue
}
if strings.HasPrefix(line, "# ") {
if listOpen {
builder.WriteString(`</ul>`)
listOpen = false
}
builder.WriteString(fmt.Sprintf(`<h1>%s</h1>`, html.EscapeString(strings.TrimSpace(strings.TrimPrefix(line, "# ")))))
continue
}
if strings.HasPrefix(line, "## ") {
if listOpen {
builder.WriteString(`</ul>`)
listOpen = false
}
builder.WriteString(fmt.Sprintf(`<h2>%s</h2>`, html.EscapeString(strings.TrimSpace(strings.TrimPrefix(line, "## ")))))
continue
}
if strings.HasPrefix(line, "> ") {
if listOpen {
builder.WriteString(`</ul>`)
listOpen = false
}
builder.WriteString(fmt.Sprintf(`<blockquote>%s</blockquote>`, html.EscapeString(strings.TrimSpace(strings.TrimPrefix(line, "> ")))))
continue
}
if strings.HasPrefix(line, "- ") {
if !listOpen {
builder.WriteString(`<ul>`)
listOpen = true
}
builder.WriteString(fmt.Sprintf(`<li>%s</li>`, html.EscapeString(strings.TrimSpace(strings.TrimPrefix(line, "- ")))))
continue
}
if listOpen {
builder.WriteString(`</ul>`)
listOpen = false
}
builder.WriteString(fmt.Sprintf(`<p>%s</p>`, html.EscapeString(line)))
}
if listOpen {
builder.WriteString(`</ul>`)
}
builder.WriteString(`</article>`)
return builder.String()
}
func parseOfficialAccountImageMarkdown(line string) (string, string) {
startAlt := strings.Index(line, "![")
mid := strings.Index(line, "](")
end := strings.LastIndex(line, ")")
if startAlt < 0 || mid < 0 || end < 0 || end <= mid+2 {
return "配图", ""
}
alt := line[startAlt+2 : mid]
src := line[mid+2 : end]
return firstNonEmpty(strings.TrimSpace(alt), "配图"), strings.TrimSpace(src)
}
func ExportOfficialAccountDocument(c *gin.Context) {
user := middleware.CurrentUser(c)
task, ok := loadAccessibleWorkerTask(c, user)
if !ok {
return
}
if task.SpecialistKey != officialAccountSpecialistKey {
web.Fail(c, web.NewNotFoundError("当前事项不属于公众号助手"))
return
}
workflow := parseOfficialAccountWorkflow(task.ContextJSON)
article, err := ensureOfficialAccountArticle(task, workflow)
if err != nil {
web.Fail(c, web.NewBadRequest("读取公众号文章状态失败"))
return
}
syncWorkflowFromOfficialAccountArticle(&workflow, article)
format := strings.TrimSpace(strings.ToLower(c.DefaultQuery("format", "preview_md")))
filename := "official-account-export.md"
contentType := "text/markdown; charset=utf-8"
body := workflow.Shared.PreviewMarkdown
switch format {
case "preview_md":
if strings.TrimSpace(body) == "" {
body = buildOfficialAccountPreviewMarkdown(&workflow)
}
filename = "official-account-preview.md"
case "preview_html":
contentType = "text/html; charset=utf-8"
body = workflow.Shared.PreviewHTML
if strings.TrimSpace(body) == "" {
body = renderOfficialAccountHTML(buildOfficialAccountPreviewMarkdown(&workflow))
}
filename = "official-account-preview.html"
case "pagemd":
body = workflow.Shared.PageMarkdown
if strings.TrimSpace(body) == "" {
body = buildOfficialAccountPageMarkdown(buildOfficialAccountPages(&workflow))
}
filename = "official-account-pagemd.md"
default:
web.Fail(c, web.NewBadRequest("导出格式不支持"))
return
}
c.Header("Content-Disposition", fmt.Sprintf("attachment; filename*=UTF-8''%s", url.PathEscape(filename)))
c.Data(200, contentType, []byte(body))
}
@@ -0,0 +1,175 @@
package api
import (
"crypto/sha1"
"encoding/base64"
"fmt"
"os"
"path/filepath"
"strings"
"github.com/gin-gonic/gin"
"eaisalestrain/backend/internal/config"
"eaisalestrain/backend/internal/model"
"eaisalestrain/backend/internal/store"
"eaisalestrain/backend/internal/web"
)
func officialAccountGeneratedImageDir() string {
cfg := config.Load()
return filepath.Join(filepath.Dir(cfg.DBPath), "official_account_generated_images")
}
func normalizeOfficialAccountMediaForResponse(task *model.WorkerTask, workflow *officialAccountWorkflowState, article *model.OfficialAccountArticle) error {
if workflow == nil || article == nil || task == nil {
return nil
}
refs, replacements, changed, err := persistOfficialAccountImageRefs(task.ID, workflow.Shared.ImageRefs)
if err != nil {
return err
}
if !changed {
return nil
}
workflow.Shared.ImageRefs = refs
workflow.Shared.PreviewMarkdown = replaceOfficialAccountURLs(workflow.Shared.PreviewMarkdown, replacements)
workflow.Shared.PreviewHTML = replaceOfficialAccountURLs(workflow.Shared.PreviewHTML, replacements)
workflow.Shared.PageMarkdown = replaceOfficialAccountURLs(workflow.Shared.PageMarkdown, replacements)
for idx := range workflow.Shared.Pages {
workflow.Shared.Pages[idx].Content = replaceOfficialAccountURLs(workflow.Shared.Pages[idx].Content, replacements)
}
syncOfficialAccountArticleFromWorkflow(article, *workflow)
task.ContextJSON = marshalOfficialAccountWorkflow(*workflow)
if err := store.DB.Save(article).Error; err != nil {
return err
}
if err := store.DB.Save(task).Error; err != nil {
return err
}
return nil
}
func persistOfficialAccountImageRefs(taskID uint, refs []officialAccountImageRef) ([]officialAccountImageRef, map[string]string, bool, error) {
if len(refs) == 0 {
return refs, map[string]string{}, false, nil
}
next := make([]officialAccountImageRef, len(refs))
replacements := map[string]string{}
changed := false
for idx, item := range refs {
next[idx] = item
if !strings.HasPrefix(strings.TrimSpace(item.URL), "data:image/") {
continue
}
localURL, err := persistOfficialAccountDataURL(taskID, item.Key, item.URL)
if err != nil {
return refs, nil, false, err
}
replacements[item.URL] = localURL
next[idx].URL = localURL
changed = true
}
return next, replacements, changed, nil
}
func persistOfficialAccountDataURL(taskID uint, key, raw string) (string, error) {
parts := strings.SplitN(strings.TrimSpace(raw), ",", 2)
if len(parts) != 2 {
return "", fmt.Errorf("图片数据格式错误")
}
header := parts[0]
body := parts[1]
if !strings.Contains(header, ";base64") {
return "", fmt.Errorf("图片数据缺少 base64 头")
}
ext := ".png"
switch {
case strings.HasPrefix(header, "data:image/jpeg"):
ext = ".jpg"
case strings.HasPrefix(header, "data:image/webp"):
ext = ".webp"
case strings.HasPrefix(header, "data:image/gif"):
ext = ".gif"
}
payload, err := base64.StdEncoding.DecodeString(body)
if err != nil {
return "", err
}
dir := officialAccountGeneratedImageDir()
if err := os.MkdirAll(dir, 0o755); err != nil {
return "", err
}
hash := sha1.Sum(payload)
filename := fmt.Sprintf("task_%d_%s_%x%s", taskID, sanitizeOfficialAccountImageKey(key), hash[:6], ext)
fullPath := filepath.Join(dir, filename)
if _, err := os.Stat(fullPath); err != nil {
if writeErr := os.WriteFile(fullPath, payload, 0o644); writeErr != nil {
return "", writeErr
}
}
return "/api/official-account/generated-images/" + filename, nil
}
func sanitizeOfficialAccountImageKey(key string) string {
trimmed := strings.TrimSpace(strings.ToLower(key))
trimmed = strings.ReplaceAll(trimmed, " ", "_")
trimmed = strings.ReplaceAll(trimmed, "/", "_")
trimmed = strings.ReplaceAll(trimmed, "\\", "_")
if trimmed == "" {
return "image"
}
return trimmed
}
func replaceOfficialAccountURLs(content string, replacements map[string]string) string {
next := content
for oldURL, newURL := range replacements {
next = strings.ReplaceAll(next, oldURL, newURL)
}
return next
}
func ServeOfficialAccountGeneratedImage(c *gin.Context) {
filename := filepath.Base(strings.TrimSpace(c.Param("filename")))
if filename == "." || filename == "" {
web.Fail(c, web.NewNotFoundError("图片不存在"))
return
}
fullPath := filepath.Join(officialAccountGeneratedImageDir(), filename)
if _, err := os.Stat(fullPath); err != nil {
web.Fail(c, web.NewNotFoundError("图片不存在"))
return
}
c.File(fullPath)
}
func compactOfficialAccountRunsForResponse(items []model.WorkerRun) []model.WorkerRun {
next := make([]model.WorkerRun, len(items))
for idx, item := range items {
next[idx] = item
if len(item.OutputJSON) > 8192 || strings.Contains(item.OutputJSON, "data:image/") {
next[idx].OutputJSON = ""
}
if len(item.InputJSON) > 4096 {
next[idx].InputJSON = ""
}
}
return next
}
func compactOfficialAccountArtifactsForResponse(items []model.WorkerArtifact) []model.WorkerArtifact {
next := make([]model.WorkerArtifact, len(items))
for idx, item := range items {
next[idx] = item
if len(item.ContentJSON) > 8192 || strings.Contains(item.ContentJSON, "data:image/") {
next[idx].ContentJSON = ""
}
if len(item.ContentText) > 20000 {
next[idx].ContentText = item.ContentText[:20000] + "\n\n[内容过长,已在画布接口中省略]"
}
}
return next
}
@@ -20,11 +20,15 @@ import (
const (
officialAccountSpecialistKey = "wechat-official-account"
officialAccountStepKeyConfig = "task_configuration"
officialAccountStepKeyTopic = "topic_recommendation"
officialAccountStepKeyConfig = "task_configuration"
officialAccountStepKeyTopic = "topic_recommendation"
officialAccountStepKeyTitle = "title_generation"
officialAccountStepKeyOutline = "outline_generation"
officialAccountStepKeyContent = "content_creation"
officialAccountStepKeyImagePrompts = "image_prompt_generation"
officialAccountStepKeyImages = "image_generation"
officialAccountStepKeyPreview = "preview_export"
officialAccountStepKeyPageMD = "page_markdown"
officialAccountStepStatusPending = "pending"
officialAccountStepStatusRunning = "running"
@@ -33,11 +37,12 @@ const (
)
type officialAccountTaskCreateReq struct {
Keyword string `json:"keyword"`
Audience string `json:"audience"`
Goal string `json:"goal"`
Tone string `json:"tone"`
Requirements string `json:"requirements"`
BusinessDomain string `json:"business_domain"`
Keyword string `json:"keyword"`
Audience string `json:"audience"`
Goal string `json:"goal"`
Tone string `json:"tone"`
Requirements string `json:"requirements"`
OutlineStyle string `json:"outline_style"`
OutlineWords int `json:"outline_words"`
ContentTargetWords int `json:"content_target_words"`
@@ -47,10 +52,13 @@ type officialAccountStepReq struct {
Topic string `json:"topic"`
Title string `json:"title"`
Outline string `json:"outline"`
Content string `json:"content"`
ContentWithPrompts string `json:"content_with_prompts"`
Requirements string `json:"requirements"`
}
type officialAccountTaskUpdateReq struct {
BusinessDomain string `json:"business_domain"`
Keyword string `json:"keyword"`
Audience string `json:"audience"`
Goal string `json:"goal"`
@@ -70,11 +78,12 @@ type officialAccountWorkflowState struct {
}
type officialAccountForm struct {
Keyword string `json:"keyword"`
Audience string `json:"audience"`
Goal string `json:"goal"`
Tone string `json:"tone"`
Requirements string `json:"requirements"`
BusinessDomain string `json:"business_domain"`
Keyword string `json:"keyword"`
Audience string `json:"audience"`
Goal string `json:"goal"`
Tone string `json:"tone"`
Requirements string `json:"requirements"`
OutlineStyle string `json:"outline_style"`
OutlineWords int `json:"outline_words"`
ContentTargetWords int `json:"content_target_words"`
@@ -87,6 +96,13 @@ type officialAccountSharedState struct {
SelectedTitle string `json:"selected_title"`
Outline string `json:"outline"`
Content string `json:"content"`
ContentWithPrompts string `json:"content_with_prompts"`
ImagePrompts []officialAccountImagePrompt `json:"image_prompts"`
ImageRefs []officialAccountImageRef `json:"image_refs"`
PreviewMarkdown string `json:"preview_markdown"`
PreviewHTML string `json:"preview_html"`
PageMarkdown string `json:"page_markdown"`
Pages []officialAccountPageChunk `json:"pages"`
}
type officialAccountStepState struct {
@@ -108,6 +124,29 @@ type officialAccountTopicCandidate struct {
Heat string `json:"heat"`
}
type officialAccountImagePrompt struct {
Key string `json:"key"`
Section string `json:"section"`
Prompt string `json:"prompt"`
}
type officialAccountImageRef struct {
Key string `json:"key"`
Section string `json:"section"`
Prompt string `json:"prompt"`
URL string `json:"url"`
ImageSize string `json:"image_size"`
RouteID string `json:"route_id,omitempty"`
RouteModel string `json:"route_model,omitempty"`
RouteProvider string `json:"route_provider,omitempty"`
}
type officialAccountPageChunk struct {
Index int `json:"index"`
Title string `json:"title"`
Content string `json:"content"`
}
type officialAccountStepExecution struct {
Summary string
TaskStatus string
@@ -229,6 +268,7 @@ func UpdateOfficialAccountTask(c *gin.Context) {
}
syncWorkflowFromOfficialAccountArticle(&oldWorkflow, article)
changed := oldWorkflow.Form.Keyword != req.Keyword ||
oldWorkflow.Form.BusinessDomain != resolveOfficialAccountBusinessDomain(req.BusinessDomain, req.Keyword) ||
oldWorkflow.Form.Audience != firstNonEmpty(req.Audience, "公众号读者") ||
oldWorkflow.Form.Goal != firstNonEmpty(req.Goal, "输出一篇可发布的公众号文章") ||
oldWorkflow.Form.Tone != firstNonEmpty(req.Tone, "专业但好懂") ||
@@ -240,6 +280,7 @@ func UpdateOfficialAccountTask(c *gin.Context) {
workflow := oldWorkflow
if changed {
workflow = newOfficialAccountWorkflow(officialAccountTaskCreateReq{
BusinessDomain: req.BusinessDomain,
Keyword: req.Keyword,
Audience: req.Audience,
Goal: req.Goal,
@@ -250,6 +291,7 @@ func UpdateOfficialAccountTask(c *gin.Context) {
ContentTargetWords: req.ContentTargetWords,
})
} else {
workflow.Form.BusinessDomain = resolveOfficialAccountBusinessDomain(req.BusinessDomain, req.Keyword)
workflow.Form.Keyword = req.Keyword
workflow.Form.Audience = firstNonEmpty(req.Audience, "公众号读者")
workflow.Form.Goal = firstNonEmpty(req.Goal, "输出一篇可发布的公众号文章")
@@ -263,6 +305,7 @@ func UpdateOfficialAccountTask(c *gin.Context) {
if changed {
task.Title = buildOfficialAccountTaskTitle(req.Keyword)
task.Summary = buildOfficialAccountTaskSummary(officialAccountTaskCreateReq{
BusinessDomain: req.BusinessDomain,
Keyword: req.Keyword,
Audience: req.Audience,
})
@@ -300,6 +343,7 @@ func UpdateOfficialAccountTask(c *gin.Context) {
configSummary := fmt.Sprintf("任务配置已更新为「%s」,流程已重置,等待重新执行。", req.Keyword)
outputJSON, _ := json.Marshal(gin.H{
"summary": configSummary,
"business_domain": resolveOfficialAccountBusinessDomain(req.BusinessDomain, req.Keyword),
"keyword": req.Keyword,
"audience": firstNonEmpty(req.Audience, "公众号读者"),
"goal": firstNonEmpty(req.Goal, "输出一篇可发布的公众号文章"),
@@ -359,6 +403,8 @@ func ExecuteOfficialAccountWorkflowStep(c *gin.Context) {
req.Topic = strings.TrimSpace(req.Topic)
req.Title = strings.TrimSpace(req.Title)
req.Outline = strings.TrimSpace(req.Outline)
req.Content = strings.TrimSpace(req.Content)
req.ContentWithPrompts = strings.TrimSpace(req.ContentWithPrompts)
req.Requirements = strings.TrimSpace(req.Requirements)
workflow := parseOfficialAccountWorkflow(task.ContextJSON)
@@ -398,6 +444,8 @@ func ExecuteOfficialAccountWorkflowStep(c *gin.Context) {
"topic": req.Topic,
"title": req.Title,
"outline": req.Outline,
"content": req.Content,
"content_with_prompts": req.ContentWithPrompts,
"requirements": req.Requirements,
})
outputPayload := cloneOutputMap(execution.Output)
@@ -465,6 +513,10 @@ func respondOfficialAccountWorkflow(c *gin.Context, task model.WorkerTask) {
workflow := parseOfficialAccountWorkflow(task.ContextJSON)
article, _ := ensureOfficialAccountArticle(task, workflow)
syncWorkflowFromOfficialAccountArticle(&workflow, article)
if err := normalizeOfficialAccountMediaForResponse(&task, &workflow, article); err != nil {
web.Fail(c, web.NewBadRequest("整理公众号图片结果失败"))
return
}
var artifacts []model.WorkerArtifact
if err := store.DB.Where("task_id = ?", task.ID).Order("created_at DESC, id DESC").Find(&artifacts).Error; err != nil {
@@ -477,6 +529,8 @@ func respondOfficialAccountWorkflow(c *gin.Context, task model.WorkerTask) {
web.Fail(c, web.NewBadRequest("查询公众号运行记录失败"))
return
}
artifacts = compactOfficialAccountArtifactsForResponse(artifacts)
runs = compactOfficialAccountRunsForResponse(runs)
web.OK(c, gin.H{
"task": task,
@@ -493,6 +547,7 @@ func newOfficialAccountWorkflow(req officialAccountTaskCreateReq) officialAccoun
WorkflowType: "wechat_official_account",
CurrentStep: officialAccountStepKeyTopic,
Form: officialAccountForm{
BusinessDomain: resolveOfficialAccountBusinessDomain(req.BusinessDomain, req.Keyword),
Keyword: req.Keyword,
Audience: firstNonEmpty(req.Audience, "公众号读者"),
Goal: firstNonEmpty(req.Goal, "输出一篇可发布的公众号文章"),
@@ -506,6 +561,9 @@ func newOfficialAccountWorkflow(req officialAccountTaskCreateReq) officialAccoun
Shared: officialAccountSharedState{
TopicCandidates: []officialAccountTopicCandidate{},
TitleCandidates: []string{},
ImagePrompts: []officialAccountImagePrompt{},
ImageRefs: []officialAccountImageRef{},
Pages: []officialAccountPageChunk{},
},
}
for _, item := range []struct {
@@ -517,6 +575,10 @@ func newOfficialAccountWorkflow(req officialAccountTaskCreateReq) officialAccoun
{officialAccountStepKeyTitle, officialAccountStepLabel(officialAccountStepKeyTitle), "等待生成标题"},
{officialAccountStepKeyOutline, officialAccountStepLabel(officialAccountStepKeyOutline), "等待生成提纲"},
{officialAccountStepKeyContent, officialAccountStepLabel(officialAccountStepKeyContent), "等待创作正文"},
{officialAccountStepKeyImagePrompts, officialAccountStepLabel(officialAccountStepKeyImagePrompts), "等待生成配图提示词"},
{officialAccountStepKeyImages, officialAccountStepLabel(officialAccountStepKeyImages), "等待生成图片"},
{officialAccountStepKeyPreview, officialAccountStepLabel(officialAccountStepKeyPreview), "等待生成预览稿"},
{officialAccountStepKeyPageMD, officialAccountStepLabel(officialAccountStepKeyPageMD), "等待生成分页 MD"},
} {
workflow.Steps[item.key] = &officialAccountStepState{
Key: item.key,
@@ -563,6 +625,9 @@ func parseOfficialAccountWorkflow(value string) officialAccountWorkflowState {
if workflow.Form.Audience == "" {
workflow.Form.Audience = "公众号读者"
}
if workflow.Form.BusinessDomain == "" {
workflow.Form.BusinessDomain = resolveOfficialAccountBusinessDomain("", workflow.Form.Keyword)
}
if workflow.Form.Goal == "" {
workflow.Form.Goal = "输出一篇可发布的公众号文章"
}
@@ -600,6 +665,15 @@ func parseOfficialAccountWorkflow(value string) officialAccountWorkflowState {
if workflow.Shared.TitleCandidates == nil {
workflow.Shared.TitleCandidates = []string{}
}
if workflow.Shared.ImagePrompts == nil {
workflow.Shared.ImagePrompts = []officialAccountImagePrompt{}
}
if workflow.Shared.ImageRefs == nil {
workflow.Shared.ImageRefs = []officialAccountImageRef{}
}
if workflow.Shared.Pages == nil {
workflow.Shared.Pages = []officialAccountPageChunk{}
}
if workflow.CurrentStep == "" {
workflow.CurrentStep = officialAccountStepKeyTopic
}
@@ -616,7 +690,7 @@ func marshalOfficialAccountWorkflow(workflow officialAccountWorkflowState) strin
func isOfficialAccountStep(stepKey string) bool {
switch stepKey {
case officialAccountStepKeyTopic, officialAccountStepKeyTitle, officialAccountStepKeyOutline, officialAccountStepKeyContent:
case officialAccountStepKeyTopic, officialAccountStepKeyTitle, officialAccountStepKeyOutline, officialAccountStepKeyContent, officialAccountStepKeyImagePrompts, officialAccountStepKeyImages, officialAccountStepKeyPreview, officialAccountStepKeyPageMD:
return true
default:
return false
@@ -633,6 +707,14 @@ func officialAccountStepLabel(stepKey string) string {
return "提纲生成"
case officialAccountStepKeyContent:
return "正文创作"
case officialAccountStepKeyImagePrompts:
return "配图提示词"
case officialAccountStepKeyImages:
return "图片生成"
case officialAccountStepKeyPreview:
return "预览与导出"
case officialAccountStepKeyPageMD:
return "分页 MD"
default:
return "工作流步骤"
}
@@ -646,8 +728,16 @@ func officialAccountNextStep(stepKey string) string {
return officialAccountStepKeyOutline
case officialAccountStepKeyOutline:
return officialAccountStepKeyContent
case officialAccountStepKeyContent:
return officialAccountStepKeyImagePrompts
case officialAccountStepKeyImagePrompts:
return officialAccountStepKeyImages
case officialAccountStepKeyImages:
return officialAccountStepKeyPreview
case officialAccountStepKeyPreview:
return officialAccountStepKeyPageMD
default:
return officialAccountStepKeyContent
return officialAccountStepKeyPageMD
}
}
@@ -724,13 +814,43 @@ func runOfficialAccountStep(stepKey string, workflow *officialAccountWorkflowSta
return officialAccountStepExecution{}, web.NewBadRequest("请先生成提纲,或手动输入提纲")
}
return executeOfficialAccountContentStep(workflow, article, user, now), nil
case officialAccountStepKeyImagePrompts:
if req.Content != "" {
workflow.Shared.Content = req.Content
}
if workflow.Shared.Content == "" {
return officialAccountStepExecution{}, web.NewBadRequest("请先生成正文,再生成配图提示词")
}
return executeOfficialAccountImagePromptStep(workflow, article, user, now), nil
case officialAccountStepKeyImages:
if req.ContentWithPrompts != "" {
workflow.Shared.ContentWithPrompts = req.ContentWithPrompts
}
if len(workflow.Shared.ImagePrompts) == 0 && workflow.Shared.ContentWithPrompts == "" {
return officialAccountStepExecution{}, web.NewBadRequest("请先生成配图提示词")
}
execution, appErr := executeOfficialAccountImageGenerationStep(workflow, article, user, now)
if appErr != nil {
return officialAccountStepExecution{}, appErr
}
return execution, nil
case officialAccountStepKeyPreview:
if workflow.Shared.Content == "" {
return officialAccountStepExecution{}, web.NewBadRequest("请先生成正文,再生成预览稿")
}
return executeOfficialAccountPreviewStep(workflow, article, user, now), nil
case officialAccountStepKeyPageMD:
if workflow.Shared.Content == "" {
return officialAccountStepExecution{}, web.NewBadRequest("请先生成正文,再生成分页 MD")
}
return executeOfficialAccountPageMarkdownStep(workflow, article, user, now), nil
default:
return officialAccountStepExecution{}, web.NewBadRequest("工作流步骤不存在")
}
}
func executeOfficialAccountTopicStep(workflow *officialAccountWorkflowState, article *model.OfficialAccountArticle, user *model.User, now time.Time) officialAccountStepExecution {
hotspots, hotspotLogs, _ := ensureOfficialAccountHotspots(false)
hotspots, hotspotLogs, _ := ensureOfficialAccountHotspots(workflow.Form, false)
hotspots = sortOfficialAccountHotspots(hotspots)
candidates := buildOfficialAccountTopicCandidatesFromHotspots(workflow.Form, hotspots)
_ = tryFillOfficialAccountTopicsWithAI(workflow.Form, hotspots, user, &candidates)
@@ -1123,8 +1243,20 @@ func buildOfficialAccountTaskSummary(req officialAccountTaskCreateReq) string {
}
func buildOfficialAccountTaskSummaryFromWorkflow(workflow officialAccountWorkflowState) string {
if workflow.Shared.Content != "" {
return "正文草稿已生成,等待人工校对。"
if workflow.Shared.PageMarkdown != "" {
return "分页 MD 已生成,可直接预览与导出。"
}
if workflow.Shared.PreviewHTML != "" {
return "图文预览稿已生成,可检查排版并导出。"
}
if len(workflow.Shared.ImageRefs) > 0 {
return fmt.Sprintf("已生成 %d 张配图,可继续预览与导出。", len(workflow.Shared.ImageRefs))
}
if len(workflow.Shared.ImagePrompts) > 0 || workflow.Shared.ContentWithPrompts != "" {
return "配图提示词已生成,可继续生成图片。"
}
if workflow.Shared.Content != "" {
return "正文草稿已生成,可继续补齐配图与预览。"
}
if workflow.Shared.Outline != "" {
return "提纲已生成,可继续创作正文。"
@@ -1225,4 +1357,3 @@ func firstNonEmpty(values ...string) string {
}
return ""
}
@@ -51,6 +51,8 @@ func RegisterRoutes(r *gin.Engine, cfg *config.Config) {
r.GET("/api/official-account/tasks/:id/workflow", middleware.Auth(cfg), GetOfficialAccountWorkflow)
r.PUT("/api/official-account/tasks/:id", middleware.Auth(cfg), UpdateOfficialAccountTask)
r.POST("/api/official-account/tasks/:id/steps/:stepKey", middleware.Auth(cfg), ExecuteOfficialAccountWorkflowStep)
r.GET("/api/official-account/tasks/:id/export", middleware.Auth(cfg), ExportOfficialAccountDocument)
r.GET("/api/official-account/generated-images/:filename", middleware.Auth(cfg), ServeOfficialAccountGeneratedImage)
r.GET("/api/connectors", middleware.Auth(cfg), ListConnectors)
r.GET("/api/connectors/:key", middleware.Auth(cfg), GetConnector)
r.POST("/api/connectors/:key/query", middleware.Auth(cfg), QueryConnector)
@@ -87,6 +87,10 @@ func GetWorkerTaskDetail(c *gin.Context) {
web.Fail(c, web.NewBadRequest("查询运行记录失败"))
return
}
if task.SpecialistKey == officialAccountSpecialistKey {
artifacts = compactOfficialAccountArtifactsForResponse(artifacts)
runs = compactOfficialAccountRunsForResponse(runs)
}
web.OK(c, gin.H{
"task": task,
@@ -2,7 +2,7 @@ package model
import "time"
// OfficialAccountArticle 公众号文章状态表,对齐工作流中的主题/标题/提纲/正文显式传递。
// OfficialAccountArticle 公众号文章状态表,对齐工作流中的主题/标题/提纲/正文/配图/预览/分页显式传递。
type OfficialAccountArticle struct {
ID uint `gorm:"primaryKey" json:"id"`
TaskID uint `gorm:"not null;uniqueIndex" json:"task_id"`
@@ -24,6 +24,13 @@ type OfficialAccountArticle struct {
Outline string `gorm:"type:text" json:"outline"`
ContentTargetWords int `gorm:"not null;default:1400" json:"content_target_words"`
Content string `gorm:"type:text" json:"content"`
ContentWithPrompts string `gorm:"type:text" json:"content_with_prompts"`
ImagePromptsJSON string `gorm:"type:text" json:"image_prompts_json"`
ImageRefsJSON string `gorm:"type:text" json:"image_refs_json"`
PreviewMarkdown string `gorm:"type:text" json:"preview_markdown"`
PreviewHTML string `gorm:"type:text" json:"preview_html"`
PageMarkdown string `gorm:"type:text" json:"page_markdown"`
PagesJSON string `gorm:"type:text" json:"pages_json"`
Status string `gorm:"size:32;not null;default:draft;index" json:"status"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
@@ -17,3 +17,11 @@ export function executeOfficialAccountWorkflowStep(taskId, stepKey, data) {
timeout: 0,
})
}
export function exportOfficialAccountDocument(taskId, format) {
return http.get(`/official-account/tasks/${taskId}/export`, {
params: { format },
responseType: 'blob',
timeout: 0,
})
}
@@ -330,6 +330,82 @@
<pre class="content-preview">{{ workflow.shared.content || '还没有生成正文草稿。' }}</pre>
</template>
<template v-else-if="id === stepKeyImagePrompts">
<div class="inline-meta">当前已提炼 {{ (workflow.shared.image_prompts || []).length }} 条配图提示词</div>
<div v-if="(workflow.shared.image_prompts || []).length" class="prompt-list">
<div v-for="(item, promptIndex) in workflow.shared.image_prompts || []" :key="item.key" class="prompt-card">
<div class="prompt-header">
<div>
<div class="prompt-badge">{{ getPromptSlotLabel(item, promptIndex) }}</div>
<div class="prompt-title">{{ item.section || getPromptSlotLabel(item, promptIndex) }}</div>
</div>
<div class="prompt-chip">用于配图</div>
</div>
<div class="prompt-meta">用途:{{ getPromptPurpose(item, promptIndex) }}</div>
<div class="prompt-body">{{ item.prompt }}</div>
</div>
</div>
<div v-else class="empty-text">当前还没有生成配图提示词。</div>
<div class="node-panel-title">原始标记稿</div>
<div class="inline-meta">下面这段是给系统继续传递用的结构化正文,不是给人直接阅读的最终文案。</div>
<pre class="content-preview prompt-raw-preview">{{ workflow.shared.content_with_prompts || '还没有生成带提示词的正文稿。' }}</pre>
</template>
<template v-else-if="id === stepKeyImages">
<div class="node-panel-title">图片结果</div>
<div class="inline-meta">当前已生成 {{ (workflow.shared.image_refs || []).length }} 张图片</div>
<div v-if="(workflow.shared.image_refs || []).length" class="image-grid">
<div v-for="item in workflow.shared.image_refs || []" :key="item.key" class="image-card">
<img class="image-thumb" :src="item.url" :alt="item.section || item.key" />
<div class="image-section">{{ item.section || item.key }}</div>
<div class="image-prompt">{{ item.prompt }}</div>
</div>
</div>
<div v-else class="empty-text">当前还没有图片结果。</div>
</template>
<template v-else-if="id === stepKeyPreview">
<div class="node-panel-title">预览与导出</div>
<div class="action-row action-row-start">
<el-button
size="small"
:loading="exportingFormat === 'preview_md'"
@click.stop="downloadExport('preview_md', 'official-account-preview.md')"
>
导出预览 MD
</el-button>
<el-button
size="small"
:loading="exportingFormat === 'preview_html'"
@click.stop="downloadExport('preview_html', 'official-account-preview.html')"
>
导出预览 HTML
</el-button>
</div>
<div v-if="workflow.shared.preview_html" class="preview-render" v-html="workflow.shared.preview_html" />
<div v-else class="empty-text">当前还没有生成预览稿。</div>
</template>
<template v-else-if="id === stepKeyPageMD">
<div class="node-panel-title">分页 MD</div>
<div class="action-row action-row-start">
<el-button
size="small"
:loading="exportingFormat === 'pagemd'"
@click.stop="downloadExport('pagemd', 'official-account-pagemd.md')"
>
导出分页 MD
</el-button>
</div>
<div class="inline-meta">当前共 {{ (workflow.shared.pages || []).length }} 页</div>
<div v-if="(workflow.shared.pages || []).length" class="page-chip-list">
<span v-for="item in workflow.shared.pages || []" :key="`${item.index}-${item.title}`" class="page-chip">
第 {{ item.index }} 页 · {{ item.title }}
</span>
</div>
<pre class="content-preview pagemd-preview">{{ workflow.shared.page_markdown || '还没有生成分页 MD。' }}</pre>
</template>
<div class="node-panel-title">节点输出</div>
<div v-if="getOutputRows(id).length" class="output-list">
<div v-for="row in getOutputRows(id)" :key="row.label" class="output-row">
@@ -372,6 +448,7 @@ import { useWorkerRuntimeStore } from '@/store/workerRuntime'
import {
createOfficialAccountTask,
executeOfficialAccountWorkflowStep,
exportOfficialAccountDocument,
getOfficialAccountWorkflow,
updateOfficialAccountTask,
} from '@/api/officialAccount'
@@ -384,6 +461,10 @@ const stepKeyTopic = 'topic_recommendation'
const stepKeyTitle = 'title_generation'
const stepKeyOutline = 'outline_generation'
const stepKeyContent = 'content_creation'
const stepKeyImagePrompts = 'image_prompt_generation'
const stepKeyImages = 'image_generation'
const stepKeyPreview = 'preview_export'
const stepKeyPageMD = 'page_markdown'
const stepKeyConfig = 'task_configuration'
const creatingTask = ref(false)
@@ -398,11 +479,12 @@ const fittedTaskId = ref('')
const canvasNodes = ref([])
const canvasEdges = ref([])
const canvasZoomOnScroll = ref(true)
const canvasTranslateExtent = [[-320, -260], [3280, 1560]]
const canvasNodeExtent = [[40, 80], [2480, 1120]]
const canvasTranslateExtent = [[-320, -260], [4680, 1680]]
const canvasNodeExtent = [[40, 80], [4320, 1320]]
const expandedHotspotKeys = ref([])
const pollingToken = ref(0)
const pollingStepKey = ref('')
const exportingFormat = ref('')
const taskForm = reactive({
keyword: '',
@@ -420,6 +502,8 @@ const stepPayload = reactive({
topic: '',
title: '',
outline: '',
content: '',
contentWithPrompts: '',
requirements: '',
})
@@ -524,6 +608,10 @@ const workflowSteps = computed(() => {
buildStepCard(3, stepKeyTitle, steps[stepKeyTitle]),
buildStepCard(4, stepKeyOutline, steps[stepKeyOutline]),
buildStepCard(5, stepKeyContent, steps[stepKeyContent]),
buildStepCard(6, stepKeyImagePrompts, steps[stepKeyImagePrompts]),
buildStepCard(7, stepKeyImages, steps[stepKeyImages]),
buildStepCard(8, stepKeyPreview, steps[stepKeyPreview]),
buildStepCard(9, stepKeyPageMD, steps[stepKeyPageMD]),
]
})
@@ -558,6 +646,18 @@ function getOutputRows(stepKey) {
if (stepKey === stepKeyContent) {
return [{ label: '正文状态', value: output.content ? '已生成' : '未生成' }]
}
if (stepKey === stepKeyImagePrompts) {
return [{ label: '提示词数量', value: String((output.image_prompts || []).length || 0) }]
}
if (stepKey === stepKeyImages) {
return [{ label: '图片数量', value: String((output.image_refs || []).length || 0) }]
}
if (stepKey === stepKeyPreview) {
return [{ label: '预览状态', value: output.preview_html ? '已生成' : '未生成' }]
}
if (stepKey === stepKeyPageMD) {
return [{ label: '分页数量', value: String((output.pages || []).length || 0) }]
}
return []
}
@@ -596,7 +696,7 @@ const flowNodes = computed(() => {
return nodes.map((item, index) => ({
id: item.id,
type: 'official-step',
position: { x: 80 + index * 456, y: 150 },
position: { x: 80 + index * 410, y: 150 },
selectable: true,
draggable: true,
data: {
@@ -612,6 +712,10 @@ const flowEdges = computed(() => {
[stepKeyTopic, stepKeyTitle],
[stepKeyTitle, stepKeyOutline],
[stepKeyOutline, stepKeyContent],
[stepKeyContent, stepKeyImagePrompts],
[stepKeyImagePrompts, stepKeyImages],
[stepKeyImages, stepKeyPreview],
[stepKeyPreview, stepKeyPageMD],
]
return chain.map(([source, target], index) => ({
id: `edge-${source}-${target}`,
@@ -670,6 +774,10 @@ function stepPreview(key) {
if (key === stepKeyTitle) return workflow.value?.shared?.selected_title || '围绕选题生成标题候选'
if (key === stepKeyOutline) return workflow.value?.shared?.outline ? firstLine(workflow.value.shared.outline) : '输出文章结构提纲'
if (key === stepKeyContent) return workflow.value?.shared?.content ? `${workflow.value.shared.content.slice(0, 28)}...` : '输出正文草稿'
if (key === stepKeyImagePrompts) return (workflow.value?.shared?.image_prompts || []).length ? `已提炼 ${(workflow.value?.shared?.image_prompts || []).length} 条配图提示词` : '从正文提炼配图提示词'
if (key === stepKeyImages) return (workflow.value?.shared?.image_refs || []).length ? `已生成 ${(workflow.value?.shared?.image_refs || []).length} 张图片` : '根据提示词生成配图'
if (key === stepKeyPreview) return workflow.value?.shared?.preview_html ? '已生成图文预览稿' : '生成图文预览与导出稿'
if (key === stepKeyPageMD) return workflow.value?.shared?.page_markdown ? '已生成分页 Markdown' : '输出分页 MD'
return '等待执行'
}
@@ -680,6 +788,10 @@ function canRunStep(key) {
if (key === stepKeyTitle) return Boolean(stepPayload.topic || workflow.value?.shared?.selected_topic)
if (key === stepKeyOutline) return Boolean(stepPayload.title || workflow.value?.shared?.selected_title)
if (key === stepKeyContent) return Boolean(stepPayload.outline || workflow.value?.shared?.outline)
if (key === stepKeyImagePrompts) return Boolean(stepPayload.content || workflow.value?.shared?.content)
if (key === stepKeyImages) return Boolean(stepPayload.contentWithPrompts || workflow.value?.shared?.content_with_prompts || (workflow.value?.shared?.image_prompts || []).length)
if (key === stepKeyPreview) return Boolean(workflow.value?.shared?.content)
if (key === stepKeyPageMD) return Boolean(workflow.value?.shared?.content)
return false
}
@@ -689,6 +801,10 @@ function stepHintText(key) {
if (key === stepKeyTitle) return '请先在上一节点确认选题'
if (key === stepKeyOutline) return '请先生成或选择标题'
if (key === stepKeyContent) return '请先生成或填写提纲'
if (key === stepKeyImagePrompts) return '请先生成正文'
if (key === stepKeyImages) return '请先生成配图提示词'
if (key === stepKeyPreview) return '请先生成正文与图片'
if (key === stepKeyPageMD) return '请先生成预览稿'
return '当前节点暂不可执行'
}
@@ -712,6 +828,10 @@ function stepLabel(key) {
if (key === stepKeyTitle) return '选题后生成标题'
if (key === stepKeyOutline) return '提纲生成'
if (key === stepKeyContent) return '正文创作'
if (key === stepKeyImagePrompts) return '配图提示词'
if (key === stepKeyImages) return '图片生成'
if (key === stepKeyPreview) return '预览与导出'
if (key === stepKeyPageMD) return '分页 MD'
return '节点'
}
@@ -762,12 +882,33 @@ function toggleNodeExpand(stepKey) {
function getNodeBubbleAlign(stepKey) {
if (stepKey === stepKeyContent) return 'end'
if (stepKey === stepKeyImagePrompts) return 'center'
if (stepKey === stepKeyImages) return 'center'
if (stepKey === stepKeyPreview) return 'end'
if (stepKey === stepKeyPageMD) return 'end'
if (stepKey === stepKeyOutline) return 'center'
return 'start'
}
function getNodeBubbleWidth(stepKey) {
return stepKey === stepKeyConfig ? 560 : 500
if (stepKey === stepKeyConfig) return 560
if (stepKey === stepKeyPreview) return 700
if (stepKey === stepKeyPageMD) return 660
if (stepKey === stepKeyImages) return 620
return 500
}
function getPromptSlotLabel(item, index) {
if (item?.key === 'cover' || index === 0) return '封面配图'
return `内容配图 ${index}`
}
function getPromptPurpose(item, index) {
if (item?.key === 'cover' || index === 0) {
return '用于文章封面或头图,先传达主题和整体氛围'
}
const sectionName = item?.section || `内容段落 ${index}`
return `用于“${sectionName}”这一段的说明配图,帮助读者更直观理解内容`
}
function handleNodePanelWheel() {
@@ -805,6 +946,10 @@ function stepShortLabel(key) {
if (key === stepKeyTitle) return '标题'
if (key === stepKeyOutline) return '提纲'
if (key === stepKeyContent) return '正文'
if (key === stepKeyImagePrompts) return '提示词'
if (key === stepKeyImages) return '图片'
if (key === stepKeyPreview) return '预览'
if (key === stepKeyPageMD) return '分页'
return '节点'
}
@@ -877,7 +1022,7 @@ async function createTask() {
}
async function loadWorkflow(taskId = selectedTaskId.value, options = {}) {
const { silent = false, preserveNodeState = false, syncRuntime = true } = options
const { silent = false, preserveNodeState = false, syncRuntime = false } = options
if (!taskId) {
workflowDetail.value = null
return
@@ -894,7 +1039,7 @@ async function loadWorkflow(taskId = selectedTaskId.value, options = {}) {
expandedNodes.value = []
}
if (syncRuntime) {
await workerRuntime.loadTaskDetail(taskId)
void workerRuntime.loadTaskDetail(taskId)
}
if (fittedTaskId.value !== String(taskId)) {
await nextTick()
@@ -920,6 +1065,8 @@ function syncSelectionsFromWorkflow() {
stepPayload.topic = shared.selected_topic || ''
stepPayload.title = shared.selected_title || ''
stepPayload.outline = shared.outline || ''
stepPayload.content = shared.content || ''
stepPayload.contentWithPrompts = shared.content_with_prompts || ''
stepPayload.requirements = workflow.value?.form?.requirements || ''
configForm.keyword = workflow.value?.form?.keyword || ''
configForm.businessDomain = workflow.value?.form?.business_domain || 'auto'
@@ -1016,14 +1163,16 @@ async function runStep(stepKey) {
runningStepKey.value = stepKey
pollingStepKey.value = stepKey
await nextTick()
ElMessage.success(`${stepLabel(stepKey)}已开始执行`)
await executeOfficialAccountWorkflowStep(selectedTaskId.value, stepKey, {
topic: stepPayload.topic,
title: stepPayload.title,
outline: stepPayload.outline,
content: stepPayload.content,
content_with_prompts: stepPayload.contentWithPrompts,
requirements: stepPayload.requirements,
})
await loadWorkflow(selectedTaskId.value, { silent: true, preserveNodeState: true, syncRuntime: false })
ElMessage.success(`${stepLabel(stepKey)}已开始执行`)
await pollWorkflowUntilStepSettled(String(selectedTaskId.value), stepKey, { notifyOnFinish: true })
} catch (error) {
runningStepKey.value = ''
@@ -1035,10 +1184,31 @@ async function runStep(stepKey) {
function findRunningStepKey(workflowState = workflow.value) {
const steps = workflowState?.steps || {}
return [stepKeyTopic, stepKeyTitle, stepKeyOutline, stepKeyContent]
return [stepKeyTopic, stepKeyTitle, stepKeyOutline, stepKeyContent, stepKeyImagePrompts, stepKeyImages, stepKeyPreview, stepKeyPageMD]
.find(key => steps?.[key]?.status === 'running') || ''
}
async function downloadExport(format, filename) {
if (!selectedTaskId.value) return
try {
exportingFormat.value = format
const blob = await exportOfficialAccountDocument(selectedTaskId.value, format)
const url = window.URL.createObjectURL(blob)
const link = document.createElement('a')
link.href = url
link.download = filename
document.body.appendChild(link)
link.click()
link.remove()
window.URL.revokeObjectURL(url)
ElMessage.success('导出文件已准备好')
} catch (error) {
ElMessage.error(error?.message || '导出失败')
} finally {
exportingFormat.value = ''
}
}
async function pollWorkflowUntilStepSettled(taskId, stepKey, options = {}) {
const { notifyOnFinish = false } = options
const token = pollingToken.value + 1
@@ -1532,12 +1702,187 @@ onMounted(async () => {
overflow: auto;
}
.pagemd-preview {
max-height: 420px;
}
.prompt-list {
display: flex;
flex-direction: column;
gap: 10px;
}
.prompt-header {
display: flex;
align-items: flex-start;
justify-content: space-between;
gap: 12px;
}
.prompt-badge {
display: inline-flex;
align-items: center;
padding: 4px 10px;
border-radius: 999px;
background: rgba(37, 99, 235, 0.1);
color: #2563eb;
font-size: 12px;
font-weight: 700;
}
.prompt-card,
.image-card {
padding: 12px;
border-radius: 14px;
border: 1px solid rgba(228, 231, 237, 0.9);
background: #f8fafc;
}
.prompt-title,
.image-section {
margin-top: 8px;
font-size: 14px;
font-weight: 700;
color: #1f2d3d;
}
.prompt-chip {
flex: none;
display: inline-flex;
align-items: center;
padding: 4px 10px;
border-radius: 999px;
background: rgba(15, 23, 42, 0.06);
color: #475569;
font-size: 12px;
font-weight: 600;
}
.prompt-meta {
margin-top: 8px;
font-size: 12px;
line-height: 1.6;
color: #64748b;
}
.prompt-body,
.image-prompt {
margin-top: 8px;
font-size: 13px;
line-height: 1.7;
color: #526071;
word-break: break-word;
}
.prompt-raw-preview {
max-height: 260px;
color: #64748b;
}
.image-grid {
display: grid;
grid-template-columns: repeat(2, minmax(0, 1fr));
gap: 12px;
}
.image-thumb {
width: 100%;
aspect-ratio: 16 / 9;
object-fit: cover;
border-radius: 12px;
border: 1px solid rgba(191, 219, 254, 0.9);
background: #e2e8f0;
}
.action-row {
margin-top: 8px;
display: flex;
justify-content: flex-end;
}
.action-row-start {
justify-content: flex-start;
gap: 10px;
}
.preview-render {
border-radius: 18px;
border: 1px solid rgba(228, 231, 237, 0.9);
background: #fff;
padding: 18px;
max-height: min(60vh, 680px);
overflow: auto;
}
.page-chip-list {
display: flex;
flex-wrap: wrap;
gap: 8px;
}
.page-chip {
padding: 6px 10px;
border-radius: 999px;
background: rgba(37, 99, 235, 0.08);
color: #2563eb;
font-size: 13px;
font-weight: 700;
}
:deep(.oa-preview) {
color: #1f2d3d;
font-size: 14px;
line-height: 1.8;
}
:deep(.oa-preview h1) {
margin: 0 0 14px;
font-size: 22px;
line-height: 1.5;
}
:deep(.oa-preview h2) {
margin: 22px 0 10px;
font-size: 18px;
line-height: 1.5;
}
:deep(.oa-preview p),
:deep(.oa-preview blockquote),
:deep(.oa-preview li) {
font-size: 14px;
line-height: 1.85;
}
:deep(.oa-preview blockquote) {
margin: 10px 0;
padding: 10px 12px;
border-left: 3px solid rgba(37, 99, 235, 0.35);
background: rgba(37, 99, 235, 0.05);
color: #475569;
}
:deep(.oa-preview ul) {
margin: 10px 0 10px 20px;
padding: 0;
}
:deep(.oa-preview-figure) {
margin: 16px 0;
}
:deep(.oa-preview-figure img) {
width: 100%;
border-radius: 14px;
display: block;
}
:deep(.oa-preview-figure figcaption) {
margin-top: 8px;
font-size: 13px;
color: #64748b;
}
.oa-node {
width: 360px;
padding: 20px;