We see her brooding and making decisions near the Tree of Harmony on the Eve of the fateful Summer Sun Celebration of the first episode.I consider it very unlikely that she was taken by surprise by the return of Nightmare Moon but much rather that she was aware of what would happen during the Summer Sun Celebration and that she allowed herself to be captured since this would be the only chance for a return of Luna. This still may have been a great risk for Celestia to take. Moreover it meant giving up whatever control she still may have had about the Elements of Harmony. Seeing how Twilight’s Cutie Mark is represented much larger and higher on the Tree of Harmony than those of Celestia and Luna, chances are that the implications of what happened that night may go way farer than we realize so far. The surroundings of the Tree of Harmony don’t look exactly like they do in the show, but I wanted a waterfall nearby and some star sky above rather than the cavern it is set in in the series.