Update gully_drs_core/ball_detection.py

gully_drs_core/ball_detection.py

@@ -2,11 +2,19 @@ import cv2
 import numpy as np
 from deep_sort_realtime.deepsort_tracker import DeepSort
 from scipy.interpolate import UnivariateSpline
+import platform
+import asyncio
+
+# Check if NumPy is available
+try:
+    import numpy
+except ImportError:
+    raise ImportError("NumPy is required but not found. Please install it with 'pip install numpy'.")
 
 # Global tracker initialization
 tracker = DeepSort(max_age=5)
 
-def detect_ball_hsv(frame, hsv_values):
+async def detect_ball_hsv(frame, hsv_values):
     """
     Detects a ball in an image frame based on HSV color filtering.
     
@@ -57,7 +65,7 @@ def detect_ball_hsv(frame, hsv_values):
         print(f"Error in HSV ball detection: {str(e)}")
         return None, frame
 
-def track_ball_and_estimate_speed(frame, ball_position, frame_rate=60):
+async def track_ball_and_estimate_speed(frame, ball_position, frame_rate=60):
     """Track ball and estimate speed with DeepSort."""
     try:
         dets = []
@@ -95,7 +103,7 @@ def track_ball_and_estimate_speed(frame, ball_position, frame_rate=60):
     except Exception as e:
         raise Exception(f"Error in ball tracking: {str(e)}")
 
-def detect_lbw(frame, ball_position):
+async def detect_lbw(frame, ball_position):
     """
     Placeholder for LBW detection logic.
     Implement based on your specific requirements.
@@ -109,4 +117,11 @@ def detect_lbw(frame, ball_position):
         return annotated_frame
     except Exception as e:
         print(f"Error in LBW detection: {str(e)}")
-        return frame
+        return frame
+
+# Pyodide compatibility for asyncio
+if platform.system() == "Emscripten":
+    asyncio.ensure_future(detect_ball_hsv(None, {}))  # Example initialization
+else:
+    if __name__ == "__main__":
+        asyncio.run(detect_ball_hsv(None, {}))