Update app.py

app.py

@@ -8,7 +8,6 @@ import streamlit as st
 from streamlit_webrtc import WebRtcMode, webrtc_streamer
 from sample_utils.turn import get_ice_servers
 from cvzone.HandTrackingModule import HandDetector
-from cvzone.SelfiSegmentationModule import SelfiSegmentation
 import os
 import time
 
@@ -22,20 +21,13 @@ st.subheader('''Turn on the webcam and use hand gestures to interact with the vi
 Use 'a' and 'd' from the keyboard to change the background.''')
 
 # Initialize modules
-detector = HandDetector(maxHands=1, detectionCon=0.85)
-segmentor = SelfiSegmentation()
+detector = HandDetector(maxHands=1, detectionCon=0.7)
 
 # Define virtual keyboard layout
 keys = [["Q", "W", "E", "R", "T", "Y", "U", "I", "O", "P"],
         ["A", "S", "D", "F", "G", "H", "J", "K", "L", ";"],
         ["Z", "X", "C", "V", "B", "N", "M", ",", ".", "/"]]
 
-class Button:
-    def __init__(self, pos, text, size=[100, 100]):
-        self.pos = pos
-        self.size = size
-        self.text = text
-
 class Detection(NamedTuple):
     label: str
     score: float
@@ -43,7 +35,7 @@ class Detection(NamedTuple):
 
 result_queue: "queue.Queue[List[Detection]]" = queue.Queue()
 
-# Load background images
+# Background image loading
 listImg = os.listdir('model/street') if os.path.exists('model/street') else []
 if not listImg:
     st.error("Error: 'street' directory is missing or empty. Please add background images.")
@@ -62,17 +54,23 @@ if "output_text" not in st.session_state:
 def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
     global indexImg, output_text
 
+    # Convert the frame to BGR
     img = frame.to_ndarray(format="bgr24")
+
+    # Process the frame with Hand Detector
     hands, img = detector.findHands(img, draw=True)
 
     detections = []
     if hands:
+        logger.info(f"Detected {len(hands)} hand(s).")
         for hand in hands:
             bbox = hand['bbox']
             label = "Hand"
             score = hand['score']
             box = np.array([bbox[0], bbox[1], bbox[0] + bbox[2], bbox[1] + bbox[3]])
             detections.append(Detection(label=label, score=score, box=box))
+    else:
+        logger.info("No hands detected.")
 
     result_queue.put(detections)
     st.session_state["output_text"] = output_text