Spaces:
Sleeping
Sleeping
File size: 2,060 Bytes
ea26998 68f0ffe ea26998 91efcfe ea26998 65dafe3 91efcfe 65dafe3 91efcfe 65dafe3 91efcfe 65dafe3 91efcfe 65dafe3 91efcfe 65dafe3 ea26998 68f0ffe ee9409b 65dafe3 68f0ffe 91efcfe 68f0ffe 91efcfe |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
# planner.py
import os
from huggingface_hub import InferenceClient
HF_TOKEN = os.getenv("HF_TOKEN")
client = InferenceClient(
model="Qwen/Qwen2.5-Coder-7B-Instruct",
token=HF_TOKEN
)
def generate_task_plan(goal: str) -> str:
# 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
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.
Goal: "{goal}"
You have exactly {total_days} days to complete everything. The final task must be done by Day {total_days}.
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]
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:"""
try:
response = client.chat.completions.create(
model="Qwen/Qwen2.5-Coder-7B-Instruct",
messages=[{"role": "user", "content": prompt}],
max_tokens=2048,
temperature=0.3,
stream=False
)
return response.choices[0].message.content.strip()
except Exception as e:
return f"Error: {str(e)}" |