update

src/langgraphagenticai/LLMS/groqllm.py

@@ -9,7 +9,7 @@ class GroqLLM:
     def get_llm_model(self):
         try:
             groq_api_key = self.user_controls_input["GROQ_API_KEY"]
-            selected_groq_model = self.user_controls_input["selected_groq_models"]
+            selected_groq_model = self.user_controls_input["selected_groq_model"]
             if groq_api_key=='' and os.environ["GROQ_API_KEY"]=='':
                 st.error("Please Enter the Groq API key")
             llm = ChatGroq(api_key=groq_api_key,model=selected_groq_model)

src/langgraphagenticai/main.py

@@ -3,6 +3,7 @@ import json
 from src.langgraphagenticai.ui.streamlitui.loadui import LoadStreamlitUI
 from src.langgraphagenticai.LLMS.groqllm import GroqLLM
 from src.langgraphagenticai.graph.graph_builder import GraphBuilder
+from src.langgraphagenticai.ui.streamlitui.display_result import DisplayResultStreamlit
 
 def load_langgraph_agenticai_app():
     """
@@ -26,30 +27,36 @@ def load_langgraph_agenticai_app():
 
     # Initializing the LLM
     if user_message:
+        
+        obj_llm_config = GroqLLM(user_controls_input=user_input)
+        model = obj_llm_config.get_llm_model()
+
+        if not model:
+            st.error("Error: LLM model could not be initialized.")
+            return 
+        
+        usecase = user_input.get("selected_usecase")
+        if not usecase:
+            st.error("Error: Usecase not selected.")
+            return
+        
+
+        # Graph Builder
+        graph_builder = GraphBuilder(model)
         try:
-            obj_llm_config = GroqLLM(user_controls_input=user_input)
-            model = obj_llm_config.get_llm_model()
-
-            if not model:
-                st.error("Error: LLM model could not be initialized.")
-                return 
-            
-            usecase = user_input.get("selected_usecase")
-            if not usecase:
-                st.error("Error: Usecase not selected.")
-                return
-            
-
-            # Graph Builder
-            graph_builder = GraphBuilder(model)
-            try:
-                graph = graph_builder.setup_graph(usecase=usecase)
-            except Exception as e:
-                raise ValueError(f"Error: Graph set up Failed - {e}")
-                return 
-            
+            graph = graph_builder.setup_graph(usecase=usecase)
         except Exception as e:
-            raise ValueError(f"Error occured with exception : {e}")
+            raise ValueError(f"Error: Graph set up Failed - {e}")
+            return 
+        
+        # Display Result
+        display_obj = DisplayResultStreamlit(
+            usecase=usecase,
+            graph=graph,
+            user_message=user_message
+        )
+        display_obj.display_result_on_ui()
+    
         
  
 

src/langgraphagenticai/nodes/basic_chatbot_node.py

@@ -6,6 +6,7 @@ class BasicChatbotNode:
     """
     def __init__(self,model):
         self.llm = model
+        
 
     def process(self,state):
         """ 

src/langgraphagenticai/state/state.py

@@ -1,10 +1,10 @@
-from typing import Annotated,Literal,Optional
+from typing import Annotated,Literal,Optional,List
 from typing_extensions import TypedDict
 from langgraph.graph.message import add_messages
-from langchain_core.messages import AIMessage,HumanMessage
+from langchain_core.messages import AIMessage,HumanMessage,BaseMessage
 
 class State(TypedDict):
     """ 
     Represents the structure of the state used in the graph
     """
-    messages = Annotated[list,add_messages]
+    messages: Annotated[list,add_messages]

src/langgraphagenticai/ui/streamlitui/display_result.py

@@ -14,11 +14,11 @@ class DisplayResultStreamlit:
         graph = self.graph
         user_message = self.user_message
         if usecase =="Basic Chatbot":
-                for event in graph.stream({'messages':("user",user_message)}):
+                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)
+                            st.write(value["messages"][-1].content)

src/langgraphagenticai/ui/uiconfigfile.ini

@@ -1,5 +1,5 @@
 [DEFAULT]
 PAGE_TITLE = LangGraph: Build stateful Agentic AI Graph 
 LLM_OPTIONS = Groq
-USECASE_OPTIONS = Basic chatbot
+USECASE_OPTIONS = Basic Chatbot
 GROQ_MODEL_OPTIONS = mixtral-8x7b-32768, llama3-8b-8192, llama3-70b-8192, gemma-7b-i