Update app.py

app.py

@@ -92,8 +92,8 @@ def process_image(image, text):
             logging.debug(f"Detected paper angle: {angle} degrees.")
 
             # Adjust padding based on the white paper's detected dimensions
-            padding_x = int((x2 - x1) * 0.15)  # 15% padding horizontally
-            padding_y = int((y2 - y1) * 0.15)  # 15% padding vertically
+            padding_x = int((x2 - x1) * 0.05)  # 5% padding horizontally
+            padding_y = int((y2 - y1) * 0.05)  # 5% padding vertically
 
             x1_padded = x1 + padding_x
             y1_padded = y1 + padding_y
@@ -118,20 +118,26 @@ def process_image(image, text):
             text_draw = ImageDraw.Draw(text_layer)
             logging.debug("Created transparent text layer.")
 
-            # Function to adjust font size until text fits within the white paper
+            # Adjust font size to fit within the white paper
             def adjust_font_size(lines, box_width, box_height, font, text_draw):
-                line_height = (text_draw.textbbox((0, 0), "Hg", font=font)[3] - text_draw.textbbox((0, 0), "Hg", font=font)[1]) * 1.2
-                total_text_height = len(lines) * line_height
-                while total_text_height > box_height:
-                    font_size = font.size - 4  # Reduce font size
-                    if font_size < 10:  # Set a minimum font size
-                        break
-                    font = ImageFont.truetype(FONT_PATH, size=font_size)
-                    line_height = (text_draw.textbbox((0, 0), "Hg", font=font)[3] - text_draw.textbbox((0, 0), "Hg", font=font)[1]) * 1.2
+                while True:
+                    line_height = (
+                        text_draw.textbbox((0, 0), "Hg", font=font)[3] - text_draw.textbbox((0, 0), "Hg", font=font)[1]
+                    ) * 1.2
                     total_text_height = len(lines) * line_height
+                    max_line_width = max(
+                        text_draw.textbbox((0, 0), line, font=font)[2] - text_draw.textbbox((0, 0), line, font=font)[0]
+                        for line in lines
+                    )
+                    if total_text_height <= box_height and max_line_width <= box_width:
+                        return font, line_height
+                    font_size = font.size - 2
+                    if font_size < 10:
+                        break  # Minimum font size
+                    font = ImageFont.truetype(FONT_PATH, size=font_size)
                 return font, line_height
 
-            # Improved text wrapping logic to fit within the white paper
+            # Split and wrap text to fit within the white paper
             words = text.split()
             lines = []
             current_line = ""
@@ -161,7 +167,9 @@ def process_image(image, text):
             logging.debug(f"Drew text within white paper: ({x1_padded}, {y1_padded}), ({x2_padded}, {y2_padded}).")
 
             # Apply rotation to align with the paper's angle
-            rotated_text_layer = text_layer.rotate(-angle, resample=Image.BICUBIC, center=(x1 + box_width / 2, y1 + box_height / 2))
+            rotated_text_layer = text_layer.rotate(
+                -angle, resample=Image.BICUBIC, center=(x1 + box_width / 2, y1 + box_height / 2)
+            )
 
             # Composite the rotated text layer onto the original image
             pil_image = Image.alpha_composite(pil_image, rotated_text_layer)
@@ -221,3 +229,4 @@ if __name__ == "__main__":
 
 
 
+