Update app.py

app.py

@@ -38,24 +38,13 @@ def generate_caption(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
-    font_size = 1
+    font_size = int(pil_image.width * 0.8)  # Adjust the multiplier to change the font size based on image width
     font = ImageFont.truetype("impact.ttf", 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)
-
+    
+    # Wrap the text to fit within the image width
+    wrapped_text = textwrap.fill(meme_text, width=int(pil_image.width / font_size * 0.8))
+    
     # 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