Upload app.py

app.py

@@ -180,12 +180,17 @@ async def tts_generate(input_text, speaker1, speaker2):
 
 async def process_podcast(input_text, language, speaker1, speaker2, api_key):
     podcast_script = generate_response(input_text, language, speaker1, speaker2, api_key)
+    speaker1_name = speaker1.split(' - ')[0]
+    speaker2_name = speaker2.split(' - ')[0]
 
     try:
         podcast_data = json.loads(podcast_script)
         podcast_text = ""
         for line in podcast_data.get("podcast", []):
-            speaker = f"speaker{line['speaker']}" if isinstance(line['speaker'], int) else line['speaker']
+            if isinstance(line['speaker'], int):
+                speaker = speaker1_name if line['speaker'] == 1 else speaker2_name
+            else:
+                speaker = line['speaker']
             podcast_text += f"{speaker}：{line['line']}\n"
     except json.JSONDecodeError:
         podcast_text = "Error: Unable to parse the podcast script."
@@ -253,7 +258,8 @@ body {
     margin: 0 !important;
 }
 .clear-button {
-    background-color: #FFF5CD !important;
+    color: black !important;
+    background-color: #FFCFB3 !important;
     padding: 5px !important;
     border-radius: 10px !important;
     margin: 0 !important;
@@ -281,7 +287,7 @@ body {
 with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as iface:
     gr.Markdown("# 🎙️ 自動生成 Podcast 節目及音檔。系統布署：江信宗。 🎙️", elem_classes="center-aligned")
 
-    input_text = gr.Textbox(label="請輸入 Podcast 話題（建議50~500字之間）", placeholder="輸入 Podcast 話題內容，越詳細劇本越佳......", elem_classes="input-background")
+    input_text = gr.Textbox(label="請輸入 Podcast 話題（建議50~500字之間）", placeholder="輸入 Podcast 話題內容越詳細劇本越佳 ......", elem_classes="input-background")
 
     with gr.Row():
         Language = gr.Dropdown(
@@ -331,7 +337,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as iface:
             elem_classes="sk2-background"
         )
 
-        clear_input_text_button = gr.Button("清除輸入文字", scale=1, elem_classes="clear-button")
+        clear_input_text_button = gr.Button("清除Podcast話題", scale=1, elem_classes="clear-button")
         clear_input_text_button.click(fn=lambda: "", inputs=None, outputs=input_text)
 
     with gr.Row():