Change the Caption function with proper version

app.py

@@ -537,23 +537,71 @@ def removeTonalMarks(string):
     return noTonalMarksStr
 
 
-def add_text_to_image(input_image, text, output_image_path="output.png", border_size=2):
+# def add_text_to_image(input_image, text, output_image_path="output.png", border_size=2):
+#     text = removeTonalMarks(text)
+#     imagearr = np.asarray(input_image) #Image.open(input_image_path)
+#     width, height = imagearr.shape[:2] #width, height = image.size
+#     img = Image.fromarray(imagearr)
+#     draw = ImageDraw.Draw(img)
+#     font = ImageFont.truetype("ShortBaby.ttf", 36) #ShortBaby-Mg2w.ttf
+#     text_width, text_height = draw.textbbox((0, 0), text, font=font)[2:] #draw.textsize(text, font)
+#     # calculate the x, y coordinates of the text box
+#     x = (width - text_width) / 2
+#     y = (height - text_height) / 2
+#     # put the text on the image with a border
+#     for dx, dy in [(0, 0), (border_size, border_size), (-border_size, -border_size), (border_size, -border_size), (-border_size, border_size)]:
+#         draw.text((x + dx, y + dy), text, font=font, fill=(255, 255, 255))
+#     draw.text((x, y), text, font=font, fill=(0, 0, 0))
+#     img.save(output_image_path, "PNG")
+#     return "output.png"
+
+
+def calculate_max_chars(image_width, font_size, font_path="ShortBaby.ttf", margin=20):
+    # Create a temporary image to calculate character width
+    img_temp = Image.new('RGB', (100, 100))
+    draw_temp = ImageDraw.Draw(img_temp)
+    font = ImageFont.truetype(font_path, font_size)
+    # Use a common character for width estimation and multiply by an adjustment factor
+    avg_char_width = draw_temp.textlength("W", font=font) #textsize("W", font=font)[0]
+    max_chars = int(image_width / avg_char_width) * 2
+    return max_chars
+
+def fitimagetexttowidth(text, image_width, font_size, font_path="ShortBaby.ttf"):
+    max_chars = calculate_max_chars(image_width, font_size, font_path)
+    words = text.split()
+    adjusted_text = ""
+    line = ""
+
+    for word in words:
+        if len(line + word) <= max_chars:
+            line += word + " "
+        else:
+            adjusted_text += line.strip() + "\n"
+            line = word + " "
+    adjusted_text += line.strip()
+
+    return adjusted_text
+
+def add_text_to_image(input_image, text, text_size=36, output_image_path="output.png", border_size=2, font_path="ShortBaby.ttf"):
     text = removeTonalMarks(text)
-    imagearr = np.asarray(input_image) #Image.open(input_image_path)
-    width, height = imagearr.shape[:2] #width, height = image.size
+    imagearr = np.asarray(input_image)
+    height, width = imagearr.shape[:2]
+    text = fitimagetexttowidth(text, width, text_size, font_path)
     img = Image.fromarray(imagearr)
+
     draw = ImageDraw.Draw(img)
-    font = ImageFont.truetype("ShortBaby.ttf", 36) #ShortBaby-Mg2w.ttf
-    text_width, text_height = draw.textbbox((0, 0), text, font=font)[2:] #draw.textsize(text, font)
-    # calculate the x, y coordinates of the text box
+    font = ImageFont.truetype(font_path, text_size)
+    # Recalculate position based on the entire block of text
+    text_width, text_height = draw.textbbox((0, 0), text, font=font)[2:]
     x = (width - text_width) / 2
     y = (height - text_height) / 2
-    # put the text on the image with a border
+
+    # Add text with border
     for dx, dy in [(0, 0), (border_size, border_size), (-border_size, -border_size), (border_size, -border_size), (-border_size, border_size)]:
         draw.text((x + dx, y + dy), text, font=font, fill=(255, 255, 255))
     draw.text((x, y), text, font=font, fill=(0, 0, 0))
     img.save(output_image_path, "PNG")
-    return "output.png"
+    return output_image_path
 
 def UnknownTrackTexttoApp(text): #Copy of def OptimisedTtAppForUNWFWO(text):
       #Buttons and labels autocreation