Update app.py

app.py

@@ -2,37 +2,45 @@ import os, json, shutil
 import pandas as pd
 import gradio as gr
 from agent import chat_with_agent
-from tools import CSV_PATH, OUT_DIR, PDF_DIR, _embed, HEADERS, PERSISTENT_BASE
+from tools import CSV_PATH, OUT_DIR, PDF_DIR, _embed, HEADERS
 
 try: _embed()
 except Exception as e: print(f">>> WARNING: Embedding pre-load failed: {e}")
 
-IS_PERSISTENT = PERSISTENT_BASE == "/data"
-STATUS_TAG = '<span style="background:#0ea5e8;color:white;padding:2px 6px;border-radius:4px;font-size:0.5em;vertical-align:middle;margin-left:8px">PERSISTENT</span>' if IS_PERSISTENT else ""
-
 CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600&family=Manrope:wght@600;700&display=swap');
-:root { --primary: #0f172a; --secondary: #712ae2; --secondary-container: #8a4cfc; --surface: #f7f9fb; }
-body, .gradio-container { font-family: 'Inter', sans-serif !important; background-color: var(--surface) !important; color: #191c1e !important; }
-.hdr { font-family: 'Manrope', sans-serif !important; font-size: 3.5rem !important; font-weight: 700; color: var(--primary); margin-bottom: 24px; letter-spacing: -0.02em; }
-.sub-hdr { font-family: 'Manrope', sans-serif !important; font-size: 1.75rem !important; font-weight: 600; margin-bottom: 16px; }
-button.primary { background: linear-gradient(135deg, var(--secondary), var(--secondary-container)) !important; color: white !important; border-radius: 12px !important; border: none !important; box-shadow: none !important; transition: opacity 0.2s; }
-button.primary:hover { opacity: 0.9; }
-button.secondary { background: transparent !important; color: var(--primary) !important; border: none !important; box-shadow: none !important; }
-button.secondary:hover { background: rgba(15,23,42,0.05) !important; }
-.sidebar { background: rgba(19, 27, 46, 0.03) !important; border-radius: 16px !important; padding: 24px !important; box-shadow: none !important; border: none !important; }
-.main-content { background: white !important; border-radius: 16px !important; padding: 32px !important; box-shadow: 0 20px 50px -12px rgba(15, 23, 42, 0.08) !important; border: none !important; }
-input, textarea { background-color: rgba(255,255,255,0.7) !important; border-radius: 0 !important; border: none !important; border-bottom: 2px solid rgba(15,23,42,0.1) !important; box-shadow: none !important; }
-input:focus, textarea:focus { border-bottom: 2px solid var(--secondary) !important; }
-.tab-nav { border-bottom: 1px solid rgba(15,23,42,0.08) !important; }
-.label-sm { font-size: 0.6875rem !important; font-weight: 500 !important; letter-spacing: 0.05em !important; text-transform: uppercase !important; }
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600&family=Plus+Jakarta+Sans:wght@600;700&display=swap');
+:root { --primary: #0f172a; --secondary: #6366f1; --accent: #8b5cf6; --bg: #f8fafc; }
+body, .gradio-container { font-family: 'Inter', sans-serif !important; background-color: var(--bg) !important; color: var(--primary) !important; }
+.hdr { font-family: 'Plus Jakarta Sans', sans-serif !important; font-size: 2.25rem !important; font-weight: 700; color: var(--primary) !important; letter-spacing: -0.04em; margin-bottom: 2px !important; }
+.sub-hdr { font-size: 0.875rem !important; color: #64748b !important; margin-bottom: 24px !important; }
+.sidebar { background: white !important; border-right: 1px solid rgba(15,23,42,0.08) !important; padding: 24px !important; }
+.main-content { background: white !important; border-radius: 12px !important; border: 1px solid rgba(15,23,42,0.05) !important; box-shadow: 0 4px 6px -1px rgba(0,0,0,0.05) !important; }
+button.primary { background: var(--primary) !important; color: white !important; border-radius: 8px !important; font-weight: 600 !important; transition: all 0.2s; }
+button.primary:hover { background: #1e293b !important; transform: translateY(-1px); }
+.phase-orb { display: flex; align-items: center; gap: 8px; font-size: 0.75rem; font-weight: 600; text-transform: uppercase; letter-spacing: 0.05em; color: #94a3b8; margin-bottom: 12px; }
+.phase-orb.active { color: var(--secondary); }
+.phase-dot { width: 8px; height: 8px; border-radius: 50%; background: #e2e8f0; }
+.phase-orb.active .phase-dot { background: var(--secondary); box-shadow: 0 0 8px var(--secondary); }
+input, textarea, .gr-box, .gr-form, label, span, p, .markdown-text { color: #0f172a !important; }
+input, textarea { border-radius: 8px !important; border: 1px solid #e2e8f0 !important; background: white !important; }
+input:focus { border-color: var(--secondary) !important; ring: 2px var(--secondary) !important; }
 """
-HDR = f'<div class="hdr">The Digital Curator{STATUS_TAG}</div>'
-PRG = '<div class="label-sm" style="margin-bottom: 24px; color: #64748b;">① Ingestion &gt; ② Algorithmic Coding &gt; ③ Thematic Abstraction &gt; ④ Taxonomy Alignment &gt; ⑥ Report Narrative</div>'
+HDR = '<div class="hdr">Digital Curator</div><div class="sub-hdr">Enterprise Agentic Research Platform</div>'
+PRG_TPL = '<div style="display: flex; flex-direction: column; gap: 4px;">' \
+          '<div class="phase-orb {a1}"><div class="phase-dot"></div>1. Ingestion</div>' \
+          '<div class="phase-orb {a2}"><div class="phase-dot"></div>2. Algorithmic Coding</div>' \
+          '<div class="phase-orb {a3}"><div class="phase-dot"></div>3. Thematic Abstraction</div>' \
+          '<div class="phase-orb {a4}"><div class="phase-dot"></div>4. Report Narration</div>' \
+          '</div>'
 
 PHASE_FILE = {"CODES": "_labels.json", "THEMES": "_themes.json", "PAJAIS": "_taxonomy.json"}
 NEXT_PHASE = {"CODES": "THEMES", "THEMES": "PAJAIS", "PAJAIS": "REPORT"}
 
+def get_phase_html(current_phase):
+    mapping = {"INGEST": "a1", "CODES": "a2", "THEMES": "a3", "PAJAIS": "a3", "REPORT": "a4"}
+    active = mapping.get(current_phase, "a1")
+    return PRG_TPL.format(**{k: "active" if k == active else "" for k in ["a1", "a2", "a3", "a4"]})
+
 def init_analysis(query, csv_file, chat_hist):
     # 1. Clear old outputs
     if os.path.exists(OUT_DIR):
@@ -91,7 +99,7 @@ def init_analysis(query, csv_file, chat_hist):
     feedback = (f"Detected Mode: {mode}\nSources:\n{source_lines}\nTotal: {total_papers} papers\n\n👉 Preparing dataset... Proceeding to Phase 2")
     
     new_hist = chat_hist + [{"role":"user","content": f"Init {mode}"}, {"role":"assistant","content": feedback + "\n\n" + res}]
-    return "", new_hist, _load_table("abstract", "CODES")
+    return "", new_hist, _load_table("abstract", "CODES"), get_phase_html("CODES")
 
 def handle_pdf_upload(files):
     if not files: return "No files selected."
@@ -153,12 +161,13 @@ def _show_chart(m, t):
     path = os.path.join(OUT_DIR, f"{m}_{t}.html")
     return f'<iframe srcdoc="{open(path, encoding="utf-8").read().replace(chr(34),"&quot;")}" width="100%" height="450"></iframe>' if os.path.exists(path) else "Run Phase 2 first."
 
-with gr.Blocks(theme=gr.themes.Base(primary_hue="violet", neutral_hue="slate"), css=CSS) as demo:
+with gr.Blocks() as demo:
     with gr.Row():
         # LEFT SIDEBAR
         with gr.Column(scale=1, elem_classes="sidebar"):
             gr.HTML(HDR)
-            gr.HTML('<div class="sub-hdr">Query / Source</div>')
+            prg_display = gr.HTML(get_phase_html("INGEST"))
+            gr.HTML('<div style="height: 24px; border-top: 1px solid #f1f5f9; margin-top: 24px;"></div>')
             input_box = gr.Textbox(placeholder="Enter research topic...", label="Research Query")
             start_btn = gr.Button("Execute Discovery", variant="primary")
             
@@ -178,8 +187,6 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="violet", neutral_hue="slate"),
             
         # RIGHT MAIN WORKSPACE
         with gr.Column(scale=3, elem_classes="main-content"):
-            gr.HTML(PRG)
-            
             with gr.Tabs(elem_classes="tab-nav"):
                 with gr.Tab("💬 Agent Copilot"):
                     chatbot = gr.Chatbot([], height=500, label="Research Assistant")
@@ -215,7 +222,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="violet", neutral_hue="slate"),
                     btn_refresh_files.click(_get_files, [], [dl])
 
     # Event Bindings
-    start_btn.click(init_analysis, [input_box, csv_up, chatbot], [input_box, chatbot, tbl])
+    start_btn.click(init_analysis, [input_box, csv_up, chatbot], [input_box, chatbot, tbl, prg_display])
     send_btn.click(handle_chat, [msg_in, chatbot, mode_btn, ph_state], [msg_in, chatbot, tbl])
     btn_ingest.click(handle_chat, [gr.State("Ingest all uploaded PDFs and Word documents into the system."), chatbot, gr.State("abstract"), gr.State("CODES")], [msg_in, chatbot, tbl])
     btn_scratch.click(handle_chat, [gr.State("Check the local scratch folder and import any papers found there."), chatbot, gr.State("abstract"), gr.State("CODES")], [msg_in, chatbot, tbl])
@@ -223,5 +230,4 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="violet", neutral_hue="slate"),
     sub_btn.click(submit_review, [tbl, chatbot, mode_btn, ph_state], [chatbot, ph_state, tbl]).then(_get_report, [], [report_box])
 
 if __name__ == "__main__":
-    server_port = int(os.getenv("PORT", 7860))
-    demo.launch(server_name="0.0.0.0", server_port=server_port)
+    demo.launch(server_port=7861, theme=gr.themes.Base(primary_hue="violet", neutral_hue="slate"), css=CSS)