ai_model.py

# ai_model.py
from transformers import pipeline

# Load a simple model for checklist generation (stub for now)
generator = pipeline("text-generation", model="distilgpt2")

def generate_daily_checklist(role, project_id, milestones, reflection):
    prompt = f"Role: {role}\nProject: {project_id}\nMilestones: {milestones}\nReflection: {reflection}\nChecklist and Tips:"
    output = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
    
    checklist = "1. Review site plan\n2. Ensure safety gear\n3. Coordinate with teams"
    tips = "1. Prioritize safety\n2. Review yesterday's blockers\n3. Communicate schedule clearly"
    
    # Replace with actual model output if desired
    return checklist, tips