Update app.py

app.py

@@ -33,7 +33,7 @@ def get_extraction_config():
     ]
     return prompt_description, examples
 
-# --- 2. Gradio 的核心处理函数 (已增加自动重试功能) ---
+# --- 2. Gradio 的核心处理函数 (最终修复版) ---
 def process_input_and_visualize(input_text, input_file):
     source_text = ""
     if input_file is not None:
@@ -53,9 +53,8 @@ def process_input_and_visualize(input_text, input_file):
     if not api_key:
         raise gr.Error("错误：服务器未配置 LANGEXTRACT_API_KEY。")
 
-    # --- 新增：自动重试逻辑 ---
     max_retries = 3
-    retry_delay = 5  # 秒
+    retry_delay = 5
     result = None
 
     for attempt in range(max_retries):
@@ -67,25 +66,24 @@ def process_input_and_visualize(input_text, input_file):
                 examples=examples,
                 model_id="gemini-2.5-flash",
                 api_key=api_key,
-                max_workers=2,
+                # --- 最终关键修复 ---
+                # 将并行工作线程数设为 1，强制顺序处理，以避免超出免费套餐的速率限制。
+                max_workers=1,
                 extraction_passes=2,
                 max_char_buffer=1500
             )
-            # 如果成功，跳出循环
             print("API call successful.")
             break
         except Exception as e:
-            # 捕获所有 langextract 相关的运行时错误
             print(f"Attempt #{attempt + 1} failed with error: {e}")
-            if "503" in str(e) and attempt < max_retries - 1:
-                # 如果是 503 错误，并且还有重试机会，则等待后重试
-                print(f"Server overloaded (503). Retrying in {retry_delay} seconds...")
+            # 同时处理 503 (服务器超载) 和 429 (速率限制) 错误
+            if ("503" in str(e) or "429" in str(e)) and attempt < max_retries - 1:
+                error_type = "503" if "503" in str(e) else "429"
+                print(f"API Error ({error_type}). Retrying in {retry_delay} seconds...")
                 time.sleep(retry_delay)
             else:
-                # 如果是其他错误，或已达到最大重试次数，则直接抛出异常
                 raise gr.Error(f"信息提取过程中发生错误: {e}")
     
-    # 如果所有重试都失败了，result 仍然会是 None，这里可以加一个检查
     if result is None:
         raise gr.Error("所有重试均失败，无法从API获取结果。请稍后再试。")