Update main.py

main.py

@@ -79,11 +79,11 @@ def get_topics_within_timeframe(subject, allowed_time, start_subtopic):
             continue
 
         for subtopic in chapter["subtopics"]:
-            if subtopic["subtopic"] == start_subtopic:
+            # Start collecting immediately if start_subtopic is empty
+            if start_subtopic == "" or subtopic["subtopic"] == start_subtopic:
                 start_collecting = True
-                continue
 
-            if start_collecting and subtopic["subtopic_hours"]:
+            if start_collecting and subtopic.get("subtopic_hours"):
                 time_spent += float(subtopic["subtopic_hours"])
                 if time_spent <= allowed_time:
                     collected_subtopics.append({
@@ -98,6 +98,7 @@ def get_topics_within_timeframe(subject, allowed_time, start_subtopic):
 
     return collected_subtopics
 
+
 @app.post("/generate_roadmap")
 async def generate_roadmap(input_data: RoadmapInput):
     try: