Update main.py

main.py

@@ -309,34 +309,33 @@ async def generate_roadmap(study_input: StudyInput):
        - The days of the week when you do revision : {study_input.revision_days}
        - The days of the week when you give tests : {study_input.test_days}
     """
-
+    output_structure = """{
+  "schedule": [
+    {
+      "dayNumber": int,
+      "subjects": [
+        {
+          "name": "string",
+          "tasks": [
+            {
+              "type": "string",
+              "topic": "string",
+              "time": "string"
+            }
+          ]
+        }
+      ]
+    }
+  ]
+}
+"""
     # Prepare system prompt
     sys_prompt = f"""
     You are required to generate a highly personalized roadmap for a student studying Physics, Chemistry, and Mathematics for the JEE Main exam.
     The roadmap should be tailored based on the following student-specific details:
 
     The roadmap must be provided in the following format:
-    {{
-      "schedule": [
-              "days": [
-                {{
-                  "dayNumber": int,
-                  "subjects": [
-                    {{
-                      "name": "string",
-                      "tasks": [
-                        {{
-                          "type": "string",
-                          "topic": "string",
-                          "time": "string"
-                        }}
-                      ]
-                    }}
-                  ]
-                }}
-              ]
-      ]
-    }}
+    {output_structure}
 
     Do not include anything other than the roadmap, and ensure the focus remains strictly on the subjects {phy}, {chem}, and {maths} and associated chapters.
     MAKE SURE THAT YOU MAKE THE ROADMAP FOR ALL THE THREE CHAPTERS EACH OF PHYSICS , CHEMISTRY AND MATHS TO COMPLETE THOSE CHAPTERS WITH 4 ASPECTS i.e "CONCEPT UNDERSTANDING","QUESTION PRACTICE","REVISION","TEST". ALSO INCLUDE TIME FOR EACH TASK THAT YOU GENERATE