new version of trex

trex/main.py

@@ -136,7 +136,7 @@ if __name__ == '__main__':
     start = time()
 
     if DEBUG: cv2.imshow('View', rgbScreen) 
-    while cv2.waitKey(1) != ord('q'):
+    while True:
         binScreen = take_binaryScreenshot(sct, playing_bbox)
         binScreen = to_bbox(binScreen)
         labels, labeledScreen = cv2.connectedComponents(binScreen.astype('uint8'))
@@ -170,6 +170,7 @@ if __name__ == '__main__':
             cv2.circle(rgbScreen, trigger_feather[::-1], radius=2, color=(255, 0, 0), thickness=2)
             cv2.circle(rgbScreen, trigger_second[::-1], radius=2, color=(0, 255, 0), thickness=2)
             cv2.imshow('View', rgbScreen)
+            if cv2.waitKey(1) == ord('q'): break