add cai ren speaker

api.py

@@ -11,8 +11,10 @@ INFO_audio_ID_dict = {
     "matsu": os.getenv("INFO_audio_matsu"),
     "ken": os.getenv("INFO_audio_ken"),
     "mana": os.getenv("INFO_audio_mana"),
+    "cai": os.getenv("INFO_audio_cai"),
+    "ren": os.getenv("INFO_audio_ren"),
 }
-
+# print(INFO_audio_ID_dict)
 
 client = OpenAI(api_key=os.getenv("gpt"))
 #
@@ -87,6 +89,8 @@ def get_embedding(text):
 )
 def audio_text(text, audio_path, speaker_id):
     # TODO change speaker
+    # print(speaker_id, ":", INFO_audio_ID_dict[speaker_id])
+
     with open(audio_path, "wb") as f:
         for chunk in audio_client.tts(
             TTSRequest(

app.py

@@ -163,7 +163,13 @@ with gr.Blocks(title="勉強会", css=custom_css, head=custom_head, js=js_func)
         with gr.Accordion("話者紹介", open=False):
             gr.Markdown(speaker_text)
         speaker_toggle = gr.Radio(
-            [("さらとが", "matsu"), ("ケンシロウ", "ken"), ("まな", "mana")],
+            [
+                ("さらとが", "matsu"),
+                ("ケンシロウ", "ken"),
+                ("まな", "mana"),
+                ("藤井", "ren"),
+                ("内山", "cai"),
+            ],
             value="matsu",
             label="話者スタイル",
         )

blog_class.py

@@ -141,7 +141,7 @@ class knowledge_class:
         # get similar info
         question_vector = get_embedding(question_text)[0]
         info_text, info_title = self.find_top_info(question_vector, speaker_flag)
-        if speaker_flag in ["matsu"]:
+        if speaker_flag in ["matsu", "cai", "ren"]:
             user_prompt = QA_Prompt_matsu_template.format(
                 q_text=question_text,
                 r_text=info_text,
@@ -223,7 +223,7 @@ class knowledge_class:
         # get similar info
         question_vector = get_embedding(rewrite_question)[0]
         info_text, info_title = self.find_top_info(question_vector, speaker_flag)
-        if speaker_flag in ["matsu"]:
+        if speaker_flag in ["matsu", "cai", "ren"]:
             user_prompt = QA_chat_Prompt_matsu_template.format(
                 h_text=chat_history_str,
                 q_text=rewrite_question,