src/services/concept-designer.ts

/**
 * 概念设计者服务 - 两阶段 AI 生成架构
 * 第一阶段：设计者/思考者 -> 场景设计
 * 第二阶段：代码生成者 -> Manim 代码
 */

import OpenAI from 'openai'
import { createLogger } from '../utils/logger'
import type { CustomApiConfig, OutputMode, PromptOverrides, ReferenceImage } from '../types'
import { createCustomOpenAIClient, initializeDefaultOpenAIClient } from './openai-client-factory'
import { generateSceneDesignStage } from './concept-designer/scene-design-stage'
import { generateCodeFromDesignStage } from './concept-designer/code-from-design-stage'

const logger = createLogger('ConceptDesigner')

const OPENAI_MODEL = process.env.OPENAI_MODEL || 'glm-4-flash'
const DESIGNER_TEMPERATURE = parseFloat(process.env.DESIGNER_TEMPERATURE || '0.8')
const CODER_TEMPERATURE = parseFloat(process.env.AI_TEMPERATURE || '0.7')
const MAX_TOKENS = parseInt(process.env.AI_MAX_TOKENS || '1200', 10)
const DESIGNER_MAX_TOKENS = parseInt(process.env.DESIGNER_MAX_TOKENS || '12000', 10)

const openaiClient: OpenAI | null = initializeDefaultOpenAIClient((error) => {
  logger.warn('OpenAI 客户端初始化失败', { error })
})

function createCustomClient(config: CustomApiConfig): OpenAI {
  return createCustomOpenAIClient(config)
}

function resolveClient(customApiConfig?: CustomApiConfig): OpenAI | null {
  return customApiConfig ? createCustomClient(customApiConfig) : openaiClient
}

function resolveModel(customApiConfig?: CustomApiConfig): string {
  return customApiConfig?.model?.trim() || OPENAI_MODEL
}

/**
 * 两阶段 AI 生成
 * 1. 设计者生成场景设计方案
 * 2. 代码生成者根据设计方案生成代码
 */
export async function generateTwoStageAIManimCode(
  concept: string,
  outputMode: OutputMode,
  customApiConfig?: CustomApiConfig,
  promptOverrides?: PromptOverrides,
  referenceImages?: ReferenceImage[],
  onCheckpoint?: () => Promise<void>
): Promise<{ code: string; sceneDesign: string }> {
  logger.info('开始两阶段 AI 生成流程', {
    concept,
    outputMode,
    hasImages: !!referenceImages?.length
  })

  const client = resolveClient(customApiConfig)
  if (!client) {
    logger.warn('OpenAI 客户端不可用')
    return { code: '', sceneDesign: '' }
  }

  const model = resolveModel(customApiConfig)
  if (onCheckpoint) await onCheckpoint()

  const sceneDesign = await generateSceneDesignStage({
    client,
    concept,
    outputMode,
    model,
    promptOverrides,
    referenceImages,
    designerTemperature: DESIGNER_TEMPERATURE,
    designerMaxTokens: DESIGNER_MAX_TOKENS,
    onCheckpoint
  })

  if (!sceneDesign) {
    logger.warn('场景设计方案生成失败，中止流程')
    return { code: '', sceneDesign: '' }
  }
  if (onCheckpoint) await onCheckpoint()

  const code = await generateCodeFromDesignStage({
    client,
    concept,
    outputMode,
    sceneDesign,
    model,
    promptOverrides,
    coderTemperature: CODER_TEMPERATURE,
    maxTokens: MAX_TOKENS,
    onCheckpoint
  })

  logger.info('两阶段 AI 生成流程完成', {
    concept,
    hasSceneDesign: !!sceneDesign,
    hasCode: !!code
  })

  return { code, sceneDesign }
}

/**
 * 检查 OpenAI 客户端是否可用
 */
export function isOpenAIAvailable(): boolean {
  return openaiClient !== null
}