🎉 Complete Project Assembly

#!/usr/bin/env python3 """ Human-Tracking Table Fan Complete system with face detection, servo tracking, display, and button control """ import cv2 import RPi.GPIO as GPIO from RPLCD.i2c import CharLCD import time from datetime import datetime # ===== CONFIGURATION ===== SERVO_PIN = 18 BUTTON_PIN = 2 LCD_ADDRESS = 0x27 # May be 0x3F - check with i2cdetect -y 1 CAMERA_WIDTH = 320 CAMERA_HEIGHT = 240 # ===== SETUP FUNCTIONS ===== def setup_face_detection(): """Initialize face detection""" face_cascade = cv2.CascadeClassifier( cv2.data.haarcascades + 'haarcascade_frontalface_default.xml' ) return face_cascade def setup_servo(): """Initialize servo motor""" GPIO.setup(SERVO_PIN, GPIO.OUT) pwm = GPIO.PWM(SERVO_PIN, 50) pwm.start(0) return pwm def setup_button(): """Initialize button""" GPIO.setup(BUTTON_PIN, GPIO.IN, pull_up_down=GPIO.PUD_UP) def setup_display(): """Initialize LCD display""" try: lcd = CharLCD(i2c_expander='PCF8574', address=LCD_ADDRESS, port=1, cols=16, rows=2, dotsize=8) return lcd except: print("LCD not found - continuing without display") return None def setup_camera(): """Initialize camera""" cap = cv2.VideoCapture(0) cap.set(cv2.CAP_PROP_FRAME_WIDTH, CAMERA_WIDTH) cap.set(cv2.CAP_PROP_FRAME_HEIGHT, CAMERA_HEIGHT) return cap # ===== CONTROL FUNCTIONS ===== def set_servo_angle(pwm, angle): """Move servo to specific angle (0-180 degrees)""" angle = max(0, min(180, angle)) # Clamp to valid range duty = 2 + (angle / 180) * 10 pwm.ChangeDutyCycle(duty) time.sleep(0.1) pwm.ChangeDutyCycle(0) # Stop signal to prevent jitter return angle def check_button_press(last_state): """Check if button was pressed, return (new_state, was_pressed)""" current_state = GPIO.input(BUTTON_PIN) was_pressed = (last_state == GPIO.HIGH and current_state == GPIO.LOW) if was_pressed: time.sleep(0.3) # Debounce return current_state, was_pressed def calculate_servo_angle(face_x, frame_width): """Convert face X position to servo angle""" angle = (face_x / frame_width) * 180 return max(0, min(180, angle)) def update_display(lcd, tracking_active, face_detected, servo_angle): """Update LCD display with current status""" if lcd is None: return lcd.clear() # Line 1: Tracking status if tracking_active: lcd.write_string("Tracking: ON") else: lcd.write_string("Tracking: OFF") # Line 2: Face and angle lcd.cursor_pos = (1, 0) if face_detected: lcd.write_string(f"Face:YES {int(servo_angle)}°") else: lcd.write_string("Face: NO") # ===== MAIN SYSTEM ===== def main(): """Main program loop""" print("Initializing Human-Tracking Fan System...") # Initialize all components GPIO.setmode(GPIO.BCM) face_cascade = setup_face_detection() servo_pwm = setup_servo() setup_button() lcd = setup_display() cap = setup_camera() # State variables tracking_active = False current_servo_angle = 90 # Start at center last_button_state = GPIO.HIGH last_display_update = 0 # Initial display if lcd: lcd.write_string("Press button") lcd.cursor_pos = (1, 0) lcd.write_string("to start") time.sleep(2) print("System ready! Press button to start tracking.") print("Press Ctrl+C to exit.") try: while True: # Check button last_button_state, button_pressed = check_button_press(last_button_state) if button_pressed: tracking_active = not tracking_active status = "STARTED" if tracking_active else "STOPPED" print(f"Tracking {status}!") if lcd: lcd.clear() lcd.write_string(f"Tracking {status}") time.sleep(1) if tracking_active: # Read camera frame ret, frame = cap.read() if ret: # Convert to grayscale gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # Detect faces faces = face_cascade.detectMultiScale(gray, 1.1, 4) if len(faces) > 0: # Face detected - track it x, y, w, h = faces[0] face_center_x = x + w // 2 # Calculate target angle target_angle = calculate_servo_angle(face_center_x, CAMERA_WIDTH) # Smooth movement toward target angle_diff = target_angle - current_servo_angle if abs(angle_diff) > 2: # Dead zone # Move in steps step_size = 5 step = step_size if angle_diff > 0 else -step_size new_angle = current_servo_angle + step new_angle = max(0, min(180, new_angle)) current_servo_angle = set_servo_angle(servo_pwm, new_angle) # Update display every 0.5 seconds current_time = time.time() if current_time - last_display_update >= 0.5: update_display(lcd, True, True, current_servo_angle) last_display_update = current_time else: # No face detected current_time = time.time() if current_time - last_display_update >= 0.5: update_display(lcd, True, False, current_servo_angle) last_display_update = current_time else: # Tracking stopped current_time = time.time() if current_time - last_display_update >= 0.5: update_display(lcd, False, False, current_servo_angle) last_display_update = current_time time.sleep(0.1) # Main loop delay except KeyboardInterrupt: print("\nShutting down...") # Cleanup cap.release() servo_pwm.stop() if lcd: lcd.clear() lcd.write_string("Shutting down") time.sleep(2) lcd.clear() GPIO.cleanup() print("System stopped. Goodbye!") # ===== RUN PROGRAM ===== if __name__ == "__main__": main()