ui tested

src/langgraphagenticai/LLMS/groqllm.py

@@ -5,18 +5,18 @@ from langchain_groq import ChatGroq
 
 class GroqLLM:
     def __init__(self,user_controls_input):
-        self.user_control_input=user_controls_input
+        self.user_controls_input=user_controls_input
     
     def get_llm_model(self):
         try:
             groq_api_key=self.user_controls_input["GROQ_API_KEY"]
-            selected_groq_model=self.user_control_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)
         
         except Exception as e:
-            raise ValueError(f"Error Occurred with xception : {e}")
+            raise ValueError(f"Error Occurred with exception : {e}")
         return llm
 

src/langgraphagenticai/ui/streamlitui/display_result.py

@@ -0,0 +1,23 @@
+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)

src/langgraphagenticai/ui/streamlitui/loadui.py

@@ -56,13 +56,13 @@ class LoadStreamlitUI:
             # Use case selection
             self.user_controls["selected_usecase"] = st.selectbox("Select Usecases", usecase_options)
 
-            if self.user_controls["selected_usecase"] =="Chatbot with Tool":
-                # API key input
-                os.environ["TAVILY_API_KEY"] = self.user_controls["TAVILY_API_KEY"] = st.session_state["TAVILY_API_KEY"] = st.text_input("TAVILY API KEY",
-                                                                                                      type="password")
-                # Validate API key
-                if not self.user_controls["TAVILY_API_KEY"]:
-                    st.warning("⚠️ Please enter your TAVILY_API_KEY key to proceed. Don't have? refer : https://app.tavily.com/home")
+            # if self.user_controls["selected_usecase"] =="Chatbot with Tool":
+            #     # API key input
+            #     os.environ["TAVILY_API_KEY"] = self.user_controls["TAVILY_API_KEY"] = st.session_state["TAVILY_API_KEY"] = st.text_input("TAVILY API KEY",
+            #                                                                                           type="password")
+            #     # Validate API key
+            #     if not self.user_controls["TAVILY_API_KEY"]:
+            #         st.warning("⚠️ Please enter your TAVILY_API_KEY key to proceed. Don't have? refer : https://app.tavily.com/home")
             
             if "state" not in st.session_state:
                 st.session_state.state = self.initialize_session()