641 lines
32 KiB
TypeScript
641 lines
32 KiB
TypeScript
/* =====================================================================
|
||
* ai.ts — AI 内容创作引擎
|
||
* - OpenAI 兼容 /chat/completions
|
||
* - Anthropic 兼容 /v1/messages
|
||
* - 流式 SSE 解析、AbortController 中断
|
||
* - 三类调用：generate / polish / chat
|
||
* 由 ai.js 迁移：逻辑几乎不变，仅改为 TS + ESM
|
||
* ===================================================================== */
|
||
import type { AiOp, ChartItem, Deck, Slide, SlideElement, ElementStyle, RichLine } from './types'
|
||
import { elementTypes, uid } from './sample'
|
||
import { store } from './store'
|
||
import { normSegments } from './richtext'
|
||
|
||
const SEP = '%%PPT_JSON%%' // 对话模式中，自然语言回复与结构化操作的分隔标记
|
||
const VALID_TYPES = ['title', 'text', 'list', 'stat', 'quote', 'image', 'shape', 'chart', 'card', 'table', 'code', 'formula'] as const
|
||
|
||
/* 智谱/通用错误码 → 可操作提示 */
|
||
const ERROR_HINTS: Record<string, string> = {
|
||
'1113': '该 Key 账户按量计费余额不足或无资源包。智谱 Coding Plan 套餐额度仅限官方工具(Claude Code 等)使用，网页应用需走按量计费——请到 bigmodel.cn 充值后重试。',
|
||
'1309': 'GLM Coding Plan 套餐已到期，需续订后使用。',
|
||
'1211': '模型名错误，请在设置里检查模型名（如 glm-4.6）。',
|
||
'1234': '模型名错误，请检查模型名拼写。',
|
||
'1123': 'API Key 无效或已过期，请检查设置中的 Key。',
|
||
'1303': '该 Key 无对应模型权限，请更换模型或 Key。'
|
||
}
|
||
const ERROR_HINTS_BY_STATUS: Record<number, string> = {
|
||
401: '鉴权失败：API Key 无效或格式错误，请检查设置。',
|
||
403: '无权限或来源被拒：Key 无该模型权限，或调用来源受限。',
|
||
404: '接口路径错误：请检查 Base URL 与所选协议是否匹配。',
|
||
429: '限流或额度不足，请稍后重试或检查账户额度。'
|
||
}
|
||
|
||
/* ============================================================
|
||
* Prompt 模板
|
||
* ============================================================ */
|
||
const SYS_BASE =
|
||
'你是「u-ppt」的内容创作助手，擅长把主题变成结构清晰、视觉现代、带入场动效的中文演示稿。\n' +
|
||
'输出必须严格遵循下面的数据模型，坐标用百分比(0-100)，字号为数字。\n\n' +
|
||
'幻灯片模型：\n' +
|
||
'{ "slides": [ { "background": "bg|panel|primary|accent|g-primary|g-deep|g-soft", "elements": [ 元素, ... ] } ] }\n' +
|
||
'背景：bg/panel=浅底；primary/accent=纯色深底；g-primary=主→强调渐变(深)；g-deep=深色渐变；g-soft=浅色渐变。\n\n' +
|
||
'元素类型 type：title 标题 | text 正文 | list 列表 | stat 数据 | quote 金句 | image 图片 | shape 形状 | chart 图表 | card 卡片 | table 表格 | code 代码 | formula 公式\n' +
|
||
'元素：{ "type":..., "x":数字,"y":数字,"w":数字,"h":数字 (0-100), "content":字符串, "style":{...} }\n' +
|
||
' - title/text/list/quote：content 为文字，list 用 \\n 分多行\n' +
|
||
' - stat：content 为大数字(如 "65%")，style.label 为说明\n' +
|
||
' - card：content 第一行=标题、其余行=正文；style.accent=顶部色条键，style.icon=emoji 图标\n' +
|
||
' - shape：style.shapeType=rect|circle|triangle，style.fill=颜色键，style.gradient=true 渐变，style.opacity=0~1\n' +
|
||
' - chart：content 为 JSON，两种格式：\n' +
|
||
' 单系列：[{"label":"","value":数字}, ...]\n' +
|
||
' 多系列：{"series":["Q1","Q2"], "items":[{"label":"华东","values":[120,150]}, ...]}\n' +
|
||
' style.max=最大值，style.chartType=bar|line|area|pie|doughnut|radar|hbar|progress（默认 bar）\n' +
|
||
' style.legend=true 显示图例，style.grid=true 显示网格线，style.stack=true 堆叠（多系列柱状）\n' +
|
||
' 图表类型选择建议：柱状(bar)对比离散类型；折线(line)显示趋势；面积(area)累计趋势；\n' +
|
||
' 饼图(pie)占比构成；环形(doughnut)轻量占比；雷达(radar)多维度对比；条形(hbar)长标签对比；进度(progress)完成率。\n' +
|
||
' - table：content 为 Markdown 管道表格字符串（首行表头，用 | 分列，换行分行），style.header=true 首行加粗\n' +
|
||
' - code：content 为代码文本（保留符原样，不要转义），style.lang=语言如 js/python（可选）\n' +
|
||
' - formula：content 为 LaTeX 公式字符串（不含 $$ 分隔符），如 "E = mc^2"、"\\sum_{i=1}^n x_i"\n' +
|
||
' - image：content 留空\n\n' +
|
||
'可选 segments 字段（结构化富文本，同一行内不同片段可有不同样式）：\n' +
|
||
' segments: [[ {"text":"华东 "}, {"text":"增长 23%","bold":true,"color":"accent"} ], ...]\n' +
|
||
' 每个 segment 支持：bold/italic/underline/strike/color(主题键或#hex)/highlight(黄底)/code/sup/sub/fontSize/link\n' +
|
||
' 仅在需要行内混合样式时用 segments；纯文本继续用 content 即可。segments 与 content 二选一。\n\n' +
|
||
'style 公共字段：fontSize 数字、color 颜色键、align left|center|right、bold 布尔、italic 布尔、label/labelColor/labelSize、radius\n' +
|
||
'颜色 color/fill/accent：一律用主题键 primary|accent|text|muted（系统按背景深浅自动反相，深底文字自动变白，不要手填 #ffffff）。\n\n' +
|
||
'入场动画 style.anim（重要，给主要元素都加上，演示时逐个入场）：\n' +
|
||
' 可选：fade-up(默认上浮) | fade | scale | zoom | slide-l | slide-r | pop(弹现) | rotate(旋转) | bounce(弹跳) | flip(翻转) | blur(聚焦)\n' +
|
||
' 建议：标题/金句用 pop 或 scale；正文/列表用 fade-up；卡片用 fade-up 或 slide-l/slide-r；装饰形状用 scale。\n\n' +
|
||
'版式规范：\n' +
|
||
'- 画布逻辑 1280×720。封面标题 y≈30 h≈18；内容标题 y≈8-10。\n' +
|
||
'- 字号：title 44-66、text 22-28、list 24-30、stat 数字 64-80、quote 40-52。\n' +
|
||
'- 一页一个观点，留白充足，列表不超过 5 条。\n' +
|
||
'- 现代版式：多用 card 分组；封面/金句/结尾用 g-primary；目录用 3-4 张卡片网格；数据页 stat+chart。\n' +
|
||
'- emoji 极克制：默认不给 card.icon，除非确有助于理解，多数卡片留空。\n\n' +
|
||
'内容准则（重要——避免「AI 味」，写得像该领域的真人）：\n' +
|
||
'- 标题写具体事实而非口号：「华东 Q3 增长 23%」而非「业绩腾飞」；「同屏字+口述记忆降 50%」而非「效率革命」。\n' +
|
||
'- 正文要有实质：具体数字、案例、步骤、来源；少用「赋能/助力/打造/引领/开启/一站式」这类空词。\n' +
|
||
'- 禁用万能金句：「少即是多」「一图胜千言」「未来已来」「让 X 更简单」「不止是 X，更是 Y」。\n' +
|
||
'- 列表每条是可执行/可验证的要点，不是三词排比（避免「清晰、简洁、高效」式口号）。\n' +
|
||
'- 有判断、有观点，像从业者写的；数据尽量给来源（人名/书名/研究）。\n' +
|
||
'- 不要使用 markdown 代码块，直接输出 JSON。'
|
||
|
||
const SYS_GENERATE =
|
||
SYS_BASE +
|
||
'\n任务：根据用户主题生成一套完整演示（封面、目录、3-5 页内容、结尾，共 6-8 页）。\n' +
|
||
'要求：每个主要元素都加 style.anim；封面与结尾用 g-primary；至少 2 张 card；数据页用 stat+chart。\n' +
|
||
'严格输出：{"action":"create_all","slides":[...]}'
|
||
|
||
const SYS_POLISH =
|
||
SYS_BASE +
|
||
'\n任务：按指令优化给定的“单页”，保持主题一致、布局合理，并为各元素补充 style.anim。\n' +
|
||
'严格输出：{"action":"update_page","slides":[ 优化后的单页 ],"note":"一句话说明改动"}'
|
||
|
||
const SYS_CHAT =
|
||
SYS_BASE +
|
||
'\n任务：与用户对话式编辑当前演示。先用中文流式回复用户（解释你将做什么），' +
|
||
'如需修改 PPT，在回复最后另起一行输出分隔标记 ' + SEP + '，紧随其后输出 JSON 操作。\n' +
|
||
'JSON 操作格式：{"action":"add_page|update_page|create_all|answer","slides":[...],"target":页码(从1开始,可选)}\n' +
|
||
'- add_page：在 target 页后插入新页；- update_page：替换 target 页；- create_all：整体替换；- answer：仅回答不改稿。\n' +
|
||
'没有改动时不要输出分隔标记，仅回复即可。'
|
||
|
||
/* ============================================================
|
||
* 上下文摘要
|
||
* ============================================================ */
|
||
function deckContext(currentIdx: number, selectedEl?: SlideElement | null): string {
|
||
const deck: Deck = store.getDeck()
|
||
const lines = ['当前主题: ' + deck.theme + '，共 ' + deck.slides.length + ' 页。']
|
||
deck.slides.forEach((s, i) => {
|
||
const types = s.elements.map(e => e.type).join('/')
|
||
const head = (s.elements[0] && s.elements[0].type === 'title') ? (' 标题:"' + (s.elements[0].content || '') + '"') : ''
|
||
lines.push('第' + (i + 1) + '页 [' + types + ']' + head + (i === currentIdx ? ' ← 当前页' : ''))
|
||
})
|
||
lines.push('\n当前页(第' + (currentIdx + 1) + '页)完整JSON:\n' + JSON.stringify(store.currentSlide.value))
|
||
// 选中元素上下文：让 AI 知道用户正在编辑哪个元素，对话直接围绕它
|
||
if (selectedEl) {
|
||
const typeLabel = selectedEl.type
|
||
const preview = (selectedEl.content || '').slice(0, 100)
|
||
lines.push('\n【用户当前选中的元素】（第' + (currentIdx + 1) + '页）')
|
||
lines.push('类型: ' + typeLabel + '，内容预览: "' + preview + '"')
|
||
lines.push('完整JSON: ' + JSON.stringify(selectedEl))
|
||
lines.push('用户接下来的对话默认针对此元素，除非明确说整页/整套。')
|
||
}
|
||
return lines.join('\n')
|
||
}
|
||
|
||
/* ============================================================
|
||
* 底层：流式请求
|
||
* ============================================================ */
|
||
interface StreamOpts {
|
||
jsonMode?: boolean
|
||
onVisible?: (text: string) => void
|
||
signal?: AbortSignal
|
||
}
|
||
|
||
interface Message { role: 'system' | 'user' | 'assistant'; content: string }
|
||
|
||
async function streamChat(messages: Message[], opts: StreamOpts): Promise<{ json: any; reply: string; op: any }> {
|
||
const cfg = store.getCfg()
|
||
const isLocal = /localhost|127\.0\.0\.1/i.test(cfg.base || '')
|
||
if (!cfg.key && !isLocal) throw new Error('未配置 API Key，请点击右上角 ⚙ 填写。')
|
||
if (cfg.protocol === 'anthropic') return runAnthropic(messages, opts, cfg)
|
||
return runOpenAI(messages, opts, cfg)
|
||
}
|
||
|
||
function apiUrl(cfg: { proxy: string; base: string }): string {
|
||
return (cfg.proxy || cfg.base || '').replace(/\/+$/, '')
|
||
}
|
||
|
||
async function postJSON(url: string, headers: Record<string, string>, body: unknown, signal?: AbortSignal): Promise<Response> {
|
||
try {
|
||
return await fetch(url, { method: 'POST', headers, body: JSON.stringify(body), signal })
|
||
} catch (e: any) {
|
||
if (e.name === 'AbortError') throw e
|
||
throw new Error('请求失败（可能是 CORS 跨域拦截）。可在设置中配置"代理 URL"。\n' + e.message)
|
||
}
|
||
}
|
||
|
||
// OpenAI 兼容
|
||
async function runOpenAI(messages: Message[], opts: StreamOpts, cfg: ReturnType<typeof store.getCfg>) {
|
||
const url = apiUrl(cfg) + '/chat/completions'
|
||
const body: Record<string, unknown> = { model: cfg.model || 'glm-4.6', messages, stream: true, temperature: 0.75 }
|
||
if (opts.jsonMode) body.response_format = { type: 'json_object' }
|
||
const resp = await postJSON(url,
|
||
{ 'Content-Type': 'application/json', 'Authorization': 'Bearer ' + cfg.key }, body, opts.signal)
|
||
return consumeStream(resp, (obj: any) => {
|
||
const ch = obj.choices && obj.choices[0]
|
||
return ch && ch.delta && ch.delta.content
|
||
}, opts)
|
||
}
|
||
|
||
// Anthropic 兼容
|
||
async function runAnthropic(messages: Message[], opts: StreamOpts, cfg: ReturnType<typeof store.getCfg>) {
|
||
const url = apiUrl(cfg) + '/v1/messages'
|
||
const sysParts: string[] = []
|
||
const msgs: Message[] = []
|
||
messages.forEach(m => {
|
||
if (m.role === 'system') { if (m.content) sysParts.push(m.content) }
|
||
else msgs.push({ role: m.role, content: m.content })
|
||
})
|
||
const body: Record<string, unknown> = {
|
||
model: cfg.model || 'glm-4.6',
|
||
messages: msgs,
|
||
max_tokens: opts.jsonMode ? 8192 : 4096,
|
||
stream: true,
|
||
temperature: 0.75
|
||
}
|
||
if (sysParts.length) body.system = sysParts.join('\n\n')
|
||
const headers = {
|
||
'Content-Type': 'application/json',
|
||
'x-api-key': cfg.key,
|
||
'anthropic-version': '2023-06-01',
|
||
'anthropic-dangerous-direct-browser-access': 'true'
|
||
}
|
||
const resp = await postJSON(url, headers, body, opts.signal)
|
||
return consumeStream(resp, (obj: any) => {
|
||
if (obj.type === 'content_block_delta' && obj.delta) return obj.delta.text || ''
|
||
return null
|
||
}, opts)
|
||
}
|
||
|
||
// 通用 SSE 消费
|
||
async function consumeStream(resp: Response, extractDelta: (obj: any) => string | null, opts: StreamOpts): Promise<{ json: any; reply: string; op: any }> {
|
||
if (!resp.ok) {
|
||
let t = ''; try { t = await resp.text() } catch (e) {}
|
||
let msg = '接口返回 ' + resp.status
|
||
let hint = ERROR_HINTS_BY_STATUS[resp.status]
|
||
try {
|
||
const err = JSON.parse(t).error
|
||
if (err) {
|
||
msg += ' [' + err.code + '] ' + (err.message || '')
|
||
if (err.code && ERROR_HINTS[err.code]) hint = ERROR_HINTS[err.code]
|
||
} else { msg += ' ' + (t || '').slice(0, 200) }
|
||
} catch (e) { msg += ' ' + (t || '').slice(0, 200) }
|
||
if (hint) msg += '\n\n💡 ' + hint
|
||
throw new Error(msg)
|
||
}
|
||
const reader = resp.body!.getReader()
|
||
const dec = new TextDecoder('utf-8')
|
||
let sseBuf = ''
|
||
let full = ''
|
||
let pending = ''
|
||
let sepMode = false
|
||
|
||
function feed(delta: string | null) {
|
||
if (delta == null) return
|
||
full += delta
|
||
if (opts.jsonMode || sepMode) return
|
||
pending += delta
|
||
const i = pending.indexOf(SEP)
|
||
if (i >= 0) {
|
||
emit(pending.slice(0, i))
|
||
sepMode = true
|
||
pending = ''
|
||
return
|
||
}
|
||
const keep = SEP.length - 1
|
||
if (pending.length > keep) {
|
||
const safe = pending.slice(0, pending.length - keep)
|
||
emit(safe)
|
||
pending = pending.slice(safe.length)
|
||
}
|
||
}
|
||
function emit(text: string) { if (opts.onVisible) opts.onVisible(text) }
|
||
|
||
for (;;) {
|
||
const chunk = await reader.read()
|
||
if (chunk.done) break
|
||
sseBuf += dec.decode(chunk.value, { stream: true })
|
||
const lines = sseBuf.split('\n')
|
||
sseBuf = lines.pop()!
|
||
for (let k = 0; k < lines.length; k++) {
|
||
const line = lines[k].trim()
|
||
if (!line || line.indexOf('data:') !== 0) continue
|
||
const payload = line.slice(5).trim()
|
||
if (!payload || payload === '[DONE]') continue
|
||
let obj: any; try { obj = JSON.parse(payload) } catch (e) { continue }
|
||
const delta = extractDelta(obj)
|
||
if (delta != null) feed(delta)
|
||
}
|
||
}
|
||
// 末帧残留 data: 行补解析
|
||
const tail = sseBuf.trim()
|
||
if (tail.indexOf('data:') === 0) {
|
||
const tp = tail.slice(5).trim()
|
||
if (tp && tp !== '[DONE]') {
|
||
let to: any; try { to = JSON.parse(tp) } catch (e) { to = null }
|
||
if (to) { const td = extractDelta(to); if (td != null) feed(td) }
|
||
}
|
||
}
|
||
if (!opts.jsonMode && !sepMode && pending) emit(pending)
|
||
|
||
if (opts.jsonMode) return { json: tryParse(full), reply: '', op: null }
|
||
const parts = full.split(SEP)
|
||
return {
|
||
json: null,
|
||
reply: (parts[0] || '').trim(),
|
||
op: parts.length > 1 ? tryParse(parts.slice(1).join(SEP)) : null
|
||
}
|
||
}
|
||
|
||
function tryParse(s: string): any {
|
||
if (!s) return null
|
||
s = String(s).replace(/```json/gi, '').replace(/```/g, '').trim()
|
||
const i = s.indexOf('{'), j = s.lastIndexOf('}')
|
||
if (i < 0 || j < 0) return null
|
||
const candidate = s.slice(i, j + 1)
|
||
try { return JSON.parse(candidate) }
|
||
catch (e) {
|
||
try { return JSON.parse(candidate.replace(/,(\s*[}\]])/g, '$1')) }
|
||
catch (e2) { return null }
|
||
}
|
||
}
|
||
|
||
/* ============================================================
|
||
* 数据规范化（AI 输出 → 可入库）
|
||
* ============================================================ */
|
||
function validColor(v: string): string | undefined {
|
||
if (!v) return v
|
||
if (v.charAt(0) === '#') return /^#[0-9a-f]{3,8}$/i.test(v) ? v : undefined
|
||
return ['primary', 'accent', 'text', 'muted'].indexOf(v) >= 0 ? v : undefined
|
||
}
|
||
|
||
function normStyle(st: any): ElementStyle {
|
||
st = st || {}
|
||
const out: any = { ...st }
|
||
if (out.fontSize != null) out.fontSize = Number(out.fontSize) || 24
|
||
if (out.bold != null) out.bold = !!out.bold
|
||
if (out.italic != null) out.italic = !!out.italic
|
||
if (out.labelSize != null) out.labelSize = Number(out.labelSize) || 16
|
||
if (out.radius != null) out.radius = Number(out.radius) || 0
|
||
if (out.max != null) out.max = Number(out.max) || 0
|
||
// 图表子类型校验
|
||
const VALID_CHART_TYPES = ['bar', 'line', 'pie', 'area', 'doughnut', 'radar', 'hbar', 'progress']
|
||
if (out.chartType != null && !VALID_CHART_TYPES.includes(out.chartType)) delete out.chartType
|
||
if (out.legend != null) out.legend = !!out.legend
|
||
if (out.stack != null) out.stack = !!out.stack
|
||
if (out.grid != null) out.grid = !!out.grid
|
||
if (out.header != null) out.header = !!out.header
|
||
if (out.inline != null) out.inline = !!out.inline
|
||
if (typeof out.lang === 'string' && out.lang.length > 16) out.lang = out.lang.slice(0, 16)
|
||
;['color', 'fill', 'accent', 'labelColor'].forEach(k => {
|
||
if (validColor(out[k]) === undefined && out[k] != null) delete out[k]
|
||
})
|
||
if (typeof out.icon === 'string' && out.icon.length > 8) out.icon = out.icon.slice(0, 8)
|
||
return out
|
||
}
|
||
|
||
function normElement(e: any): SlideElement | null {
|
||
if (!e || typeof e !== 'object') return null
|
||
const type = (VALID_TYPES as readonly string[]).indexOf(e.type) >= 0 ? e.type : 'text'
|
||
const out: SlideElement = {
|
||
id: e.id || uid('el'),
|
||
type: type as SlideElement['type'],
|
||
x: clampNum(e.x, 0, 100, 10),
|
||
y: clampNum(e.y, 0, 100, 10),
|
||
w: clampNum(e.w, 3, 100, 60),
|
||
h: clampNum(e.h, 3, 100, 15),
|
||
content: (e.content == null ? '' : String(e.content)),
|
||
style: normStyle(e.style || {})
|
||
}
|
||
// 可选 segments（结构化富文本）
|
||
if (e.segments) {
|
||
const segs = normSegments(e.segments)
|
||
if (segs) (out as any).segments = segs
|
||
}
|
||
return out
|
||
}
|
||
|
||
const VALID_BGS = ['bg', 'panel', 'primary', 'accent', 'g-primary', 'g-deep', 'g-soft']
|
||
|
||
function normSlide(s: any): Slide | null {
|
||
if (!s || typeof s !== 'object') return null
|
||
const bg = VALID_BGS.indexOf(s.background) >= 0 ? s.background
|
||
: (typeof s.background === 'string' && s.background.charAt(0) === '#' ? s.background : 'bg')
|
||
const els = (Array.isArray(s.elements) ? s.elements : []).map(normElement).filter(Boolean) as SlideElement[]
|
||
return { id: uid('s'), background: bg as Slide['background'], elements: els }
|
||
}
|
||
|
||
function normSlides(arr: any[]): Slide[] {
|
||
return (Array.isArray(arr) ? arr : []).map(normSlide).filter((s): s is Slide => s !== null)
|
||
}
|
||
|
||
function clampNum(v: number, lo: number, hi: number, dflt: number): number {
|
||
const n = Number(v)
|
||
if (isNaN(n)) return dflt
|
||
return Math.max(lo, Math.min(hi, n))
|
||
}
|
||
|
||
/* ============================================================
|
||
* 高层 API
|
||
* ============================================================ */
|
||
|
||
/** 生成整套 */
|
||
export async function generate(opts: { topic: string; count?: number; signal?: AbortSignal }): Promise<{ action: 'create_all'; slides: Slide[] }> {
|
||
const count = opts.count || 7
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_GENERATE },
|
||
{ role: 'user', content: '主题：' + opts.topic + '\n请生成约 ' + count + ' 页（含封面与结尾），中文内容。' }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 输出无法解析为 JSON，请重试。')
|
||
const slides = normSlides(r.json.slides || r.json)
|
||
if (!slides.length) throw new Error('AI 未生成有效幻灯片，请重试或换一个主题。')
|
||
return { action: 'create_all', slides }
|
||
}
|
||
|
||
/** 文档内容 → AI 分析后生成幻灯片 */
|
||
export async function generateFromDocument(opts: {
|
||
text: string
|
||
filename?: string
|
||
signal?: AbortSignal
|
||
}): Promise<{ action: 'create_all'; slides: Slide[] }> {
|
||
const SYS_DOC =
|
||
SYS_BASE +
|
||
'\n任务：分析下方提供的文档内容，将其提炼为 5-10 页专业演示稿（含封面 1 页 + 内容 3-7 页 + 结尾 1 页）。\n' +
|
||
'要求：\n' +
|
||
'- 提取文档中的核心观点、关键数据、重要结论，不要遗漏重要信息\n' +
|
||
'- 每页只有一个主题，标题具体、有信息量，如「Q3 营收增长 23%」而非「业绩回顾」\n' +
|
||
'- 正文有实质内容，包含具体数字、案例、来源；避免空泛口号\n' +
|
||
'- 合理运用 card（对比/分组）、stat（突出数据）、chart（趋势/分布）等元素\n' +
|
||
'- 每个主要元素都加 style.anim 入场动画\n' +
|
||
'- 封面用 g-primary 渐变，结尾可用 g-primary 或 g-deep，内容页用 bg/panel 浅底\n' +
|
||
'严格输出 JSON：{"action":"create_all","slides":[...]}'
|
||
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_DOC },
|
||
{ role: 'user', content: '文档' + (opts.filename ? '（' + opts.filename + '）' : '') + '内容如下：\n\n' + opts.text }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 输出无法解析为 JSON，请重试。')
|
||
const slides = normSlides(r.json.slides || r.json)
|
||
if (!slides.length) throw new Error('AI 未生成有效幻灯片，请重试或检查文档内容。')
|
||
return { action: 'create_all', slides }
|
||
}
|
||
|
||
/** 润色单页 */
|
||
export async function polish(opts: { slide: Slide; instruction?: string; signal?: AbortSignal }): Promise<{ action: 'update_page'; slide: Slide; note: string }> {
|
||
const instruction = opts.instruction || '让这页内容更有吸引力、表达更精炼'
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_POLISH },
|
||
{ role: 'user', content: '指令：' + instruction + '\n当前页JSON：\n' + JSON.stringify(opts.slide) }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 输出无法解析，请重试。')
|
||
const arr = r.json.slides || (r.json.slide ? [r.json.slide] : (r.json.elements ? [r.json] : []))
|
||
const slides = normSlides(arr)
|
||
if (!slides.length) throw new Error('AI 未返回有效页面。')
|
||
return { action: 'update_page', slide: slides[0], note: r.json.note || '' }
|
||
}
|
||
|
||
/** 对话式编辑 */
|
||
export async function chat(opts: {
|
||
history: Message[]
|
||
input: string
|
||
onVisible?: (text: string) => void
|
||
signal?: AbortSignal
|
||
/** 当前选中的元素（可选，注入上下文让对话围绕它） */
|
||
selectedElement?: SlideElement | null
|
||
}): Promise<{ reply: string; op: AiOp | null }> {
|
||
const messages: Message[] = [{ role: 'system', content: SYS_CHAT + '\n\n' + deckContext(store.getCurrentIndex(), opts.selectedElement || null) }]
|
||
;(opts.history || []).forEach(m => messages.push(m))
|
||
messages.push({ role: 'user', content: opts.input })
|
||
const r = await streamChat(messages, { jsonMode: false, onVisible: opts.onVisible, signal: opts.signal })
|
||
return { reply: r.reply, op: normalizeOp(r.op) }
|
||
}
|
||
|
||
function normalizeOp(json: any): AiOp | null {
|
||
if (!json) return null
|
||
let action: AiOp['action'] = json.action || 'answer'
|
||
const slides = normSlides(json.slides)
|
||
const target = json.target != null ? (Number(json.target) - 1) : null
|
||
if (action === 'update_page' && slides.length) action = 'update_page'
|
||
if (action === 'add_page' && slides.length) action = 'add_page'
|
||
if ((action === 'update_page' || action === 'add_page') && !slides.length) action = 'answer'
|
||
return { action, slides, target: isNaN(target as number) ? null : target, note: json.note || '' }
|
||
}
|
||
|
||
export function isConfigured(): boolean { return !!(store.getCfg().key) }
|
||
|
||
/* ============================================================
|
||
* AI 深化能力：大纲 / 逐页生成 / 配图 / 主题建议 / 一键美化
|
||
* ============================================================ */
|
||
|
||
import type { Outline, OutlineItem, ThemeSuggestion, ImageGenResult } from './types'
|
||
|
||
/* ---------- Prompt 模板 ---------- */
|
||
|
||
const SYS_OUTLINE =
|
||
'你是「u-ppt」的演示策划助手。用户给你一个主题，你先制定大纲，不要直接写完整幻灯片。\n\n' +
|
||
'大纲要素：先拟定一个具体、有信息量的整份标题（不要「关于 X 的分享」这类空泛标题）；\n' +
|
||
'然后拆成 5-8 页，每页给出：页类型 kind、标题 title、3-5 条要点 points、可选 hint。\n' +
|
||
'kind 取值：cover(封面) / toc(目录) / content(内容) / quote(金句) / end(结尾)。\n' +
|
||
'封面与结尾各 1 页，目录可选。\n' +
|
||
'hint 用来给后续生成幻灯片的 AI 补充指令，例如「数据页：用 stat+chart」「对比页：双 card 并置」「引言页：深色背景金句」。\n\n' +
|
||
'内容准则：\n' +
|
||
'- 标题写具体事实而非口号（「华东 Q3 增长 23%」而非「业绩腾飞」）。\n' +
|
||
'- 要点每条是可验证的论点或具体数字，不要三词排比。\n' +
|
||
'- 有判断、有观点，像该领域从业者写的；数据尽量给来源。\n' +
|
||
'- 禁用万能金句：「少即是多」「一图胜千言」「未来已来」。\n\n' +
|
||
'严格输出 JSON：\n' +
|
||
'{ "title": "整份标题", "items": [ { "kind":"cover", "title":"页标题", "points":["要点 1","要点 2"], "hint":"可选" } ] }\n' +
|
||
'不要使用 markdown 代码块，直接输出 JSON。'
|
||
|
||
const SYS_GEN_PAGE =
|
||
SYS_BASE +
|
||
'\n任务：根据大纲中的一条，生成「一页」幻灯片。严格遵循要点与 hint，不要偏离主题。\n' +
|
||
'严格输出：{ "background":"...", "elements":[ ... ] }（单个 slide 对象，不要数组）。\n' +
|
||
'kind=cover 用 g-primary 背景；kind=quote 用 g-deep；kind=end 用 g-primary；kind=content 数据页用 stat+chart。'
|
||
|
||
const SYS_THEME =
|
||
'你是配色设计师。根据主题关键词，推荐一套现代、专业的 6 色配色（primary 主色、accent 强调色、bg 背景、panel 面板、text 正文、muted 次要文字）。\n\n' +
|
||
'要求：\n' +
|
||
'- 主色饱和度适中，不刺眼；强调色与主色形成对比但不冲突。\n' +
|
||
'- bg 与 panel 接近白色，差异轻微。\n' +
|
||
'- text 深于 primary，muted 为中性灰。\n' +
|
||
'- 所有颜色用 #hex 格式。\n\n' +
|
||
'可选背景推荐：bg/panel（浅底）/primary/accent（深底）/g-primary（主→强调渐变）/g-deep（深色渐变）。\n' +
|
||
'内容页优先 bg/panel；封面/结尾优先 g-primary；金句页优先 g-deep。\n\n' +
|
||
'严格输出 JSON：\n' +
|
||
'{ "name":"主题名","primary":"#...","accent":"#...","bg":"#...","panel":"#...","text":"#...","muted":"#...","recommendBg":"g-primary","reason":"一句话设计理由" }\n' +
|
||
'不要使用 markdown 代码块。'
|
||
|
||
const SYS_BEAUTIFY =
|
||
SYS_BASE +
|
||
'\n任务：优化给定页面的「排版」（位置、尺寸、字号、对齐、装饰元素），保持文字内容不变。\n' +
|
||
'如果布局合理则微调；如果有明显问题（元素重叠、字号失衡、留白不足）则重排。\n' +
|
||
'严格输出：{ "background":"原背景或更合适的键", "elements":[ ... ], "note":"一句话说明" }\n' +
|
||
'不要改动文字内容本身，只优化视觉。'
|
||
|
||
/* ---------- 1. 大纲生成 ---------- */
|
||
|
||
export async function outline(opts: { topic: string; count?: number; signal?: AbortSignal }): Promise<Outline> {
|
||
const count = opts.count || 7
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_OUTLINE },
|
||
{ role: 'user', content: '主题：' + opts.topic + '\n请拟定约 ' + count + ' 页的大纲（含封面与结尾），中文。' }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 未返回有效大纲，请重试。')
|
||
const items: OutlineItem[] = (Array.isArray(r.json.items) ? r.json.items : []).map((it: any, i: number) => ({
|
||
id: 'ol-' + Date.now() + '-' + i,
|
||
kind: ['cover', 'toc', 'content', 'quote', 'end'].includes(it.kind) ? it.kind : 'content',
|
||
title: String(it.title || '未命名').slice(0, 80),
|
||
points: Array.isArray(it.points) ? it.points.map((p: any) => String(p).slice(0, 200)).filter(Boolean).slice(0, 6) : [],
|
||
hint: it.hint ? String(it.hint).slice(0, 120) : undefined,
|
||
done: false
|
||
}))
|
||
if (!items.length) throw new Error('大纲为空，请重试或换一个主题。')
|
||
return { title: String(r.json.title || opts.topic).slice(0, 80), topic: opts.topic, items }
|
||
}
|
||
|
||
/* ---------- 2. 按大纲条目生成单页 ---------- */
|
||
|
||
export async function generatePage(opts: { item: OutlineItem; index: number; total: number; signal?: AbortSignal }): Promise<Slide> {
|
||
const user =
|
||
'大纲第 ' + (opts.index + 1) + '/' + opts.total + ' 页：\n' +
|
||
'类型：' + opts.item.kind + '\n' +
|
||
'标题：' + opts.item.title + '\n' +
|
||
'要点：' + opts.item.points.map((p, i) => (i + 1) + '. ' + p).join('\n') +
|
||
(opts.item.hint ? '\n补充指令：' + opts.item.hint : '')
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_GEN_PAGE },
|
||
{ role: 'user', content: user }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 未返回有效页面。')
|
||
const slides = normSlides([r.json])
|
||
if (!slides.length) throw new Error('AI 输出无法解析为页面。')
|
||
return slides[0]
|
||
}
|
||
|
||
/* ---------- 3. AI 配图 ---------- */
|
||
|
||
/** 图像生成——当前支持 DALL-E 协议（OpenAI /images/generations）与通用返回 URL 的网关 */
|
||
export async function generateImage(opts: { prompt: string; signal?: AbortSignal }): Promise<ImageGenResult> {
|
||
const cfg = store.getCfg()
|
||
const imgBase = (cfg.imgBase || cfg.base || '').replace(/\/+$/, '')
|
||
const imgKey = cfg.imgKey || cfg.key
|
||
const imgModel = cfg.imgModel || 'dall-e-3'
|
||
if (!imgKey) throw new Error('未配置 API Key，无法生成图片。')
|
||
const url = imgBase + '/images/generations'
|
||
const body = {
|
||
model: imgModel,
|
||
prompt: opts.prompt,
|
||
n: 1,
|
||
size: '1024x1024',
|
||
response_format: 'b64_json'
|
||
}
|
||
let resp: Response
|
||
try {
|
||
resp = await fetch(url, {
|
||
method: 'POST',
|
||
headers: { 'Content-Type': 'application/json', 'Authorization': 'Bearer ' + imgKey },
|
||
body: JSON.stringify(body),
|
||
signal: opts.signal
|
||
})
|
||
} catch (e: any) {
|
||
if (e.name === 'AbortError') throw e
|
||
throw new Error('图片生成请求失败（可能是 CORS）。\n' + e.message)
|
||
}
|
||
if (!resp.ok) {
|
||
let t = ''; try { t = await resp.text() } catch (e) {}
|
||
throw new Error('图片生成失败 ' + resp.status + '：' + t.slice(0, 200))
|
||
}
|
||
const json = await resp.json()
|
||
const item = json.data && json.data[0]
|
||
if (!item) throw new Error('图片生成返回为空。')
|
||
const dataUrl = item.b64_json
|
||
? 'data:image/png;base64,' + item.b64_json
|
||
: (item.url || '')
|
||
if (!dataUrl) throw new Error('图片生成返回无图像数据。')
|
||
return { url: dataUrl, revisedPrompt: item.revised_prompt }
|
||
}
|
||
|
||
/** 判断是否配置了图像生成能力（当前以主 Key 是否存在为信号） */
|
||
export function isImageConfigured(): boolean {
|
||
const cfg = store.getCfg()
|
||
return !!(cfg.imgKey || cfg.key)
|
||
}
|
||
|
||
/* ---------- 4. AI 主题/配色建议 ---------- */
|
||
|
||
export async function suggestTheme(opts: { topic: string; signal?: AbortSignal }): Promise<ThemeSuggestion> {
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_THEME },
|
||
{ role: 'user', content: '主题关键词：' + opts.topic + '\n请推荐一套配色，并说明设计理由（一句话）。' }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 未返回有效配色方案。')
|
||
const j = r.json
|
||
// 基本校验，非法 hex 回退中性灰
|
||
const safeHex = (v: string, dflt: string) => /^#[0-9a-f]{6}$/i.test(v) ? v : dflt
|
||
return {
|
||
name: String(j.name || '自定义').slice(0, 20),
|
||
primary: safeHex(j.primary, '#4f46e5'),
|
||
accent: safeHex(j.accent, '#06b6d4'),
|
||
bg: safeHex(j.bg, '#ffffff'),
|
||
panel: safeHex(j.panel, '#f8fafc'),
|
||
text: safeHex(j.text, '#1e293b'),
|
||
muted: safeHex(j.muted, '#64748b'),
|
||
recommendBg: j.recommendBg || 'g-primary',
|
||
reason: j.reason ? String(j.reason).slice(0, 200) : undefined
|
||
}
|
||
}
|
||
|
||
/* ---------- 5. 一键美化单页（排版优化，内容不变） ---------- */
|
||
|
||
export async function beautifyPage(opts: { slide: Slide; signal?: AbortSignal }): Promise<{ slide: Slide; note: string }> {
|
||
const messages: Message[] = [
|
||
{ role: 'system', content: SYS_BEAUTIFY },
|
||
{ role: 'user', content: '当前页 JSON：\n' + JSON.stringify(opts.slide) }
|
||
]
|
||
const r = await streamChat(messages, { jsonMode: true, signal: opts.signal })
|
||
if (!r.json) throw new Error('AI 未返回有效页面。')
|
||
const slides = normSlides([r.json])
|
||
if (!slides.length) throw new Error('AI 输出无法解析。')
|
||
return { slide: slides[0], note: r.json.note || '' }
|
||
}
|
||
|
||
/** 导出 ChartItem 类型供组件用 */
|
||
export type { ChartItem, Outline, OutlineItem, ThemeSuggestion, ImageGenResult }
|