Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import os
-import yaml
 import shutil
 import subprocess
 import sys
 import queue
 import threading
+import uuid
 from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from typing import Iterable
@@ -15,61 +15,43 @@ from gradio.themes.utils import colors, fonts, sizes
 # 自动安装 Playwright 浏览器（如果不存在）
 try:
     import playwright
-    # 检查是否已经安装了浏览器，没有则下载
     subprocess.run(["playwright", "install", "chromium"], check=True)
 except Exception as e:
     print(f"Playwright setup failed: {e}")
 
-# Initialize environment paths
+# ==========================================
+# --- 📁 全局目录配置 (修改为 Session 基础目录) ---
+# ==========================================
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-PAPERS_DIR = os.path.join(BASE_DIR, "papers")
-CONFIG_PATH = os.path.join(BASE_DIR, "config.yaml")
-OUTPUT_DIR = os.path.join(BASE_DIR, "mineru_outputs")
-ZIP_OUTPUT_PATH = os.path.join(BASE_DIR, "mineru_results.zip")
-
-os.makedirs(PAPERS_DIR, exist_ok=True)
-os.makedirs(OUTPUT_DIR, exist_ok=True)
+SESSIONS_BASE_DIR = os.path.join(BASE_DIR, "user_sessions")
+os.makedirs(SESSIONS_BASE_DIR, exist_ok=True)
+
+def get_user_dirs(session_id):
+    """根据 Session ID 生成用户专属的隔离目录"""
+    user_base = os.path.join(SESSIONS_BASE_DIR, session_id)
+    papers_dir = os.path.join(user_base, "papers")
+    output_dir = os.path.join(user_base, "mineru_outputs")
+    zip_path = os.path.join(user_base, "mineru_results.zip")
+    
+    os.makedirs(papers_dir, exist_ok=True)
+    os.makedirs(output_dir, exist_ok=True)
+    return papers_dir, output_dir, zip_path
 
 # ==========================================
 # --- 🎨 Custom Purple Theme Definition ---
 # ==========================================
 colors.purple = colors.Color(
-    name="purple",
-    c50="#FAF5FF",
-    c100="#F3E8FF",
-    c200="#E9D5FF",
-    c300="#DAB2FF",
-    c400="#C084FC",
-    c500="#A855F7",
-    c600="#9333EA",
-    c700="#7E22CE",
-    c800="#6B21A8",
-    c900="#581C87",
-    c950="#3B0764",
+    name="purple", c50="#FAF5FF", c100="#F3E8FF", c200="#E9D5FF",
+    c300="#DAB2FF", c400="#C084FC", c500="#A855F7", c600="#9333EA",
+    c700="#7E22CE", c800="#6B21A8", c900="#581C87", c950="#3B0764",
 )
 
 class PurpleTheme(Soft):
-    def __init__(
-        self,
-        *,
-        primary_hue: colors.Color | str = colors.gray,
-        secondary_hue: colors.Color | str = colors.purple,
-        neutral_hue: colors.Color | str = colors.slate,
-        text_size: sizes.Size | str = sizes.text_lg,
-        font: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("Outfit"), "Arial", "sans-serif",
-        ),
-        font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace",
-        ),
-    ):
+    def __init__(self, **kwargs):
         super().__init__(
-            primary_hue=primary_hue,
-            secondary_hue=secondary_hue,
-            neutral_hue=neutral_hue,
-            text_size=text_size,
-            font=font,
-            font_mono=font_mono,
+            primary_hue=colors.gray, secondary_hue=colors.purple, neutral_hue=colors.slate,
+            font=(fonts.GoogleFont("Outfit"), "Arial", "sans-serif"),
+            font_mono=(fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace"),
         )
         super().set(
             background_fill_primary="*primary_50",
@@ -77,26 +59,14 @@ class PurpleTheme(Soft):
             body_background_fill="linear-gradient(135deg, *primary_200, *primary_100)",
             body_background_fill_dark="linear-gradient(135deg, *primary_900, *primary_800)",
             button_primary_text_color="white",
-            button_primary_text_color_hover="white",
             button_primary_background_fill="linear-gradient(90deg, *secondary_500, *secondary_600)",
             button_primary_background_fill_hover="linear-gradient(90deg, *secondary_600, *secondary_700)",
-            button_primary_background_fill_dark="linear-gradient(90deg, *secondary_600, *secondary_700)",
-            button_primary_background_fill_hover_dark="linear-gradient(90deg, *secondary_500, *secondary_600)",
             button_secondary_text_color="black",
-            button_secondary_text_color_hover="white",
             button_secondary_background_fill="linear-gradient(90deg, *primary_300, *primary_300)",
-            button_secondary_background_fill_hover="linear-gradient(90deg, *primary_400, *primary_400)",
-            button_secondary_background_fill_dark="linear-gradient(90deg, *primary_500, *primary_600)",
-            button_secondary_background_fill_hover_dark="linear-gradient(90deg, *primary_500, *primary_500)",
             slider_color="*secondary_500",
-            slider_color_dark="*secondary_600",
-            block_title_text_weight="600",
             block_border_width="3px",
             block_shadow="*shadow_drop_lg",
             button_primary_shadow="*shadow_drop_lg",
-            button_large_padding="11px",
-            color_accent_soft="*primary_100",
-            block_label_background_fill="*primary_200",
         )
 
 purple_theme = PurpleTheme()
@@ -106,19 +76,15 @@ purple_theme = PurpleTheme()
 # ==========================================
 
 def get_tree_html(dir_path):
-    """递归生成可折叠的 HTML 目录树"""
     if not os.path.exists(dir_path):
         return "<div style='margin-left: 15px; color: #888;'><i>Directory missing</i></div>"
         
     def build_html(current_path):
         html = ""
-        try:
-            items = sorted(os.listdir(current_path))
-        except Exception:
-            return ""
+        try: items = sorted(os.listdir(current_path))
+        except Exception: return ""
         
-        if not items:
-            return ""
+        if not items: return ""
             
         for item in items:
             item_path = os.path.join(current_path, item)
@@ -134,13 +100,14 @@ def get_tree_html(dir_path):
     content = build_html(dir_path)
     return content if content else "<div style='margin-left: 15px; color: #888;'><i>Empty directory</i></div>"
 
-def get_debug_info():
-    """返回渲染好的目录树 HTML 容器"""
-    papers_tree = get_tree_html(PAPERS_DIR)
-    output_tree = get_tree_html(OUTPUT_DIR)
+def get_debug_info(session_id):
+    papers_dir, output_dir, _ = get_user_dirs(session_id)
+    papers_tree = get_tree_html(papers_dir)
+    output_tree = get_tree_html(output_dir)
     
     html = f"""
     <div style="font-family: 'IBM Plex Mono', monospace; font-size: 13px; background-color: #1e1e1e; border: 1px solid #C084FC; border-radius: 8px; padding: 16px; max-height: 400px; overflow-y: auto;">
+        <div style="color: #888; margin-bottom: 8px;">Session ID: {session_id[:8]}...</div>
         <details open style="margin-bottom: 12px; cursor: pointer;">
             <summary style="outline: none; font-size: 15px; color: #A855F7;">📁 <b>papers/</b></summary>
             {papers_tree}
@@ -153,153 +120,128 @@ def get_debug_info():
     """
     return html
 
-def save_api_settings(api_key, api_base_url=None):
+def save_api_settings(api_key, api_base_url, session_id):
+    """不再写入全局 yaml，而是保存在 Gradio State 并在运行时通过环境变量注入"""
     if not api_key: 
-        return "❌ Key cannot be empty", get_debug_info(), False
-    try:
-        config = {}
-        if os.path.exists(CONFIG_PATH):
-            with open(CONFIG_PATH, "r", encoding="utf-8") as f:
-                config = yaml.safe_load(f) or {}
-        config.setdefault("api_keys", {})["gemini_api_key"] = api_key
-        if api_base_url:
-            config["api_base_url"] = api_base_url
-        with open(CONFIG_PATH, "w", encoding="utf-8") as f:
-            yaml.dump(config, f, allow_unicode=True)
-            
-        success_msg = "✅ Key saved"
-        if api_base_url:
-            success_msg += ", Base URL updated"
-            
-        return success_msg, get_debug_info(), True
-    except Exception as e:
-        return f"❌ Error: {str(e)}", get_debug_info(), False
+        return "❌ Key cannot be empty", get_debug_info(session_id), False, "", ""
+    
+    success_msg = "✅ Key saved securely in session memory"
+    if api_base_url: success_msg += ", Base URL updated"
+        
+    return success_msg, get_debug_info(session_id), True, api_key, api_base_url
 
-def save_pdf(file):
-    if file is None: return "❌ Please select a PDF first", get_debug_info(), False
+def save_pdf(file, session_id):
+    if file is None: return "❌ Please select a PDF first", get_debug_info(session_id), False
     try:
-        for f in os.listdir(PAPERS_DIR):
-            file_to_del = os.path.join(PAPERS_DIR, f)
-            if os.path.isfile(file_to_del):
-                os.remove(file_to_del)
+        papers_dir, _, _ = get_user_dirs(session_id)
+        
+        # 清理旧文件
+        for f in os.listdir(papers_dir):
+            file_to_del = os.path.join(papers_dir, f)
+            if os.path.isfile(file_to_del): os.remove(file_to_del)
                 
-        file_path = os.path.join(PAPERS_DIR, os.path.basename(file.name))
+        file_path = os.path.join(papers_dir, os.path.basename(file.name))
         shutil.copy(file.name, file_path)
-        return f"✅ Saved: {os.path.basename(file.name)}", get_debug_info(), True
+        return f"✅ Saved: {os.path.basename(file.name)}", get_debug_info(session_id), True
     except Exception as e:
-        return f"❌ Error: {str(e)}", get_debug_info(), False
+        return f"❌ Error: {str(e)}", get_debug_info(session_id), False
 
-def clear_pdf():
-    """清空 PDF 并锁定后续所有步骤"""
+def clear_pdf(session_id):
     try:
-        deleted_files = []
-        deleted_dirs = []
-        
-        pdf_names = [f for f in os.listdir(PAPERS_DIR) if f.endswith('.pdf')]
-        
-        for f in os.listdir(PAPERS_DIR):
-            file_to_del = os.path.join(PAPERS_DIR, f)
-            if os.path.isfile(file_to_del):
-                os.remove(file_to_del)
-                deleted_files.append(f)
-                
-        if os.path.exists(OUTPUT_DIR):
-            for pdf_name in pdf_names:
-                base_name = os.path.splitext(pdf_name)[0]
-                base_name_us = base_name.replace(" ", "_")
-                
-                for item in os.listdir(OUTPUT_DIR):
-                    if item in [base_name, base_name_us, pdf_name]:
-                        dir_to_del = os.path.join(OUTPUT_DIR, item)
-                        if os.path.isdir(dir_to_del):
-                            shutil.rmtree(dir_to_del)
-                            deleted_dirs.append(item)
-                            
-        if os.path.exists(ZIP_OUTPUT_PATH):
-            os.remove(ZIP_OUTPUT_PATH)
+        user_base = os.path.join(SESSIONS_BASE_DIR, session_id)
+        if os.path.exists(user_base):
+            shutil.rmtree(user_base) # 直接删除该用户的整个工作区
             
         disable_btn = gr.update(interactive=False)
-        if deleted_files or deleted_dirs:
-            return f"🗑️ Workspace cleared", get_debug_info(), False, disable_btn, disable_btn, disable_btn, disable_btn
-        return "ℹ️ Workspace is already empty", get_debug_info(), False, disable_btn, disable_btn, disable_btn, disable_btn
+        return f"🗑️ Workspace cleared", get_debug_info(session_id), False, disable_btn, disable_btn, disable_btn, disable_btn
     except Exception as e:
-        return f"❌ Error deleting file: {str(e)}", get_debug_info(), False, gr.update(), gr.update(), gr.update(), gr.update()
-
-def run_mineru_parsing_and_dag_gen():
+        return f"❌ Error deleting: {str(e)}", 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()
+    env["MINERU_FORMULA_ENABLE"] = "false"
+    env["MINERU_TABLE_ENABLE"] = "false"
+    env["MINERU_DEVICE_MODE"] = "cpu"
+    env["MINERU_VIRTUAL_VRAM_SIZE"] = "8"
+    
+    # 注入专属 API 配置和路径
+    if api_key: env["GEMINI_API_KEY"] = api_key
+    if api_base_url: env["GEMINI_API_BASE_URL"] = api_base_url
+    env["USER_PAPERS_DIR"] = papers_dir
+    env["USER_OUTPUT_DIR"] = output_dir
+    return env
+
+def run_mineru_parsing_and_dag_gen(session_id, api_key, api_base_url, progress=gr.Progress()):
     no_change = gr.update()
     disable_btn = gr.update(interactive=False)
+    papers_dir, output_dir, _ = get_user_dirs(session_id)
     
-    # 进度条显示 (visible=True)
-    show_progress = gr.update(visible=True)
-    hide_progress = gr.update(visible=False)
-    
-    if not os.path.exists(PAPERS_DIR) or not any(f.endswith('.pdf') for f in os.listdir(PAPERS_DIR)):
-        yield "❌ No PDF file found", get_debug_info(), "No execution logs.", no_change, no_change, no_change, no_change, hide_progress
+    if not os.path.exists(papers_dir) or not any(f.endswith('.pdf') for f in os.listdir(papers_dir)):
+        yield "❌ No PDF file found", get_debug_info(session_id), "No execution logs.", no_change, no_change, no_change, no_change
         return
 
     full_log = ""
     try:
-        env = os.environ.copy()
-        env["MINERU_FORMULA_ENABLE"] = "false"
-        env["MINERU_TABLE_ENABLE"] = "false"
-        env["MINERU_DEVICE_MODE"] = "cpu"
-        env["MINERU_VIRTUAL_VRAM_SIZE"] = "8"
-
-        command_mineru = ["mineru", "-p", PAPERS_DIR, "-o", OUTPUT_DIR]
-        full_log += "--- Mineru Executing ---\n"
+        env = build_user_env(api_key, api_base_url, papers_dir, output_dir)
+
+        command_mineru = ["mineru", "-p", papers_dir, "-o", output_dir]
+        full_log += f"--- Mineru Executing (Session: {session_id[:8]}) ---\n"
         
-        # 激活显示全局进度条
-        yield "⏳ Executing Mineru parsing...", get_debug_info(), full_log, no_change, no_change, no_change, no_change, show_progress
+        progress(0.1, desc="启动 Mineru 解析...")
+        yield "⏳ Executing Mineru parsing...", 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
         )
+        
+        progress(0.3, desc="Mineru 正在解析 PDF...")
         for line in iter(process_mineru.stdout.readline, ''):
             full_log += line
-            yield "⏳ Executing Mineru parsing...", get_debug_info(), full_log, no_change, no_change, no_change, no_change, show_progress
+            yield "⏳ Executing Mineru parsing...", 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:
-            yield f"❌ Mineru parsing failed (Exit Code: {returncode_mineru})", get_debug_info(), full_log, disable_btn, disable_btn, disable_btn, disable_btn, hide_progress
+            progress(1.0, desc="Mineru 解析失败")
+            yield f"❌ Mineru parsing failed (Exit Code: {returncode_mineru})", get_debug_info(session_id), full_log, disable_btn, disable_btn, disable_btn, disable_btn
             return
 
         command_dag = [sys.executable, "gen_dag.py"]
         full_log += "\n--- DAG Gen Executing ---\n"
-        yield "⏳ Mineru parsing complete, executing DAG generation...", get_debug_info(), full_log, no_change, no_change, no_change, no_change, show_progress
+        progress(0.6, desc="执行 DAG 生成...")
+        yield "⏳ Executing DAG generation...", get_debug_info(session_id), full_log, no_change, no_change, no_change, no_change
         
         process_dag = subprocess.Popen(
-            command_dag, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1
+            command_dag, env=env, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1
         )
+        
+        progress(0.8, desc="构建图结构中...")
         for line in iter(process_dag.stdout.readline, ''):
             full_log += line
-            yield "⏳ Executing DAG generation...", get_debug_info(), full_log, no_change, no_change, no_change, no_change, show_progress
+            yield "⏳ Executing DAG generation...", 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:
-            status = "✅ PDF parsing & DAG generation fully completed"
+            progress(1.0, desc="解析与构建完成！")
             enable_btn = gr.update(interactive=True)
-            yield status, get_debug_info(), full_log, enable_btn, enable_btn, enable_btn, enable_btn, hide_progress
+            yield "✅ PDF parsing & DAG generation fully completed", get_debug_info(session_id), full_log, enable_btn, enable_btn, enable_btn, enable_btn
         else:
-            status = f"❌ DAG generation failed (Exit Code: {returncode_dag})"
-            yield status, get_debug_info(), full_log, disable_btn, disable_btn, disable_btn, disable_btn, hide_progress
+            progress(1.0, desc="DAG 生成失败")
+            yield f"❌ DAG generation failed", get_debug_info(session_id), full_log, disable_btn, disable_btn, disable_btn, disable_btn
 
     except Exception as e:
-        error_log = full_log + f"\n[Global Exception] Exception occurred:\n{str(e)}"
-        yield "❌ Execution Exception", get_debug_info(), error_log, disable_btn, disable_btn, disable_btn, disable_btn, hide_progress
+        progress(1.0, desc="发生异常")
+        error_log = full_log + f"\n[Global Exception]:\n{str(e)}"
+        yield "❌ Execution Exception", get_debug_info(session_id), error_log, disable_btn, disable_btn, disable_btn, disable_btn
 
 
-def run_final_generation(task_type="all"):
-    """
-    使用队列和多线程实时获取子进程的流式输出，并通过 yield 推送到前端界面。
-    """
-    show_progress = gr.update(visible=True)
-    hide_progress = gr.update(visible=False)
-
-    if not os.path.exists(OUTPUT_DIR):
-        yield "❌ Please run the parsing step first", get_debug_info(), "No output folder found.", gr.update(visible=False), hide_progress
+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 "❌ Please run parsing first", get_debug_info(session_id), "No output folder found.", gr.update(visible=False)
         return
 
     scripts_to_run = []
@@ -307,16 +249,14 @@ def run_final_generation(task_type="all"):
     elif task_type == "poster": scripts_to_run = ["gen_poster.py"]
     elif task_type == "pr": scripts_to_run = ["gen_pr.py"]
     elif task_type == "all": scripts_to_run = ["gen_ppt.py", "gen_poster.py", "gen_pr.py"]
-    else: 
-        yield "❌ Unknown task type", get_debug_info(), "Invalid task_type.", gr.update(visible=False), hide_progress
-        return
-
-    full_log = f"🚀 Preparing to start {len(scripts_to_run)} tasks...\n"
-    # 激活全局进度条
-    yield f"⏳ Starting {task_type.upper()} generation...", get_debug_info(), full_log, gr.update(visible=False), show_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 f"⏳ Starting {task_type.upper()} generation...", get_debug_info(session_id), full_log, gr.update(visible=False)
 
     q = queue.Queue()
     processes = []
+    env = build_user_env(api_key, api_base_url, papers_dir, output_dir)
     
     def enqueue_output(out, script_name):
         for line in iter(out.readline, ''):
@@ -326,11 +266,7 @@ def run_final_generation(task_type="all"):
     try:
         for script in scripts_to_run:
             p = subprocess.Popen(
-                [sys.executable, script], 
-                stdout=subprocess.PIPE, 
-                stderr=subprocess.STDOUT, 
-                text=True, 
-                bufsize=1
+                [sys.executable, script], env=env, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, bufsize=1
             )
             processes.append((script, p))
             t = threading.Thread(target=enqueue_output, args=(p.stdout, script))
@@ -338,36 +274,38 @@ def run_final_generation(task_type="all"):
             t.start()
 
         active_processes = len(processes)
+        progress(0.5, desc=f"正在并行生成 {task_type.upper()}...")
+        
         while active_processes > 0 or not q.empty():
             try:
                 script_name, line = q.get(timeout=0.1)
                 full_log += f"[{script_name}] {line}"
-                yield f"⏳ Generating {task_type.upper()}...", get_debug_info(), full_log, gr.update(visible=False), show_progress
+                yield f"⏳ Generating {task_type.upper()}...", 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)
 
-        success = True
-        for script, p in processes:
-            if p.returncode != 0:
-                success = False
-                full_log += f"\n❌ [Error] {script} returned non-zero exit code (Exit Code: {p.returncode})\n"
+        success = all(p.returncode == 0 for _, p in processes)
 
         if not success: 
-            yield f"❌ {task_type.upper()} contains failed tasks, please check logs", get_debug_info(), full_log, gr.update(visible=False), hide_progress
+            progress(1.0, desc="生成失败")
+            yield f"❌ Tasks failed, check logs", get_debug_info(session_id), full_log, gr.update(visible=False)
             return
 
         full_log += "\n📦 Zipping output directory...\n"
-        yield f"⏳ Zipping outputs...", get_debug_info(), full_log, gr.update(visible=False), show_progress
+        progress(0.9, desc="打包压缩结果...")
+        yield f"⏳ Zipping outputs...", get_debug_info(session_id), full_log, gr.update(visible=False)
         
-        zip_base_name = ZIP_OUTPUT_PATH.replace(".zip", "")
-        shutil.make_archive(zip_base_name, 'zip', OUTPUT_DIR)
+        zip_base_name = zip_path.replace(".zip", "")
+        shutil.make_archive(zip_base_name, 'zip', output_dir)
         
-        full_log += "✅ All tasks and zipping completed successfully.\n"
-        yield f"✅ {task_type.upper()} generated and zipped successfully", get_debug_info(), full_log, gr.update(value=ZIP_OUTPUT_PATH, visible=True), hide_progress
+        full_log += "✅ All tasks completed successfully.\n"
+        progress(1.0, desc="全部完成！")
+        yield f"✅ {task_type.upper()} successfully generated", get_debug_info(session_id), full_log, gr.update(value=zip_path, visible=True)
         
     except Exception as e:
-        error_log = full_log + f"\n[Global Exception] Exception occurred:\n{str(e)}"
-        yield "❌ Global exception during final generation", get_debug_info(), error_log, gr.update(visible=False), hide_progress
+        progress(1.0, desc="发生全局异常")
+        error_log = full_log + f"\n[Global Exception]:\n{str(e)}"
+        yield "❌ Global exception", get_debug_info(session_id), error_log, gr.update(visible=False)
 
 
 # ==========================================
@@ -401,29 +339,10 @@ body, .gradio-container {
 
 @keyframes gradient-shift { 0%, 100% { background-position: 0% 50%; } 50% { background-position: 100% 50%; } }
 
-#subtitle { text-align: center !important; margin-bottom: 1.5rem; }
+#subtitle { text-align: center !important; margin-bottom: 2rem; }
 #subtitle p { margin: 0 auto; color: #666; font-size: 1.1rem; font-weight: 500; }
 .dark #subtitle p { color: #DAB2FF; }
 
-/* ======== CSS FOR GLOBAL PROGRESS BAR ======== */
-#global-progress { margin-bottom: 20px !important; transition: opacity 0.3s ease; }
-.progress-container {
-    width: 100%; background-color: rgba(243, 232, 255, 0.8);
-    border: 1px solid #C084FC; border-radius: 16px; overflow: hidden;
-    position: relative; height: 38px; display: flex; align-items: center; justify-content: center;
-    box-shadow: 0 4px 15px rgba(168, 85, 247, 0.15);
-}
-.progress-bar {
-    height: 100%; width: 40%;
-    background: linear-gradient(90deg, transparent, rgba(168, 85, 247, 0.6), rgba(192, 132, 252, 0.8), rgba(168, 85, 247, 0.6), transparent);
-    position: absolute; left: -40%; top: 0;
-    animation: loading-slide 1.5s infinite linear;
-}
-@keyframes loading-slide { 0% { left: -40%; } 100% { left: 100%; } }
-.progress-text { position: relative; z-index: 1; color: #6B21A8; font-weight: 600; font-size: 15px; letter-spacing: 0.5px; }
-.dark .progress-container { background-color: rgba(30, 30, 30, 0.9); border-color: rgba(168, 85, 247, 0.4); }
-.dark .progress-text { color: #DAB2FF; }
-
 .gradio-group {
     background: rgba(255, 255, 255, 0.9) !important;
     border: 2px solid #E9D5FF !important; 
@@ -498,7 +417,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; 
@@ -566,7 +485,11 @@ details > summary:hover { color: #E9D5FF !important; }
 """
 
 with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
-    # 定义全局状态变量，用于实现依赖解锁
+    # 核心：为每个连接的用户生成唯一的 Session ID 和存放凭证的状态
+    session_id_state = gr.State(lambda: str(uuid.uuid4()))
+    user_api_key_state = gr.State("")
+    user_api_base_state = gr.State("")
+    
     api_saved_state = gr.State(False)
     pdf_ready_state = gr.State(False)
 
@@ -574,17 +497,8 @@ with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
         gr.Markdown("# **PaperX Platform**", elem_id="main-title")
         gr.Markdown("One-click parsing of academic PDFs, DAG structuring, and multi-modal asset generation.", elem_id="subtitle")
         
-        # 🟢 新增的全局加载动画条，默认隐藏
-        global_progress = gr.HTML(
-            '<div class="progress-container"><div class="progress-bar"></div><div class="progress-text">⏳ 加载中... (Task in progress, please wait)</div></div>',
-            visible=False,
-            elem_id="global-progress"
-        )
-        
         with gr.Row():
-            # ================= LEFT COLUMN: SETTINGS & ACTIONS =================
             with gr.Column(scale=1):
-                
                 # 1. API Configuration
                 with gr.Group(elem_classes="gradio-group"):
                     gr.Markdown("### ⚙️ 1. Global API Configuration")
@@ -592,112 +506,88 @@ with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
                         key_input = gr.Textbox(label="API Key", type="password", placeholder="sk-...", scale=1)
                         api_base_url_input = gr.Textbox(label="Base URL (Optional)", placeholder="https://api.example.com", scale=1)
                     key_btn = gr.Button("💾 Save API Configuration")
-                    
-                    api_status = gr.Textbox(
-                        show_label=False, placeholder="Waiting for API configuration...", lines=1, interactive=False, elem_classes="status-text"
-                    )
-                
+                    api_status = gr.Textbox(show_label=False, interactive=False, elem_classes="status-text")
+
                 # 2. Document Parsing
                 with gr.Group(elem_classes="gradio-group"):
                     gr.Markdown("### 📄 2. Document Parsing")
-                    
-                    pdf_input = gr.File(
-                        label="Upload Document", 
-                        file_types=[".pdf"], 
-                        elem_id="pdf-upload-box"
-                    )
-                    
-                    # 默认锁定
+                    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_status = gr.Textbox(
-                        show_label=False, placeholder="Waiting for document upload...", lines=1, interactive=False, elem_classes="status-text"
-                    )
+                    parse_status = gr.Textbox(show_label=False, interactive=False, elem_classes="status-text")
 
                 # 3. Asset Generation
                 with gr.Group(elem_classes="gradio-group"):
                     gr.Markdown("### 🎯 3. Asset Generation")
-                    gr.Markdown("Generate final formats based on DAG structure:")
-                    
                     with gr.Row(elem_classes="action-row"):
-                        # 默认锁定
                         gen_ppt_btn = gr.Button("📊 Gen PPT", elem_classes="action-btn", interactive=False)
                         gen_poster_btn = gr.Button("🖼️ Gen Poster", elem_classes="action-btn", interactive=False)
                         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)
 
-            # ================= RIGHT COLUMN: OUTPUTS & LOGS =================
             with gr.Column(scale=1):
-                
                 # 4. Results & Downloads
                 with gr.Group(elem_classes="gradio-group"):
                     gr.Markdown("### 📦 Generation Results & Download")
-                    gen_status = gr.Textbox(
-                        show_label=False, placeholder="No generation task currently...", lines=2, interactive=False, elem_classes="status-text"
-                    )
+                    gen_status = gr.Textbox(show_label=False, interactive=False, elem_classes="status-text")
                     download_file = gr.File(label="📥 Get Final Zip Archive", interactive=False, visible=False)
 
-                # 5. Debugging & Terminal
+                # 5. Debugging
                 with gr.Group(elem_classes="gradio-group"):
                     gr.Markdown("### 🛠️ Developer Monitoring (Debug Only)")
                     with gr.Tabs():
                         with gr.Tab("📜 Terminal Stream"):
-                            cmd_logs = gr.Textbox(
-                                label="Stdout / Stderr", placeholder="Waiting for task to start...", lines=14, interactive=False, elem_classes="log-box" 
-                            )
-                        
+                            cmd_logs = gr.Textbox(show_label=False, lines=14, interactive=False, elem_classes="log-box")
                         with gr.Tab("🔍 System Snapshot"):
                             refresh_btn = gr.Button("🔄 Refresh Directory Tree") 
-                            debug_view = gr.HTML(
-                                value=get_debug_info()
-                            )
+                            debug_view = gr.HTML()
+
+    # 初始化加载空的 Debug 信息
+    demo.load(fn=get_debug_info, inputs=session_id_state, outputs=debug_view)
 
     # ================= LOGIC BINDINGS =================
     
-    # [步骤 1] 配置 API
     key_btn.click(
         fn=save_api_settings, 
-        inputs=[key_input, api_base_url_input], 
-        outputs=[api_status, debug_view, api_saved_state]
+        inputs=[key_input, api_base_url_input, session_id_state], 
+        outputs=[api_status, debug_view, api_saved_state, user_api_key_state, user_api_base_state]
     )
     
-    # [步骤 2.1] 上传 PDF
-    pdf_input.upload(fn=save_pdf, inputs=pdf_input, outputs=[parse_status, debug_view, pdf_ready_state])
+    pdf_input.upload(
+        fn=save_pdf, 
+        inputs=[pdf_input, session_id_state], 
+        outputs=[parse_status, debug_view, pdf_ready_state]
+    )
     
-    # [步骤 2.2] 清除 PDF
     pdf_input.clear(
         fn=clear_pdf, 
+        inputs=[session_id_state],
         outputs=[parse_status, debug_view, pdf_ready_state, gen_ppt_btn, gen_poster_btn, gen_pr_btn, gen_all_btn]
     )
     
-    # [依赖检查] 当 api 状态或 pdf 状态发生变化时，检查是否解锁"开始解析"按钮
     def check_parse_btn_ready(api_ready, pdf_ready):
         return gr.update(interactive=(api_ready and pdf_ready))
         
     api_saved_state.change(fn=check_parse_btn_ready, inputs=[api_saved_state, pdf_ready_state], outputs=parse_btn)
     pdf_ready_state.change(fn=check_parse_btn_ready, inputs=[api_saved_state, pdf_ready_state], outputs=parse_btn)
 
-    # [步骤 2.3] 执行解析，绑定 8 个 outputs (新增控制 global_progress)
     parse_btn.click(
         fn=run_mineru_parsing_and_dag_gen, 
-        outputs=[parse_status, debug_view, cmd_logs, gen_ppt_btn, gen_poster_btn, gen_pr_btn, gen_all_btn, global_progress]
+        inputs=[session_id_state, user_api_key_state, user_api_base_state],
+        outputs=[parse_status, debug_view, cmd_logs, gen_ppt_btn, gen_poster_btn, gen_pr_btn, gen_all_btn]
     )
 
-    # [步骤 3] 资产生成
-    def trigger_gen_ppt(): yield from run_final_generation("ppt")
-    def trigger_gen_poster(): yield from run_final_generation("poster")
-    def trigger_gen_pr(): yield from run_final_generation("pr")
-    def trigger_gen_all(): yield from run_final_generation("all")
-
-    # [步骤 3 绑定] 绑定 5 个 outputs (新增控制 global_progress)
-    gen_ppt_btn.click(fn=trigger_gen_ppt, outputs=[gen_status, debug_view, cmd_logs, download_file, global_progress])
-    gen_poster_btn.click(fn=trigger_gen_poster, outputs=[gen_status, debug_view, cmd_logs, download_file, global_progress])
-    gen_pr_btn.click(fn=trigger_gen_pr, outputs=[gen_status, debug_view, cmd_logs, download_file, global_progress])
-    gen_all_btn.click(fn=trigger_gen_all, outputs=[gen_status, debug_view, cmd_logs, download_file, global_progress])
+    def trigger_gen_ppt(sid, ak, ab, progress=gr.Progress()): yield from run_final_generation("ppt", sid, ak, ab, progress)
+    def trigger_gen_poster(sid, ak, ab, progress=gr.Progress()): yield from run_final_generation("poster", sid, ak, ab, progress)
+    def trigger_gen_pr(sid, ak, ab, progress=gr.Progress()): yield from run_final_generation("pr", sid, ak, ab, progress)
+    def trigger_gen_all(sid, ak, ab, progress=gr.Progress()): yield from run_final_generation("all", sid, ak, ab, progress)
+
+    # 绑定需要传递所有专属状态
+    gen_ppt_btn.click(fn=trigger_gen_ppt, inputs=[session_id_state, user_api_key_state, user_api_base_state], outputs=[gen_status, debug_view, cmd_logs, download_file])
+    gen_poster_btn.click(fn=trigger_gen_poster, inputs=[session_id_state, user_api_key_state, user_api_base_state], outputs=[gen_status, debug_view, cmd_logs, download_file])
+    gen_pr_btn.click(fn=trigger_gen_pr, inputs=[session_id_state, user_api_key_state, user_api_base_state], outputs=[gen_status, debug_view, cmd_logs, download_file])
+    gen_all_btn.click(fn=trigger_gen_all, inputs=[session_id_state, user_api_key_state, user_api_base_state], outputs=[gen_status, debug_view, cmd_logs, download_file])
     
-    refresh_btn.click(fn=get_debug_info, outputs=debug_view)
+    refresh_btn.click(fn=get_debug_info, inputs=[session_id_state], outputs=debug_view)
 
 if __name__ == "__main__":
     demo.launch()