load ui, main, app

app.py

@@ -0,0 +1,5 @@
+from src.langgraphagenticai.main import load_langgraph_agenticai_app
+
+
+if __name__=="__main__":
+    load_langgraph_agenticai_app()

src/langgraphagenticai/LLMS/groqllm.py

@@ -0,0 +1 @@
+pass

src/langgraphagenticai/main.py

@@ -0,0 +1,59 @@
+import streamlit as st
+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():
+    """
+    Loads and runs the LangGraph AgenticAI application with Streamlit UI.
+    This function initializes the UI, handles user input, configures the LLM model,
+    sets up the graph based on the selected use case, and displays the output while 
+    implementing exception handling for robustness.
+    """
+   
+    # Load UI
+    ui = LoadStreamlitUI()
+    user_input = ui.load_streamlit_ui()
+
+    if not user_input:
+        st.error("Error: Failed to load user input from the UI.")
+        return
+
+    # Text input for user message
+    if st.session_state.IsFetchButtonClicked:
+        user_message = st.session_state.timeframe 
+    else :
+        user_message = st.chat_input("Enter your message:")
+
+    # if user_message:
+    #         try:
+    #             # Configure LLM
+    #             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
+
+    #             # Initialize and set up the graph based on use case
+    #             usecase = user_input.get('selected_usecase')
+    #             if not usecase:
+    #                 st.error("Error: No use case selected.")
+    #                 return
+                
+
+    #             ### Graph Builder
+    #             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/loadui.py

@@ -8,5 +8,65 @@ from src.langgraphagenticai.ui.uiconfigfile import Config
 
 class LoadStreamlitUI:
     def __init__(self):
-        self.config=Config() # config
-        self.user_controls={}
+        self.config =  Config() # config
+        self.user_controls = {}
+
+    def initialize_session(self):
+        return {
+        "current_step": "requirements",
+        "requirements": "",
+        "user_stories": "",
+        "po_feedback": "",
+        "generated_code": "",
+        "review_feedback": "",
+        "decision": None
+    }
+  
+
+
+    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 = ''
+        st.session_state.IsFetchButtonClicked = False
+        st.session_state.IsSDLC = False
+        
+        
+
+        with st.sidebar:
+            # Get options from config
+            llm_options = self.config.get_llm_options()
+            usecase_options = self.config.get_usecase_options()
+
+            # LLM selection
+            self.user_controls["selected_llm"] = st.selectbox("Select LLM", llm_options)
+
+            if self.user_controls["selected_llm"] == 'Groq':
+                # Model selection
+                model_options = self.config.get_groq_model_options()
+                self.user_controls["selected_groq_model"] = st.selectbox("Select Model", model_options)
+                # API key input
+                self.user_controls["GROQ_API_KEY"] = st.session_state["GROQ_API_KEY"] = st.text_input("API Key",
+                                                                                                      type="password")
+                # Validate API key
+                if not self.user_controls["GROQ_API_KEY"]:
+                    st.warning("⚠️ Please enter your GROQ API key to proceed. Don't have? refer : https://console.groq.com/keys ")
+                   
+            
+            # 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 "state" not in st.session_state:
+                st.session_state.state = self.initialize_session()
+            
+            
+        
+        return self.user_controls

src/langgraphagenticai/ui/uiconfigfile.py

@@ -11,7 +11,7 @@ class Config:
         self.config = ConfigParser()
         self.config.read(config_file)
     
-    def get_llm_option(self):
+    def get_llm_options(self):
         return self.config["DEFAULT"].get("LLM_OPTIONS").split(", ")
     
     def get_usecase_options(self):