complete project modular

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")
 

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
 
 # MAIN Function START
 def load_langgraph_agenticai_app():
@@ -48,9 +49,11 @@ def load_langgraph_agenticai_app():
                 graph_builder=GraphBuilder(model)
                 try:
                     graph = graph_builder.setup_graph(usecase)
+                    DisplayResultStreamlit(usecase,graph,user_message).display_result_on_ui()
                 except Exception as e:
                     st.error(f"Error: Graph setup failed - {e}")
                     return
+                
 
             except Exception as e:
                  raise ValueError(f"Error Occurred with Exception : {e}")

src/langgraphagenticai/ui/streamlitui/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_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)