Update app.py

app.py

@@ -42,14 +42,20 @@ def generate_caption(image):
     font_size = 60
     font = ImageFont.truetype("impact.ttf", font_size)
 
-    # Calculate the width for text wrap
-    wrap_width = len(meme_text) // 2
-
-    # Wrap the text to fit within the image width and have a maximum of 2 lines
-    wrapped_text = textwrap.wrap(meme_text, width=wrap_width)
-    text_lines = wrapped_text
-
-    y = 10
+    # Calculate the average character width for the font
+    alphabet = "ABCEMOPQRSTWXZ"
+    total_char_width = sum(draw.textlength(char, font=font) for char in alphabet)
+    average_char_width = total_char_width / len(alphabet)
+    
+    # Calculate the number of characters that fit within the image width
+    chars_per_line = int(pil_image.width / average_char_width)
+    
+    # Wrap the text to fit within the image width
+    wrapped_text = textwrap.fill(meme_text, width=chars_per_line)
+    
+    # Calculate the position to place the text at the top and center horizontally
+    text_lines = wrapped_text.split('\n')
+    y = 10  # Adjust this value to add more or less padding from the top
     for line in text_lines:
         line_width = draw.textlength(line, font=font)
         line_mask = font.getmask(line)