Update main.py

main.py

@@ -16,12 +16,11 @@ class RoadmapInput(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
     class_std: str
-    months_of_study: int
     completed_phy_chapters: List[str]
     completed_chem_chapters: List[str]
     completed_maths_chapters: List[str]
@@ -31,7 +30,6 @@ class RoadmapInput(BaseModel):
     userleft_topic_phy: List[dict]
     userleft_topic_chem: List[dict]
     userleft_topic_maths: List[dict]
-    total_time: int
 
 output_structure = """
 {
@@ -118,11 +116,11 @@ async def generate_roadmap(input_data: RoadmapInput):
         phy = remove_completed_chapters(phy, input_data.completed_phy_chapters)
         chem = remove_completed_chapters(chem, input_data.completed_chem_chapters)
         maths = remove_completed_chapters(maths, input_data.completed_maths_chapters)
-       
+       total_time=10;
         # Get topics within timeframe
-        physics_topics = get_topics_within_timeframe(phy, input_data.total_time, input_data.startphyRoadmapFrom)
-        chemistry_topics = get_topics_within_timeframe(chem, input_data.total_time, input_data.startchemRoadmapFrom)
-        mathematics_topics = get_topics_within_timeframe(maths, input_data.total_time, input_data.startmathRoadmapFrom)
+        physics_topics = get_topics_within_timeframe(phy, total_time, input_data.startphyRoadmapFrom)
+        chemistry_topics = get_topics_within_timeframe(chem, total_time, input_data.startchemRoadmapFrom)
+        mathematics_topics = get_topics_within_timeframe(maths, total_time, input_data.startmathRoadmapFrom)
      
         for topic in reversed(input_data.userleft_topic_phy):  
             physics_topics.insert(0, topic)