update

app.py

@@ -7,8 +7,13 @@ from PIL import Image
 from itertools import product
 
 def show_preview(image):
-    """Show uploaded image as-is."""
-    return image
+    """Show uploaded image, converted to RGB for reliable preview rendering."""
+    if image is None:
+        return None
+    try:
+        return image.convert("RGB")
+    except Exception:
+        return image
 
 def process(image):
     if image is None:
@@ -21,10 +26,23 @@ def process(image):
         outputs = run_pipeline_on_image(str(img_path), tmpdir, save_artifacts=True)
 
         # Assemble displays
-        overlay = outputs.get('Overlay')
-        mask = outputs.get('Mask')
+        def load_pil(path_str):
+            try:
+                if not path_str:
+                    return None
+                im = Image.open(path_str)
+                im = im.convert('RGB')
+                # Copy to memory so it survives after tmpdir is removed
+                copied = im.copy()
+                im.close()
+                return copied
+            except Exception:
+                return None
+
+        overlay = load_pil(outputs.get('Overlay'))
+        mask = load_pil(outputs.get('Mask'))
         order = ['NDVI', 'ARI', 'GNDVI']
-        gallery_items = [outputs[k] for k in order if k in outputs]
+        gallery_items = [load_pil(outputs[k]) for k in order if k in outputs]
         stats_text = outputs.get('StatsText', '')
         return overlay, mask, gallery_items, stats_text
 
@@ -40,8 +58,8 @@ with gr.Blocks() as demo:
             preview = gr.Image(type="pil", label="Uploaded Image Preview", interactive=False)
 
     with gr.Row():
-        overlay_img = gr.Image(type="filepath", label="Segmentation Overlay", interactive=False)
-        mask_img = gr.Image(type="filepath", label="Mask", interactive=False)
+        overlay_img = gr.Image(type="pil", label="Segmentation Overlay", interactive=False)
+        mask_img = gr.Image(type="pil", label="Mask", interactive=False)
 
     gallery = gr.Gallery(label="Vegetation Indices", columns=3, height="auto")
     stats = gr.Textbox(label="Statistics", lines=4)