From d4060d731e1c19df653744c1184689065dbfed12 Mon Sep 17 00:00:00 2001
From: Don Restarone <35935196+donrestarone@users.noreply.github.com>
Date: Mon, 24 Feb 2025 18:02:25 -0500
Subject: [PATCH] Update vilib.py

---
 vilib/vilib.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/vilib/vilib.py b/vilib/vilib.py
index a36c592..19c08cb 100644
--- a/vilib/vilib.py
+++ b/vilib/vilib.py
@@ -17,6 +17,7 @@
 os.environ['LIBCAMERA_LOG_LEVELS'] = '*:ERROR'
 from picamera2 import Picamera2
 import libcamera
+from libcamera import controls
 
 import cv2
 import numpy as np
@@ -233,6 +234,8 @@ def camera():
 
         # init picamera
         picam2 = Picamera2()
+            # auto white balance and continuous autofocus for camera module v3
+        picam2.set_controls({"AwbMode": controls.AwbModeEnum.Auto, "AfMode": controls.AfModeEnum.Continuous})
 
         preview_config = picam2.preview_configuration
         # preview_config.size = (800, 600)