Max attempts

app.py

@@ -5,10 +5,14 @@ import json
 import shutil
 from pathlib import Path
 import base64
-
+import logging
 from main import generate_paper_poster
 
 
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+
 def process_paper_to_poster(
     pdf_file, 
     model_choice, 
@@ -64,6 +68,11 @@ def process_paper_to_poster(
             output=""  # 不再使用
         )
         
+        # 检查返回值是否为None
+        if poster is None and html is None:
+            yield None, None, None, "❌ Failed to generate poster! The paper processing returned no results. Please check:\n- PDF file format and content\n- Figure detection service availability\n- API key validity\n- Model configuration\n- Network connectivity"
+            return
+        
         # 图片处理完毕，开始生成JSON
         yield None, None, None, "🖼️ Image processing completed! Generating JSON structure..."
         
@@ -268,7 +277,7 @@ def create_interface():
         - Generated Poster is recommended to be viewed in fullscreen mode or download the HTML file to view in browser
         - Requires a valid OpenAI API key with sufficient balance
         - Figure detection service URL is required for extracting images from PDFs
-        - Processing time depends on paper length and complexity (usually 1-3 minutes)
+        - Processing time depends on paper length and complexity (usually 3-6 minutes)
         - Ensure the model name is correct and supported by your API
         - Download the HTML file and open it in your browser for the best viewing experience
         

main.py

@@ -74,6 +74,9 @@ def generate_paper_poster(
         # with open(figures_cap_cache, "w") as f:
         #     json.dump(figures_cap, f, ensure_ascii=False)
 
+    print("开始生成海报...")
+    max_attempts = 3
+    attempt = 0
     while True:
         try:
             result = generate_poster_v3(
@@ -114,6 +117,10 @@ def generate_paper_poster(
             ):
                 raise
             print(f"处理文件 {pdf} 时出错: {e}")
+            attempt += 1
+            if attempt > max_attempts:
+                return None, None
+
 
 
 if __name__ == "__main__":