feat(config): add discovery mode

adutchak · Oct 9, 2023 · 9e94671 · 9e94671
1 parent 2be1bd3
commit 9e94671
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -96,7 +96,7 @@ func main() {
 				continue
 			}
 			if len(output.FaceDetails) == 0 {
-				l.Warn("no faces detected")
+				l.Errorf("no faces detected in the image: %s", configuration.TargetImagePath)
 				if !configuration.DiscoveryMode {
 					publishMqttMessage(mqttClient, configuration.MqttTopic, configuration.MqttNotRecognizedMessage)
 					continue
@@ -175,6 +175,8 @@ func main() {
 					publishMqttMessage(mqttClient, configuration.MqttTopic, configuration.MqttNotRecognizedMessage)
 				}
 			}
+
+			mqttClient.Disconnect(250)
 		}
 	}(doneChan)
 

diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -108,5 +108,8 @@ func Parse(args []string) (*Config, error) {
 			conf.ConfidencesNotMoreThanNormalized[s[0]] = s[1]
 		}
 	}
+	if conf.DiscoveryMode {
+		l.Warn("RUNNING APPLICATION IN DISCOVERY MODE")
+	}
 	return conf, nil
 }