Update main.py

main.py

@@ -43,8 +43,8 @@ async def generate_roadmap_endpoint(course_request: CourseRequest):
     diagram = create_diagram(roadmap_text)
 
     # Render the diagram to a PNG image
-    diagram_path = "/tmp/roadmap"
+    diagram_path = "/tmp/roadmap.png"
     diagram.render(diagram_path, format='png', cleanup=True)
 
-    with open(diagram_path + ".png", "rb") as f:
+    with open(diagram_path, "rb") as f:
         return Response(content=f.read(), media_type="image/png")