Hi Sarah,
DWP advise that they can consider only problems caused by a physical condition in the first section and problems caused by a psychological/mental problem in the second part. However our recommendation, as always, is that you make the form work for you and put things into whichever descriptor/s that you believe are appropriate.
Nothing on this board constitutes legal advice - always consult a professional about specific problems