当女职工怀孕后，单位需要承担一系列的责任和义务，以确保女职工的权益得到充分保障。以下是一些单位在女职工怀孕期间应做的关键事项：

首先，关于产假期间的待遇，如果女职工已经参加了生育保险，那么她在产假期间的生育津贴将按照用人单位上年度职工月平均工资的标准由生育保险基金支付。如果未参加生育保险，则按照女职工产假前工资的标准由用人单位支付。同时，女职工生育或流产的医疗费用，也根据是否参加生育保险，分别由生育保险基金或用人单位支付。

其次，单位不得因女职工怀孕、生育、哺乳而降低其工资、予以辞退或与其解除劳动或聘用合同。这是法律对女职工的明确保护，单位必须严格遵守。

再者，如果女职工在孕期不能适应原劳动，单位应根据医疗机构的证明，减轻其劳动量或安排其他能够适应的劳动。对于怀孕7个月以上的女职工，单位不得延长其劳动时间或安排夜班劳动，并应在劳动时间内安排一定的休息时间。

此外，怀孕女职工在劳动时间内进行产前检查的时间应计入劳动时间，单位不得因此扣除其工资或视为缺勤。

关于产假天数，女职工生育享受98天产假，其中产前可以休假15天。难产的，增加产假15天；生育多胞胎的，每多生育1个婴儿，增加产假15天。如果女职工怀孕未满4个月流产，享受15天产假；怀孕满4个月流产，享受42天产假。

对于哺乳未满1周岁婴儿的女职工，单位同样不得延长其劳动时间或安排夜班劳动，并应在每天的劳动时间内为其安排1小时哺乳时间。如果女职工生育多胞胎，每多哺乳1个婴儿，每天应增加1小时哺乳时间。

最后，如果单位中女职工较多，还应根据女职工的需要，建立女职工卫生室、孕妇休息室、哺乳室等设施，以妥善解决女职工在生理卫生、哺乳方面的困难。

综上所述，单位在女职工怀孕期间应承担的责任和义务是多方面的，包括提供产假待遇、保障工作条件、安排产前检查时间、提供产假和哺乳假等。单位必须严格遵守相关法律法规，确保女职工的权益得到充分保障。