added display result ui

src/langgraphagenticai/graph/graph_builder.py

@@ -4,7 +4,6 @@ from langchain_core.prompts import ChatPromptTemplate
 from src.langgraphagenticai.state.state import State
 from src.langgraphagenticai.nodes.basic_chatbot_node import BasicChatbotNode
 
-
 class GraphBuilder:
     def __init__(self, model):
         self.llm = model

src/langgraphagenticai/main.py

@@ -1,6 +1,7 @@
 import streamlit as st 
 import json
 
+from src.langgraphagenticai.ui.streamlit.display_result import DisplayResultStreamlit
 from src.langgraphagenticai.ui.streamlit.load_ui import LoadStreamlitUI
 from src.langgraphagenticai.llms.groq_llm import GroqChatLLM
 from src.langgraphagenticai.graph.graph_builder import GraphBuilder
@@ -48,11 +49,10 @@ def load_langgraph_agenticai_app():
             graph_builder = GraphBuilder(model)
             try:
                 graph = graph_builder.setup_graph(usecase)
+                DisplayResultStreamlit(usecase, graph, user_message).display_result_in_ui()
             except Exception as e:
-                st.error(f"Error: Graph setup failed - {e}")
+                st.error(f"Error: Graph setup failed {e}")
                 return
 
         except Exception as e:
-            raise ValueError(f"Error Occurred with Exception: {e}")
-
-    print(user_input)
+            raise ValueError(f"Error Occurred with Exception: {e}")

src/langgraphagenticai/nodes/basic_chatbot_node.py

@@ -11,4 +11,4 @@ class BasicChatbotNode:
         """
         Processes the input state and generates a chatbot response.
         """
-        return {"message": self.llm.invoke(state['messages'])}
+        return {"messages": self.llm.invoke(state['messages'])}

src/langgraphagenticai/ui/streamlit/display_result.py

@@ -0,0 +1,24 @@
+import streamlit as st
+from langchain_core.messages import HumanMessage, AIMessage
+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_in_ui(self):
+        usecase = self.usecase
+        graph = self.graph
+        user_message = self.user_message
+        if usecase == "Basic Chatbot":
+            print(usecase, graph, 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)

src/langgraphagenticai/ui/uiconfigfile.ini

@@ -2,4 +2,4 @@
 PAGE_TITLE = LangGraph: Build Stateful Agentic AI graph 
 LLM_OPTIONS = Groq, OPENAI
 USECASE_OPTIONS = Basic Chatbot, Voice Analyzer
-GROQ_MODEL_OPTIONS = mixtral-8x7b-32768, llama3-8b-8192, gemma-7b-i
+GROQ_MODEL_OPTIONS = mixtral-8x7b-32768, llama-3.3-70b-versatile, gemma2-9b-it

src/langgraphagenticai/ui/uiconfigfile.py

@@ -8,13 +8,13 @@ class Config:
 
         
     def get_llm_options(self):
-        return self.config["DEFAULT"].get("LLM_OPTIONS").split(",")
+        return self.config["DEFAULT"].get("LLM_OPTIONS").split(", ")
     
     def get_usecase_options(self):
-        return self.config["DEFAULT"].get("USECASE_OPTIONS").split(",")
+        return self.config["DEFAULT"].get("USECASE_OPTIONS").split(", ")
 
     def get_groq_model_options(self):
-        return self.config["DEFAULT"].get("GROQ_MODEL_OPTIONS").split(",")
+        return self.config["DEFAULT"].get("GROQ_MODEL_OPTIONS").split(", ")
 
     def get_page_title(self):
         return self.config["DEFAULT"].get("PAGE_TITLE")