На лечении или на лечение, как правильно?

1 случай - на лечениИ ( то есть нахождение где- то)

Пример : Моя бабушка сейчас на лечении .

Падеж - Предложный

2 случай- на лечениЕ (на что- то )

Пример : Давайте вместе соберём на лечение

Падеж - Винительный

То есть определяем падеж и по падёжным окончаниям пишем верно.