Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 import pandas as pd
+import datetime
 
-# 1. Aligned Data Structure (Reflecting your specific sources)
+# 1. DATA STRUCTURES
 subtask_map = {
     "Security": ["Industry/peers","Find/Inquiries about Stock information", "Presentation/Stakeholders","Manage/Exploit exisiting customers", "Educate/Market on security services"],
     "Leasing": ["Legal/Credit/Accounting paperwork/policy","Manage existing customers", "Proposal/Marketing report","Financial analysis", "Credit-Nhóm nợ vay", "Interview-Asset checkup-Quantitative", "Understand/Market/Educate leasing/financial services"],
@@ -14,75 +15,90 @@ subtask_map = {
     "Interview": ["Applicant-Checkup CV", "Root-cause analyze", "STAR method for interview"]
 }
 
-author = {
-    "Accounting": ["NXB Kinh Te"],
-    "Psychology": ["Malcom Gladwell","Robert Greene", "Dr.Nguyen", "Ms. Bui Tran Phuong"],
-    "Learning/Cognitive": ["Barbara Oakley"],
-    "ADHD": ["Dr Tracy", "Psychology Nation"],
-    "Technology/Data": ["SQL star", "Mysterious"],
-    "Entrepreneur": ["Brian Tracy"]
-}
+# Persistent session storage
+session_records = []
+
+# 2. LOGIC FUNCTIONS
 def update_subtasks(cat):
-    return gr.Dropdown(choices=subtask_map.get(cat, []), value=subtask_map.get(cat, [])[0] if subtask_map.get(cat) else None)
-def update_author(cat):
-    return gr.Dropdown(choices=author.get(cat, []), value=author.get(cat, [])[0] if author.get(cat) else None)
-def add_to_table(cat, task, current_data):
-    new_row = {"Date": "28/01", "Category": cat, "Task": task, "Subtask": "Subtask"}
-    df = pd.concat([current_data, pd.DataFrame([new_row])], ignore_index=True)
-    return df
-def save_a3_analysis(bg, curr, target, root, confirm, follow):
-    new_row = {"Job Role": role,"Background": bg, "Current Condition": curr, "Target": target, "Root Cause": root, "Confirmation": confirm, "Follow-up": follow
-}
-    df = pd.concat([current_data, pd.DataFrame([new_row])], ignore_index=True)
-    return df
+    choices = subtask_map.get(cat, [])
+    return gr.Dropdown(choices=choices, value=choices[0] if choices else None)
+
+def save_and_update_table(cat, task, bg, curr, target, root, confirm, follow):
+    global session_records
+    
+    new_row = {
+        "Date": datetime.date.today().strftime("%d/%m"),
+        "Category": cat,
+        "Task": task,
+        "Subtask": "Deep Work",
+        "Background": bg,
+        "Current condition": curr,
+        "Goal/Target condition": target,
+        "Root Cause Analysis": root,
+        "Confirmation": confirm,
+        "Follow-up": follow
+    }
+    
+    session_records.append(new_row)
+    df = pd.DataFrame(session_records)
+    
+    # Save to Excel
+    file_path = "Job_Strategy_A3.xlsx"
+    df.to_excel(file_path, index=False)
+    
+    # Update the Vault Text
+    vault_entry = f"--- {new_row['Date']} | {cat}: {task} ---\nROOT: {root}\nFOLLOW-UP: {follow}\n\n"
     
-# 3. Building the Dashboard
+    return df, file_path, vault_entry
+
+# 3. BUILDING THE DASHBOARD
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🍅 Track your Knowledge & Reflection")
 
     with gr.Row():
-        # LEFT: Task Input
+        # --- LEFT PANEL: INPUTS ---
         with gr.Column(scale=1):
             category = gr.Radio(choices=list(subtask_map.keys()), label="1. Category", value="Accountancy")
             task_select = gr.Dropdown(choices=subtask_map["Accountancy"], label="2. Task")
-            add_btn = gr.Button("➕ Add to Timeline", variant="secondary")
-
+            
             gr.Markdown("---")
+            gr.Markdown("### 🛠 A3 Deep Analysis")
             bg = gr.Textbox(label="Background", placeholder="Bối cảnh vấn đề...")
             curr = gr.Textbox(label="Current Condition", placeholder="Tình trạng hiện tại...")
-            target = gr.Textbox(label="Goal / Target Condition", placeholder="Mục tiêu sau khi giải quyết...")
+            target = gr.Textbox(label="Goal / Target Condition", placeholder="Mục tiêu...")
+            
+            with gr.Accordion("Deep Root Cause", open=False):
+                root = gr.Textbox(label="Root Cause Analysis", placeholder="5 Whys...")
+                confirm = gr.Textbox(label="Confirmation")
+                follow = gr.Textbox(label="Follow-up")
 
-            with gr.Column(): 
-                root = gr.Textbox(label="Root Cause Analysis", placeholder="Nguyên nhân cốt lõi (5 Whys)...")
-                confirm = gr.Textbox(label="Confirmation", placeholder="Đánh giá kết quả thực tế...")
-                follow = gr.Textbox(label="Follow-up", placeholder="Hành động tiếp theo/Phòng ngừa...")
+            save_reflect = gr.Button("💾 Save Reflection & Log", variant="primary")
 
-            save_reflect = gr.Button("💾 Save Reflection", variant="primary") # Added the missing button definition
-            
-    df = pd.DataFrame(session_records)
-    
-    file_path = "Job_Strategy_A3.xlsx"
-    df.to_excel(file_path, index=False)
-    
-    return df, file_path  
-        # RIGHT: Visualization
-        with gr.Column(scale=2): 
+        # --- RIGHT PANEL: VISUALIZATION ---
+        with gr.Column(scale=2):
             gr.Markdown("### 📅 Study Log")
             task_table = gr.DataFrame(
                 headers=["Date", "Category", "Task", "Subtask", "Background", "Current condition", "Goal/Target condition", "Root Cause Analysis", "Confirmation", "Follow-up"],
-                datatype="str",  
+                datatype="str",
                 wrap=True,
                 interactive=True
             )
             
+            excel_file = gr.File(label="Download Excel report")
+            
             with gr.Accordion("View Knowledge Vault", open=False):
                 gr.Markdown("Your reflections will be archived here for 'Deep Thinking' sessions.")
                 vault_display = gr.Textbox(label="History Log", interactive=False, lines=10)
-    
-    # Save Reflection Logic
-    def save_log(bg_val, curr_val, target_val, root_val, confirm_val, follow_val):
-        return f"BACKGROUND: {bg_val}\nCURRENT: {curr_val}\nTARGET: {target_val}\nROOT CAUSE: {root_val}\nCONFIRMATION: {confirm_val}\nFOLLOW-UP: {follow_val}\n---"
 
-    save_reflect.click(save_log, inputs=[bg, curr, target, root, confirm, follow], outputs=vault_display)
+    # 4. EVENT LISTENERS
+    # Dynamic Dropdown
+    category.change(update_subtasks, inputs=[category], outputs=[task_select])
+    
+    # Save Logic
+    save_reflect.click(
+        fn=save_and_update_table,
+        inputs=[category, task_select, bg, curr, target, root, confirm, follow],
+        outputs=[task_table, excel_file, vault_display]
+    )
 
-demo.launch(share=True)
+demo.launch()