update Streamlit UI: To display chat conversation for Tools calls for - Chatbot with Tools

src/basicchatbot/ui/streamlitui/display_result.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from typing import Literal
 from langgraph.graph.state import CompiledStateGraph
-from langchain_core.messages import AIMessage
+from langchain_core.messages import AIMessage,HumanMessage,ToolMessage
 
 
 class DisplayResultStreamlit:
@@ -49,9 +49,48 @@ class DisplayResultStreamlit:
         except Exception as e:
             st.error(f"Error processing response: {str(e)}")
 
+    from langchain_core.messages import AIMessage, HumanMessage, ToolMessage
+
     def _handle_chatbot_with_tool(self) -> None:
         """Handle the tool-enhanced chatbot interaction flow."""
-        pass  # Placeholder for future implementation
+        
+        # Display chat history before processing new input
+        self._display_chat_history()
+
+        # Append user message to session state and display it
+        # st.session_state.message_history.append({"role": "user", "message": self.user_message})
+        # with st.chat_message("user"):
+        #     st.markdown(self.user_message)
+        # self._display_message("user", self.user_message)
+        
+        try:
+            initial_state = {"messages": self.user_message}
+
+            result = self.graph.invoke(input=initial_state)
+
+            for message in result["messages"]:
+                if isinstance(message, HumanMessage):
+                    st.session_state.message_history.append({"role": "user", "message": message.content})
+                    self._display_message("user", message.content)
+                    
+                elif isinstance(message, AIMessage) and message.content:
+                    st.session_state.message_history.append({"role": "assistant", "message": message.content})
+                    self._display_message("assistant", message.content)
+                    
+                elif isinstance(message, ToolMessage):
+                    # Display tool-specific response
+                    tool_message = f"**Tool Response:**\n{message.content}"
+                    
+                    st.session_state.message_history.append({"role": "assistant", "message": tool_message})
+                    with st.chat_message("assistant"):
+                        st.markdown("🔧 **Tool Call Start**")
+                        st.markdown(message.content)
+                        st.markdown("🔧 **Tool Call End**")
+
+        except Exception as e:
+            st.error(f"Error processing response: {str(e)}")
+
+
 
     def display_result_on_ui(self) -> None:
         usecase_handlers = {