Update app.py

app.py

@@ -155,11 +155,47 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
         e = text.find(end, s) if end else len(text)
         return text[s + len(start):e].strip() if s != -1 else ""
 
+    # Extract the checklist and suggestions
     checklist = extract_between(result, "Checklist:\n", "Suggestions:")
     suggestions = extract_between(result, "Suggestions:\n", None)
 
+    # If checklist or suggestions are empty, generate fallback content
+    if not checklist.strip():
+        checklist = generate_fallback_checklist(role, milestones)
+    if not suggestions.strip():
+        suggestions = generate_fallback_suggestions(reflection)
+
     return checklist, suggestions
 
+# Fallback generation for checklist and suggestions
+def generate_fallback_checklist(role, milestones):
+    checklist_items = []
+
+    # If milestones are provided, add them to the checklist directly
+    if milestones and milestones.strip():
+        kpis = [kpi.strip() for kpi in milestones.split(",")]
+        for kpi in kpis:
+            checklist_items.append(f"- Ensure progress on {kpi}")
+    else:
+        checklist_items.append("- Perform daily safety inspection")
+    
+    return "\n".join(checklist_items)
+
+def generate_fallback_suggestions(reflection):
+    suggestions_items = []
+    reflection_lower = reflection.lower()
+    if "student" in reflection_lower or "learning" in reflection_lower:
+        suggestions_items.append("- Ensure students are logging incidents consistently")
+        suggestions_items.append("- Provide guidance on timely incident recording")
+    if "incident" in reflection_lower:
+        suggestions_items.append("- Follow up on reported incidents with corrective actions")
+    
+    if not suggestions_items:
+        suggestions_items.append("- Monitor team coordination")
+        suggestions_items.append("- Review safety protocols with the team")
+    
+    return "\n".join(suggestions_items)
+
 # Interface
 def create_interface():
     roles = get_roles_from_salesforce()