Update app.py

app.py

@@ -80,9 +80,9 @@ def server(input, output, session: Session):
                     "filename": file["name"],
                     "image": img_base64,
                     **prediction,
-                    "viable": classes.count(1),
-                    "non-viable": classes.count(2),
-                    "empty": classes.count(3),
+                    "viable": classes.count(0),
+                    "non-viable": classes.count(1),
+                    "empty": classes.count(2),
                     "total": len(classes)
                 })
 
@@ -97,30 +97,30 @@ def server(input, output, session: Session):
             return ui.div("No results yet. Upload images and click 'Analyze'.",
                           class_="text-muted")
 
-        print(results)
         return ui.div(
             [ui.div(
                 ui.row(
                     ui.column(4, ui.img(src=f"data:image/png;base64,{r['image']}")),
-                    ui.column(4, ui.img(src=f"data:image/png;base64,{r['image']}")),
+                    ui.column(4, ui.output_plot("output", result=r)),
                 ),
                 ui.h5(r['filename'], style="margin-top: 15px;"),
                 ui.div(
-                    ui.span(f"Viable = {r.get('viable', '?')}"),
-                    ui.span(f"Nonviable = {r.get('nonviable', '?')}", style="margin: 0 15px;"),
-                    ui.span(f"Empty = {r.get('empty', '?')}"),
-                    ui.span(f"Total = {r.get('total', '?')}"),
+                    ui.span(f"Viable = {r.get('viable', '? ')}", style="margin: 0 15px;"),
+                    ui.span(f"Nonviable = {r.get('nonviable', '? ')}", 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"
             ) for r in results]
         )
 
-    # @render.plot
-    # def output_image():
-    #         results.
-            
-    #         return fig
+    @render.plot
+    def output(result):
+        fig = plt.imshow(cv2.imread(r['image']))
+        print(result)
+        
+        return fig
 
     @session.download()
     def download():