From ba004950c5d57656894418de0b8d8a13cb301d24 Mon Sep 17 00:00:00 2001
From: samiabat <samuelabatneh20@gmail.com>
Date: Wed, 16 Apr 2025 17:44:12 +0300
Subject: [PATCH] add the folder structure

---
 api.py                                              |   6 ++++--
 {saotome => idols/saotome}/refs/ref1.wav            | Bin
 {saotome => idols/saotome}/refs/ref2.wav            | Bin
 {saotome => idols/saotome}/refs/ref3.wav            | Bin
 {saotome => idols/saotome}/refs/ref4.wav            | Bin
 .../saotome-6s.wav => idols/saotome/refs/ref5.wav   | Bin
 .../saotome-10s.wav => idols/saotome/saotome.wav    | Bin
 7 files changed, 4 insertions(+), 2 deletions(-)
 rename {saotome => idols/saotome}/refs/ref1.wav (100%)
 rename {saotome => idols/saotome}/refs/ref2.wav (100%)
 rename {saotome => idols/saotome}/refs/ref3.wav (100%)
 rename {saotome => idols/saotome}/refs/ref4.wav (100%)
 rename saotome/saotome-6s.wav => idols/saotome/refs/ref5.wav (100%)
 rename saotome/saotome-10s.wav => idols/saotome/saotome.wav (100%)

diff --git a/api.py b/api.py
index fe404c67..08ee67e8 100644
--- a/api.py
+++ b/api.py
@@ -1089,7 +1089,6 @@ async def tts_endpoint(request: Request):
 
 @app.get("/")
 async def tts_endpoint(
-        refer_wav_path: str = "saotome/saotome-6s.wav",
         prompt_text: str = "今日は友達と一緒に映画を見に行く予定ですが、天気が悪くて少し心配です。",
         prompt_language: str = "all_ja",
         character: str = "saotome",
@@ -1100,10 +1099,13 @@ async def tts_endpoint(
         top_p: float = 1.0,
         temperature: float = 1.0,
         speed: float = 1.0,
-        inp_refs: list = Query(default=["saotome/saotome-10s.wav", "saotome/refs/ref1.wav", "saotome/refs/ref2.wav", "saotome/refs/ref3.wav", "saotome/refs/ref4.wav"]),
         sample_steps: int = 32,
         if_sr: bool = False
 ):
+    refer_wav_path = f"idols/{character}.wav"
+    inp_refs = [f"idols/{character}/refs/{file}" for file in os.listdir(f"idols/{character}/refs") if file.endswith('.wav')]
+
+    
     print(f"the base path is {refer_wav_path}")
     return handle(refer_wav_path, prompt_text, prompt_language, text, text_language, cut_punc, top_k, top_p, temperature, speed, inp_refs, sample_steps, if_sr)
 
diff --git a/saotome/refs/ref1.wav b/idols/saotome/refs/ref1.wav
similarity index 100%
rename from saotome/refs/ref1.wav
rename to idols/saotome/refs/ref1.wav
diff --git a/saotome/refs/ref2.wav b/idols/saotome/refs/ref2.wav
similarity index 100%
rename from saotome/refs/ref2.wav
rename to idols/saotome/refs/ref2.wav
diff --git a/saotome/refs/ref3.wav b/idols/saotome/refs/ref3.wav
similarity index 100%
rename from saotome/refs/ref3.wav
rename to idols/saotome/refs/ref3.wav
diff --git a/saotome/refs/ref4.wav b/idols/saotome/refs/ref4.wav
similarity index 100%
rename from saotome/refs/ref4.wav
rename to idols/saotome/refs/ref4.wav
diff --git a/saotome/saotome-6s.wav b/idols/saotome/refs/ref5.wav
similarity index 100%
rename from saotome/saotome-6s.wav
rename to idols/saotome/refs/ref5.wav
diff --git a/saotome/saotome-10s.wav b/idols/saotome/saotome.wav
similarity index 100%
rename from saotome/saotome-10s.wav
rename to idols/saotome/saotome.wav