diff --git a/synthesis/text/text_block_generator/paragraph.py b/synthesis/text/text_block_generator/paragraph.py index bc5908a..9c3afd9 100644 --- a/synthesis/text/text_block_generator/paragraph.py +++ b/synthesis/text/text_block_generator/paragraph.py @@ -2,30 +2,47 @@ import textwrap from typing import List from faker import Faker -from funcy import identity, iterate, take, last +from funcy import iterate, take, last +from cv_analysis.logging import debug_log from cv_analysis.utils import star from cv_analysis.utils.rectangle import Rectangle from synthesis.random import rnd -from synthesis.text.text_block_generator.text_block_generator import TextBlockGenerator +from synthesis.text.line_formatter.identity import IdentityLineFormatter +from synthesis.text.line_formatter.line_formatter import LineFormatter from synthesis.text.line_formatter.paragraph import ParagraphLineFormatter +from synthesis.text.text_block_generator.text_block_generator import TextBlockGenerator class ParagraphGenerator(TextBlockGenerator): def __init__(self): self.line_formatter = ParagraphLineFormatter(blank_line_percentage=rnd.uniform(0, 0.5)) - def __call__(self, rectangle, n_sentences): + @debug_log(exits=False) + def __call__(self, rectangle, n_sentences=None): return self.generate_paragraph(rectangle, n_sentences) - def generate_paragraph(self, rectangle, n_sentences): + @debug_log(exits=False) + def generate_paragraph(self, rectangle, n_sentences=None): lines = generate_random_text_lines(rectangle, self.line_formatter, n_sentences) return lines -def generate_random_text_lines(rectangle: Rectangle, line_formatter=identity, n_sentences=3000) -> List[str]: +@debug_log(exits=False) +def generate_random_text_lines( + rectangle: Rectangle, + line_formatter: LineFormatter = None, + n_sentences=None, +) -> List[str]: + n_sentences = n_sentences or 3000 # TODO: De-hardcode. + line_formatter = line_formatter or IdentityLineFormatter() + text = Faker().paragraph(nb_sentences=n_sentences, variable_nb_sentences=False, ext_word_list=None) - unformatted_lines = textwrap.wrap(text, width=rectangle.width, break_long_words=False) + unformatted_lines = textwrap.wrap( + text, + width=rectangle.width // rnd.uniform(4, 5), # TODO: De-hardcode. + break_long_words=True, + ) # each iteration of the line formatter function formats one more line and adds it to the back of the list formatted_lines_generator = iterate(star(line_formatter), (unformatted_lines, True)) # hence do as many iterations as there are lines in the rectangle