Upload app.py

app.py

@@ -5,6 +5,8 @@ import gradio as gr
 import openai
 import os
 import re
+import json
+
 def create_client(api_key=None):
     if api_key:
         openai.api_key = api_key
@@ -76,7 +78,29 @@ Follow this example structure, MUST be in {language} language:
     )
     try:
         podcast_match = re.search(r'{(.*?)}', response.choices[0].message.content, re.DOTALL)
-        print(podcast_match)
+        if podcast_match:
+            podcast_content = podcast_match.group(0)
+            podcast_dict = json.loads(podcast_content)
+            
+            # 創建新的 JSON 格式
+            podcast_json = {
+                "topic": input_text[:30] + "..." if len(input_text) > 30 else input_text,
+                "podcast": []
+            }
+            
+            speaker_map = {"speaker1": 1, "speaker2": 2}
+            
+            for speaker, line in podcast_dict.items():
+                podcast_json["podcast"].append({
+                    "speaker": speaker_map[speaker],
+                    "line": line
+                })
+            
+            return json.dumps(podcast_json, ensure_ascii=False, indent=2)
+        else:
+            raise gr.Error("Failed to generate podcast script. Please try again.")
+    except json.JSONDecodeError:
+        raise gr.Error("Invalid JSON format in the generated response. Please try again.")
     except Exception as e:
         if "API key not valid" in str(e):
             raise gr.Error("Invalid API key. Please provide a valid API key.")
@@ -84,12 +108,11 @@ Follow this example structure, MUST be in {language} language:
             raise gr.Error("Rate limit exceeded for the API key. Please try again later or provide your own API key.")
         else:
             raise gr.Error(f"Failed to generate podcast script: {e}")
-    return response.choices[0].message.content
 
 with gr.Blocks() as iface:
     gr.Markdown("# 🎙️ Generated Podcast Audio. Deployed by 江信宗")
     
-    input_text = gr.Textbox(label="請輸入探究的話題（建議50~500字之間）")
+    input_text = gr.Textbox(label="請輸入 Podcast 話題（建議50~500字之間）")
     
     with gr.Row():
         Language = gr.Dropdown(