Update main.py

main.py

@@ -239,8 +239,8 @@ class StudyInput(BaseModel):
     new_topic_approach: str
     old_topic_approach: str
     topic_ratio: str
-    hours_of_study: int
-    hours_of_study_weekends: int
+    hours_of_study: str
+    hours_of_study_weekends: str
     revision_days: str
     test_days: str
     physicsStartIndex: int
@@ -249,7 +249,6 @@ class StudyInput(BaseModel):
     completed_phy_chapters: List[str]
     completed_chem_chapters: List[str]
     completed_maths_chapters: List[str]
-    total_time: int
 
 # Function to remove completed chapters
 def remove_completed_chapters(subject_data, completed_chapters):
@@ -284,9 +283,6 @@ async def generate_roadmap(study_input: StudyInput):
     phy = get_data_at_index(phy, study_input.physicsStartIndex)
     chem = get_data_at_index(chem, study_input.chemistryStartIndex)
     maths = get_data_at_index(maths, study_input.mathematicsStartIndex)
-    print(phy)
-    print(chem)
-    print(maths)
     # Prepare user persona
     user_persona = f"""
     You are required to generate a highly personalized roadmap for a student studying Physics, Chemistry, and Mathematics for the JEE Main exam.