Update app.py

app.py

@@ -148,7 +148,8 @@ def handle_parse_text(task_text, user_id):
     """Call Groq to classify a typed task, return pre-filled fields."""
     if not task_text.strip():
         return (gr.update(visible=False),
-                "", "", "Work", "Not Urgent", "Important", "Easy", 30, str(date.today()), False, "Daily", "")
+                "", "", "Work", "Not Urgent", "Important", "Easy", 30, str(date.today()), False, "Daily", "",
+                gr.update(visible=False), [], "")
 
     ctx   = _ensure_context(user_id)
     goals = get_goals(user_id)
@@ -158,6 +159,7 @@ def handle_parse_text(task_text, user_id):
 
     clarifications = result.get("clarifications_needed", [])
     clar_html = ""
+    has_clarifications = bool(clarifications)
     if clarifications:
         items = "".join(f"<li style='margin:4px 0'>{q}</li>" for q in clarifications)
         clar_html = (
@@ -184,7 +186,62 @@ def handle_parse_text(task_text, user_id):
         str(date.today()),
         False,
         "Daily",
-        _ok("AI classified your task"),
+        _ok("AI classified your task — please answer the clarification questions above." if has_clarifications else "AI classified your task"),
+        gr.update(visible=has_clarifications),  # clar_reply_row
+        clarifications,                          # clar_questions_state
+        task_text,                               # original_task_state
+    )
+
+
+def handle_clarification_reply(user_reply, original_task, clarifications, user_id):
+    """Re-run Groq classification with the original task + clarification Q&A appended."""
+    if not user_reply.strip():
+        return (gr.update(), gr.update(), gr.update(), gr.update(),
+                gr.update(), gr.update(), gr.update(), gr.update(visible=True), "")
+
+    ctx   = _ensure_context(user_id)
+    goals = get_goals(user_id)
+    areas = get_life_area_names(user_id) if user_id else []
+
+    # Build enriched task description: original + each clarification + the user's answer
+    q_block = "\n".join(f"Q: {q}" for q in clarifications)
+    enriched = (
+        f"{original_task}\n\n"
+        f"Additional context from user (answers to clarification questions):\n"
+        f"{q_block}\n"
+        f"A: {user_reply}"
+    )
+
+    result = parse_task_with_groq(enriched, ctx, goals, areas)
+
+    # If AI still has clarifications, show them; otherwise hide the clarification row
+    remaining = result.get("clarifications_needed", [])
+    if remaining:
+        items = "".join(f"<li style='margin:4px 0'>{q}</li>" for q in remaining)
+        clar_html_val = (
+            f'<div style="margin:10px 0;padding:12px;background:#0c0f1a;'
+            f'border-left:3px solid #a78bfa;border-radius:8px">'
+            f'<div style="color:#a78bfa;font-size:11px;font-weight:600;margin-bottom:6px">⚠ Please clarify:</div>'
+            f'<ul style="color:#94a3b8;font-size:13px;margin:0;padding-left:16px">{items}</ul></div>'
+        )
+        clar_row_visible = True
+    else:
+        clar_html_val    = '<span style="color:#4ade80;font-size:13px">✓ Clarified — classification updated!</span>'
+        clar_row_visible = False
+
+    area_choices = areas if areas else ["Work","Health","Finance","Learning","Personal","Family","Other"]
+    area_val     = result.get("life_area") or (area_choices[0] if area_choices else "Work")
+
+    return (
+        result.get("title", original_task),
+        clar_html_val,
+        area_val,
+        result.get("urgency", "Not Urgent"),
+        result.get("importance", "Important"),
+        result.get("state_of_mind", "Easy"),
+        int(result.get("time_estimate") or 30),
+        gr.update(visible=clar_row_visible),
+        remaining,
     )
 
 
@@ -593,6 +650,9 @@ with gr.Blocks(title="🧠 Second Brain") as demo:
     journal_hist_state  = gr.State([])
     journal_tasks_state = gr.State([])
     journal_active      = gr.State(False)
+    # Clarification states
+    clar_questions_state = gr.State([])
+    original_task_state  = gr.State("")
 
     # ═════════════════════════════════════════════════════════════════════════
     # AUTH SECTION
@@ -671,6 +731,15 @@ with gr.Blocks(title="🧠 Second Brain") as demo:
                     with gr.Column(visible=False) as parsed_panel:
                         parse_status = gr.HTML("")
                         clar_html    = gr.HTML("")
+                        # ── Clarification reply row (shown only when AI has questions) ──
+                        with gr.Column(visible=False) as clar_reply_row:
+                            gr.HTML('<div class="sec-label" style="margin-top:8px">Your Answer</div>')
+                            clar_reply_input = gr.Textbox(
+                                label="",
+                                placeholder="e.g. It's for a client deadline, very high priority, should take about 2 hours",
+                                lines=2
+                            )
+                            btn_clar_submit = gr.Button("🔄 Re-classify with my answer", elem_classes="btn-accent")
                         gr.HTML('<div class="sec-label" style="margin-top:12px">Review & Confirm</div>')
                         with gr.Row():
                             f_title = gr.Textbox(label="Title", scale=3)
@@ -908,7 +977,18 @@ with gr.Blocks(title="🧠 Second Brain") as demo:
         handle_parse_text,
         [task_input, user_id_state],
         [parsed_panel, f_title, clar_html, f_area, f_urgency, f_importance,
-         f_state, f_time, f_date, f_is_habit, f_interval, parse_status]
+         f_state, f_time, f_date, f_is_habit, f_interval, parse_status,
+         clar_reply_row, clar_questions_state, original_task_state]
+    )
+
+    btn_clar_submit.click(
+        handle_clarification_reply,
+        [clar_reply_input, original_task_state, clar_questions_state, user_id_state],
+        [f_title, clar_html, f_area, f_urgency, f_importance,
+         f_state, f_time, clar_reply_row, clar_questions_state]
+    ).then(
+        lambda: "",
+        outputs=[clar_reply_input]
     )
 
     btn_transcribe.click(