Orange flames and burner not staying on sounds like your intake air filter is clogged. Pull it out, vacuum it and vacuum everything under the water heater, and reinstall it all.

Thermocouple failure would mean the burner doesn’t light at all. The main valve is shut until the pilot light heats up the thermocouple.