summaryrefslogtreecommitdiff
path: root/internal/comic/artist.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/comic/artist.go')
-rw-r--r--internal/comic/artist.go342
1 files changed, 342 insertions, 0 deletions
diff --git a/internal/comic/artist.go b/internal/comic/artist.go
new file mode 100644
index 0000000..c21e633
--- /dev/null
+++ b/internal/comic/artist.go
@@ -0,0 +1,342 @@
+package comic
+
+import (
+ "context"
+ "fmt"
+ "os"
+ "path/filepath"
+ "strings"
+ "time"
+
+ "codeberg.org/snonux/comicforge/internal/provider"
+)
+
+// ArtistConfig configures comic page generation.
+type ArtistConfig struct {
+ ImageProvider provider.ImageProvider
+ TextProvider provider.TextProvider
+ Prompts PromptRenderer
+ OutputDir string
+ Style string
+ Theme string
+ Language string
+ Script string
+ UltraRealistic bool
+ StoryPages int
+ GalleryPages int
+ PanelsPerPage int
+}
+
+// Artist generates comic-book pages.
+type Artist struct {
+ imageProvider provider.ImageProvider
+ textProvider provider.TextProvider
+ prompts PromptRenderer
+ outputDir string
+ style string
+ theme string
+ language string
+ script string
+ ultraRealistic bool
+ storyPages int
+ galleryPages int
+ panelsPerPage int
+ initErr error
+}
+
+// NewArtist creates an Artist.
+func NewArtist(cfg *ArtistConfig) *Artist {
+ a := &Artist{
+ outputDir: ".",
+ language: "Bulgarian",
+ script: "Cyrillic",
+ storyPages: storyPagesInScript,
+ galleryPages: 5,
+ panelsPerPage: storyPanelsPerPage,
+ ultraRealistic: true,
+ }
+ if cfg == nil {
+ a.initErr = fmt.Errorf("artist config is required")
+ return a
+ }
+
+ a.imageProvider = cfg.ImageProvider
+ a.textProvider = cfg.TextProvider
+ a.prompts = cfg.Prompts
+ a.outputDir = orDefault(cfg.OutputDir, a.outputDir)
+ a.style = cfg.Style
+ a.theme = cfg.Theme
+ a.language = orDefault(cfg.Language, a.language)
+ a.script = orDefault(cfg.Script, a.script)
+ a.ultraRealistic = cfg.UltraRealistic
+ if cfg.StoryPages > 0 {
+ a.storyPages = cfg.StoryPages
+ }
+ if cfg.GalleryPages > 0 {
+ a.galleryPages = cfg.GalleryPages
+ }
+ if cfg.PanelsPerPage > 0 {
+ a.panelsPerPage = cfg.PanelsPerPage
+ }
+
+ if a.imageProvider == nil {
+ a.initErr = fmt.Errorf("%w: image provider", ErrMissingProvider)
+ }
+ if a.prompts == nil {
+ a.initErr = errorsJoin(a.initErr, fmt.Errorf("%w: prompt renderer", ErrMissingProvider))
+ }
+ return a
+}
+
+// DrawComicPages renders the cover, story pages, gallery pages, and back cover.
+func (a *Artist) DrawComicPages(ctx context.Context, storyText, bible, titleSlug string, entries []WordEntry, panelScript [][]string) ([]string, error) {
+ if err := a.ready(); err != nil {
+ return nil, err
+ }
+ style := a.style
+ if style == "" {
+ style = pickStyle(nil, a.ultraRealistic)
+ }
+ fmt.Printf(" Comic style: %s\n", style)
+
+ resolvedBible, blurb, err := a.resolveHelperTexts(ctx, storyText, bible)
+ if err != nil {
+ return nil, err
+ }
+
+ var paths []string
+ if p, err := a.renderPage(ctx, titleSlug+"_cover", coverPromptTemplate, a.coverPromptData(storyText, style, resolvedBible), "cover page"); err == nil && p != "" {
+ paths = append(paths, p)
+ }
+
+ sections := splitIntoSections(storyText, a.storyPages)
+ for i, section := range sections {
+ pageNum := i + 1
+ data := a.storyPagePromptData(section, pageNum, style, resolvedBible, entries, panelScript)
+ if p, err := a.renderPage(ctx, fmt.Sprintf("%s_page_%d", titleSlug, pageNum), storyPagePromptTemplate, data, fmt.Sprintf("story page %d", pageNum)); err == nil && p != "" {
+ paths = append(paths, p)
+ }
+ }
+
+ for i := 0; i < a.galleryPages; i++ {
+ galleryNum := i + 1
+ data := a.galleryPromptData(style, resolvedBible, galleryNum)
+ if p, err := a.renderPage(ctx, fmt.Sprintf("%s_gallery_%d", titleSlug, galleryNum), galleryPagePromptTemplate, data, fmt.Sprintf("gallery page %d/%d", galleryNum, a.galleryPages)); err == nil && p != "" {
+ paths = append(paths, p)
+ }
+ }
+
+ if p, err := a.renderPage(ctx, titleSlug+"_back", backCoverPromptTemplate, a.backPromptData(storyText, style, resolvedBible, blurb), "back cover"); err == nil && p != "" {
+ paths = append(paths, p)
+ }
+ return paths, nil
+}
+
+func (a *Artist) ready() error {
+ if a == nil {
+ return fmt.Errorf("artist is nil")
+ }
+ if a.initErr != nil {
+ return a.initErr
+ }
+ return nil
+}
+
+func (a *Artist) renderPage(ctx context.Context, fileName, templateName string, data map[string]any, label string) (string, error) {
+ path := filepath.Join(a.outputDir, fileName+".png")
+ if _, err := os.Stat(path); err == nil {
+ fmt.Printf(" Skipping %s (already exists)\n", filepath.Base(path))
+ return path, nil
+ }
+ prompt, err := a.prompts.RenderPrompt(templateName, data)
+ if err != nil {
+ return "", fmt.Errorf("render %s prompt: %w", label, err)
+ }
+ if err := a.generateWithRetry(ctx, prompt, path, label); err != nil {
+ return "", err
+ }
+ return path, nil
+}
+
+func (a *Artist) generateWithRetry(ctx context.Context, prompt, outputFile, label string) error {
+ attempts := pageMaxRetries
+ for attempt := 1; attempt <= attempts; attempt++ {
+ callCtx, cancel := withTimeout(ctx, helperTimeout)
+ err := a.imageProvider.GenerateImage(callCtx, prompt, outputFile)
+ cancel()
+ if err == nil {
+ return nil
+ }
+ if attempt < attempts {
+ pause := pageRetryBase * time.Duration(attempt)
+ fmt.Printf(" Warning: %s attempt %d/%d failed (%v), retrying in %s...\n", label, attempt, attempts, err, pause)
+ time.Sleep(pause)
+ continue
+ }
+ return fmt.Errorf("%s failed after %d attempts: %w", label, attempts, err)
+ }
+ return nil
+}
+
+func (a *Artist) resolveHelperTexts(ctx context.Context, storyText, prebuiltBible string) (string, string, error) {
+ bible := strings.TrimSpace(prebuiltBible)
+ if bible != "" {
+ fmt.Printf(" Character bible ready (%d chars)\n", len(bible))
+ }
+ blurb := ""
+ if a.textProvider == nil {
+ return bible, blurb, nil
+ }
+ systemPrompt, err := a.prompts.RenderPrompt(blurbSystemTemplate, map[string]any{
+ "StoryText": storyText,
+ })
+ if err != nil {
+ return "", "", fmt.Errorf("render blurb prompt: %w", err)
+ }
+ prompt := systemPrompt + "\n\n" + storyText
+ callCtx, cancel := withTimeout(ctx, helperTimeout)
+ defer cancel()
+ text, err := a.textProvider.GenerateText(callCtx, prompt)
+ if err != nil {
+ fmt.Printf(" Warning: back-cover blurb generation failed: %v\n", err)
+ return bible, blurb, nil
+ }
+ blurb = strings.TrimSpace(text)
+ if blurb != "" {
+ fmt.Printf(" Back-cover blurb ready (%d chars)\n", len(blurb))
+ }
+ return bible, blurb, nil
+}
+
+func (a *Artist) coverPromptData(storyText, style, bible string) map[string]any {
+ return map[string]any{
+ "Language": a.language,
+ "Script": a.script,
+ "Style": style,
+ "Bible": bible,
+ "Subtitle": "ComicForge Adventures",
+ "StoryText": storyText,
+ "RenderingRequirement": a.renderingRequirement(),
+ "RenderingRequirementEnd": a.renderingRequirementEnd(),
+ }
+}
+
+func (a *Artist) storyPagePromptData(section string, pageNum int, style, bible string, entries []WordEntry, panelScript [][]string) map[string]any {
+ return map[string]any{
+ "Language": a.language,
+ "Script": a.script,
+ "Style": style,
+ "Bible": bible,
+ "Words": buildWordList(entries, ""),
+ "PageNum": pageNum,
+ "TotalPages": a.storyPages,
+ "PanelLayout": buildPanelLayout(section, pageScriptForPage(panelScript, pageNum-1)),
+ "RenderingRequirement": a.renderingRequirement(),
+ "RenderingRequirementEnd": a.renderingRequirementEnd(),
+ }
+}
+
+func (a *Artist) galleryPromptData(style, bible string, galleryNum int) map[string]any {
+ return map[string]any{
+ "Language": a.language,
+ "Script": a.script,
+ "Style": style,
+ "Bible": bible,
+ "Pose": galleryPoses[(galleryNum-1)%len(galleryPoses)],
+ "RenderingRequirement": a.renderingRequirement(),
+ "RenderingRequirementEnd": a.renderingRequirementEnd(),
+ }
+}
+
+func (a *Artist) backPromptData(storyText, style, bible, blurb string) map[string]any {
+ return map[string]any{
+ "Language": a.language,
+ "Script": a.script,
+ "Style": style,
+ "Bible": bible,
+ "BlurbBox": blurbBoxInstruction(blurb),
+ "SeriesTitle": "ComicForge Adventures",
+ "StoryText": storyText,
+ "RenderingRequirement": a.renderingRequirement(),
+ "RenderingRequirementEnd": a.renderingRequirementEnd(),
+ }
+}
+
+func (a *Artist) renderingRequirement() string {
+ if a.ultraRealistic {
+ text, err := a.prompts.RenderPrompt(renderingRequirementPrompt, nil)
+ if err == nil {
+ return text
+ }
+ }
+ return ""
+}
+
+func (a *Artist) renderingRequirementEnd() string {
+ if a.ultraRealistic {
+ text, err := a.prompts.RenderPrompt(renderingRequirementEndPrompt, nil)
+ if err == nil {
+ return text
+ }
+ }
+ return ""
+}
+
+func pageScriptForPage(panelScript [][]string, idx int) []string {
+ if idx < 0 || idx >= len(panelScript) {
+ return nil
+ }
+ return panelScript[idx]
+}
+
+func buildPanelLayout(section string, pagePanels []string) string {
+ labels := [4]string{"TOP-LEFT", "TOP-RIGHT", "BOTTOM-LEFT", "BOTTOM-RIGHT"}
+ if len(pagePanels) == 4 && pagePanels[0] != "" && pagePanels[1] != "" && pagePanels[2] != "" && pagePanels[3] != "" {
+ var sb strings.Builder
+ sb.WriteString("MANDATORY PANEL LAYOUT — divide the image into exactly 4 panels in a 2×2 grid.\n")
+ sb.WriteString("Draw each panel EXACTLY as described below — these are the precise scenes to illustrate:\n")
+ for i, label := range labels {
+ sb.WriteString(fmt.Sprintf(" • %s panel: %s\n", label, pagePanels[i]))
+ }
+ return sb.String()
+ }
+
+ excerpt := strings.TrimSpace(section)
+ if len(excerpt) > comicPromptMaxChars {
+ excerpt = excerpt[:comicPromptMaxChars]
+ if idx := strings.LastIndex(excerpt, " "); idx > 0 {
+ excerpt = excerpt[:idx]
+ }
+ excerpt += "…"
+ }
+ return "MANDATORY PANEL LAYOUT — divide the image into exactly 4 panels in a 2×2 grid:\n" +
+ " • TOP-LEFT panel: scene 1 from the excerpt\n" +
+ " • TOP-RIGHT panel: scene 2 from the excerpt\n" +
+ " • BOTTOM-LEFT panel: scene 3 from the excerpt\n" +
+ " • BOTTOM-RIGHT panel: scene 4 from the excerpt\n" +
+ "Story excerpt (ALL panels must illustrate THIS excerpt only):\n\n" + excerpt + "\n"
+}
+
+func blurbBoxInstruction(blurb string) string {
+ if strings.TrimSpace(blurb) == "" {
+ return "a rectangular text box (white or cream background, thin black border) near the bottom, styled like a classic back-cover synopsis box"
+ }
+ return fmt.Sprintf("a rectangular text box (white or cream background, thin black border) near the bottom displaying this blurb text in italic type:\n %q", blurb)
+}
+
+func errorsJoin(errs ...error) error {
+ var filtered []error
+ for _, err := range errs {
+ if err != nil {
+ filtered = append(filtered, err)
+ }
+ }
+ switch len(filtered) {
+ case 0:
+ return nil
+ case 1:
+ return filtered[0]
+ default:
+ return fmt.Errorf("%v; %w", filtered[0], filtered[1])
+ }
+}
generated by cgit v1.2.3 (git 2.25.1) at 2026-08-07 03:26:53 +0000