The head of the UN nuclear watchdog says they are close to signing a deal with Iran on an investigation into its atomic activity, after talks in Tehran.