Update app.py

app.py

@@ -80,7 +80,7 @@ def process_video(input_video):
 
     fps = cap.get(cv2.CAP_PROP_FPS)
 
-    # Define the output video path in the current directory
+    # Define the output video path
     output_video_path = os.path.join(os.getcwd(), "processed_video.mp4")
 
     writer = imageio.get_writer(output_video_path, fps=fps, codec="h264")
@@ -102,11 +102,14 @@ def process_video(input_video):
 
         # Convert the resized frame to grayscale
         gray_frame = cv2.cvtColor(resized_frame, cv2.COLOR_BGR2GRAY)
-        # Convert grayscale to 3 channels (RGB format)
-        input_frame = cv2.merge([gray_frame, gray_frame, gray_frame])  # Convert grayscale to 3 channels
 
+        # Convert grayscale to RGB (3 channels)
+        input_frame = cv2.merge([gray_frame, gray_frame, gray_frame])
+
+        # Get predictions from the model
         results = model.predict(input_frame)
 
+        # Drawing bounding boxes and adding text
         for result in results:
             for box in result.boxes:
                 if box.conf[0] >= confidence_threshold:
@@ -122,14 +125,18 @@ def process_video(input_video):
                     timestamps.append(timestamp)
                     classes_detected.append(class_id)
 
-        # Ensure the frame is in RGB format before appending to the writer
+        # Convert the frame to RGB (BGR to RGB)
         rgb_frame = cv2.cvtColor(resized_frame, cv2.COLOR_BGR2RGB)
-        
-        # Ensure the frame has the correct shape (640x640x3)
+
+        # Log the shape of the frame to check it's correct
+        print(f"Frame shape: {rgb_frame.shape}")  # This should print (640, 640, 3)
+
+        # Check if the frame has 3 channels (RGB)
         if rgb_frame.shape[2] != 3:
             print(f"Warning: Unexpected frame shape {rgb_frame.shape}, adjusting...")
             rgb_frame = cv2.cvtColor(resized_frame, cv2.COLOR_BGR2RGB)
 
+        # Append the frame to the video writer
         writer.append_data(rgb_frame)
 
     cap.release()
@@ -145,6 +152,7 @@ def process_video(input_video):
 
 
 
+
 def process_image(input_image):
     # Convert image from RGB to BGR for OpenCV processing
     bgr_frame = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)