Update app.py

app.py

@@ -27,19 +27,14 @@ LLM_MODEL_TEMPERATURE = float(config["llm_model_temperature"])
 
 
 class KadiBot:
-    def __init__(self):
-        vectorstore = get_chroma_vectorstore(get_SFR_Code_embedding_model(), vectorstore_path)
-        llm = get_groq_llm(LLM_MODEL_NAME, LLM_MODEL_TEMPERATURE, GROQ_API_KEY)
-
+    def __init__(self, llm, vectorstore):
         self.kadiAPY_ragchain = KadiApyRagchain(llm, vectorstore)
 
     def handle_chat(self, chat_history):
         if not chat_history:
             return chat_history
         
-        # Get the last user query from the chat history
         user_query = chat_history[-1][0]
-    
         response = self.kadiAPY_ragchain.process_query(user_query, chat_history)
         chat_history[-1] = (user_query, response)
 
@@ -47,6 +42,7 @@ class KadiBot:
 
 
 
+
 def add_text_to_chat_history(chat_history, user_input):
     chat_history = chat_history + [(user_input, None)]
     return chat_history, ""
@@ -61,8 +57,11 @@ def reset_all():
 
 
 def main():
-    kadi_bot = KadiBot()  # Initialize the KadiBot class
-
+    vectorstore = get_chroma_vectorstore(get_SFR_Code_embedding_model(), vectorstore_path)
+    llm = get_groq_llm(LLM_MODEL_NAME, LLM_MODEL_TEMPERATURE, GROQ_API_KEY)
+    
+    kadi_bot = KadiBot(llm, vectorstore)
+    
     with gr.Blocks() as demo:
         gr.Markdown("## KadiAPY - AI Coding-Assistant")
         gr.Markdown("AI Coding-Assistnat for KadiAPY based on RAG architecture powered by LLM")