Update app.py

app.py

@@ -71,7 +71,7 @@ def process_image(image, text):
             pil_warped = Image.fromarray(cv2.cvtColor(warped, cv2.COLOR_BGRA2RGBA))
             draw = ImageDraw.Draw(pil_warped)
 
-            # Calculate dynamic font size and write text
+            # Calculate dynamic font size and adjust for multi-line text
             font_size = max(10, int(height * 0.2))  # Adjust size relative to height
             try:
                 font = ImageFont.truetype(FONT_PATH, size=font_size)
@@ -80,20 +80,29 @@ def process_image(image, text):
                 logging.warning(f"Error loading font. Using default. {e}")
                 font = ImageFont.load_default()
 
-            bbox = draw.textbbox((0, 0), text, font=font)
-            text_width = bbox[2] - bbox[0]
-            text_height = bbox[3] - bbox[1]
-
-            text_position = ((width - text_width) // 2, (height - text_height) // 2)
-
-            # Debugging rectangle for text position
-            draw.rectangle(
-                [text_position, (text_position[0] + text_width, text_position[1] + text_height)],
-                outline="red",  # Debugging rectangle color
-                width=2
-            )
-
-            draw.text(text_position, text, fill=(0, 0, 0, 255), font=font)
+            # Split text into multiple lines to fit within the bounding box
+            words = text.split()
+            lines = []
+            current_line = ""
+            for word in words:
+                test_line = f"{current_line} {word}".strip()
+                bbox = draw.textbbox((0, 0), test_line, font=font)
+                if bbox[2] - bbox[0] <= width:
+                    current_line = test_line
+                else:
+                    lines.append(current_line)
+                    current_line = word
+            if current_line:
+                lines.append(current_line)
+
+            # Draw each line of text within the bounding box
+            line_height = draw.textbbox((0, 0), "Ag", font=font)[3] - draw.textbbox((0, 0), "Ag", font=font)[1]
+            start_y = (height - line_height * len(lines)) // 2
+            for i, line in enumerate(lines):
+                text_width = draw.textbbox((0, 0), line, font=font)[2] - draw.textbbox((0, 0), line, font=font)[0]
+                text_x = (width - text_width) // 2
+                text_y = start_y + i * line_height
+                draw.text((text_x, text_y), line, fill=(0, 0, 0, 255), font=font)
 
             # Convert back to OpenCV and reapply the perspective transformation
             warped_with_text = cv2.cvtColor(np.array(pil_warped), cv2.COLOR_RGBA2BGRA)
@@ -151,3 +160,4 @@ if __name__ == "__main__":
     interface.launch(share=True)
 
 
+