Update app.py

app.py

@@ -1,39 +1,23 @@
 import gradio as gr
-from PIL import Image, ImageOps
+import tempfile
+from plant_analysis_demo.wrapper import run_pipeline_on_image
 
-def demo_process(image):
-    """Simple placeholder:
-    - returns original
-    - grayscale version
-    - inverted colors
-    """
+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"
+        image.save(img_path)
+        outputs = run_pipeline_on_image(str(img_path), tmpdir, save_artifacts=True)
+        return list(outputs.values())
 
-    # Original
-    original = image
-
-    # Grayscale
-    gray = ImageOps.grayscale(image)
-
-    # Invert
-    inverted = ImageOps.invert(image.convert("RGB"))
-
-    return [original, gray, inverted]
-
-# Build Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌿 Plant Analysis Demo\nUpload an image and see example outputs.")
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            inp = gr.Image(type="pil", label="Upload Image")
-            run = gr.Button("Run Demo", variant="primary")
-        with gr.Column(scale=2):
-            # Removed .style() for compatibility
-            gallery = gr.Gallery(label="Outputs")
-
-    run.click(demo_process, inputs=inp, outputs=gallery)
+    gr.Markdown("# 🌿 Sorghum Single-Image Demo")
+    inp = gr.Image(type="pil", label="Upload Image")
+    run = gr.Button("Run Pipeline", variant="primary")
+    gallery = gr.Gallery(label="Outputs")
+    run.click(process, inputs=inp, outputs=gallery)
 
 if __name__ == "__main__":
     demo.launch()