Update app.py

app.py

@@ -160,7 +160,6 @@ def save_pdf(file, session_id):
                 
         file_path = os.path.join(papers_dir, os.path.basename(file.name))
         shutil.copy(file.name, file_path)
-        # 上传成功后显示进度条以反馈状态
         return gr.update(value=create_progress_html(100, f"✅ PDF Uploaded: {os.path.basename(file.name)}", "success"), visible=True), get_debug_info(session_id), True
     except Exception as e:
         return gr.update(value=create_progress_html(0, f"❌ Error: {str(e)}", "error"), visible=True), get_debug_info(session_id), False
@@ -172,7 +171,6 @@ def clear_pdf(session_id):
             shutil.rmtree(user_base) 
             
         disable_btn = gr.update(interactive=False)
-        # 清除时隐藏全部进度条
         return gr.update(visible=False), gr.update(visible=False), get_debug_info(session_id), False, disable_btn, disable_btn, disable_btn, disable_btn
     except Exception as e:
         return gr.update(value=create_progress_html(0, f"❌ Clear Error: {str(e)}", "error"), visible=True), gr.update(), get_debug_info(session_id), False, gr.update(), gr.update(), gr.update(), gr.update()
@@ -216,6 +214,10 @@ def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=g
         progress(0.3, desc="Mineru 正在解析 PDF...")
         for line in iter(process_mineru.stdout.readline, ''):
             full_log += line
+            # ==========================================
+            # 🔔 双端输出 1：Mineru 阶段
+            # ==========================================
+            print(f"[Mineru | {session_id[:6]}] {line}", end="", flush=True)
             yield gr.update(value=create_progress_html(30, "⏳ Mineru parsing PDF...", "active"), visible=True), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
         process_mineru.stdout.close()
         returncode_mineru = process_mineru.wait()
@@ -240,6 +242,10 @@ def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=g
         progress(0.8, desc="构建图结构中...")
         for line in iter(process_dag.stdout.readline, ''):
             full_log += line
+            # ==========================================
+            # 🔔 双端输出 2：DAG 生成阶段
+            # ==========================================
+            print(f"[DAG | {session_id[:6]}] {line}", end="", flush=True)
             yield gr.update(value=create_progress_html(80, "⏳ Building DAG...", "active"), visible=True), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
         process_dag.stdout.close()
         returncode_dag = process_dag.wait()
@@ -255,6 +261,7 @@ def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=g
     except Exception as e:
         progress(1.0, desc="发生异常")
         error_log = full_log + f"\n[Global Exception]:\n{str(e)}"
+        print(f"[Exception | {session_id[:6]}] {str(e)}", flush=True)
         yield gr.update(value=create_progress_html(0, "❌ Execution Exception", "error"), visible=True), get_debug_info(session_id), error_log, disable_btn, disable_btn, disable_btn, disable_btn
 
 def run_final_generation(task_type, session_id, api_key, api_base_url, progress=gr.Progress()):
@@ -271,6 +278,8 @@ def run_final_generation(task_type, session_id, api_key, api_base_url, progress=
     elif task_type == "all": scripts_to_run = ["gen_ppt.py", "gen_poster.py", "gen_pr.py"]
     
     full_log = f"🚀 Starting {len(scripts_to_run)} tasks for session {session_id[:8]}...\n"
+    print(f"[GEN Start | {session_id[:6]}] Starting {task_type.upper()}", flush=True)
+    
     progress(0.1, desc=f"启动 {task_type.upper()} 生成任务...")
     yield gr.update(value=create_progress_html(10, f"⏳ Starting {task_type.upper()}...", "active"), visible=True), get_debug_info(session_id), full_log, gr.update(visible=False)
 
@@ -300,6 +309,10 @@ def run_final_generation(task_type, session_id, api_key, api_base_url, progress=
             try:
                 script_name, line = q.get(timeout=0.1)
                 full_log += f"[{script_name}] {line}"
+                # ==========================================
+                # 🔔 双端输出 3：生成阶段
+                # ==========================================
+                print(f"[{script_name.upper()} | {session_id[:6]}] {line}", end="", flush=True)
                 yield gr.update(value=create_progress_html(50, f"⏳ Generating {task_type.upper()}...", "active"), visible=True), get_debug_info(session_id), full_log, gr.update(visible=False)
             except queue.Empty:
                 active_processes = sum(1 for _, p in processes if p.poll() is None)
@@ -325,6 +338,7 @@ def run_final_generation(task_type, session_id, api_key, api_base_url, progress=
     except Exception as e:
         progress(1.0, desc="发生全局异常")
         error_log = full_log + f"\n[Global Exception]:\n{str(e)}"
+        print(f"[Exception | {session_id[:6]}] {str(e)}", flush=True)
         yield gr.update(value=create_progress_html(0, "❌ Global exception", "error"), visible=True), get_debug_info(session_id), error_log, gr.update(visible=False)
 
 # ==========================================
@@ -667,4 +681,5 @@ with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
     )
 
 if __name__ == "__main__":
+    # 并发放宽至 20
     demo.queue(default_concurrency_limit=20).launch()