Update app.py

app.py

@@ -43,9 +43,12 @@ def generate_caption(image):
     font_size = int(max_width / (len(meme_text) / 2))
     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.fill(meme_text, width=int(max_width / font.getsize("A")[0]))
-    text_lines = wrapped_text.split('\n')
+    wrapped_text = textwrap.wrap(meme_text, width=wrap_width)
+    text_lines = wrapped_text
 
     y = 10
     for line in text_lines: