Fix indentation issues in main function for proper script execution
- Corrected inconsistent indentation in the main function to avoid runtime `IndentationError`. - Ensured the MQTT client properly connects and publishes when processing images from the command line.
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user