diff --git a/dayglo_detector/dayglo_detector.py b/dayglo_detector/dayglo_detector.py index 440cd1d..793de00 100644 --- a/dayglo_detector/dayglo_detector.py +++ b/dayglo_detector/dayglo_detector.py @@ -89,8 +89,8 @@ async def process_snapshot(client, payload): # Attempt to read the saved image with OpenCV image = cv2.imread(temp_image_path) - if image is not None: - rating = calculate_dayglo_rating(image) + if image is not None: + rating = calculate_dayglo_rating(image) print("Dayglo Rating calculated:", rating) await publish_rating(client, rating) else: @@ -119,19 +119,19 @@ async def main(): # Handle command line argument for image file if len(sys.argv) > 1: image_file = sys.argv[1] - if os.path.exists(image_file): + if os.path.exists(image_file): client = DaygloDetectorMQTTClient("dayglo_detector") client.set_auth_credentials(MQTT_USERNAME, MQTT_PASSWORD) - await client.connect(MQTT_BROKER, MQTT_PORT, keepalive=60) + await client.connect(MQTT_BROKER, MQTT_PORT, keepalive=60) print(f"Processing image from file: {image_file}") image = cv2.imread(image_file) if image is not None: rating = calculate_dayglo_rating(image) print("Dayglo Rating calculated from file:", rating) - await publish_rating(client, rating) + await publish_rating(client, rating) else: print("Invalid image file provided.") - else: + else: print(f"File not found: {image_file}") # Set up MQTT client for normal operation