fshow the count of ripe and unripe tomatoes

app.py

@@ -93,58 +93,66 @@ interface_image = gr.Interface(
 )
 
 def show_preds_video(video_path):
+    results = model.track(source=video_path, persist=True, tracker="bytetrack.yaml", verbose=False, stream=True)
+    
+    ripe_ids = set()
+    unripe_ids = set()
+    
+    # Get video frame dimensions for centering text
     cap = cv2.VideoCapture(video_path)
-    while(cap.isOpened()):
-        ret, frame = cap.read()
-        if ret:
-            frame_copy = frame.copy()
-            outputs = model.predict(source=frame)
-            results = outputs[0].cpu().numpy()
-            
-            boxes = results.boxes
-            confidences = boxes.conf
-            classes = boxes.cls
+    if not cap.isOpened():
+        print("Error: Could not open video.")
+        return
+    frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    cap.release()
+
+    for output in results:
+        frame = output.orig_img
+        
+        if output.boxes and output.boxes.id is not None:
             names = model.model.names
+            boxes = output.boxes
+            ids = boxes.id.cpu().numpy().astype(int)
+            classes = boxes.cls.cpu().numpy().astype(int)
 
-            for box, conf, cls in zip(boxes.xyxy, confidences, classes):
+            for box, cls, track_id in zip(boxes.xyxy, classes, ids):
                 x1, y1, x2, y2 = map(int, box)
-
-                # Determine color based on class
-                class_name = names[int(cls)]
-                if class_name.lower() == 'ripe':
-                    color = (0, 0, 255)  # Red for ripe
+                class_name = names[cls]
+
+                # Define BGR colors directly for OpenCV functions
+                if class_name.lower() == "ripe":
+                    # To get RED in Gradio (RGB), you need to use (255, 0, 0) BGR
+                    # Note: You were using (0, 0, 255) which is Blue in RGB after conversion.
+                    color = (0, 0, 255)
+                    ripe_ids.add(track_id)
                 else:
-                    color = (0, 255, 0)  # Green for unripe
-
-                # Draw rectangle around object
-                cv2.rectangle(
-                    frame_copy,
-                    (x1, y1),
-                    (x2, y2),
-                    color=color,
-                    thickness=2,
-                    lineType=cv2.LINE_AA
-                )
-
-                # Display class label on top of rectangle with capitalized class name
-                label = f"{class_name.capitalize()}: {conf:.2f}"
-                cv2.putText(
-                    frame_copy,
-                    label,
-                    (x1, y1 - 10),  # Position slightly above the top of the rectangle
-                    cv2.FONT_HERSHEY_SIMPLEX,
-                    0.5,
-                    color,  # Use the same color as the rectangle
-                    1,
-                    cv2.LINE_AA
-                )
-
-            # Convert frame to RGB (Gradio expects RGB format)
-            yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
-        else:
-            break
-
-    cap.release()
+                    # To get GREEN in Gradio (RGB), you need to use (0, 255, 0) BGR.
+                    # This color is already correct.
+                    color = (0, 255, 0)
+                    unripe_ids.add(track_id)
+
+                cv2.rectangle(frame, (x1, y1), (x2, y2), color, 2)
+                cv2.putText(frame, f"{class_name.capitalize()} ID:{track_id}",
+                            (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
+
+        ripe_count_text = f"Ripe: {len(ripe_ids)}"
+        unripe_count_text = f"Unripe: {len(unripe_ids)}"
+        full_text = f"{ripe_count_text} | {unripe_count_text}"
+
+        # Get text size to center it
+        (text_width, text_height), baseline = cv2.getTextSize(full_text, cv2.FONT_HERSHEY_SIMPLEX, 1, 2)
+        text_x = (frame_width - text_width) // 2
+        text_y = 40 # A fixed position at the top
+
+        # Display the counts at the top center
+        cv2.putText(frame, full_text, (text_x, text_y),
+                    cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
+        
+        # This line is crucial for the fix.
+        # It correctly converts the frame from BGR to RGB for Gradio.
+        yield cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    
+    print(f"Final Counts → Ripe: {len(ripe_ids)}, Unripe: {len(unripe_ids)}")
 
 inputs_video = [
     gr.components.Video(label="Input Video"),