Show cropped regions with preserved aspect ratio and dimensions

app.py

@@ -1153,7 +1153,14 @@ def _draw_selected_region_boxes(image, boxes):
     return img_out
 
 def _region_gallery_items(regions):
-    return [(r["image"], f"Region {i}") for i, r in enumerate(regions, 1)]
+    items = []
+    for i, r in enumerate(regions, 1):
+        img = r["image"]
+        label = f"Region {i}"
+        if isinstance(img, Image.Image):
+            label = f"{label} ({img.width}x{img.height})"
+        items.append((img, label))
+    return items
 
 def _label_gallery_items(items, prefix=None):
     labeled = []
@@ -1164,6 +1171,8 @@ def _label_gallery_items(items, prefix=None):
             img, label = item, f"Item {i}"
         if prefix:
             label = f"{prefix} - {label}"
+        if isinstance(img, Image.Image):
+            label = f"{label} ({img.width}x{img.height})"
         labeled.append((img, label))
     return labeled
 
@@ -1392,7 +1401,14 @@ with gr.Blocks(title="DeepSeek-OCR-2") as demo:
                 add_region_btn = gr.Button("Add Region", variant="secondary")
                 clear_regions_btn = gr.Button("Clear Regions")
             selection_status = gr.Textbox(label="Region Selection Status", value="No saved regions.", interactive=False, visible=False)
-            selected_regions_gallery = gr.Gallery(label="Selected Regions", show_label=True, columns=2, height=190, visible=False)
+            selected_regions_gallery = gr.Gallery(
+                label="Selected Regions",
+                show_label=True,
+                columns=2,
+                height=190,
+                visible=False,
+                object_fit="contain",
+            )
 
             with gr.Accordion("Advanced Options", open=False):
                 equation_zoom = gr.Checkbox(label="Equation Zoom (multipass)", value=False)
@@ -1410,7 +1426,7 @@ with gr.Blocks(title="DeepSeek-OCR-2") as demo:
                 with gr.Tab("Boxes", id="tab_boxes"):
                     img_out = gr.Image(type="pil", height=560, show_label=False)
                 with gr.Tab("Cropped Images", id="tab_crops"):
-                    gallery = gr.Gallery(show_label=False, columns=3, height=420)
+                    gallery = gr.Gallery(show_label=False, columns=3, height=420, object_fit="contain")
                 with gr.Tab("Raw Text", id="tab_raw"):
                     raw_out = gr.Textbox(lines=20, buttons=["copy"], show_label=False)
             download_btn = gr.DownloadButton("Download Markdown", visible=False, variant="secondary")