Update app.py

app.py

@@ -6,16 +6,93 @@ import subprocess
 import sys
 from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from typing import Iterable
+from gradio.themes import Soft
+from gradio.themes.utils import colors, fonts, sizes
 
 # Initialize environment paths
 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") # Zip path
+ZIP_OUTPUT_PATH = os.path.join(BASE_DIR, "mineru_results.zip")
 
 os.makedirs(PAPERS_DIR, exist_ok=True)
 
+# ==========================================
+# --- 🎨 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",
+)
+
+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",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
+        super().set(
+            background_fill_primary="*primary_50",
+            background_fill_primary_dark="*primary_900",
+            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()
+
+# ==========================================
+# --- ⚙️ Backend Logic & Functions ---
+# ==========================================
 def get_debug_info():
     """Read server file system status"""
     now = datetime.now().strftime("%H:%M:%S")
@@ -29,7 +106,7 @@ def get_debug_info():
             for name in files_in_out:
                 all_output_items.append(os.path.join(os.path.relpath(root, OUTPUT_DIR), name))
         output_detail = f"Found {len(all_output_items)} files: {all_output_items[:5]}..." if all_output_items else "Directory exists but is EMPTY"
-
+    
     return f"[{now}] 📁 papers/ Content: {files}\n\n[{now}] 📂 mineru_outputs Status: {output_detail}"
 
 def save_pdf(file):
@@ -68,7 +145,6 @@ def save_api_settings(api_key, api_base_url=None):
     except Exception as e:
         return f"❌ Error: {str(e)}", get_debug_info()
 
-
 def run_mineru_parsing_and_dag_gen():
     """Execute PDF parsing and DAG generation (supports real-time streaming)"""
     if not os.path.exists(PAPERS_DIR) or not any(f.endswith('.pdf') for f in os.listdir(PAPERS_DIR)):
@@ -89,7 +165,6 @@ def run_mineru_parsing_and_dag_gen():
         full_log += "--- Mineru Executing ---\n"
         yield "⏳ Executing Mineru parsing...", get_debug_info(), full_log
 
-        # 1. Use Popen for real-time streaming
         process_mineru = subprocess.Popen(
             command_mineru, 
             env=env, 
@@ -99,11 +174,9 @@ def run_mineru_parsing_and_dag_gen():
             bufsize=1 
         )
 
-        # 2. Read output line by line and yield to Gradio
         for line in iter(process_mineru.stdout.readline, ''):
             full_log += line
             yield "⏳ Executing Mineru parsing...", get_debug_info(), full_log
-
         process_mineru.stdout.close()
         returncode_mineru = process_mineru.wait()
 
@@ -129,7 +202,6 @@ def run_mineru_parsing_and_dag_gen():
         for line in iter(process_dag.stdout.readline, ''):
             full_log += line
             yield "⏳ Executing DAG generation...", get_debug_info(), full_log
-
         process_dag.stdout.close()
         returncode_dag = process_dag.wait()
         
@@ -145,10 +217,7 @@ def run_mineru_parsing_and_dag_gen():
         yield "❌ Execution Exception", get_debug_info(), error_log
 
 def run_final_generation(task_type="all"):
-    """
-    Execute generation scripts and zip results
-    task_type supports: 'ppt', 'poster', 'pr', 'all'
-    """
+    """Execute generation scripts and zip results"""
     if not os.path.exists(OUTPUT_DIR):
         return "❌ Please run the parsing step first", get_debug_info(), "No output folder found.", None
 
@@ -169,12 +238,7 @@ def run_final_generation(task_type="all"):
 
     def execute_script(script):
         command = [sys.executable, script]
-        result = subprocess.run(
-            command,
-            capture_output=True,
-            text=True,
-            timeout=600 
-        )
+        result = subprocess.run(command, capture_output=True, text=True, timeout=600)
         return script, result
 
     try:
@@ -206,7 +270,7 @@ def run_final_generation(task_type="all"):
         # Zip the mineru_outputs folder
         zip_base_name = ZIP_OUTPUT_PATH.replace(".zip", "")
         shutil.make_archive(zip_base_name, 'zip', OUTPUT_DIR)
-
+        
         success_msg = f"✅ {task_type.upper()} generated and zipped successfully"
         return success_msg, get_debug_info(), full_log, ZIP_OUTPUT_PATH
 
@@ -215,19 +279,38 @@ def run_final_generation(task_type="all"):
         return "❌ Global exception during final generation", get_debug_info(), error_log, None
 
 # ==========================================
-# --- 🚀 UI Configuration (Left-Right Layout) ---
+# --- 🚀 UI Configuration & Advanced CSS ---
 # ==========================================
-
 custom_css = """
-/* Prominent Title Animation from Reference */
+@import url('https://fonts.googleapis.com/css2?family=Outfit:wght@400;500;600;700&family=IBM+Plex+Mono:wght@400;500;600&display=swap');
+
+body, .gradio-container {
+    background-color: #FAF5FF !important;
+    background-image: linear-gradient(#E9D5FF 1px, transparent 1px), linear-gradient(90deg, #E9D5FF 1px, transparent 1px) !important;
+    background-size: 40px 40px !important;
+    font-family: 'Outfit', sans-serif !important;
+}
+
+.dark body, .dark .gradio-container {
+    background-color: #1a1a1a !important;
+    background-image: linear-gradient(rgba(168, 85, 247, .1) 1px, transparent 1px), linear-gradient(90deg, rgba(168, 85, 247, .1) 1px, transparent 1px) !important;
+    background-size: 40px 40px !important;
+}
+
+#col-container {
+    margin: 0 auto;
+    max-width: 1200px;
+}
+
 #main-title {
     text-align: center !important;
-    padding: 1rem 0 0.5rem 0;
+    padding: 1.5rem 0 0.5rem 0;
 }
+
 #main-title h1 {
     font-size: 2.6em !important;
     font-weight: 800 !important;
-    background: linear-gradient(135deg, #4f46e5 0%, #818cf8 50%, #4338ca 100%);
+    background: linear-gradient(135deg, #A855F7 0%, #C084FC 50%, #9333EA 100%);
     background-size: 200% 200%;
     -webkit-background-clip: text;
     -webkit-text-fill-color: transparent;
@@ -235,14 +318,17 @@ custom_css = """
     animation: gradient-shift 4s ease infinite;
     letter-spacing: -0.02em;
 }
+
 @keyframes gradient-shift {
     0%, 100% { background-position: 0% 50%; }
     50% { background-position: 100% 50%; }
 }
+
 #subtitle {
     text-align: center !important;
-    margin-bottom: 1.5rem;
+    margin-bottom: 2rem;
 }
+
 #subtitle p {
     margin: 0 auto;
     color: #666;
@@ -250,98 +336,141 @@ custom_css = """
     font-weight: 500;
 }
 
+.dark #subtitle p { color: #DAB2FF; }
+
+/* Glassmorphism Cards */
+.gradio-group {
+    background: rgba(255, 255, 255, 0.9) !important;
+    border: 2px solid #E9D5FF !important;
+    border-radius: 12px !important;
+    box-shadow: 0 4px 24px rgba(168, 85, 247, 0.08) !important;
+    backdrop-filter: blur(10px);
+    transition: all 0.3s ease;
+    padding: 15px !important;
+}
+
+.gradio-group:hover {
+    box-shadow: 0 8px 32px rgba(168, 85, 247, 0.12) !important;
+    border-color: #C084FC !important;
+}
+
+.dark .gradio-group {
+    background: rgba(30, 30, 30, 0.9) !important;
+    border-color: rgba(168, 85, 247, 0.3) !important;
+}
+
 /* Terminal Log Style */
 .log-box textarea {
-    font-family: 'Courier New', Consolas, monospace !important; 
+    font-family: 'IBM Plex Mono', Consolas, monospace !important; 
     font-size: 13px !important; 
     background-color: #1e1e1e !important; 
-    color: #4AF626 !important; 
+    color: #DAB2FF !important; 
+    border: 1px solid #C084FC !important;
+    border-radius: 8px !important;
 }
+
 /* Borderless Status Text */
 .status-text textarea {
     background-color: transparent !important;
     border: none !important;
     box-shadow: none !important;
-    font-weight: bold;
+    font-weight: 600 !important;
+    color: #6B21A8 !important;
 }
+
+.dark .status-text textarea {
+    color: #C084FC !important;
+}
+
+/* Custom Scrollbars */
+::-webkit-scrollbar { width: 8px; height: 8px; }
+::-webkit-scrollbar-track { background: rgba(168, 85, 247, 0.05); border-radius: 4px; }
+::-webkit-scrollbar-thumb { background: linear-gradient(135deg, #A855F7, #C084FC); border-radius: 4px; }
+::-webkit-scrollbar-thumb:hover { background: linear-gradient(135deg, #9333EA, #A855F7); }
 """
 
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=custom_css) as demo:
-    gr.Markdown("# **PaperX / Mineru Parsing Platform**", elem_id="main-title")
-    gr.Markdown("One-click parsing of academic PDFs, DAG structuring, and multi-modal asset generation.", elem_id="subtitle")
-    
-    with gr.Row():
-        # ================= LEFT COLUMN: SETTINGS & ACTIONS =================
-        with gr.Column(scale=1):
-            
-            # 1. API Configuration
-            with gr.Accordion("⚙️ 1. Global API Configuration", open=False):
-                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 (Optional)", placeholder="https://api.example.com", scale=1)
-                key_btn = gr.Button("💾 Save API Configuration")
-
-            # 2. Document Parsing
-            with gr.Group():
-                gr.Markdown("### 📄 2. Document Parsing")
-                pdf_input = gr.File(label="Drag and drop or click to upload PDF", file_types=[".pdf"])
-                
-                parse_btn = gr.Button("🚀 Start Mineru & DAG Extraction", variant="primary", size="lg")
-                
-                parse_status = gr.Textbox(
-                    show_label=False, 
-                    placeholder="Waiting for document upload...", 
-                    lines=1, 
-                    interactive=False, 
-                    elem_classes="status-text"
-                )
-
-            # 3. Asset Generation
-            with gr.Group():
-                gr.Markdown("### 🎯 3. Asset Generation")
-                gr.Markdown("Generate final formats based on DAG structure:")
+with gr.Blocks(theme=purple_theme, css=custom_css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("# **PaperX / Mineru Parsing Platform**", elem_id="main-title")
+        gr.Markdown("One-click parsing of academic PDFs, DAG structuring, and multi-modal asset generation.", elem_id="subtitle")
+        
+        with gr.Row():
+            # ================= LEFT COLUMN: SETTINGS & ACTIONS =================
+            with gr.Column(scale=1):
                 
-                with gr.Row():
-                    gen_ppt_btn = gr.Button("📊 Generate PPT")
-                    gen_poster_btn = gr.Button("🖼️ Generate Poster")
-                    gen_pr_btn = gr.Button("📰 Generate PR Article")
+                # 1. API Configuration
+                with gr.Accordion("⚙️ 1. Global API Configuration", open=False, elem_classes="gradio-group"):
+                    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 (Optional)", placeholder="https://api.example.com", scale=1)
+                    key_btn = gr.Button("💾 Save API Configuration")
                 
-                gen_all_btn = gr.Button("✨ Generate All Assets (ALL)", variant="primary")
+                # 2. Document Parsing
+                with gr.Group(elem_classes="gradio-group"):
+                    gr.Markdown("### 📄 2. Document Parsing")
+                    pdf_input = gr.File(label="Drag and drop or click to upload PDF", file_types=[".pdf"])
+                    
+                    parse_btn = gr.Button("🚀 Start Mineru & DAG Extraction", variant="primary", size="lg")
+                    
+                    parse_status = gr.Textbox(
+                        show_label=False, 
+                        placeholder="Waiting for document upload...", 
+                        lines=1, 
+                        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():
+                        gen_ppt_btn = gr.Button("📊 Generate PPT")
+                        gen_poster_btn = gr.Button("🖼️ Generate Poster")
+                        gen_pr_btn = gr.Button("📰 Generate PR Article")
+                    
+                    gen_all_btn = gr.Button("✨ Generate All Assets (ALL)", variant="primary")
 
-        # ================= RIGHT COLUMN: OUTPUTS & LOGS =================
-        with gr.Column(scale=1):
-            
-            # 4. Results & Downloads
-            with gr.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"
-                )
-                download_file = gr.File(label="📥 Get Final Zip Archive", interactive=False, visible=False)
-
-            # 5. Debugging & Terminal
-            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=18, 
+            # ================= 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="log-box" 
+                        elem_classes="status-text"
                     )
-                
-                with gr.Tab("🔍 System Snapshot"):
-                    refresh_btn = gr.Button("🔄 Refresh Directory Tree") 
-                    debug_view = gr.Textbox(label="Workspace Files", lines=17, interactive=False, value=get_debug_info())
+                    download_file = gr.File(label="📥 Get Final Zip Archive", interactive=False, visible=False)
+
+                # 5. Debugging & Terminal
+                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" 
+                            )
+                        
+                        with gr.Tab("🔍 System Snapshot"):
+                            refresh_btn = gr.Button("🔄 Refresh Directory Tree") 
+                            debug_view = gr.Textbox(
+                                label="Workspace Files", 
+                                lines=13, 
+                                interactive=False, 
+                                value=get_debug_info(),
+                                elem_classes="log-box"
+                            )
 
     # ================= LOGIC BINDINGS =================
-    
     key_btn.click(fn=save_api_settings, inputs=[key_input, api_base_url_input], outputs=[parse_status, debug_view])
     
     pdf_input.upload(fn=save_pdf, inputs=pdf_input, outputs=[parse_status, debug_view])