allow more lines to be written

app_dialogue.py

@@ -356,6 +356,41 @@ def resize_with_ratio(image: PIL.Image.Image, fixed_width: int) -> PIL.Image.Ima
     return resized_img
 
 
+def make_new_lines(draw, image, font, text_is_too_long, lines, num_lines, num_loops):
+    max_len_increment = 0
+    while text_is_too_long and max_len_increment < 10:
+        new_lines = lines.copy()
+        last_line_with_backslash = insert_backslash(
+            new_lines[-1],
+            max_length=(len(new_lines[-1]) + max_len_increment)
+            // (num_lines - num_loops),
+        )
+        penultimate_line, last_line = (
+            last_line_with_backslash.split("\n")[0],
+            last_line_with_backslash.split("\n")[1],
+        )
+        new_lines.pop(-1)
+        new_lines.append(penultimate_line)
+        new_lines.append(last_line)
+        # If the we haven't reached the last line, we split it again
+        if len(new_lines) < num_lines:
+            new_lines, text_width, text_is_too_long = make_new_lines(
+                draw=draw,
+                image=image,
+                font=font,
+                text_is_too_long=text_is_too_long,
+                lines=new_lines,
+                num_lines=num_lines,
+                num_loops=num_loops + 1,
+            )
+        text_width = max([draw.textlength(line, font) for line in new_lines])
+        text_is_too_long = text_width > image.width
+        max_len_increment += 1
+    if not text_is_too_long:
+        lines = new_lines
+    return lines, text_width, text_is_too_long
+
+
 def test_font_size(
     draw,
     image,
@@ -368,75 +403,25 @@ def test_font_size(
 ):
     text_width = draw.textlength(text, font)
     text_is_too_long = True
-
-    if num_lines == 1:
-        while font.size > min_font and text_is_too_long:
-            font = ImageFont.truetype(
-                f"fonts/{font_meme_text}.ttf", size=font.size - font_size_reduction
-            )
+    lines = [text]
+    while font.size > min_font and text_is_too_long:
+        font = ImageFont.truetype(
+            f"fonts/{font_meme_text}.ttf", size=font.size - font_size_reduction
+        )
+        if num_lines == 1:
             text_width = draw.textlength(text, font)
             text_is_too_long = text_width > image.width
-
-    elif num_lines == 2:
-        while font.size > min_font and text_is_too_long:
-            font = ImageFont.truetype(
-                f"fonts/{font_meme_text}.ttf", size=font.size - font_size_reduction
+        else:
+            lines, text_width, text_is_too_long = make_new_lines(
+                draw=draw,
+                image=image,
+                font=font,
+                text_is_too_long=text_is_too_long,
+                lines=lines,
+                num_lines=num_lines,
+                num_loops=0,
             )
-            max_len_increment = 0
-            while (
-                text_is_too_long
-                and max_len_increment < 10
-                and max_len_increment < (len(text)) // 2
-            ):
-                temp_text = insert_backslash(
-                    text, max_length=(len(text) + max_len_increment) // 2
-                )
-                first_line, second_line = (
-                    temp_text.split("\n")[0],
-                    temp_text.split("\n")[1],
-                )
-                text_width = max(
-                    draw.textlength(first_line, font),
-                    draw.textlength(second_line, font),
-                )
-                text_is_too_long = text_width > image.width
-                max_len_increment += 1
-
-    elif num_lines == 3:
-        while font.size > min_font and text_is_too_long:
-            font = ImageFont.truetype(
-                f"fonts/{font_meme_text}.ttf", size=font.size - font_size_reduction
-            )
-            max_len_incr_1_split = 0
-            while text_is_too_long and max_len_incr_1_split < 10:
-                first_temp_text = insert_backslash(
-                    text, max_length=(len(text) + max_len_incr_1_split) // 3
-                )
-                first_line, second_line = (
-                    first_temp_text.split("\n")[0],
-                    first_temp_text.split("\n")[1],
-                )
-                max_len_incr_2_split = 0
-                while text_is_too_long and max_len_incr_2_split < 10:
-                    temp_text_second_line = insert_backslash(
-                        second_line,
-                        max_length=(len(second_line) + max_len_incr_2_split) // 2,
-                    )
-                    second_line_1, second_line_2 = (
-                        temp_text_second_line.split("\n")[0],
-                        temp_text_second_line.split("\n")[1],
-                    )
-                    temp_text = first_line + "\n" + second_line_1 + "\n" + second_line_2
-                    text_width = max(
-                        draw.textlength(first_line, font),
-                        draw.textlength(second_line_1, font),
-                        draw.textlength(second_line_2, font),
-                    )
-                    text_is_too_long = text_width > image.width
-                    max_len_incr_2_split += 1
-                max_len_incr_1_split += 1
-    else:
-        raise (ValueError("num_lines can only be 1, 2 or 3"))
+            temp_text = "\n".join(lines)
 
     if not text_is_too_long and num_lines > 1:
         text = temp_text
@@ -471,7 +456,7 @@ def make_meme_image(
         initial_font_size = 80
     text_is_too_long = True
     num_lines = 0
-    while text_is_too_long and num_lines < 3:
+    while text_is_too_long and num_lines < 8:
         num_lines += 1
         font = ImageFont.truetype(f"fonts/{font_meme_text}.ttf", size=initial_font_size)
         text, font, text_width, text_is_too_long = test_font_size(
@@ -494,7 +479,7 @@ def make_meme_image(
 
     outline_width = 2
     text_x = (image_width - text_width) / 2
-    text_y = image_height - num_lines * font.size - 10 - num_lines
+    text_y = image_height - num_lines * font.size - 10 - 2 * num_lines
     if text_at_the_top:
         text_y = 0
 
@@ -583,16 +568,16 @@ chatbot = gr.Chatbot(
     visible=False,
     avatar_images=[None, BOT_AVATAR],
 )
-css='''
+css = """
 .gradio-container{max-width: 970px!important}
 h1{display: flex;align-items: center;justify-content: center;gap: .25em}
-'''
+"""
 with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as demo:
     with gr.Row(scale=0.5):
         gr.HTML(
             """<h1 align="center">AI Meme Generator <span style="font-size: 13px;">powered by <a href="https://huggingface.co/blog/idefics">IDEFICS</a></span><img width=40 height=40 src="https://cdn-uploads.huggingface.co/production/uploads/624bebf604abc7ebb01789af/v770xGti5vH1SYLBgyOO_.png" /></h1>"""
         )
-        
+
     with gr.Row(elem_id="model_selector_row"):
         model_selector = gr.Dropdown(
             choices=MODELS,
@@ -896,7 +881,9 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
                 full_text += acc_text
                 acc_text = ""
 
-    textbox.submit(fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False).then(
+    textbox.submit(
+        fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False
+    ).then(
         fn=model_inference,
         inputs=[
             model_selector,
@@ -915,8 +902,14 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
         ],
         outputs=[textbox, generated_memes_gallery, chatbot],
     )
-    imagebox.upload(fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False).then(
-        fn=lambda: "Write a meme about this image.", inputs=[], outputs=[textbox], queue=False).then(
+    imagebox.upload(
+        fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False
+    ).then(
+        fn=lambda: "Write a meme about this image.",
+        inputs=[],
+        outputs=[textbox],
+        queue=False,
+    ).then(
         fn=model_inference,
         inputs=[
             model_selector,
@@ -939,7 +932,9 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
             chatbot,
         ],
     )
-    submit_btn.click(fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False).then(
+    submit_btn.click(
+        fn=lambda: "", inputs=[], outputs=[generated_memes_gallery], queue=False
+    ).then(
         fn=model_inference,
         inputs=[
             model_selector,