I think they had no idea what to do with her character at the end

Errr no. Since she lived in real life and was sent to prison and he died I think it was the directors choice to reverse the roles and have what actually should have happened go on screen.