Update app.py

app.py

@@ -236,6 +236,13 @@ custom_css = """
     background-color: #1e1e1e !important; 
     color: #4AF626 !important; 
 }
+/* 可选：进一步弱化状态框的边框感，使其更像文字标签 */
+.status-text textarea {
+    background-color: transparent !important;
+    border: none !important;
+    box-shadow: none !important;
+    font-weight: bold;
+}
 """
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=custom_css) as demo:
@@ -251,39 +258,54 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=custom_css) as de
                 with gr.Row():
                     key_input = gr.Textbox(label="API Key", type="password", placeholder="sk-...", scale=1)
                     api_base_url_input = gr.Textbox(label="Base URL (可选)", placeholder="https://api.example.com", scale=1)
-                key_btn = gr.Button("💾 保存 API 配置") # 统一尺寸
+                key_btn = gr.Button("💾 保存 API 配置")
 
             # 2. 核心上传与解析
             with gr.Group():
                 gr.Markdown("### 📄 2. 文档解析")
+                # 去掉了原来的“加载文件”按钮，改为拖拽自动上传
                 pdf_input = gr.File(label="拖拽或点击上传 PDF", file_types=[".pdf"])
                 
-                with gr.Row():
-                    # 去掉 size="sm"，使其与右侧 primary 按钮等高对齐
-                    pdf_btn = gr.Button("📥 1. 加载文件") 
-                    parse_btn = gr.Button("🚀 2. 执行 Mineru & DAG 抽取", variant="primary")
+                # 让解析按钮独占一行，并放大尺寸，成为视觉焦点
+                parse_btn = gr.Button("🚀 开始执行 Mineru & DAG 抽取", variant="primary", size="lg")
                 
-                parse_status = gr.Textbox(label="解析进度", lines=1, interactive=False)
+                # 弱化状态框视觉（去掉了 Label）
+                parse_status = gr.Textbox(
+                    show_label=False, 
+                    placeholder="等待上传文档...", 
+                    lines=1, 
+                    interactive=False, 
+                    elem_classes="status-text"
+                )
 
             # 3. 最终产物生成
             with gr.Group():
                 gr.Markdown("### 🎯 3. 产物生成")
                 gr.Markdown("基于 DAG 结构生成最终所需格式：")
+                
+                # 采用 2x2 网格布局，结构更紧凑平衡
+                with gr.Row():
+                    gen_ppt_btn = gr.Button("📊 单独生成 PPT")
+                    gen_poster_btn = gr.Button("🖼️ 单独生成 Poster")
                 with gr.Row():
-                    # 去掉 size="sm"，让这三个按钮等高且填满当前行
-                    gen_ppt_btn = gr.Button("📊 PPT")
-                    gen_poster_btn = gr.Button("🖼️ Poster")
-                    gen_pr_btn = gr.Button("📰 PR 文章")
-                # 下方的 ALL 按钮也是默认尺寸，整体模块视觉统一
-                gen_all_btn = gr.Button("✨ 一键生成全部 (ALL)", variant="primary")
+                    gen_pr_btn = gr.Button("📰 单独生成 PR 文章")
+                    gen_all_btn = gr.Button("✨ 一键生成全部 (ALL)", variant="primary")
 
         # ================= 右侧：状态与日志监控区 =================
         with gr.Column(scale=5):
             with gr.Tabs():
                 # Tab 1: 结果与下载
                 with gr.Tab("📦 生成结果 & 下载"):
-                    gen_status = gr.Textbox(label="当前生成状态", lines=4, interactive=False)
-                    download_file = gr.File(label="📥 获取最终压缩包", interactive=False)
+                    # 弱化状态框视觉
+                    gen_status = gr.Textbox(
+                        show_label=False, 
+                        placeholder="当前暂无生成任务...", 
+                        lines=4, 
+                        interactive=False,
+                        elem_classes="status-text"
+                    )
+                    # 默认隐藏下载框 (visible=False)，生成成功后再动态展示
+                    download_file = gr.File(label="📥 获取最终压缩包", interactive=False, visible=False)
                 
                 # Tab 2: 实时终端
                 with gr.Tab("📜 终端流 (Terminal)"):
@@ -292,28 +314,38 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=custom_css) as de
                         placeholder="等待任务开始...", 
                         lines=22, 
                         interactive=False,
-                        elem_classes="log-box" # 应用自定义 CSS
+                        elem_classes="log-box" 
                     )
                 
                 # Tab 3: 文件系统快照
                 with gr.Tab("🔍 系统快照 (Debug)"):
-                    refresh_btn = gr.Button("🔄 刷新目录树") # 统一尺寸
+                    refresh_btn = gr.Button("🔄 刷新目录树") 
                     debug_view = gr.Textbox(label="Workspace Files", lines=20, interactive=False, value=get_debug_info())
 
     # ================= 逻辑绑定 =================
     
     key_btn.click(fn=save_api_settings, inputs=[key_input, api_base_url_input], outputs=[parse_status, debug_view])
-    pdf_btn.click(fn=save_pdf, inputs=pdf_input, outputs=[parse_status, debug_view])
+    
+    # 🎯 优化：绑定上传和清除事件，实现自动化
+    pdf_input.upload(fn=save_pdf, inputs=pdf_input, outputs=[parse_status, debug_view])
+    pdf_input.clear(fn=lambda: ("ℹ️ 已清空文件", get_debug_info()), outputs=[parse_status, debug_view])
     
     parse_btn.click(
         fn=run_mineru_parsing_and_dag_gen, 
         outputs=[parse_status, debug_view, cmd_logs]
     )
 
-    gen_ppt_btn.click(fn=lambda: run_final_generation("ppt"), outputs=[gen_status, debug_view, cmd_logs, download_file])
-    gen_poster_btn.click(fn=lambda: run_final_generation("poster"), outputs=[gen_status, debug_view, cmd_logs, download_file])
-    gen_pr_btn.click(fn=lambda: run_final_generation("pr"), outputs=[gen_status, debug_view, cmd_logs, download_file])
-    gen_all_btn.click(fn=lambda: run_final_generation("all"), outputs=[gen_status, debug_view, cmd_logs, download_file])
+    # 🎯 优化：增加包装函数，用于动态控制下载组件的显示与隐藏
+    def trigger_gen(task):
+        status, debug, logs, file_path = run_final_generation(task)
+        # 如果 file_path 存在，显示下载组件并赋值；否则保持隐藏
+        file_update = gr.update(value=file_path, visible=True) if file_path else gr.update(visible=False)
+        return status, debug, logs, file_update
+
+    gen_ppt_btn.click(fn=lambda: trigger_gen("ppt"), outputs=[gen_status, debug_view, cmd_logs, download_file])
+    gen_poster_btn.click(fn=lambda: trigger_gen("poster"), outputs=[gen_status, debug_view, cmd_logs, download_file])
+    gen_pr_btn.click(fn=lambda: trigger_gen("pr"), outputs=[gen_status, debug_view, cmd_logs, download_file])
+    gen_all_btn.click(fn=lambda: trigger_gen("all"), outputs=[gen_status, debug_view, cmd_logs, download_file])
     
     refresh_btn.click(fn=get_debug_info, outputs=debug_view)