Update app.py

app.py

@@ -74,11 +74,16 @@ def server(input, output, session: Session):
 
                 # Run prediction with original BGR image
                 prediction = predictor(im)
+                classes = pred["instances"].pred_classes.tolist()
 
                 results.append({
                     "filename": file["name"],
                     "image": img_base64,
-                    **prediction
+                    **prediction,
+                    "viable": classes.count(1),
+                    "non-viable": classes.count(2),
+                    "empty": classes.count(3),
+                    "total": len(classes)
                 })
 
         # Update reactive value
@@ -104,12 +109,19 @@ def server(input, output, session: Session):
                     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', '?')}"),
                     class_="results-text"
                 ),
                 class_="card p-3"
             ) for r in results]
         )
 
+    # @render.plot
+    # def output_image():
+    #         results.
+            
+    #         return fig
+
     @session.download()
     def download():
         results = analysis_results.get()
@@ -117,7 +129,8 @@ def server(input, output, session: Session):
             "Filename": r["filename"],
             "Viable": r.get("viable", ""),
             "Nonviable": r.get("nonviable", ""),
-            "Empty": r.get("empty", "")
+            "Empty": r.get("empty", ""),
+            "Total": r.get("total", "")
         } for r in results])
 
         # Create in-memory CSV file