From 43920fda4c250e3f327e1267b2905b966bfa5dcb Mon Sep 17 00:00:00 2001
From: w-okada <you@example.com>
Date: Sat, 26 Aug 2023 16:01:30 +0900
Subject: [PATCH] handle pipeline initialization

---
 server/voice_changer/VoiceChangerV2.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/server/voice_changer/VoiceChangerV2.py b/server/voice_changer/VoiceChangerV2.py
index 721d9089..c6466b8b 100644
--- a/server/voice_changer/VoiceChangerV2.py
+++ b/server/voice_changer/VoiceChangerV2.py
@@ -32,6 +32,7 @@ from Exceptions import (
     NoModeLoadedException,
     NotEnoughDataExtimateF0,
     ONNXInputArgumentException,
+    PipelineNotInitializedException,
     VoiceChangerIsNotSelectedException,
 )
 from voice_changer.utils.VoiceChangerParams import VoiceChangerParams
@@ -316,6 +317,9 @@ class VoiceChangerV2(VoiceChangerIF):
         except DeviceCannotSupportHalfPrecisionException:
             # RVC.pyでfallback処理をするので、ここはダミーデータ返すだけ。
             return np.zeros(1).astype(np.int16), [0, 0, 0]
+        except PipelineNotInitializedException:
+            logger.warn("[Voice Changer] Waiting generate pipeline...")
+            return np.zeros(1024).astype(np.int16), [0, 0, 0]
         except Exception as e:
             logger.warn(f"[Voice Changer] VC PROCESSING EXCEPTION!!! {e}")
             logger.exception(e)