Calm before disaster

rpi/main.py

@@ -1,15 +1,38 @@
 #!/usr/bin/env python3
+import cv2
 
 from hardware.screen.screen import Screen
 from hardware.rfid.reader import RfidReader
+from services.detection_service import DetectionService
 from services.mqtt_service import MQTTService
 
 screen = Screen()
-service = MQTTService("127.0.0.1", 1883)
+mqtt_service = MQTTService("127.0.0.1", 1883)
+detection_service = DetectionService()
 
 def publish(chip_id : int) :
     screen.displayMessage(str(chip_id))
-    service.publish("app", "/board-mate/rfid/scan", str(chip_id), 0)
+    mqtt_service.publish("app", "/board-mate/rfid/scan", str(chip_id), 0)
+
+def detect() :
+    cap = cv2.VideoCapture(0)
+    cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
+    cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
+    cap.set(cv2.CAP_PROP_FPS, 30)
+
+    print("Initialized")
+    if not cap.isOpened():
+        print("Error: Could not open camera")
+        exit()
+
+    cv2.namedWindow("Predictions", cv2.WINDOW_NORMAL)
+
+    while True:
+        ret, frame = cap.read()
+        detection_service.get_fen(frame)
+        if not ret:
+            print("Error: Failed to grab frame")
+            break
 
 if __name__ == "__main__":
     screen.enableBackground()
@@ -18,6 +41,8 @@ if __name__ == "__main__":
     reader.subscribe(publish)
     reader.start()
 
+    detect()
+
     while True:
         pass