Update main.py

main.py

@@ -52,7 +52,6 @@
 #     except Exception as e:
 #         raise HTTPException(status_code=500, detail=f"Error merging roadmaps: {e}")
 
-
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from typing import List
@@ -67,7 +66,11 @@ class MergeRoadmapRequest(BaseModel):
 class MergeRoadmapResponse(BaseModel):
     merged_roadmap: str
 
-def merge_roadmaps(regular_roadmap, test_roadmaps):
+def merge_roadmaps(regular_roadmap_str, test_roadmaps_str_list):
+    # Convert input strings to JSON
+    regular_roadmap = json.loads(regular_roadmap_str)
+    test_roadmaps = [json.loads(roadmap) for roadmap in test_roadmaps_str_list]
+
     # Helper to convert a roadmap's schedule into a dictionary indexed by dates
     def schedule_to_dict(schedule):
         return {day["date"]: day for day in schedule}
@@ -101,15 +104,10 @@ def merge_roadmaps(regular_roadmap, test_roadmaps):
 @app.post("/merge-roadmaps", response_model=MergeRoadmapResponse)
 def merge_roadmaps_endpoint(request: MergeRoadmapRequest):
     try:
-        # Parse input strings into Python dictionaries
-        regular_roadmap = json.loads(request.regular_roadmap)
-        test_roadmaps = [json.loads(roadmap) for roadmap in request.test_roadmaps]
-
         # Merge the roadmaps
-        merged_roadmap = merge_roadmaps(regular_roadmap, test_roadmaps)
+        merged_roadmap = merge_roadmaps(request.regular_roadmap, request.test_roadmaps)
 
         # Return the merged roadmap as a JSON string
         return MergeRoadmapResponse(merged_roadmap=json.dumps(merged_roadmap))
     except Exception as e:
         raise HTTPException(status_code=400, detail=f"Error merging roadmaps: {str(e)}")
-