Update planner.py

planner.py

@@ -1,75 +1,54 @@
 # planner.py
 import os
-import re
 from huggingface_hub import InferenceClient
 
 HF_TOKEN = os.getenv("HF_TOKEN")
-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
+client = InferenceClient(
+    model="Qwen/Qwen2.5-Coder-7B-Instruct",
+    token=HF_TOKEN
+)
 
 def generate_task_plan(goal: str) -> str:
-    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:"""
+    """
+    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.
+
+Goal: "{goal}"
+
+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
+
+Output only a numbered list like this:
+
+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.
+...
+
+Now generate the plan for the goal above. Use the full available time wisely.
+"""
 
     try:
         response = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-7B-Instruct",
             messages=[{"role": "user", "content": prompt}],
-            max_tokens=2048,
-            temperature=0.2,  # Very low for precision
-            top_p=0.9,
+            max_tokens=800,
+            temperature=0.4,  # Balanced: clear but not robotic
             stream=False
         )
-        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
-
+        return response.choices[0].message.content.strip()
     except Exception as e:
-        raise Exception(f"Planning failed: {str(e)}")
+        return f"Error: {str(e)}"