Update planner.py

planner.py

@@ -9,44 +9,54 @@ client = InferenceClient(
 )
 
 def generate_task_plan(goal: str) -> str:
-    """
-    Generate a realistic task plan that respects the timeline in the goal.
-    Example: "Launch a product in 4 weeks" → use up to Day 28.
-    """
-    # Simple, clear prompt — no over-engineering
-    prompt = f"""
-You are an experienced product manager. Break down the following goal into a clear, practical list of tasks.
+    # Simple: tell the model the exact number of days
+    # We'll assume user includes "X weeks" or "X days"
+    # For reliability, we hard-code the logic for common cases
+    goal_lower = goal.lower()
+    total_days = 14  # default
 
-Goal: "{goal}"
+    if "1 week" in goal_lower:
+        total_days = 7
+    elif "2 weeks" in goal_lower:
+        total_days = 14
+    elif "3 weeks" in goal_lower:
+        total_days = 21
+    elif "4 weeks" in goal_lower:
+        total_days = 28
+    elif "5 weeks" in goal_lower:
+        total_days = 35
+    else:
+        # Try to extract number + "days"
+        import re
+        if match := re.search(r'(\d+)\s*days?', goal_lower):
+            total_days = int(match.group(1))
+
+    prompt = f"""You are a senior product manager. Break down this goal into a realistic task plan.
 
-Important:
-- If the goal mentions a timeframe (e.g., "in 4 weeks"), convert it to days (4 weeks = 28 days).
-- All tasks must be completed within that total number of days.
-- Distribute work realistically across the full timeline — don’t cram everything early.
-- Include key phases: planning, design, development, testing, marketing, launch, and post-launch.
-- For each task, specify:
-    • A short name
-    • A due date as "Due: Day N" (N must be ≤ total days)
-    • Dependencies (e.g., "Depends on: Task 3" or "None")
-    • A one-sentence description
+Goal: "{goal}"
+You have exactly {total_days} days to complete everything. The final task must be done by Day {total_days}.
 
-Output only a numbered list like this:
+Instructions:
+- Create 8 to 15 actionable tasks.
+- Spread work across all {total_days} days — don’t finish early.
+- Include: planning, design, development, testing, marketing, launch, and feedback.
+- For each task, write:
+    [Number]. [Task name] - Due: Day [N] - Depends on: [Task numbers or "None"]
+       Description: [One clear sentence]
 
-1. Define product requirements - Due: Day 2 - Depends on: None
-   Description: Finalize core features and target users.
-2. Design user interface - Due: Day 6 - Depends on: Task 1
-   Description: Create wireframes and high-fidelity mockups.
-...
+Important: 
+- NEVER use a day greater than {total_days}.
+- Do NOT add introductions, summaries, or extra text.
+- Output ONLY the numbered list.
 
-Now generate the plan for the goal above. Use the full available time wisely.
-"""
+Now generate the plan:"""
 
     try:
         response = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-7B-Instruct",
             messages=[{"role": "user", "content": prompt}],
-            max_tokens=800,
-            temperature=0.4,  # Balanced: clear but not robotic
+            max_tokens=2048,
+            temperature=0.3,
             stream=False
         )
         return response.choices[0].message.content.strip()