Update app.py

app.py

@@ -151,7 +151,7 @@ def save_api_settings(api_key, api_base_url, session_id):
 
 def save_pdf(file, session_id):
     if file is None: 
-        return empty_progress_html("❌ Please select a PDF first"), get_debug_info(session_id), False
+        return gr.update(visible=False), get_debug_info(session_id), False
     try:
         papers_dir, _, _ = get_user_dirs(session_id)
         for f in os.listdir(papers_dir):
@@ -160,9 +160,10 @@ 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 create_progress_html(100, f"✅ PDF Uploaded: {os.path.basename(file.name)}", "success"), get_debug_info(session_id), True
+        # 上传成功后显示进度条以反馈状态
+        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 create_progress_html(0, f"❌ Error: {str(e)}", "error"), get_debug_info(session_id), False
+        return gr.update(value=create_progress_html(0, f"❌ Error: {str(e)}", "error"), visible=True), get_debug_info(session_id), False
 
 def clear_pdf(session_id):
     try:
@@ -171,9 +172,10 @@ def clear_pdf(session_id):
             shutil.rmtree(user_base) 
             
         disable_btn = gr.update(interactive=False)
-        return empty_progress_html("🗑️ Workspace cleared"), empty_progress_html("No generation task currently..."), get_debug_info(session_id), False, disable_btn, disable_btn, disable_btn, disable_btn
+        # 清除时隐藏全部进度条
+        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 create_progress_html(0, f"❌ Clear Error: {str(e)}", "error"), gr.update(), get_debug_info(session_id), False, gr.update(), gr.update(), gr.update(), gr.update()
+        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()
 
 def build_user_env(api_key, api_base_url, papers_dir, output_dir):
     env = os.environ.copy()
@@ -193,7 +195,7 @@ def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=g
     papers_dir, output_dir, _ = get_user_dirs(session_id)
     
     if not os.path.exists(papers_dir) or not any(f.endswith('.pdf') for f in os.listdir(papers_dir)):
-        yield create_progress_html(0, "❌ No PDF file found", "error"), get_debug_info(session_id), "No execution logs.", no_change, no_change, no_change, no_change
+        yield gr.update(value=create_progress_html(0, "❌ No PDF file found", "error"), visible=True), get_debug_info(session_id), "No execution logs.", no_change, no_change, no_change, no_change
         return
 
     full_log = ""
@@ -204,7 +206,7 @@ def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=g
         
         # 10%
         progress(0.1, desc="启动 Mineru 解析...")
-        yield create_progress_html(10, "⏳ Starting Mineru parsing...", "active"), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
+        yield gr.update(value=create_progress_html(10, "⏳ Starting Mineru parsing...", "active"), visible=True), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
 
         process_mineru = subprocess.Popen(
             command_mineru, env=env, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1
@@ -214,13 +216,13 @@ 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
-            yield create_progress_html(30, "⏳ Mineru parsing PDF...", "active"), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
+            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()
 
         if returncode_mineru != 0:
             progress(1.0, desc="Mineru 解析失败")
-            yield create_progress_html(30, f"❌ Mineru failed (Code {returncode_mineru})", "error"), get_debug_info(session_id), full_log, disable_btn, disable_btn, disable_btn, disable_btn
+            yield gr.update(value=create_progress_html(30, f"❌ Mineru failed (Code {returncode_mineru})", "error"), visible=True), get_debug_info(session_id), full_log, disable_btn, disable_btn, disable_btn, disable_btn
             return
 
         command_dag = [sys.executable, "gen_dag.py"]
@@ -228,7 +230,7 @@ def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=g
         
         # 60%
         progress(0.6, desc="执行 DAG 生成...")
-        yield create_progress_html(60, "⏳ Executing DAG generation...", "active"), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
+        yield gr.update(value=create_progress_html(60, "⏳ Executing DAG generation...", "active"), visible=True), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
         
         process_dag = subprocess.Popen(
             command_dag, env=env, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1
@@ -238,28 +240,28 @@ 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
-            yield create_progress_html(80, "⏳ Building DAG...", "active"), get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
+            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()
         
         if returncode_dag == 0:
             progress(1.0, desc="解析与构建完成！")
             enable_btn = gr.update(interactive=True)
-            yield create_progress_html(100, "✅ Fully completed", "success"), get_debug_info(session_id), full_log, enable_btn, enable_btn, enable_btn, enable_btn
+            yield gr.update(value=create_progress_html(100, "✅ Fully completed", "success"), visible=True), get_debug_info(session_id), full_log, enable_btn, enable_btn, enable_btn, enable_btn
         else:
             progress(1.0, desc="DAG 生成失败")
-            yield create_progress_html(80, "❌ DAG generation failed", "error"), get_debug_info(session_id), full_log, disable_btn, disable_btn, disable_btn, disable_btn
+            yield gr.update(value=create_progress_html(80, "❌ DAG generation failed", "error"), visible=True), get_debug_info(session_id), full_log, disable_btn, disable_btn, disable_btn, disable_btn
 
     except Exception as e:
         progress(1.0, desc="发生异常")
         error_log = full_log + f"\n[Global Exception]:\n{str(e)}"
-        yield create_progress_html(0, "❌ Execution Exception", "error"), get_debug_info(session_id), error_log, disable_btn, disable_btn, disable_btn, disable_btn
+        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()):
     papers_dir, output_dir, zip_path = get_user_dirs(session_id)
     
     if not os.path.exists(output_dir):
-        yield create_progress_html(0, "❌ Please run parsing first", "error"), get_debug_info(session_id), "No output folder found.", gr.update(visible=False)
+        yield gr.update(value=create_progress_html(0, "❌ Please run parsing first", "error"), visible=True), get_debug_info(session_id), "No output folder found.", gr.update(visible=False)
         return
 
     scripts_to_run = []
@@ -270,7 +272,7 @@ def run_final_generation(task_type, session_id, api_key, api_base_url, progress=
     
     full_log = f"🚀 Starting {len(scripts_to_run)} tasks for session {session_id[:8]}...\n"
     progress(0.1, desc=f"启动 {task_type.upper()} 生成任务...")
-    yield create_progress_html(10, f"⏳ Starting {task_type.upper()}...", "active"), get_debug_info(session_id), full_log, gr.update(visible=False)
+    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)
 
     q = queue.Queue()
     processes = []
@@ -298,7 +300,7 @@ 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}"
-                yield create_progress_html(50, f"⏳ Generating {task_type.upper()}...", "active"), get_debug_info(session_id), full_log, gr.update(visible=False)
+                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)
 
@@ -306,24 +308,24 @@ def run_final_generation(task_type, session_id, api_key, api_base_url, progress=
 
         if not success: 
             progress(1.0, desc="生成失败")
-            yield create_progress_html(50, "❌ Tasks failed", "error"), get_debug_info(session_id), full_log, gr.update(visible=False)
+            yield gr.update(value=create_progress_html(50, "❌ Tasks failed", "error"), visible=True), get_debug_info(session_id), full_log, gr.update(visible=False)
             return
 
         full_log += "\n📦 Zipping output directory...\n"
         progress(0.9, desc="打包压缩结果...")
-        yield create_progress_html(90, "⏳ Zipping outputs...", "active"), get_debug_info(session_id), full_log, gr.update(visible=False)
+        yield gr.update(value=create_progress_html(90, "⏳ Zipping outputs...", "active"), visible=True), get_debug_info(session_id), full_log, gr.update(visible=False)
         
         zip_base_name = zip_path.replace(".zip", "")
         shutil.make_archive(zip_base_name, 'zip', output_dir)
         
         full_log += "✅ All tasks completed successfully.\n"
         progress(1.0, desc="全部完成！")
-        yield create_progress_html(100, f"✅ {task_type.upper()} Generated", "success"), get_debug_info(session_id), full_log, gr.update(value=zip_path, visible=True)
+        yield gr.update(value=create_progress_html(100, f"✅ {task_type.upper()} Generated", "success"), visible=True), get_debug_info(session_id), full_log, gr.update(value=zip_path, visible=True)
         
     except Exception as e:
         progress(1.0, desc="发生全局异常")
         error_log = full_log + f"\n[Global Exception]:\n{str(e)}"
-        yield create_progress_html(0, "❌ Global exception", "error"), get_debug_info(session_id), error_log, gr.update(visible=False)
+        yield gr.update(value=create_progress_html(0, "❌ Global exception", "error"), visible=True), get_debug_info(session_id), error_log, gr.update(visible=False)
 
 # ==========================================
 # --- 🚀 UI Configuration & Advanced CSS ---
@@ -468,7 +470,7 @@ body, .gradio-container {
     margin-right: auto !important; 
     margin-top: 10px !important;
     margin-bottom: 10px !important;
-    display: block !important;      
+    display: block !important;     
     transition: transform 0.3s cubic-bezier(0.175, 0.885, 0.32, 1.275), box-shadow 0.3s ease !important;
     box-shadow: 0 4px 15px rgba(126, 34, 206, 0.3) !important;
     cursor: pointer !important; 
@@ -562,8 +564,8 @@ with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
                     pdf_input = gr.File(label="Upload Document", file_types=[".pdf"], elem_id="pdf-upload-box")
                     parse_btn = gr.Button("🚀 Start Mineru & DAG Extraction", elem_classes="primary-action-btn", interactive=False)
                     
-                    # 进度条替换原本的文本框
-                    parse_progress = gr.HTML(value=empty_progress_html("Waiting for document upload..."))
+                    # 默认隐藏进度条
+                    parse_progress = gr.HTML(visible=False)
 
                 # 3. Asset Generation
                 with gr.Group(elem_classes="gradio-group"):
@@ -574,15 +576,14 @@ with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
                         gen_pr_btn = gr.Button("📰 Gen PR", elem_classes="action-btn", interactive=False)
                     gen_all_btn = gr.Button("✨ Generate All Assets (ALL)", elem_classes="primary-action-btn", interactive=False)
                     
-                    # 新增进度的展示条
-                    gen_progress = gr.HTML(value=empty_progress_html("No generation task currently..."))
+                    # 默认隐藏进度条
+                    gen_progress = gr.HTML(visible=False)
 
             with gr.Column(scale=1):
                 # 4. Results & Downloads
                 with gr.Group(elem_classes="gradio-group"):
                     gr.Markdown("### 📦 Generation Results & Download")
                     
-                    # --- 新增：无输出时的美化空状态提示词 ---
                     download_placeholder = gr.HTML(
                         '''
                         <div style="display: flex; flex-direction: column; align-items: center; justify-content: center; height: 160px; border: 2px dashed rgba(192, 132, 252, 0.6); border-radius: 16px; background-color: rgba(250, 245, 255, 0.5); color: #9333EA; text-align: center; transition: all 0.3s ease;">
@@ -657,7 +658,6 @@ with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
     refresh_btn.click(fn=get_debug_info, inputs=[session_id_state], outputs=debug_view)
 
     def toggle_empty_placeholder(file_val):
-        # 如果获取到了文件（值不为空），就隐藏占位提示词；否则保持显示
         return gr.update(visible=(file_val is None))
 
     download_file.change(