Update planner.py

planner.py

@@ -1,52 +1,75 @@
 # planner.py
 import os
+import re
 from huggingface_hub import InferenceClient
 
-# Get HF token from environment (set as Secret in HF Spaces)
 HF_TOKEN = os.getenv("HF_TOKEN")
+client = InferenceClient(model="Qwen/Qwen2.5-Coder-7B-Instruct", token=HF_TOKEN)
 
-# Use the SAME model as your working chatbot
-client = InferenceClient(
-    model="Qwen/Qwen2.5-Coder-7B-Instruct",
-    token=HF_TOKEN
-)
+def extract_total_days(goal: str) -> int:
+    """Extract number of days from goal like 'in 3 weeks' → 21"""
+    goal_lower = goal.lower()
+    # Match "X days", "X weeks", etc.
+    if m := re.search(r'(\d+)\s*days?', goal_lower):
+        return int(m.group(1))
+    if m := re.search(r'(\d+)\s*weeks?', goal_lower):
+        return int(m.group(1)) * 7
+    # Default to 14 if unclear
+    return 14
 
 def generate_task_plan(goal: str) -> str:
-    """
-    Uses Qwen2.5-Coder (instruct model) via chat API to break down a goal into tasks.
-    """
-    messages = [
-        {
-            "role": "user",
-            "content": (
-                "You are a strict project planner. Break down the following goal into actionable tasks.\n\n"
-                "RULES:\n"
-                "- The project must be COMPLETE within the timeframe stated in the goal.\n"
-                "- If the goal says 'in 2 weeks', you have EXACTLY 14 days — NO task can be due after Day 14.\n"
-                "- All deadlines must be between Day 1 and Day 14 (inclusive).\n"
-                "- Do NOT invent extra days. If the goal says 'in X days/weeks', convert it to total days and respect it.\n"
-                "- Number tasks clearly (1., 2., 3., ...).\n"
-                "- For each task, include: 'Due: Day N', 'Depends on: ...', and a short description.\n"
-                "- Keep tasks realistic and parallelizable where possible.\n"
-                "- NEVER output a task with 'Due: Day 15' or beyond if the limit is 14 days.\n\n"
-                "Example (for a 5-day goal):\n"
-                "1. Draft requirements - Due: Day 1 - Depends on: None\n"
-                "   Description: ...\n\n"
-                f"GOAL TO PLAN: \"{goal}\"\n\n"
-                "Now generate the plan. STRICTLY FOLLOW THE TIME LIMIT."
-            )
-        }
-    ]
+    total_days = extract_total_days(goal)
+    
+    prompt = f"""You are a professional project manager. Break down the goal into a STRICT task list.
+
+GOAL: "{goal}"
+TOTAL DAYS AVAILABLE: {total_days}
+
+RULES (VIOLATING ANY = FAILURE):
+1. Output ONLY a numbered list of tasks (1., 2., 3., ...). NO introduction, NO summary, NO extra text.
+2. Every task must have:
+   - Task name
+   - "Due: Day N" where 1 ≤ N ≤ {total_days}
+   - "Depends on: [Task # or 'None']"
+   - "Description: [one-sentence explanation]"
+3. Distribute work across all {total_days} days — do NOT cram everything into early days.
+4. Allow parallel work (e.g., marketing can start while dev is ongoing).
+5. Final launch and post-launch must happen ON or BEFORE Day {total_days}.
+6. NEVER say "Day 14" if total days is {total_days}. Use the correct max day.
+
+FORMAT EXAMPLE (for 5-day goal):
+1. Draft requirements - Due: Day 1 - Depends on: None
+   Description: Define core features and user stories.
+2. Design UI - Due: Day 3 - Depends on: Task 1
+   Description: Create mockups for main screens.
+3. Develop backend - Due: Day 4 - Depends on: Task 1
+   Description: Build API and database.
+4. Launch - Due: Day 5 - Depends on: Tasks 2 and 3
+   Description: Deploy and announce product.
+
+NOW GENERATE THE PLAN FOR {total_days} DAYS. OUTPUT ONLY THE NUMBERED LIST:"""
 
     try:
         response = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-7B-Instruct",
-            messages=messages,
+            messages=[{"role": "user", "content": prompt}],
             max_tokens=2048,
-            temperature=0.3,
-            top_p=0.95,
+            temperature=0.2,  # Very low for precision
+            top_p=0.9,
             stream=False
         )
-        return response.choices[0].message.content.strip()
+        raw = response.choices[0].message.content.strip()
+
+        # Optional: Remove any leading/trailing prose (safety net)
+        lines = raw.split('\n')
+        cleaned = []
+        for line in lines:
+            if line.strip() and (line.strip()[0].isdigit() or line.strip().startswith(('1.', '2.', '3.', '4.', '5.', '6.', '7.', '8.', '9.', '10.', '11.', '12.', '13.', '14.', '15.', '16.', '17.', '18.', '19.', '20.', '21.', '22.', '23.', '24.', '25.'))):
+                cleaned.append(line)
+            elif cleaned:  # Stop if we've started and hit non-task text
+                break
+
+        return '\n'.join(cleaned) if cleaned else raw
+
     except Exception as e:
-        raise Exception(f"LLM inference failed: {str(e)}")
+        raise Exception(f"Planning failed: {str(e)}")