refractoring codebase

app.py

@@ -0,0 +1,10 @@
+import os
+import sys
+
+# Add the project root directory to Python's module search path
+sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
+
+from src.langgraph.main import load_app
+
+if __name__=="__main__":
+    load_app()

src/langgraph/UI/streamlit/loadui.py

@@ -3,8 +3,8 @@ import os
 from dotenv import load_dotenv
 from datetime import date
 
-from langchain_core import AIMessage, HumanMessage, SystemMessage
-from src.langgraph.UI.uiconfigfile import Config
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+from ...UI.uiconfigfile import Config
 
 class LoadStreamlitUI:
     def __init__(self):
@@ -50,22 +50,30 @@ class LoadStreamlitUI:
             #Creating a dropdown for the LLM options
             st.session_state.selected_llm = st.selectbox("Select LLM", LLM_OPTIONS)
 
+            # Initialize the user_controls dictionary to store user selections
+            user_controls = {}
+            
             if st.session_state.selected_llm == "Groq":
                 model = self.config.get_groq_model()
                 st.session_state.selected_model = st.selectbox("Select Model", model)
                 st.session_state.selected_model_api_key = st.text_input("Enter API Key", type="password")
 
+                # Store the Groq-specific values in user_controls
+                user_controls["groq_api_key"] = st.session_state.selected_model_api_key
+                user_controls["selected_groq_models"] = st.session_state.selected_model
+
                 #validating the API key
                 if not st.session_state.selected_model_api_key:
                     st.error("Please enter the correct API key. refer :  https://console.groq.com/keys")
 
             #Creating a dropdown for the use case options
             st.session_state.selected_usecase = st.selectbox("Select Use Case", USE_CASE_OPTIONS)
+            user_controls["selected_use_case"] = st.session_state.selected_usecase
 
             if "state" not in st.session_state:
                 st.session_state.state = self.initialize_state()
             #self.render_requirements()
 
-        return self.user_controls
+        return user_controls
 
             

src/langgraph/graph/graph_builder.py

@@ -1,8 +1,8 @@
 from langgraph.graph import StateGraph, START,END, MessagesState
 from langgraph.prebuilt import tools_condition,ToolNode
 from langchain_core.prompts import ChatPromptTemplate
-from src.langgraphagenticai.state.state import State
-from src.langgraphagenticai.nodes.basic_chatbot_node import BasicChatbotNode
+from src.langgraph.state.state import State
+from src.langgraph.node.basic_chatbot import BasicChatbotNode
 
 
 

src/langgraph/main.py

@@ -1,12 +1,12 @@
 import streamlit as st
 import json
-from src.langgraph.UI.streamlit.loadui import LoadStreamlitUI
-from src.langgraph.llm.groqllm import GroqLLM
-from src.langgraph.graph.graph_builder import GraphBuilder
-from src.langgraph.UI.streamlit.displayresult import DisplayResultStreamlit
+from .UI.streamlit.loadui import LoadStreamlitUI
+from .llm.groqllm import GroqLLM
+from .graph.graph_builder import GraphBuilder
+from .UI.streamlit.displayresult import DisplayResultStreamlit
 
 
-def __main__():
+def load_app():
     """
     
     Loads and runs the LangGranph Agentic AI POC application.
@@ -46,8 +46,8 @@ def __main__():
             #Graph Builder
             graph_builder = GraphBuilder(llm=model, use_case=use_case)
             try:
-                graph = graph_builder.setup_graph(usecase)
-                DisplayResultStreamlit(usecase,graph,user_message).display_result_on_ui()
+                graph = graph_builder.setup_graph(use_case)
+                DisplayResultStreamlit(use_case, graph, user_message).display_result_on_ui()
             except Exception as e:
                 st.error(f"Error: Graph setup failed - {e}")
                 return