Update app.py

app.py

@@ -107,6 +107,25 @@ def server(input, output, session: Session):
             id = "image_" + str(i)
             opacity = ui.input_slider(id, "Opacity", 0, 1.0, 0.5),
             
+            ui_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))),
+                    ),
+                    opacity,
+                    ui.h5(r['filename'], style="margin-top: 15px;"),
+                    ui.div(
+                        ui.span(f"Viable = {r.get('viable', '? ')}", style="margin: 0 15px;"),
+                        ui.span(f"Nonviable = {r.get('non-viable', '? ')}", style="margin: 0 15px;"),
+                        ui.span(f"Empty = {r.get('empty', '? ')}", style="margin: 0 15px;"),
+                        ui.span(f"Total = {r.get('total', '? ')}", style="margin: 0 15px;"),
+                        class_="results-text"
+                    ),
+                    class_="card p-3"
+                )
+            )
+
             @output(id="plot_" + str(i))
             @render.plot
             def plot_prediction():
@@ -135,26 +154,7 @@ def server(input, output, session: Session):
                 ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
         
                 return fig
-            
-            ui_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))),
-                    ),
-                    opacity,
-                    ui.h5(r['filename'], style="margin-top: 15px;"),
-                    ui.div(
-                        ui.span(f"Viable = {r.get('viable', '? ')}", style="margin: 0 15px;"),
-                        ui.span(f"Nonviable = {r.get('non-viable', '? ')}", style="margin: 0 15px;"),
-                        ui.span(f"Empty = {r.get('empty', '? ')}", style="margin: 0 15px;"),
-                        ui.span(f"Total = {r.get('total', '? ')}", style="margin: 0 15px;"),
-                        class_="results-text"
-                    ),
-                    class_="card p-3"
-                )
-            )
-        # output(create_plot(r, input[id]()), id=plot_name)
+                
         return ui.div(ui_output)      
 
     @session.download()