ai-manga-factory / src /lib /dirtyGeneratedPanelsParser.ts
jbilcke-hf's picture
jbilcke-hf HF staff
added suggestion from user + work on the progressive loader
faf4ba4
raw
history blame
825 Bytes
import { GeneratedPanel } from "@/types"
import { cleanJson } from "./cleanJson"
import { parseBadJSON } from "./parseBadJSON"
export function dirtyGeneratedPanelsParser(input: string): GeneratedPanel[] {
if (input.includes("```")) {
input = input.split("```")[0]
}
// we only keep what's after the first [
let jsonOrNot = cleanJson(input)
const jsonData = parseBadJSON(jsonOrNot) as GeneratedPanel[]
const results = jsonData.map((item, i) => {
let panel = i
let caption = item.caption ? item.caption.trim() : ''
let instructions = item.instructions ? item.instructions.trim() : ''
if (!instructions && caption) {
instructions = caption
}
if (!caption && instructions) {
caption = instructions
}
return { panel, caption, instructions }
})
return results
}