AndroidRemoteController/nodejs/ef-compiler/actions/fun/ai/text2img.js

const fs = require('fs')
const path = require('path')
const { runWithModel, resolveSavePath } = require('./shared')

/** 入参：prompt, model, savePath（可选，图片保存路径；有则返回 b64 并写入文件，否则返回 url） */
async function executeText2img ({ prompt, model, savePath, folderPath }) {
  const p = prompt != null ? String(prompt).trim() : ''
  const outPath = savePath ? resolveSavePath(savePath, folderPath) : null
  try {
    const result = await runWithModel('text2img', 'doubao_text2img', [p, outPath], model)
    if (!result.success) return { success: false, error: result.error || 'text2img 失败' }
    const data = result.data
    const item = data?.data?.[0]
    if (!item) return { success: false, error: 'text2img 无返回数据' }
    if (item.b64_json && outPath) {
      const dir = path.dirname(outPath)
      if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true })
      fs.writeFileSync(outPath, Buffer.from(item.b64_json, 'base64'))
      return { success: true, value: outPath, path: outPath }
    }
    const url = item.url || ''
    return { success: true, value: url, path: url || undefined }
  } catch (e) {
    return { success: false, error: (e && (e.message || String(e))) || 'text2img 异常' }
  }
}

module.exports = { executeText2img }