Upload app.py

app.py

@@ -301,12 +301,23 @@ class PodcastGenerator:
             else:
                 raise gr.Error(f"Failed to generate podcast script: {str(e)}")
 
-        # 嘗試解析JSON，如果失敗則返回原始文本
+        # 嘗試解析JSON，如果失敗則嘗試從原始文本中提取對話
         try:
             return json.loads(generated_text)
         except json.JSONDecodeError:
-            logger.warning("Generated text is not valid JSON. Returning raw text.")
-            return {"raw_text": generated_text}
+            logger.warning("Generated text is not valid JSON. Attempting to extract dialogue.")
+            lines = generated_text.split('\n')
+            podcast = []
+            current_speaker = 1
+            for line in lines:
+                line = line.strip()
+                if line:
+                    podcast.append({
+                        "speaker": current_speaker,
+                        "line": line
+                    })
+                    current_speaker = 3 - current_speaker  # Switch between 1 and 2
+            return {"podcast": podcast}
 
     async def tts_generate(self, text: str, speaker: int, speaker1: str, speaker2: str) -> str:
         """
@@ -388,11 +399,21 @@ class PodcastGenerator:
         # 生成Podcast劇本
         gr.Info("Generating podcast script...")
         start_time = time.time()
-        podcast_json = await self.generate_script(input_text, language, api_key)
+        script_result = await self.generate_script(input_text, language, api_key)
         end_time = time.time()
         
-        if "error" in podcast_json:
-            gr.Error(f"Failed to generate podcast script: {podcast_json['error']}")
+        if "error" in script_result:
+            gr.Error(f"Failed to generate podcast script: {script_result['error']}")
+            return None
+        
+        if "raw_text" in script_result:
+            gr.Warning("Generated text is not in the expected JSON format. Attempting to process raw text.")
+            # Here you might want to implement a fallback method to process raw text
+            # For now, we'll just return None
+            return None
+        
+        if "podcast" not in script_result:
+            gr.Error("Generated script does not contain a 'podcast' key.")
             return None
         
         gr.Info(f"Successfully generated podcast script in {(end_time - start_time):.2f} seconds!")
@@ -400,7 +421,7 @@ class PodcastGenerator:
         # 生成Podcast音訊檔案
         gr.Info("Generating podcast audio files...")
         start_time = time.time()
-        audio_files = await asyncio.gather(*[self.tts_generate(item['line'], item['speaker'], speaker1, speaker2) for item in podcast_json['podcast']])
+        audio_files = await asyncio.gather(*[self.tts_generate(item['line'], item['speaker'], speaker1, speaker2) for item in script_result['podcast']])
         end_time = time.time()
         gr.Info(f"Successfully generated podcast audio files in {(end_time - start_time):.2f} seconds!")