Update src/streamlit_app.py

src/streamlit_app.py

@@ -104,20 +104,24 @@ if "current_chat_id" in st.session_state:
         with st.chat_message("assistant"):
             response_placeholder = st.empty()
             full_response = ""
-
+            tool_name = "" 
+            tool_download_msg = ""
             for message, metadata in workflow.stream(
                 {"messages": [system, HumanMessage(user_input)]},
                 config={"configurable": {"thread_id": st.session_state.current_chat_id}},
                 stream_mode="messages",
             ):
+                tool_link = ""
                 if isinstance(message, AIMessage):
                     full_response += message.content or ""
+                    if(message.tool_calls):
+                        tool_name = message.tool_calls[0]['name'] 
                 elif isinstance(message, ToolMessage):
                     st.info("Using Appropriate tool")
-                    tool_name = message.tool_calls[0]['name'] 
+                  
                     if(tool_name == "read_file" or tool_name == "write_file"):
-                        create_download_link(message.content) 
+                        tool_download_msg += message.content 
+                        tool_link = create_download_link(tool_download_msg) 
                     
 
-                response_placeholder.markdown(full_response + "|")
-        st.rerun()
+                response_placeholder.markdown(full_response + "|")