changes

app.py

@@ -59,7 +59,11 @@ def detect_components(image):
                     
                     # ✅ Draw bounding box
                     draw_final.rectangle([x_min_full, y_min_full, x_max_full, y_max_full], outline="green", width=2)
-        
+
+        print(f"\nCounts after pass {pass_num}:")
+        for label, count in total_counts.items():
+            print(f"{label}: {count}")
+
         return final_image, total_counts, detected_boxes
 
     # ✅ First pass detection
@@ -88,7 +92,13 @@ def detect_components(image):
     for label in set(counts_pass1) | set(counts_pass2) | set(counts_pass3):
         final_counts[label] = counts_pass1.get(label, 0) + counts_pass2.get(label, 0) + counts_pass3.get(label, 0)
 
-    return image_after_pass1, image_after_pass2, image_after_pass3, dict(final_counts)
+    # ✅ Return counts after each pass
+    return (
+        image_after_pass1, counts_pass1,  
+        image_after_pass2, counts_pass2,  
+        image_after_pass3, counts_pass3,  
+        final_counts
+    )
 
 # ✅ Gradio Interface
 interface = gr.Interface(
@@ -96,9 +106,12 @@ interface = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=[
         gr.Image(type="pil", label="Detection Pass 1"),
+        gr.JSON(label="Counts After Pass 1"),  # ✅ Displays first pass counts
         gr.Image(type="pil", label="Detection Pass 2"),
+        gr.JSON(label="Counts After Pass 2"),  # ✅ Displays second pass counts
         gr.Image(type="pil", label="Detection Pass 3"),
-        gr.Label(label="Final Detected Components"),  # ✅ Gradio now properly receives numbers
+        gr.JSON(label="Counts After Pass 3"),  # ✅ Displays third pass counts
+        gr.JSON(label="Final Detected Components")  # ✅ Displays total counts
     ],
     title="HVAC Component Detector",
     description="Upload an image to detect HVAC components using Roboflow API across three passes."