Improved lyrics rendering

folkugat_web/services/lilypond.py

@@ -1,17 +1,17 @@
 import asyncio
 import dataclasses
-import itertools
 import os
 import re
 from pathlib import Path
 from typing import Literal
 
 import aiofiles
-from folkugat_web.model.lilypond import (LyricsColumn, LyricsParagraph,
+from folkugat_web.model.lilypond import (LyricsLine, LyricsParagraph,
                                          LyricsText, RenderError)
 from folkugat_web.model.temes import Tema
 from folkugat_web.services import files as files_service
 from folkugat_web.templates import templates
+from folkugat_web.utils import batched
 
 SCORE_BEGINNING = "% --- SCORE BEGINNING --- %"
 
@@ -23,6 +23,9 @@ def source_to_single_score(source: str, tema: Tema) -> str:
         lyrics_text = build_lyrics(tema.lyrics[0].content)
     else:
         lyrics_text = None
+    print("AAAAAA")
+    print(lyrics_text)
+    print("AAAAAA")
     return templates.get_template("lilypond/single_score.ly").render(
         score_beginning=SCORE_BEGINNING,
         score_source=source,
@@ -77,13 +80,10 @@ async def render(source: str, fmt: RenderFormat, output_filename: Path | None =
     return output_filename, []
 
 
-def build_lyrics(text: str, max_cols: int = 2, min_pars: int = 2) -> LyricsText:
-    paragraphs = [LyricsParagraph(lines=par_str.splitlines()) for par_str in text.split("\n\n")]
-    n_cols = next(filter(lambda nc: len(paragraphs) // nc >= min_pars, range(max_cols, 0, -1)), 1)
-    n_long_cols = len(paragraphs) % n_cols
-    pars_per_col = [len(paragraphs) // n_cols + 1] * n_long_cols + [len(paragraphs) // n_cols] * (n_cols - n_long_cols)
-    acc_pars_per_col = [0] + list(itertools.accumulate(pars_per_col))
-    return LyricsText(columns=[
-        LyricsColumn(paragraphs=paragraphs[acc_pars_per_col[i]:acc_pars_per_col[i+1]])
-        for i in range(n_cols)
+def build_lyrics(text: str, max_cols: int = 3) -> LyricsText:
+    paragraphs = [LyricsParagraph(lines=par_str.splitlines()) for par_str in text.split("\n\n") if par_str]
+    print(list(batched(paragraphs, max_cols)))
+    return LyricsText(lines=[
+        LyricsLine(paragraphs=list(line_pars))
+        for line_pars in batched(paragraphs, max_cols)
     ])