Update src/streamlit_app.py

src/streamlit_app.py

@@ -93,8 +93,10 @@ def loadchats():
         elif isinstance(message, ToolMessage):
             with st.chat_message("assistant"):
                 st.info("Using Appropriate tool")
+                if(message.name == "plot_graph" and "Filepath" in message.content):
+                    st.image(message.content.split(":")[1] , "📊")
                 if("Filepath" in message.content):
-                    st.markdown(create_download_link(message.content.split(':')[1]) , unsafe_allow_html = True) 
+                    st.markdown(create_download_link(message.content.split(":")[1]) , True) 
                     
     return messages