Update virtualpainter.py

Author: 770navyasharma

OpenCV Projects/VirtualPainter/virtualpainter.py

@@ -1,52 +1,61 @@
-
-#### File: `virtual_painter.py`
-```python
 import cv2
 import numpy as np
 
-# Set up the color ranges for detection (e.g., blue pen)
-lower_blue = np.array([100, 150, 0])
-upper_blue = np.array([140, 255, 255])
+# HSV color range for detecting the color object (adjust this range for different colors)
+lower_bound = np.array([0, 120, 70])
+upper_bound = np.array([10, 255, 255])
+
+# Initialize variables
+my_points = []  # List to store points for drawing
+
+# Function to detect color and return the coordinates of the detected object
+def find_color(img, lower_bound, upper_bound):
+    hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)  # Convert to HSV color space
+    mask = cv2.inRange(hsv_img, lower_bound, upper_bound)  # Create mask for specific color
+
+    # Find contours in the masked image
+    contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+    x, y, w, h = 0, 0, 0, 0
+
+    for contour in contours:
+        area = cv2.contourArea(contour)
+        if area > 500:  # Filter by area size to remove noise
+            x, y, w, h = cv2.boundingRect(contour)
+            cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)  # Draw a rectangle around the object
+            return x + w // 2, y
 
-# Initialize a blank canvas to draw on
-canvas = np.zeros((480, 640, 3), dtype="uint8")
+    return None
 
-# Start video capture
+# Function to draw on canvas based on detected points
+def draw_on_canvas(points, img):
+    for point in points:
+        cv2.circle(img, (point[0], point[1]), 10, (0, 0, 255), cv2.FILLED)  # Draw red circles at each point
+
+# Capture video from webcam
 cap = cv2.VideoCapture(0)
 
 while True:
-    ret, frame = cap.read()
-    if not ret:
+    success, img = cap.read()  # Read frame from webcam
+    if not success:
         break
 
-    # Convert to HSV color space for color detection
-    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
+    # Find the object in the current frame
+    new_point = find_color(img, lower_bound, upper_bound)
 
-    # Create a mask for detecting the blue color
-    mask = cv2.inRange(hsv, lower_blue, upper_blue)
-    
-    # Find contours of the detected blue object
-    contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    
-    if contours:
-        # Get the largest contour (assuming it's the pen)
-        max_contour = max(contours, key=cv2.contourArea)
-
-        # Get the center of the contour (i.e., the tip of the pen)
-        (x, y), radius = cv2.minEnclosingCircle(max_contour)
-        center = (int(x), int(y))
-        
-        # Draw on the canvas at the detected position
-        cv2.circle(canvas, center, 5, (255, 0, 0), -1)
-
-    # Merge the original frame with the canvas
-    combined = cv2.add(frame, canvas)
-
-    # Display the result
-    cv2.imshow('Virtual Painter', combined)
-    
+    # If a new point is detected, add it to the list
+    if new_point:
+        my_points.append(new_point)
+
+    # Draw on the canvas using the points stored
+    draw_on_canvas(my_points, img)
+
+    # Display the frame
+    cv2.imshow("Virtual Painter", img)
+
+    # Break the loop on 'q' key press
     if cv2.waitKey(1) & 0xFF == ord('q'):
         break
 
+# Release the webcam and close windows
 cap.release()
 cv2.destroyAllWindows()