From 85f3f1ba9f9b7892b4c36d737880e025852bc722 Mon Sep 17 00:00:00 2001
From: Terrence <terrence@tenclass.com>
Date: Thu, 12 Jun 2025 14:58:33 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dmodel=E5=88=86=E5=8C=BA?=
 =?UTF-8?q?=E6=9C=AA=E5=88=9D=E5=A7=8B=E5=8C=96=E5=AF=BC=E8=87=B4=E9=87=8D?=
 =?UTF-8?q?=E5=90=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main/audio_processing/afe_wake_word.cc | 4 ++++
 main/audio_processing/esp_wake_word.cc | 5 ++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/main/audio_processing/afe_wake_word.cc b/main/audio_processing/afe_wake_word.cc
index 77955970..cfd77986 100644
--- a/main/audio_processing/afe_wake_word.cc
+++ b/main/audio_processing/afe_wake_word.cc
@@ -35,6 +35,10 @@ void AfeWakeWord::Initialize(AudioCodec* codec) {
     int ref_num = codec_->input_reference() ? 1 : 0;
 
     srmodel_list_t *models = esp_srmodel_init("model");
+    if (models == nullptr || models->num == -1) {
+        ESP_LOGE(TAG, "Failed to initialize wakenet model");
+        return;
+    }
     for (int i = 0; i < models->num; i++) {
         ESP_LOGI(TAG, "Model %d: %s", i, models->model_name[i]);
         if (strstr(models->model_name[i], ESP_WN_PREFIX) != NULL) {
diff --git a/main/audio_processing/esp_wake_word.cc b/main/audio_processing/esp_wake_word.cc
index 17744d6a..c5d4ed32 100644
--- a/main/audio_processing/esp_wake_word.cc
+++ b/main/audio_processing/esp_wake_word.cc
@@ -27,7 +27,10 @@ void EspWakeWord::Initialize(AudioCodec* codec) {
     codec_ = codec;
 
     wakenet_model_ = esp_srmodel_init("model");
-
+    if (wakenet_model_ == nullptr || wakenet_model_->num == -1) {
+        ESP_LOGE(TAG, "Failed to initialize wakenet model");
+        return;
+    }
     if(wakenet_model_->num > 1) {
         ESP_LOGW(TAG, "More than one model found, using the first one");
     } else if (wakenet_model_->num == 0) {