Update ai_model.py

ai_model.py

@@ -30,7 +30,13 @@ def generate_daily_checklist(role, project_id, milestones, reflection):
     )
     prompt += f"Milestones: {milestones}\nChecklist and Tips:\n"
 
-    output = generator(prompt, max_length=150, num_return_sequences=1)[0]['generated_text']
+    output = generator(
+        prompt,
+        max_new_tokens=100,      # generate up to 100 tokens after prompt
+        num_return_sequences=1,
+        truncation=True          # explicitly truncate prompt if too long
+    )[0]['generated_text']
+
 
     # Parse output based on keywords "Checklist:" and "Tips:"
     checklist = ""