Update app.py

app.py

@@ -106,41 +106,15 @@ def server(input, output, session: Session):
 
             id = "image_" + str(i)
             opacity = ui.input_slider(id, "Opacity", 0, 1.0, 0.5)
-            
-            @render.plot
-            def plot_predicitons():
-                fig, ax = plt.subplots()
-    
-                ax = plt.Axes(fig, [0., 0., 1., 1.])
-                ax.set_axis_off()
-                fig.add_axes(ax)
-                
-                v = Visualizer(r["image"][:, :, ::-1],
-                               scale=1, instance_mode=ColorMode.SEGMENTATION, font_size_scale=1)
-    
-                colours = []
-                for cls in r["instances"].pred_classes:
-                    if cls == 0:
-                        colours.append([1,0,0])
-                    elif cls == 1:
-                        colours.append([1,1,0])
-                    elif cls == 2:
-                        colours.append([0,0,0])
-    
-                out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
-                                          assigned_colors = colours, 
-                                          alpha = opacity)
-                ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
-
-                return fig
 
-            plot_predicitons.__name__ = "plot_" + str(i)
+            plot_name = f"plot_{i}"
+            output(create_plot(r), id=plotname)
                                 
             output.append(
                 ui.div(
                     ui.row(
                         ui.column(4, ui.img(src=f"data:image/png;base64,{r['image_base64']}")),
-                        ui.column(4, ui.output_plot("plot_" + str(i))),
+                        ui.column(4, ui.output_plot(plot_name)),
                     ),
                     opacity,
                     ui.h5(r['filename'], style="margin-top: 15px;"),
@@ -154,7 +128,36 @@ def server(input, output, session: Session):
                     class_="card p-3"
                 )
             )
-        return ui.div(output)        
+        return ui.div(output)      
+
+    def create_plot(r):
+        @render.plot
+        def plot_predicitons():
+            fig, ax = plt.subplots()
+    
+            ax = plt.Axes(fig, [0., 0., 1., 1.])
+            ax.set_axis_off()
+            fig.add_axes(ax)
+            
+            v = Visualizer(r["image"][:, :, ::-1],
+                           scale=1, instance_mode=ColorMode.SEGMENTATION, font_size_scale=1)
+    
+            colours = []
+            for cls in r["instances"].pred_classes:
+                if cls == 0:
+                    colours.append([1,0,0])
+                elif cls == 1:
+                    colours.append([1,1,0])
+                elif cls == 2:
+                    colours.append([0,0,0])
+    
+            out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
+                                      assigned_colors = colours, 
+                                      alpha = opacity)
+            ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
+    
+            return fig
+        return plot_predicitons
 
     @session.download()
     def download():