Atlas/web/src/utils/generationSummary.ts

import { serviceTypeLabel } from '@/constants/providerCompat'
import { parseImageUrl } from '@/utils/output'

export interface GenerationRow {
  service_type: string
  input_json: string
  output_json: string
}

export function summarizeGeneration(g: GenerationRow): string {
  if (g.service_type === 'chat') {
    try {
      return JSON.parse(g.input_json).content ?? g.input_json
    } catch {
      return g.input_json
    }
  }
  try {
    const input = JSON.parse(g.input_json)
    if (input.prompt) return input.prompt
  } catch { /* ignore */ }
  try {
    const out = JSON.parse(g.output_json)
    const url = parseImageUrl(out)
    if (url) return url
    return out.content ?? g.output_json.slice(0, 80)
  } catch {
    return g.output_json.slice(0, 80)
  }
}

export function formatGenerationType(type: string): string {
  return serviceTypeLabel(type)
}

export function previewImageFromGeneration(g: GenerationRow): string {
  try {
    return parseImageUrl(JSON.parse(g.output_json))
  } catch {
    return ''
  }
}