Updated Agent-4

app.py

@@ -22,6 +22,7 @@ from fastapi.responses import HTMLResponse
 # Session State replacement
 session_state = {
     "data": None,
+    "test_results": None,
     "full_roadmap": None,
     "report_data": None,
     "final_report": None,
@@ -43,6 +44,7 @@ async def lifespan(app: FastAPI):
             session_state["dependencies"] = json.load(file)
         # Process tasks as incomplete
         process_task_data()
+        load_ag4_data()
         print("✅ Roadmaps loaded successfully.")
     except Exception as e:
         print(f"❌ Error loading roadmaps: {e}")
@@ -1629,134 +1631,135 @@ def agent3(query: str = Query(..., description="User's message to the chatbot"))
         }}
     
 # --- AGENT 4: Accountability and Analysis Agent
-df = pd.DataFrame([
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Concept Understanding",
-        "Question Complexity": "Easy",
-        "Question": "The vector projection of a vector on y-axis is",
-        "Options Given": "(a) 5 (b) 4 (c) 3 (d) Zero",
-        "Correct Option": "d",
-        "Selected Option": "d",
-        "Correct?": True,
-        "ELO Score": 1200
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Concept Understanding",
-        "Question Complexity": "Easy",
-        "Question": "Position of a particle in a rectangular coordinate system is (3, 2, 5). Then its position vector will be",
-        "Options Given": "(a) 3i + 2j + 5k (b) 2i + 3j + 5k (c) 5i + 2j + 3k (d) None of these",
-        "Correct Option": "a",
-        "Selected Option": "b",
-        "Correct?": False,
-        "ELO Score": 1170
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Question Practice",
-        "Question Complexity": "Easy",
-        "Question": "If a particle moves from point P (2,3,5) to point Q (3,4,5), its displacement vector is",
-        "Options Given": "(a) i + j (b) i + j + k (c) 2i + 2j (d) None of these",
-        "Correct Option": "a",
-        "Selected Option": "a",
-        "Correct?": True,
-        "ELO Score": 1185
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Concept Understanding",
-        "Question Complexity": "Medium",
-        "Question": "A particle moving eastwards with 5 m/s. In 10 s the velocity changes to 5 m/s northwards. The average acceleration is",
-        "Options Given": "(a) 1/2 m/s² NE (b) 1/2 m/s² North (c) 1/2 m/s² NW (d) Zero",
-        "Correct Option": "a",
-        "Selected Option": "a",
-        "Correct?": True,
-        "ELO Score": 1225
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Question Practice",
-        "Question Complexity": "Medium",
-        "Question": "A river is flowing west to east with 5 m/min. A man can swim in still water at 10 m/min. Which direction should he swim to reach the south bank in shortest path?",
-        "Options Given": "(a) 30° E of S (b) 60° E of N (c) South (d) 30° W of N",
-        "Correct Option": "c",
-        "Selected Option": "a",
-        "Correct?": False,
-        "ELO Score": 1190
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Revision",
-        "Question Complexity": "Medium",
-        "Question": "Find a vector perpendicular to A = 2i + 3j in the same plane.",
-        "Options Given": "(a) -3i + 2j (b) 3i - 2j (c) i + j (d) None",
-        "Correct Option": "a",
-        "Selected Option": "a",
-        "Correct?": True,
-        "ELO Score": 1230
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Revision",
-        "Question Complexity": "Medium",
-        "Question": "Two vectors A and B have same magnitude. If their resultant is perpendicular to A, what is the angle between A and B?",
-        "Options Given": "(a) 60° (b) 120° (c) 135° (d) None",
-        "Correct Option": "b",
-        "Selected Option": "b",
-        "Correct?": True,
-        "ELO Score": 1260
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Test",
-        "Question Complexity": "Hard",
-        "Question": "A man walks 500 m and turns by 60° five times. What is displacement after 5th turn?",
-        "Options Given": "(a) 500 m (b) 1000 m (c) 500√3 m (d) None",
-        "Correct Option": "a",
-        "Selected Option": "c",
-        "Correct?": False,
-        "ELO Score": 1210
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Test",
-        "Question Complexity": "Hard",
-        "Question": "Rain is falling vertically at 3 m/s and a man moves north at 4 m/s. Direction to hold umbrella?",
-        "Options Given": "(a) 37° N of vertical (b) 37° S of vertical (c) 53° N of vertical (d) 53° S of vertical",
-        "Correct Option": "c",
-        "Selected Option": "c",
-        "Correct?": True,
-        "ELO Score": 1265
-    },
-    {
-        "Subject": "Physics",
-        "Chapter": "Vectors",
-        "Question Tag": "Test",
-        "Question Complexity": "Hard",
-        "Question": "Which set of forces cannot be in equilibrium?",
-        "Options Given": "(a) 10N,10N,5N (b) 5N,7N,9N (c) 8N,4N,13N (d) 9N,6N,5N",
-        "Correct Option": "c",
-        "Selected Option": "c",
-        "Correct?": True,
-        "ELO Score": 1310
-    }
-])
-df.sort_values(by=['Correct?'], ascending=False, inplace=True)
+def load_ag4_data():
+    df = pd.DataFrame([
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Concept Understanding",
+            "Question Complexity": "Easy",
+            "Question": "The vector projection of a vector on y-axis is",
+            "Options Given": "(a) 5 (b) 4 (c) 3 (d) Zero",
+            "Correct Option": "d",
+            "Selected Option": "d",
+            "Correct?": True,
+            "ELO Score": 1200
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Concept Understanding",
+            "Question Complexity": "Easy",
+            "Question": "Position of a particle in a rectangular coordinate system is (3, 2, 5). Then its position vector will be",
+            "Options Given": "(a) 3i + 2j + 5k (b) 2i + 3j + 5k (c) 5i + 2j + 3k (d) None of these",
+            "Correct Option": "a",
+            "Selected Option": "b",
+            "Correct?": False,
+            "ELO Score": 1170
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Question Practice",
+            "Question Complexity": "Easy",
+            "Question": "If a particle moves from point P (2,3,5) to point Q (3,4,5), its displacement vector is",
+            "Options Given": "(a) i + j (b) i + j + k (c) 2i + 2j (d) None of these",
+            "Correct Option": "a",
+            "Selected Option": "a",
+            "Correct?": True,
+            "ELO Score": 1185
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Concept Understanding",
+            "Question Complexity": "Medium",
+            "Question": "A particle moving eastwards with 5 m/s. In 10 s the velocity changes to 5 m/s northwards. The average acceleration is",
+            "Options Given": "(a) 1/2 m/s² NE (b) 1/2 m/s² North (c) 1/2 m/s² NW (d) Zero",
+            "Correct Option": "a",
+            "Selected Option": "a",
+            "Correct?": True,
+            "ELO Score": 1225
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Question Practice",
+            "Question Complexity": "Medium",
+            "Question": "A river is flowing west to east with 5 m/min. A man can swim in still water at 10 m/min. Which direction should he swim to reach the south bank in shortest path?",
+            "Options Given": "(a) 30° E of S (b) 60° E of N (c) South (d) 30° W of N",
+            "Correct Option": "c",
+            "Selected Option": "a",
+            "Correct?": False,
+            "ELO Score": 1190
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Revision",
+            "Question Complexity": "Medium",
+            "Question": "Find a vector perpendicular to A = 2i + 3j in the same plane.",
+            "Options Given": "(a) -3i + 2j (b) 3i - 2j (c) i + j (d) None",
+            "Correct Option": "a",
+            "Selected Option": "a",
+            "Correct?": True,
+            "ELO Score": 1230
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Revision",
+            "Question Complexity": "Medium",
+            "Question": "Two vectors A and B have same magnitude. If their resultant is perpendicular to A, what is the angle between A and B?",
+            "Options Given": "(a) 60° (b) 120° (c) 135° (d) None",
+            "Correct Option": "b",
+            "Selected Option": "b",
+            "Correct?": True,
+            "ELO Score": 1260
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Test",
+            "Question Complexity": "Hard",
+            "Question": "A man walks 500 m and turns by 60° five times. What is displacement after 5th turn?",
+            "Options Given": "(a) 500 m (b) 1000 m (c) 500√3 m (d) None",
+            "Correct Option": "a",
+            "Selected Option": "c",
+            "Correct?": False,
+            "ELO Score": 1210
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Test",
+            "Question Complexity": "Hard",
+            "Question": "Rain is falling vertically at 3 m/s and a man moves north at 4 m/s. Direction to hold umbrella?",
+            "Options Given": "(a) 37° N of vertical (b) 37° S of vertical (c) 53° N of vertical (d) 53° S of vertical",
+            "Correct Option": "c",
+            "Selected Option": "c",
+            "Correct?": True,
+            "ELO Score": 1265
+        },
+        {
+            "Subject": "Physics",
+            "Chapter": "Vectors",
+            "Question Tag": "Test",
+            "Question Complexity": "Hard",
+            "Question": "Which set of forces cannot be in equilibrium?",
+            "Options Given": "(a) 10N,10N,5N (b) 5N,7N,9N (c) 8N,4N,13N (d) 9N,6N,5N",
+            "Correct Option": "c",
+            "Selected Option": "c",
+            "Correct?": True,
+            "ELO Score": 1310
+        }
+    ])
+    df.sort_values(by=['Correct?'], ascending=False, inplace=True)
+    session_state["test_results"] = df
 
 @app.get("/agent4")
 def agent4(
-    df: pd.DataFrame = Query(df, description="Dataframe of the Results of the students test"),
-    current_dayNumber: int = Query(2, description="Today's day number for scheduling tasks"),
+    current_dayNumber: int = Query(2, description="Today's day number for scheduling tasks")
 ):
     """
     Agent 4 - Accountability and Analysis Agent.
@@ -1765,7 +1768,7 @@ def agent4(
         before = session_state['updated_roadmap']
     else:
         before = session_state['data']
-    ag4_update_roadmap(df, before, session_state["task_analysis"], current_dayNumber)
+    ag4_update_roadmap(session_state["test_results"], before, session_state["task_analysis"], current_dayNumber)
 
     return {
         "original_roadmap": before,