feat: improve task generation logic with line-by-line parsing, refined system prompts, and enhanced state tracking for raw task history

app.py

@@ -111,18 +111,33 @@ async def generate_atomic_task(goal: str, previous_failures: int, history: list
             temperature=0.3,
             stop=["\n", "<extra_id_1>"]
         )
-        content = response['choices'][0]['text'].strip()
+        raw_content = response['choices'][0]['text'].strip()
         
-        # Check if output is empty, repeats prompt headers, or does not represent a task step
-        is_invalid = (
-            not content or 
-            any(header in content for header in ["Goal:", "Completed Tasks:", "Failures:", "Output the NEXT step", "<extra_id"])
-        )
+        # Parse content line-by-line to extract the actual task step
+        content = ""
+        lines = [line.strip() for line in raw_content.split("\n")]
+        for line in lines:
+            if not line:
+                continue
+            has_header = any(
+                header in line for header in [
+                    "Goal:", "Completed Tasks:", "Failures:", "Output the NEXT step", 
+                    "<extra_id", "System", "Assistant", "User"
+                ]
+            )
+            if not has_header:
+                words = line.split()
+                if 1 <= len(words) <= 12:
+                    content = line
+                    break
+        
+        # Check if output is empty
+        is_invalid = not content
         
         if is_invalid:
-            print(f"NEMOTRON OUTPUT INVALID ('{content}'). FALLING BACK TO MINICPM...", flush=True)
+            print(f"NEMOTRON OUTPUT INVALID ('{raw_content}'). FALLING BACK TO MINICPM...", flush=True)
             messages = [
-                {"role": "system", "content": "You are a cognitive pacemaker. Break down goals into extremely tiny, atomic physical actions under 8 words. Return ONLY the single physical next step."},
+                {"role": "system", "content": "You are a cognitive pacemaker. Focus strictly on the physical goal and break it down into an extremely tiny, atomic physical action under 8 words. Ignore mental blocks or feelings of being stuck and output ONLY the single physical action step."},
                 {"role": "user", "content": f"Goal: {goal}\nCompleted Tasks: {history_str}\nFailures: {previous_failures}\nOutput the NEXT step."}
             ]
             fallback_res = await run_blocking(
@@ -153,8 +168,33 @@ async def apply_activation_style(task: str, style="direct") -> str:
         if style == "direct":
             return f"{task.capitalize()}."
             
+        if style == "encouraging":
+            system_prompt = (
+                "You are a cognitive pacemaker. Rewrite the given task into an encouraging, supportive tone.\n"
+                "RULES:\n"
+                "1. Keep it as an action/command the user MUST do now. Start or end with an encouraging phrase like 'You got this!', 'Let's do it!', or 'Go ahead and...'.\n"
+                "2. Do NOT write in the past tense, do NOT congratulate the user, and do NOT treat the task as already completed.\n"
+                "3. Keep it brief (under 12 words) and output ONLY the final rewritten task."
+            )
+        elif style == "calm":
+            system_prompt = (
+                "You are a cognitive pacemaker. Rewrite the given task into a calm, gentle tone.\n"
+                "RULES:\n"
+                "1. Keep it as an action/command. Use gentle prefixes like 'When you are ready, ...' or 'Take your time and ...'.\n"
+                "2. Do NOT write in the past tense and do NOT congratulate the user.\n"
+                "3. Keep it brief (under 12 words) and output ONLY the final rewritten task."
+            )
+        else:
+            system_prompt = (
+                f"You are a cognitive pacemaker. Rewrite the given task into a {style} tone.\n"
+                "RULES:\n"
+                "1. Keep it as an action/command the user MUST do now.\n"
+                "2. Do NOT write in the past tense, do NOT congratulate the user, and do NOT treat the task as already completed.\n"
+                "3. Keep it brief (under 12 words) and output ONLY the final rewritten task."
+            )
+
         messages = [
-            {"role": "system", "content": f"You are a helpful assistant. Rewrite the following task into a {style} tone. Only output the final task text and nothing else."},
+            {"role": "system", "content": system_prompt},
             {"role": "user", "content": f"Task: {task}"}
         ]
         response = await run_blocking(
@@ -196,8 +236,10 @@ async def websocket_endpoint(websocket: WebSocket):
         "goal": "",
         "style": "direct",
         "current_task": "",
+        "current_raw_task": "",
         "too_hard_count": 0,
         "history": [],
+        "raw_history": [],
         "trace": [],
     }
     
@@ -213,8 +255,10 @@ async def websocket_endpoint(websocket: WebSocket):
                 state["goal"] = payload.get("goal")
                 state["style"] = payload.get("style", "direct")
                 state["current_task"] = ""
+                state["current_raw_task"] = ""
                 state["too_hard_count"] = 0
                 state["history"] = []
+                state["raw_history"] = []
                 state["trace"] = [{"event": "start", "goal": state["goal"], "style": state["style"]}]
                 
             elif action == "too_hard":
@@ -229,6 +273,8 @@ async def websocket_endpoint(websocket: WebSocket):
                 state["too_hard_count"] = 0
                 if payload.get("last_task"):
                     state["history"].append(payload.get("last_task"))
+                    raw_task_done = state.get("current_raw_task") or payload.get("last_task")
+                    state["raw_history"].append(raw_task_done)
                     state["trace"].append({"event": "done", "task": payload.get("last_task")})
             
             # HARD CIRCUIT BREAKER
@@ -250,11 +296,12 @@ async def websocket_endpoint(websocket: WebSocket):
             raw_task = await generate_atomic_task(
                 state["goal"],
                 state["too_hard_count"],
-                history=state["history"],
+                history=state["raw_history"],
                 rejected_task=rejected_task,
             )
             styled_task = await apply_activation_style(raw_task, state["style"])
             state["current_task"] = styled_task
+            state["current_raw_task"] = raw_task
             state["trace"].append({
                 "event": "generated_step",
                 "raw_task": raw_task,