Upload app.py

app.py

@@ -256,15 +256,19 @@ class PodcastGenerator:
         user_prompt = f"Please generate a podcast script based on the following user input:\n{prompt}"
 
         # 配置 SambaNova API client
-        if not api_key:
-            api_key = os.getenv("YOUR_API_TOKEN")
+        if api_key:
+            openai.api_key = api_key
+        else:
+            openai.api_key = os.getenv("YOUR_API_TOKEN")
+        
         client = openai.OpenAI(
-            api_key=api_key,
+            api_key=openai.api_key,
             base_url="https://api.sambanova.ai/v1",
         )
 
         # 嘗試生成內容
         try:
+            logger.info(f"Sending request to SambaNova API with prompt: {user_prompt[:100]}...")
             response = client.chat.completions.create(
                 model='Meta-Llama-3.1-405B-Instruct',
                 messages=[
@@ -274,13 +278,14 @@ class PodcastGenerator:
                 temperature=1,
                 max_tokens=8192
             )
-            logger.info(f"API Response: {response}")
+            logger.info(f"Received response from API: {response}")
             
             if response.choices and len(response.choices) > 0:
                 generated_text = response.choices[0].message.content
+                logger.info(f"Generated text: {generated_text[:100]}...")
             else:
                 logger.warning("No content generated from the API")
-                raise ValueError("No content generated from the API")
+                return {"error": "No content generated from the API"}
 
         except Exception as e:
             logger.error(f"Error generating script: {str(e)}")
@@ -292,14 +297,11 @@ class PodcastGenerator:
             else:
                 raise gr.Error(f"Failed to generate podcast script: {str(e)}")
 
-        # 列印生成的Podcast指令碼
-        print(f"Generated podcast script:\n{generated_text}")
-
         # 嘗試解析JSON，如果失敗則返回原始文本
         try:
             return json.loads(generated_text)
         except json.JSONDecodeError:
-            print("Warning: Generated text is not valid JSON. Returning raw text.")
+            logger.warning("Generated text is not valid JSON. Returning raw text.")
             return {"raw_text": generated_text}
 
     async def tts_generate(self, text: str, speaker: int, speaker1: str, speaker2: str) -> str:
@@ -384,6 +386,11 @@ class PodcastGenerator:
         start_time = time.time()
         podcast_json = 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']}")
+            return None
+        
         gr.Info(f"Successfully generated podcast script in {(end_time - start_time):.2f} seconds!")
 
         # 生成Podcast音訊檔案
@@ -472,6 +479,9 @@ async def process_input(input_text: str, input_file, language: str, speaker1: st
     # 建立PodcastGenerator實例並生成Podcast
     podcast_generator = PodcastGenerator()
     podcast = await podcast_generator.generate_podcast(input_text, language, speaker1, speaker2, api_key)
+    
+    if podcast is None:
+        return None
 
     # 計算總耗時並顯示資訊
     end_time = time.time()