Agent 4

app.py

@@ -503,15 +503,15 @@ def shift_the_roadmap(roadmap, max_hours_per_day, ratio=(80, 20), dependencies=N
     return roadmap, extra_day_tasks
 
 def update_roadmap(current_roadmap, current_dayNumber, max_hours_per_day, dependencies, no_of_revision_days = 2):
-    if current_dayNumber <= 1 or current_dayNumber > len(current_roadmap):
-        return current_roadmap
+    if current_dayNumber <= 1 or current_dayNumber > len(current_roadmap['schedule']):
+        session_state["updated_roadmap"] = current_roadmap
     current_roadmap = copy.deepcopy(current_roadmap)
     day_index = current_dayNumber-2
     test_index = None
 
-    if "supplementary_tasks" in current_roadmap[day_index]:
-        current_roadmap[day_index+1]['supplementary_tasks'] = []
-        for subject in current_roadmap[day_index]["supplementary_tasks"]:
+    if "supplementary_tasks" in current_roadmap['schedule'][day_index]:
+        current_roadmap['schedule'][day_index+1]['supplementary_tasks'] = []
+        for subject in current_roadmap['schedule'][day_index]["supplementary_tasks"]:
             subject_name = subject["name"]
             tasks = subject["tasks"]
 
@@ -523,7 +523,7 @@ def update_roadmap(current_roadmap, current_dayNumber, max_hours_per_day, depend
                 task['rescheduled'] += 1
             # Move incomplete tasks to next day
             if incomplete_tasks:
-                current_roadmap[day_index+1]['supplementary_tasks'].append({
+                current_roadmap['schedule'][day_index+1]['supplementary_tasks'].append({
                                                                             "name":subject_name, 
                                                                             "tasks":incomplete_tasks
                                                                           })