Update app.py

app.py

@@ -35,9 +35,26 @@ if uploaded_file:
     
     # Show the rendered result
     st.image(rendered_image, caption="Detection Results", use_column_width=True)
+    # Count the number of each cell type
+    cell_counts = {"RBC": 0, "WBC": 0, "Platelets": 0}
+
 
     # Display details of detected boxes
     st.write("Detection Results:")
+     for box in results[0].boxes:
+        class_index = int(box.cls)  # Get the class index
+        if class_index == 1:  # RBC
+            cell_counts["RBC"] += 1
+        elif class_index == 2:  # WBC
+            cell_counts["WBC"] += 1
+        elif class_index == 0:  # Platelets
+            cell_counts["Platelets"] += 1
+
+    # Display the counts of each cell type
+    st.write("Cell Type Counts:")
+    st.write(pd.DataFrame.from_dict(cell_counts, orient='index', columns=['Count']))
+
+else:
     for box in results[0].boxes:
         st.write(f"Bounding box: {box.xyxy}")
         st.write(f"Confidence: {box.conf}")