Update app.py

app.py

@@ -27,41 +27,40 @@ hf_api = HfApi()
 LLM_MODEL_NAME = config["llm_model_name"]
 LLM_MODEL_TEMPERATURE = float(config["llm_model_temperature"])
 
-def initialize():
-    global kadiAPY_ragchain
 
-    vectorstore = get_chroma_vectorstore(get_SFR_Code_embedding_model(), vectorstore_path)
-    llm = get_groq_llm(LLM_MODEL_NAME, LLM_MODEL_TEMPERATURE, GROQ_API_KEY)
+class KadiBot:
+    def __init__(self):
+        # Initialize vector store and language model
+        vectorstore = get_chroma_vectorstore(get_SFR_Code_embedding_model(), vectorstore_path)
+        llm = get_groq_llm(LLM_MODEL_NAME, LLM_MODEL_TEMPERATURE, GROQ_API_KEY)
 
-    kadiAPY_ragchain = KadiApyRagchain(llm, vectorstore)
+        # Initialize RAG chain
+        self.kadiAPY_ragchain = KadiApyRagchain(llm, vectorstore)
 
-initialize()
+    def bot_kadi(self, chat_history):
+        user_query = chat_history[-1][0]   
+        response = self.kadiAPY_ragchain.process_query(user_query, chat_history)
+        chat_history[-1] = (user_query, response)
 
+        return chat_history  
 
 
-def bot_kadi(chat_history):
-    user_query = chat_history[-1][0]   
-    response = kadiAPY_ragchain.process_query(user_query, chat_history)
-    chat_history[-1] = (user_query, response)
-
-    return chat_history  
-
-
-import gradio as gr
-
 def add_text_to_chat_history(chat_history, user_input):
-    
     chat_history = chat_history + [(user_input, None)]
     return chat_history, ""
 
-    
+
 def show_history(chat_history):
     return chat_history
-    
+
+
 def reset_all():
     return [], "", ""
-    
+
+
 def main():
+    kadi_bot = KadiBot()  # Initialize the KadiBot class
+
     with gr.Blocks() as demo:
         gr.Markdown("## KadiAPY - AI Coding-Assistant")
         gr.Markdown("AI assistant for KadiAPY based on RAG architecture powered by LLM")
@@ -95,10 +94,10 @@ def main():
                     )
         
         # Use the state to persist chat history between interactions
-        user_txt.submit(add_text_to_chat_history, [chat_history, user_txt], [chat_history, user_txt]).then(show_history,[chat_history], [chatbot])\
-                  .then(bot_kadi, [chat_history], [chatbot])          
-        submit_btn.click(add_text_to_chat_history, [chat_history, user_txt], [chat_history, user_txt]).then(show_history,[chat_history], [chatbot])\
-                  .then(bot_kadi, [chat_history], [chatbot])        
+        user_txt.submit(add_text_to_chat_history, [chat_history, user_txt], [chat_history, user_txt]).then(show_history, [chat_history], [chatbot])\
+                  .then(kadi_bot.bot_kadi, [chat_history], [chatbot])          
+        submit_btn.click(add_text_to_chat_history, [chat_history, user_txt], [chat_history, user_txt]).then(show_history, [chat_history], [chatbot])\
+                  .then(kadi_bot.bot_kadi, [chat_history], [chatbot])        
         clear_btn.click(
             reset_all, 
             None,  
@@ -107,5 +106,6 @@ def main():
         )
     demo.launch()
 
+
 if __name__ == "__main__":
     main()