3

app.py

@@ -1,42 +1,49 @@
 import gradio as gr
 from gtts import gTTS
+import json
 
-# 1. 將對話轉換為語音的函數
 def dialogue_to_audio(dialogue):
     text = " ".join([item["text"] for item in dialogue])
-    tts = gTTS(text=text, lang='en')  # 如果你的對話是中文，改成 lang='zh'
+    tts = gTTS(text=text, lang='en')
     file_path = "temp_audio.mp3"
     tts.save(file_path)
     return file_path
 
-# 2. 生成對話的函數
-def generate_dialogue(rounds: int, method: str) -> list:
+def generate_dialogue(rounds: int, method: str, role1: str, role2: str) -> list:
+    roles = [role1, role2]
     dialogue = []
     for i in range(rounds):
         if method == "auto":
-            dialogue.append({"role": "A" if i % 2 == 0 else "B", "text": f"自動文本 {i + 1}"})
+            dialogue.append({"role": roles[i % 2], "text": f"自動文本 {i + 1}"})
         else:
-            dialogue.append({"role": "A" if i % 2 == 0 else "B", "text": ""})
+            dialogue.append({"role": roles[i % 2], "text": ""})
     return dialogue
 
-# 3. 修改 Gradio 函數來返回音頻和格式化的對話文本
-def generate_dialogue_with_audio(rounds: int, method: str):
-    dialogue = generate_dialogue(rounds, method)
+def generate_dialogue_with_audio(rounds: int, method: str, role1: str, role2: str):
+    dialogue = generate_dialogue(rounds, method, role1, role2)
     audio_path = dialogue_to_audio(dialogue)
     formatted_text = "\n".join([f"{item['role']}: {item['text']}" for item in dialogue])
-    return audio_path, formatted_text
+    json_output = json.dumps(dialogue, ensure_ascii=False, indent=4)
+    
+    # 儲存對話為 JSON 文件
+    with open("dialogue_output.json", "w", encoding="utf-8") as f:
+        f.write(json_output)
+
+    return audio_path, formatted_text, "dialogue_output.json"
 
-# 4. 創建 Gradio 界面並啟動
 def main_interface():
     interface = gr.Interface(
         fn=generate_dialogue_with_audio,
         inputs=[
             gr.components.Slider(minimum=2, maximum=6, step=2, default=2, label="對話輪數"),
             gr.components.Dropdown(choices=["auto", "manual"], label="生成方式"),
+            gr.components.Textbox(default="A", label="角色 1 名稱"),
+            gr.components.Textbox(default="B", label="角色 2 名稱"),
         ],
         outputs=[
             gr.components.Audio(type="filepath", label="對話朗讀"),
-            gr.components.Textbox(label="生成的對話")
+            gr.components.Textbox(label="生成的對話"),
+            gr.components.File(label="下載對話 JSON 文件")
         ]
     )
     interface.launch()