Update app.py

app.py

@@ -45,6 +45,13 @@ def detect_and_crop_lines(image):
         draw.text((x1, y1 - 10), f"Line {idx}", fill="blue")
 
         cropped_line = image.crop((x1, y1, x2, y2))
+
+        # Resize the cropped line to a smaller fixed width while maintaining aspect ratio
+        fixed_width = 200
+        aspect_ratio = cropped_line.height / cropped_line.width
+        new_height = int(fixed_width * aspect_ratio)
+        cropped_line = cropped_line.resize((fixed_width, new_height), Image.LANCZOS)
+
         cropped_lines.append(cropped_line)
 
     return original_image, cropped_lines
@@ -62,22 +69,18 @@ with gr.Blocks() as iface:
             gr.Markdown("### Annotated Image with Detected Lines")
             output_annotated = gr.Image(type="pil", label="Detected Text Lines")
 
-    gr.Markdown("### Cropped Text Lines (Displayed Row by Row)")
-    
-    cropped_output_rows = []
-    for i in range(20):
-        with gr.Row():
-            cropped_output_rows.append(gr.Image(type="pil", label=f"Line {i+1}"))
+    gr.Markdown("### Cropped Text Lines (Small Fixed-Size Previews)")
+
+    cropped_output = gr.Gallery(label="Detected Text Lines", columns=1, preview=True)
 
     def process_and_display(image):
         annotated_img, cropped_imgs = detect_and_crop_lines(image)
-        cropped_imgs += [None] * (20 - len(cropped_imgs))
-        return [annotated_img] + cropped_imgs[:20]
+        return annotated_img, cropped_imgs  
 
     image_input.upload(
         process_and_display,
         inputs=image_input,
-        outputs=[output_annotated] + cropped_output_rows
+        outputs=[output_annotated, cropped_output]
     )
 
 iface.launch()