Upload app.py

app.py

@@ -37,7 +37,7 @@ class PodcastGenerator:
             gr.Error: 如果 API 金鑰或速率限制出現問題。
 
         此方法使用 SambaNova API 根據使用者的輸入生成Podcast劇本。
-        它處理語言選擇，使用適當的配置設定 AI 模型，並處理生成的響應。
+        它處理語言選擇，使用適當配置設定 AI 模型，並處理生成的響應。
         """
         # 定義一個示例JSON結構，用於指導AI生成類似格式的Podcast劇本
         example = """
@@ -268,6 +268,14 @@ class PodcastGenerator:
 
         # 嘗試生成內容
         try:
+            # Calculate the available tokens for generation
+            prompt_tokens = len(user_prompt.split())  # This is a rough estimate
+            system_tokens = len(system_prompt.split())  # This is a rough estimate
+            max_tokens = 4096 - prompt_tokens - system_tokens - 100  # 100 is a safety margin
+
+            if max_tokens <= 0:
+                return {"error": "Input prompt is too long. Please provide a shorter prompt."}
+
             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',
@@ -276,7 +284,7 @@ class PodcastGenerator:
                     {"role": "user", "content": user_prompt}
                 ],
                 temperature=1,
-                max_tokens=1975  # Reduced from 8192 to stay within the 4096 token limit
+                max_tokens=max_tokens
             )
             logger.info(f"Received response from API: {response}")
             
@@ -512,6 +520,12 @@ async def process_input(input_text: str, input_file, language: str, speaker1: st
             gr.Error(f"Selected voices may not be compatible with the chosen language: {language}")
             return None
 
+    # Check input text length
+    max_input_length = 1000  # Adjust this value as needed
+    if len(input_text) > max_input_length:
+        gr.Error(f"Input text is too long. Please limit your input to {max_input_length} characters.")
+        return None
+
     # 如果提供了輸入檔案，則從檔案中提取文字
     if input_file:
         input_text = await TextExtractor.extract_text(input_file.name)