Add Plot

app.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 from typing import List, Dict, Tuple
 import matplotlib.colors as mpl_colors
+import matplotlib.pyplot as plt
 import pandas as pd
 import seaborn as sns
 import shinyswatch
@@ -22,9 +23,10 @@ app_ui = ui.page_fillable(
             ui.input_file("image_input", "Upload image: ", multiple=True),
         ),
         ui.output_image("image"),
-        ui.output_image("image_output"),
-        ui.output_image("prediction"),
-        ui.output_image("prob")
+        # ui.output_image("image_output"),
+        ui.output_plot("plot"),
+        # ui.output_image("prediction"),
+        # ui.output_image("prob")
     ),
 )
 
@@ -52,6 +54,33 @@ def server(input: Inputs, output: Outputs, session: Session):
             return img
         return None
 
+    @render.plot(alt="A histogram")  
+    def plot(): 
+        if input.image_input():
+            new_image = input.image_input()[0]['datapath']
+
+            pred_prob, pred_prediction = run.pred(new_image)
+            
+            fig, axes = plt.subplots(1, 3, figsize=(15, 5))
+
+            axes[0].imshow(new_image, cmap='gray')
+            axes[0].set_title("Image")
+
+            im = axes[1].imshow(pred_prob) 
+            axes[1].set_title("Probability Map")
+            cbar = fig.colorbar(im, ax=axes[1])
+
+            axes[2].imshow(pred_prediction, cmap='gray')
+            axes[2].set_title("Prediction")
+
+            for ax in axes:
+                ax.set_xticks([])
+                ax.set_yticks([])
+                ax.set_xticklabels([])
+                ax.set_yticklabels([])
+
+        return fig  
+
 
 app = App(
     app_ui,

run.py

@@ -48,4 +48,4 @@ def pred(src):
     pred_prediction = (pred_prob > PROBABILITY_THRES).astype(np.uint8)
     
     x=1
-    return x
+    return pred_prob, pred_prediction