Project complete

src/langgraph_agenticAI/UI/streamlitui/display_result.py

@@ -0,0 +1,43 @@
+
+import streamlit as st
+from langchain_core.messages import HumanMessage,AIMessage,ToolMessage
+import json
+
+
+class DisplayResultStreamlit:
+    def __init__(self,usecase,graph,user_message):
+        self.usecase= usecase
+        self.graph = graph
+        self.user_message = user_message
+
+    def display_result_on_ui(self):
+        usecase= self.usecase
+        graph = self.graph
+        user_message = self.user_message
+        if usecase =="Basic Chatbot":
+                for event in graph.stream({'messages':("user",user_message)}):
+                    print(event.values())
+                    for value in event.values():
+                        print(value['messages'])
+                        with st.chat_message("user"):
+                            st.write(user_message)
+                        with st.chat_message("assistant"):
+                            st.write(value["messages"].content)
+
+        elif usecase=="Chatbot with Tool":
+             # Prepare state and invoke the graph
+            initial_state = {"messages": [user_message]}
+            res = graph.invoke(initial_state)
+            for message in res['messages']:
+                if type(message) == HumanMessage:
+                    with st.chat_message("user"):
+                        st.write(message.content)
+                elif type(message)==ToolMessage:
+                    with st.chat_message("ai"):
+                        st.write("Tool Call Start")
+                        st.write(message.content)
+                        st.write("Tool Call End")
+                elif type(message)==AIMessage and message.content:
+                    with st.chat_message("assistant"):
+                        st.write(message.content)
+             

src/langgraph_agenticAI/main.py

@@ -3,7 +3,7 @@ import json
 from src.langgraph_agenticAI.UI.streamlitui.load_ui import LoadStreamLitUI
 from src.langgraph_agenticAI.LLMs.GroqLLM import GroqLLM
 from src.langgraph_agenticAI.Graphs.graph_builder import GraphBuilder
-#from src.langgraph_agenticAI.UI.streamlitui.display_result import DisplayResultStreamlit
+from src.langgraph_agenticAI.UI.streamlitui.display_result import DisplayResultStreamlit
 
 # MAIN Function START
 def load_langgraph_agenticai_app():