running out of options, short term disability?

I have had a pretty rough pregnancy and my place of employment is making even harder. I have had dibilating migraines since I was 14 weeks. I have an appointment with a neurologist next week. I'm to the point where I don't feel like I can keep pushing through them. I ended up in the hospital a few weeks back for losing my balance and almost passing out and my sister rushed me to the hospital and said I wasn't making any sense with my speech. They did nothing for me at the hospital. I slept for 2 days after that (explained to my supervisor I needed to miss 2 days; the only 2 days I have missed in months) and they began the write up process for the time I missed. I was and still am a nervous wreck. I physically and mentally am drained. I have pushed myself to my break point. I planned to work up until my water broke but when I can't even open my eyes near lights and my vision is impaired when I have perfect vision, I know something just isn't right. 
I was thinking about trying to apply for short term disability. I have it through my employer but I have no idea what I'm doing or going to do ESPECIALLY if I lose my job. I've been here for a long time and sadly will not make the same salary or hourly rate anywhere near me 😔 (I'm in the US) 
💙also, I am 25 weeks pregnant. Due in February. 
If they resolve, that would be amazing but right now I'm really struggling with performing at work and, even though they haven't mentioned it yet, I know what my goals are and I'm a bit behind when I used to be exceeding so I could be put on a warning. I also am not allowed to miss any time that isn't legal or supported by a medical condition through HR and HR said short term disability is probably my best option as I was planning to use my FMLA for when I had baby because I need surgery 6 weeks after and need recovery time 😔