Error when selecting the diagram tool in the streamlit app

[zensim]

I tried to generate a diagram. My query was "generate a diagram to show an s3 bucket to connect to a EC2 instance". Got the error below. Please advise. Thanks.

AttributeError: 'NoneType' object has no attribute 'save'
Traceback:
File "...\amazon-bedrock-agents-quickstart-main\streamlit_agent\claude3_agent_st.py", line 63, in
main()
File "....\amazon-bedrock-agents-quickstart-main\streamlit_agent\claude3_agent_st.py", line 59, in main
app()
File "....\amazon-bedrock-agents-quickstart-main\streamlit_agent\claude3_agent_st.py", line 36, in app
base64_string = tools.pil_to_base64(answer)
File ".....\amazon-bedrock-agents-quickstart-main\streamlit_agent\claude3_tools.py", line 44, in pil_to_base64
image.save(buffer, format)

[zensim]

Can someone please advise on the error? Thanks