This study improves the Agro-C model to better represent how crops respond to changing soil moisture. The updated model adjusts carbon allocation between roots and leaves according to soil moisture and growth stage, and also accounts for leaf aging under water stress. Tests with field data for maize and wheat showed better predictions of plant growth above and below ground. The results can improve estimates of crop production and soil carbon storage under future drought and climate change.