Update app.py

app.py

@@ -35,18 +35,27 @@ def generate_caption(image):
     meme_text = response.choices[0].message.content
     print(meme_text)
 
-    #put the meme text on the image
+     # put the meme text on the image
     draw = ImageDraw.Draw(pil_image)
 
+    # Set the maximum number of lines to 2
+    max_lines = 2
+
     # Calculate font size
-    text_length = len(meme_text)
-    font_size = 24
+    font_size = 1
     font = ImageFont.truetype("impact.ttf", font_size)
-    text_width, text_height = draw.textsize(meme_text, font=font)
-    
-    # Wrap the text to fit within the image width
-    wrapped_text = textwrap.fill(meme_text, width=int(pil_image.width / font_size))
-    
+
+    while True:
+        text_width, text_height = draw.textsize(meme_text, font=font)
+        wrapped_text = textwrap.fill(meme_text, width=int(pil_image.width / font_size))
+        num_lines = len(wrapped_text.split('\n'))
+
+        if num_lines <= max_lines:
+            break
+
+        font_size += 1
+        font = ImageFont.truetype("impact.ttf", font_size)
+
     # 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