added the llm classes to load right llm based on the selected provider

.gitignore

@@ -1 +1,13 @@
-venv/
+venv/
+env.txt/
+__pycache__/
+
+# Ignore __pycache__ directories everywhere
+**/__pycache__/
+
+# Specific paths to ignore
+llms/__pycache__/
+ui/__pycache__/
+
+# Environment files
+env.txt

src/llms/groq.py

@@ -0,0 +1,25 @@
+import streamlit as st
+from langchain_groq import ChatGroq
+
+
+class GroqLLM:
+
+  def __init__(self, user_controls_input):
+    self.user_controls_input = user_controls_input
+
+  def get_llm_model(self):
+    try:
+      groq_api_key = self.user_controls_input['API Key']
+      groq_model = self.user_controls_input['Selected Model']
+
+      if groq_api_key== "":
+        st.error('Please enter your Groq API key to proceed.')
+
+      llm = ChatGroq(model=groq_model, api_key=groq_api_key)
+      return llm
+    
+    except KeyError as e:
+      st.error(f"Missing configuration: {e}")
+      return None
+    
+    

src/llms/openai.py

@@ -0,0 +1,22 @@
+import os
+import streamlit as st
+from langchain_openai import ChatOpenAI
+
+class OpenAILLM:
+  def __init__(self, user_controls_input):
+    self.user_controls_input = user_controls_input
+    
+  def get_llm_model(self):
+    try:
+      openai_api_key = self.user_controls_input['API Key']
+      openai_model = self.user_controls_input['Selected Model']
+
+      if openai_api_key == "":
+        st.error('Please enter your OpenAI API key to proceed.')
+
+      llm = ChatOpenAI(model=openai_model, openai_api_key=openai_api_key)
+      return llm
+
+    except KeyError as e:
+      st.error(f"Missing configuration: {e}")
+      return None

src/main.py

@@ -2,6 +2,8 @@ import streamlit as st
 import json
 
 from src.ui.load import LoadStreamlitUI
+from src.llms.groq import GroqLLM
+
 
 def load_app():
     """
@@ -15,9 +17,26 @@ def load_app():
         st.error("Failed to load the UI. Please check your configuration.")
         return
     
-    if st.session_state.isFetchButtonClicked:
+    if st.session_state.IsFetchButtonClicked:
         user_message = st.session_state.timeframe
     
     else:
         user_message = st.text_input("Enter your message:")
-        
+        
+    if user_message:
+        try:
+            llm = GroqLLM(user_controls_input=user_input)
+            groq_model = llm.get_llm_model()
+            
+            if not groq_model:
+                st.error("Failed to initialize the Groq model. Please check your API key and model selection.")
+                return
+            
+            use_case = user_input.get('use_case')
+
+            if use_case:
+                st.error("Error: No usecase selected.")
+                return 
+        
+        except Exception as e:
+            raise ValueError(f"An error occurred while processing the input: {e}")

src/ui/config.py

@@ -10,34 +10,34 @@ class LoadConfig:
     Get the list of LLM options from the configuration file.
     """
     
-    return self.config.get['DEFAULT'].get('LLM_options').split(', ')
+    return self.config.get('DEFAULT', 'LLM_options').split(', ')
   
   def get_use_case(self):
     """
     Get the use case from config file.
     """
 
-    return self.config.get['DEFAULT'].get('USE_CASE').split(', ')
+    return self.config.get('DEFAULT', 'USE_CASE').split(', ')
   
   def get_title(self):
     """
     Get the title from config file.
     """
 
-    return self.config.get['DEFAULT'].get('Title')
+    return self.config.get('DEFAULT', 'Title')
   
   def get_groq_models(self):
     """
     Get groq models from the config file.
     """
 
-    return self.config.get['DEFAULT'].get('GROQ_MODEL').split(', ')
+    return self.config.get('DEFAULT', 'GROQ_MODEL').split(', ')
 
   def get_openai_models(self):
     """
     Get OpenAI models from config file.
     """  
 
-    return self.config.get['DEFAULT'].get('OPENAI_MODEL').split(', ')
+    return self.config.get('DEFAULT', 'OPENAI_MODEL').split(', ')
 
     

src/ui/load.py

@@ -26,7 +26,7 @@ class LoadStreamlitUI:
     st.set_page_config(page_title=" 🤖" + self.config.get_title(), page_icon=":robot_face:", layout="wide")
     st.header(" 🤖" + self.config.get_title())
     st.session_state.timeframe = ''
-    st.session_state.IsFetchbuttonClicked = False
+    st.session_state.IsFetchButtonClicked = False
     st.session_state.IsSDLC = False
 
     with st.sidebar:
@@ -44,7 +44,7 @@ class LoadStreamlitUI:
         self.user_controls['API Key'] = st.text_input('Enter Groq API Key', type='password')
 
         if not self.user_controls['API Key']:
-            st.warning('⛔ Please enter a valid Groq API key to proceed ⛔. If you don\'t have an API key, please visit https://console.groq.com to create one.')
+            st.warning('Please enter a valid Groq API key to proceed . If you don\'t have an API key, please visit https://console.groq.com to create one.')
 
       
       if self.user_controls['Selected LLM'] == 'OpenAI':
@@ -55,7 +55,7 @@ class LoadStreamlitUI:
         self.user_controls['API Key'] = st.text_input('Enter OpenAI API Key', type='password')
 
         if not self.user_controls['API Key']:
-            st.warning('⛔ Please enter a valid OpenAI API key to proceed ⛔. If you don\'t have an API key, please visit https://platform.openai.com/account/api-keys to create one.')
+            st.warning('Please enter a valid OpenAI API key to proceed . If you don\'t have an API key, please visit https://platform.openai.com/account/api-keys to create one.')
 
       use_case = self.config.get_use_case()