Iran will have the capacity to begin enriching uranium again in "a matter of months", the UN's nuclear watchdog boss has said.