Update core/ai_engine.py

core/ai_engine.py

@@ -54,6 +54,8 @@ TASK_CAPTURE_PROMPT = """You are a task classification assistant for a productiv
 
 Take the user's raw task description and return a structured JSON object.
 
+TODAY'S DATE: {today}
+
 Dimensions:
 1. title (string): Clean, action-oriented task title.
 2. life_area (string): ONE of: Work, Health, Learning, Finance, Personal, Family, Other
@@ -61,9 +63,13 @@ Dimensions:
 4. importance (string): ONE of: Move the Needle | Important | Not Important
 5. state_of_mind (string): ONE of: Flow | Easy | Quick | Personal
 6. time_estimate (integer): Realistic minutes to complete.
-7. clarifications_needed (array): Short specific questions if uncertain about any dimension. Return [] if confident.
+7. deadline_date (string | null): If the user mentions ANY deadline, due date, or "by X" — extract as YYYY-MM-DD.
+   Examples: "by Friday" -> next Friday's date, "due March 1" -> "2026-03-01", "end of month" -> last day of current month.
+   If no deadline mentioned, return null.
+8. clarifications_needed (array): Short specific questions if uncertain about any dimension. Return [] if confident.
 
-RETURN: Valid JSON only. No markdown, no prose. All 7 fields always present."""
+RETURN: Valid JSON only. No markdown, no prose. All 8 fields always present.
+CRITICAL: If user says "by [date]", "due [date]", "before [date]", "deadline [date]" — ALWAYS populate deadline_date."""
 
 
 def parse_task_with_groq(raw_text: str, user_context: dict = None,
@@ -77,10 +83,12 @@ def parse_task_with_groq(raw_text: str, user_context: dict = None,
     if life_areas:
         context_hint += f"\nUser's life areas: {', '.join(life_areas)}"
 
+    today_str = date.today().isoformat()
+    system_prompt = TASK_CAPTURE_PROMPT.replace("{today}", today_str) + context_hint
     response = _groq().chat.completions.create(
         model=GROQ_MODEL,
         messages=[
-            {"role": "system", "content": TASK_CAPTURE_PROMPT + context_hint},
+            {"role": "system", "content": system_prompt},
             {"role": "user",   "content": f"Parse this task: {raw_text}"}
         ],
         max_tokens=512,
@@ -93,6 +101,7 @@ def parse_task_with_groq(raw_text: str, user_context: dict = None,
             "title": raw_text,
             "life_area": None, "urgency": None, "importance": None,
             "state_of_mind": None, "time_estimate": None,
+            "deadline_date": None,
             "clarifications_needed": ["Could you give more details? Which area, how urgent, how long?"]
         }
     return result
@@ -102,8 +111,8 @@ def parse_task_with_groq(raw_text: str, user_context: dict = None,
 
 SMART_SCHEDULER_PROMPT = """You are an intelligent task scheduler for Second Brain.
 
-You assign UNSCHEDULED TASKS to specific future dates, like a smart personal assistant who understands the
-user's rhythms, goals, energy patterns, and the current time.
+You assign UNSCHEDULED TASKS to specific future dates AND time blocks, like a smart personal assistant
+who understands the user's rhythms, goals, energy patterns, and current time.
 
 REASONING PROCESS:
 1. Read the user request carefully — honour it above all else
@@ -114,12 +123,14 @@ REASONING PROCESS:
 2. Never assign to a date/time in the past (current datetime is given)
 3. If current time is past 18:00, treat today as unavailable unless user explicitly asks
 4. Prioritise by: deadline proximity > urgency > importance > goal alignment
-5. Match tasks to days by state_of_mind:
-   - Flow = peak days (Mon-Thu mornings if peak=Morning)
-   - Quick/Easy = any day, fill gaps
-   - Habit = today or tomorrow
-6. Spread load — don't stack everything on one day
-7. Tasks with deadlines must land BEFORE that deadline
+5. Match tasks to time blocks by state_of_mind:
+   - Flow = peak focus window (morning if peak=Morning), 60-90 min blocks
+   - Easy = mid-morning or early afternoon, 20-45 min
+   - Quick = any gap, 10-20 min
+   - Personal = end of day or lunch, flexible
+6. Respect wake/sleep times. Add 10min breaks between tasks. Don't exceed max_flow_block.
+7. Spread load sensibly — don't overstack one day
+8. Tasks with deadlines must land BEFORE that deadline
 
 RETURN FORMAT (JSON only, no markdown):
 {
@@ -128,7 +139,9 @@ RETURN FORMAT (JSON only, no markdown):
       "task_id": 123,
       "title": "Task title",
       "assigned_date": "YYYY-MM-DD",
-      "reasoning": "1 sentence why this date"
+      "start_time": "09:00",
+      "end_time": "10:30",
+      "reasoning": "1 sentence why this slot"
     }
   ],
   "skipped": [
@@ -138,10 +151,11 @@ RETURN FORMAT (JSON only, no markdown):
       "reason": "why not assigned"
     }
   ],
-  "summary": "2-3 sentence plain-English explanation of what was scheduled and why"
+  "summary": "2-3 sentence plain-English explanation of what was scheduled and why",
+  "warnings": ["any overload or conflict warnings"]
 }
 
-NEVER assign to a past date. NEVER ignore an explicit user instruction about when to schedule."""
+NEVER assign to a past date/time. NEVER ignore explicit user scheduling instructions."""
 
 
 def smart_schedule_tasks(