The First Minister of Scotland will visit the set of a new Channel 4 drama, with £2 million pledged for the screen sector in an upcoming budget.