Update app.py

app.py

@@ -71,10 +71,12 @@ def save_api_settings(api_key, api_base_url=None):
 
 
 def run_mineru_parsing_and_dag_gen():
-    """执行 PDF 解析并捕获完整日志，随后执行DAG生成流程"""
+    """执行 PDF 解析并捕获完整日志，随后执行DAG生成流程（支持实时流式输出）"""
     if not os.path.exists(PAPERS_DIR) or not any(f.endswith('.pdf') for f in os.listdir(PAPERS_DIR)):
-        return "❌ 未发现 PDF 文件", get_debug_info(), "No execution logs."
+        yield "❌ 未发现 PDF 文件", get_debug_info(), "No execution logs."
+        return
 
+    full_log = ""
     try:
         # ================= 第一步：执行 Mineru 解析 =================
         env = os.environ.copy()
@@ -85,49 +87,64 @@ def run_mineru_parsing_and_dag_gen():
 
         command_mineru = ["mineru", "-p", PAPERS_DIR, "-o", OUTPUT_DIR]
         
-        result_mineru = subprocess.run(
+        full_log += "--- Mineru 执行中 ---\n"
+        yield "⏳ 正在执行 Mineru 解析...", get_debug_info(), full_log
+
+        # 1. 使用 Popen 替代 run，开启实时流
+        process_mineru = subprocess.Popen(
             command_mineru, 
             env=env, 
-            capture_output=True, 
+            stdout=subprocess.PIPE, 
+            stderr=subprocess.STDOUT, # 将 stderr 错误流合并到 stdout 一起输出
             text=True,
-            timeout=300 
+            bufsize=1 # 开启行缓冲
         )
 
-        # 记录 Mineru 的日志
-        full_log = f"--- Mineru STDOUT ---\n{result_mineru.stdout}\n\n--- Mineru STDERR ---\n{result_mineru.stderr}\n"
-        
-        # 如果解析失败，直接返回，不执行后续 DAG 生成
-        if result_mineru.returncode != 0:
-            status = f"❌ Mineru 解析失败 (Exit Code: {result_mineru.returncode})"
-            return status, get_debug_info(), full_log
+        # 2. 逐行读取输出并实时 yield 给 Gradio 界面
+        for line in iter(process_mineru.stdout.readline, ''):
+            full_log += line
+            yield "⏳ 正在执行 Mineru 解析...", get_debug_info(), full_log
+
+        process_mineru.stdout.close()
+        returncode_mineru = process_mineru.wait()
+
+        # 如果解析失败，直接 yield 返回
+        if returncode_mineru != 0:
+            status = f"❌ Mineru 解析失败 (Exit Code: {returncode_mineru})"
+            yield status, get_debug_info(), full_log
+            return
 
         # ================= 第二步：执行 DAG 生成 =================
-        # 使用 sys.executable 确保使用当前虚拟环境的 Python 解释器
         command_dag = [sys.executable, "gen_dag.py"]
         
-        result_dag = subprocess.run(
+        full_log += "\n--- DAG Gen 执行中 ---\n"
+        yield "⏳ Mineru 解析完成，正在执行 DAG 生成...", get_debug_info(), full_log
+        
+        process_dag = subprocess.Popen(
             command_dag, 
-            capture_output=True, 
+            stdout=subprocess.PIPE, 
+            stderr=subprocess.STDOUT, 
             text=True,
-            timeout=300  # 可以根据你 DAG 生成的时间需求调整超时
+            bufsize=1
         )
 
-        # 追加 DAG 生成的日志
-        full_log += f"\n--- DAG Gen STDOUT ---\n{result_dag.stdout}\n\n--- DAG Gen STDERR ---\n{result_dag.stderr}"
+        for line in iter(process_dag.stdout.readline, ''):
+            full_log += line
+            yield "⏳ 正在执行 DAG 生成...", get_debug_info(), full_log
+
+        process_dag.stdout.close()
+        returncode_dag = process_dag.wait()
         
-        if result_dag.returncode == 0:
+        if returncode_dag == 0:
             status = "✅ PDF解析与DAG生成全部完成"
         else:
-            status = f"❌ DAG生成失败 (Exit Code: {result_dag.returncode})"
+            status = f"❌ DAG生成失败 (Exit Code: {returncode_dag})"
             
-        return status, get_debug_info(), full_log
+        yield status, get_debug_info(), full_log
 
-    except subprocess.TimeoutExpired as e:
-        error_log = f"Execution timed out:\n{str(e)}"
-        return "❌ 运行超时", get_debug_info(), error_log
     except Exception as e:
-        error_log = f"Exception occurred during execution:\n{str(e)}"
-        return "❌ 运行异常", get_debug_info(), error_log
+        error_log = full_log + f"\n[全局异常] Exception occurred:\n{str(e)}"
+        yield "❌ 运行异常", get_debug_info(), error_log
 
 def run_final_generation(task_type="all"):
     """