Update app.py

app.py

@@ -29,8 +29,6 @@ Suggestions:
 Quote:
 - Your motivational quote here
 
-Now, generate the checklist, suggestions, and quote for the following inputs:
-
 Inputs:
 Role: {role}
 Project: {project_id}
@@ -76,41 +74,38 @@ def generate_outputs(role, project_id, milestones, reflection):
     )
     
     # Generate with attention_mask
-    outputs = model.generate(
-        inputs["input_ids"],
-        attention_mask=inputs["attention_mask"],
-        max_length=1500,
-        num_return_sequences=1,
-        no_repeat_ngram_size=2,
-        do_sample=True,
-        top_p=0.9,
-        temperature=0.8,
-        pad_token_id=tokenizer.eos_token_id
-    )
+    with torch.no_grad():
+        outputs = model.generate(
+            inputs["input_ids"],
+            attention_mask=inputs["attention_mask"],
+            max_length=1000,
+            num_return_sequences=1,
+            no_repeat_ngram_size=2,
+            do_sample=True,
+            top_p=0.9,
+            temperature=0.8,
+            pad_token_id=tokenizer.eos_token_id
+        )
     
     # Decode generated text
     generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    # Parse the output using labels
+    # Parse the output
     checklist = "No checklist generated."
     suggestions = "No suggestions generated."
     quote = "No quote generated."
     
-    # Look for sections using labels
+    # Extract sections using labels
     if "Checklist:" in generated_text:
         checklist_start = generated_text.find("Checklist:") + len("Checklist:")
-        suggestions_start = generated_text.find("Suggestions:")
-        if suggestions_start == -1:
-            suggestions_start = len(generated_text)
+        suggestions_start = generated_text.find("Suggestions:") if "Suggestions:" in generated_text else len(generated_text)
         checklist = generated_text[checklist_start:suggestions_start].strip()
         if not checklist:
             checklist = "No checklist generated."
     
     if "Suggestions:" in generated_text:
         suggestions_start = generated_text.find("Suggestions:") + len("Suggestions:")
-        quote_start = generated_text.find("Quote:")
-        if quote_start == -1:
-            quote_start = len(generated_text)
+        quote_start = generated_text.find("Quote:") if "Quote:" in generated_text else len(generated_text)
         suggestions = generated_text[suggestions_start:quote_start].strip()
         if not suggestions:
             suggestions = "No suggestions generated."
@@ -126,36 +121,40 @@ def generate_outputs(role, project_id, milestones, reflection):
         checklist_items = []
         milestone_list = [m.strip() for m in milestones.split(",")]
         for i, milestone in enumerate(milestone_list, 1):
-            # Assign times based on sample patterns (e.g., 8 AM, 10 AM, etc.)
-            time = 8 + (i-1)*2 if i <= 3 else 4  # PM for last item
+            time = 8 + (i-1)*2 if i <= 3 else 4  # Follow sample timing (8 AM, 10 AM, 12 PM, 4 PM)
             period = "AM" if i <= 3 else "PM"
             checklist_items.append(f"- {milestone} by {time} {period}.")
-        # Add a generic task based on role or reflection
+        # Add context-specific task
         if "safety" in reflection.lower():
-            checklist_items.append("- Review safety compliance by end of day.")
+            checklist_items.append("- Review safety compliance by 4 PM.")
         else:
-            checklist_items.append("- Check equipment status by end of day.")
+            checklist_items.append("- Check equipment status by 4 PM.")
         checklist = "\n".join(checklist_items)
     
     if suggestions == "No suggestions generated.":
         suggestions_items = []
-        if "safety" in reflection.lower():
-            suggestions_items.append("- Address safety concerns with a team briefing.")
-        if "delay" in reflection.lower() or "late" in reflection.lower():
+        reflection_lower = reflection.lower()
+        if "safety" in reflection_lower:
+            suggestions_items.append("- Address safety concerns with a team briefing on proper procedures.")
+        if "delay" in reflection_lower or "late" in reflection_lower:
             suggestions_items.append("- Follow up with suppliers to prevent future delays.")
-        if "weather" in reflection.lower():
+        if "weather" in reflection_lower:
             suggestions_items.append("- Monitor weather updates and plan contingencies.")
-        # Add a generic suggestion
+        if "equipment" in reflection_lower:
+            suggestions_items.append("- Schedule equipment maintenance to avoid future issues.")
+        if "suppliers" in reflection_lower:
+            suggestions_items.append("- Set up a morning call with suppliers to confirm timelines.")
+        # Add generic suggestion
         suggestions_items.append("- Brief the team on tomorrow’s goals during the daily huddle.")
         suggestions = "\n".join(suggestions_items if suggestions_items else ["- Coordinate with the team.", "- Plan for contingencies."])
     
     if quote == "No quote generated.":
-        # Select a quote based on reflection context
-        if "safety" in reflection.lower():
+        reflection_lower = reflection.lower()
+        if "safety" in reflection_lower:
             quote = "- Build with care—safety today ensures success tomorrow!"
-        elif "delay" in reflection.lower():
+        elif "delay" in reflection_lower or "late" in reflection_lower:
             quote = "- Keep moving forward—every challenge is a step toward success!"
-        elif "weather" in reflection.lower():
+        elif "weather" in reflection_lower:
             quote = "- Steady progress leads to great achievements, no matter the weather!"
         else:
             quote = "- Success is built one solid step at a time!"