FInal changes3

src/langgraph_agenticAI/UI/streamlitui/load_ui.py

@@ -2,14 +2,15 @@ import streamlit as st
 import os
 from datetime import date
 
-#from langchain_core import AIMessage, HumanMessage   #because msg we are going to mention in UI is either AImsg or Humanmsg
-from src.langgraph_agenticAI.UI.uiconfigfile import Config    #Config is class
+from langchain_core.messages import AIMessage,HumanMessage
+from src.langgraph_agenticAI.UI.uiconfigfile import Config
 
-class LoadStreamLitUI:
+
+class LoadStreamlitUI:
     def __init__(self):
-        self.config=Config()    #for this cofig we created ini file i.e, uiconfig.ini (text file to store configs in form of key-value pairs) like yaml config. 
-        self.user_controls={}    #dictionary
-        
+        self.config =  Config() # config
+        self.user_controls = {}
+
     def initialize_session(self):
         return {
         "current_step": "requirements",
@@ -20,18 +21,10 @@ class LoadStreamLitUI:
         "review_feedback": "",
         "decision": None
     }
-    """ def render_requirements(self):
-        st.markdown('### Requirement Submission')
-        st. session_state.state["requirements"]= st.text_area(
-            "Enter your Requirements:",
-            height=200,
-            key="req-input"
-        )
-        if st.button("Submit requirements", key="req-input"):
-            st.session_state.state["current_step"] = " generate_User_stories" """
+  
 
 
-    def load_streamlit_ui(self):   
+    def load_streamlit_ui(self):
         st.set_page_config(page_title= "🤖 " + self.config.get_page_title(), layout="wide")
         st.header("🤖 " + self.config.get_page_title())
         st.session_state.timeframe = ''
@@ -40,7 +33,7 @@ class LoadStreamLitUI:
         
         
 
-        with st.sidebar:     #sidebar for left side code
+        with st.sidebar:
             # Get options from config
             llm_options = self.config.get_llm_options()
             usecase_options = self.config.get_usecase_options()
@@ -74,8 +67,6 @@ class LoadStreamLitUI:
             if "state" not in st.session_state:
                 st.session_state.state = self.initialize_session()
             
-            #self.render_requirements()    #this is to load right side of the page with left sidebar
+            
         
         return self.user_controls
-
-

src/langgraph_agenticAI/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, Chatbot with Tool
 GROQ_MODEL_OPTIONS= mixtral-8x7b-32768, llama3-8b-8192, llama-70b-8192, gemma-7b-i  

src/langgraph_agenticAI/main.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import json
-from src.langgraph_agenticAI.UI.streamlitui.load_ui import LoadStreamLitUI
+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
@@ -15,7 +15,7 @@ def load_langgraph_agenticai_app():
     """
    
     # Load UI
-    ui = LoadStreamLitUI()
+    ui = LoadStreamlitUI()
     user_input = ui.load_streamlit_ui()
 
     if not user_input: