Upload app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 from pydub import AudioSegment
-import google.generativeai as genai
-from google.generativeai.types import HarmCategory, HarmBlockThreshold
 import json
 import uuid
 import io
@@ -13,6 +11,11 @@ import os
 import time
 from typing import List, Dict, Tuple
 import openai
+import logging
+
+# At the beginning of your script, set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 class PodcastGenerator:
     def __init__(self):
@@ -253,12 +256,10 @@ class PodcastGenerator:
         user_prompt = f"Please generate a podcast script based on the following user input:\n{prompt}"
 
         # 配置 SambaNova API client
-        if api_key:
-            openai.api_key = api_key
-        else:
-            openai.api_key = os.getenv("YOUR_API_TOKEN")
+        if not api_key:
+            api_key = os.getenv("YOUR_API_TOKEN")
         client = openai.OpenAI(
-            api_key=openai.api_key,
+            api_key=api_key,
             base_url="https://api.sambanova.ai/v1",
         )
 
@@ -270,24 +271,36 @@ class PodcastGenerator:
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": user_prompt}
                 ],
-                temperature=0.7,
+                temperature=1,
                 max_tokens=8192
             )
-            generated_text = response.choices[0].message.content
+            logger.info(f"API Response: {response}")
+            
+            if response.choices and len(response.choices) > 0:
+                generated_text = response.choices[0].message.content
+            else:
+                logger.warning("No content generated from the API")
+                raise ValueError("No content generated from the API")
+
         except Exception as e:
+            logger.error(f"Error generating script: {str(e)}")
             # 處理可能的錯誤
             if "API key not valid" in str(e):
                 raise gr.Error("Invalid API key. Please provide a valid SambaNova API key.")
             elif "rate limit" in str(e).lower():
                 raise gr.Error("Rate limit exceeded for the API key. Please try again later or provide your own SambaNova API key.")
             else:
-                raise gr.Error(f"Failed to generate podcast script: {e}")
+                raise gr.Error(f"Failed to generate podcast script: {str(e)}")
 
         # 列印生成的Podcast指令碼
         print(f"Generated podcast script:\n{generated_text}")
 
-        # 返回解析後的JSON資料
-        return json.loads(generated_text)
+        # 嘗試解析JSON，如果失敗則返回原始文本
+        try:
+            return json.loads(generated_text)
+        except json.JSONDecodeError:
+            print("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:
         """