update

app.py

@@ -3,12 +3,21 @@ import tempfile
 from pathlib import Path
 from wrapper import run_pipeline_on_image
 
+def show_preview(image):
+    """Show the original uploaded image as-is."""
+    if image is None:
+        return None
+    # Return original image without conversion
+    return image
+
 def process(image):
     if image is None:
         return []
     with tempfile.TemporaryDirectory() as tmpdir:
-        # Save PIL image to temp file
-        img_path = Path(tmpdir) / "input.png"
+        # Save PIL image preserving original format
+        # Determine extension from image format
+        ext = image.format.lower() if image.format else 'png'
+        img_path = Path(tmpdir) / f"input.{ext}"
         image.save(img_path)
         outputs = run_pipeline_on_image(str(img_path), tmpdir, save_artifacts=True)
         # Keep order consistent: return exactly the 7 images
@@ -31,7 +40,7 @@ with gr.Blocks() as demo:
     gallery = gr.Gallery(label="Analysis Results", columns=3, height="auto")
     
     # Update preview when image is uploaded
-    inp.change(fn=lambda x: x, inputs=inp, outputs=preview)
+    inp.change(fn=show_preview, inputs=inp, outputs=preview)
     run.click(process, inputs=inp, outputs=gallery)
 
 if __name__ == "__main__":

requirements.txt

@@ -9,4 +9,6 @@ scikit-image
 scikit-learn
 scipy
 matplotlib
-plantcv
+plantcv
+kornia
+timm