Faits: A. P._, née en 1974, mariée et mère de deux enfants, travaillait à plein temps comme ouvrière au service de l'entreprise X._ SA. Le 4 avril 2004, elle a été victime d'un accident de la circulation en tant que passagère arrière droite d'une automobile conduite par son mari. Celui-ci, qui circulait sur la voie de dépassement d'une autoroute à une vitesse en dessous de 100 km/h, a brusquement dû se rabattre vers la droite pour éviter un autre véhicule roulant en sens inverse. Au cours de cette manoeuvre, la voiture a dérapé et percuté la glissière de sécurité. P._ a été projetée contre la portière droite qu'elle a heurtée de la tête et de l'épaule; elle a subi une commotion cérébrale, une distorsion cervicale, des contusions aux mains, à la cuisse gauche et à l'épaule droite. La Caisse nationale suisse d'assurance en cas d'accidents (CNA), auprès de laquelle elle était assurée, a pris en charge le cas. P._ a pu reprendre son travail à 35% le 7 mars 2005, puis à 50% dès le mois d'août 2005. Le 14 mars 2005, elle a déposé une demande de prestations à l'assurance-invalidité. L'Office de l'assurance-invalidité du canton de Berne (ci-après : l'office AI) a recueilli le dossier médical de l'assureur-accidents, dont notamment une expertise pluridisciplinaire du 14 juin 2007 établie, sur mandat de la CNA, par le Centre Z._. Dans ce rapport, les experts du Centre Z._ ont posé les diagnostics de traumatisme cranio-cérébral mineur, distorsion cervicale de degré 1 à 2 selon la Québec Task Force, contusion de l'épaule, syndrome post-commotionnel, cervico-brachialgies, état de stress post-traumatique, deuil compliqué ainsi que trouble somatoforme douloureux persistant. Ils ont estimé que si les troubles apparus immédiatement après l'accident correspondaient typiquement à un syndrome post-commotionnel et à un status post distorsion cervicale, l'importance des troubles et leur répercussion sur la capacité de travail étaient inhabituels au-delà de deux à trois ans après l'événement accidentel compte tenu de son déroulement. La persistance des plaintes était probablement à mettre sur le compte d'un état psychique fragilisé. En ce qui concernait la capacité de travail de P._, les experts du Centre Z._ ont conclu qu'elle était entière sur le plan somatique (dans une activité légère) mais qu'elle ne dépassait pas 50% dans toute activité professionnelle sous l'angle strictement psychique. L'office AI a soumis ce rapport d'expertise à son service médical régional (SMR) pour appréciation. Dans un document du 31 juillet 2007, la doctoresse T._, psychiatre du SMR, a estimé que le diagnostic d'état de stress post-traumatique retenu par l'expert-psychiatre du Centre Z._ n'était pas convaincant; il n'y avait pas lieu non plus d'admettre une incapacité de travail à raison du trouble somatoforme douloureux dès lors que celui-ci n'était associé à aucune comorbidité psychiatrique et que l'assurée disposait encore, selon l'expert précité, de bonnes ressources psychiques. Se fondant sur cet avis, l'office AI a rendu le 13 juin 2008 deux décisions par lesquelles il a alloué à P._ trois-quarts de rente du 1er avril au 31 août 2005, puis une demi-rente du 1er septembre 2005 au 31 mars 2006, prestations assorties des rentes complémentaires pour ses deux enfants. De son côté, la CNA a mis un terme à ses prestations au 31 octobre 2007 (décision sur opposition du 3 janvier 2008; voir également le ju-gement du 13 janvier 2009 du Tribunal administratif du canton de Berne, confirmé par arrêt du Tribunal fédéral de ce jour; cause 8C_182/ 2009). B. L'assurée a déféré les décisions de l'office AI à la Cour des affaires de langue française du Tribunal administratif du canton de Berne, qui a très partiellement admis son recours en ce sens que la demi-rente devait être supprimée au 30 juin 2006 (jugement du 14 janvier 2009). C. P._ interjette un recours en matière de droit public contre ce jugement dont elle requiert l'annulation. Sous suite de frais et dépens, elle conclut à ce que la demi-rente continue de lui être versée; subsidiairement, à ce que la cause soit renvoyée à l'autorité cantonale pour nouveau jugement au sens des considérants. L'office AI conclut au rejet du recours. L'Office fédéral des assurances sociales a renoncé à se déterminer. D. Par lettre du 26 novembre 2009, la recourante a demandé une suspension de la procédure pour lui permettre de déposer un rapport du docteur I._ qui a procédé à un nouvel examen de son cas.