Update app.py

app.py

@@ -9,7 +9,6 @@ from vectorstore import get_chroma_vectorstore
 from embeddings import get_SFR_Code_embedding_model
 from kadiApy_ragchain import KadiApyRagchain
 
-# Load environment variables from .env file
 load_dotenv()
 
 vectorstore_path = "data/vectorstore"
@@ -23,26 +22,29 @@ with open("config.json", "r") as file:
 login(HF_TOKEN)
 hf_api = HfApi()
 
-# Access the values
 LLM_MODEL_NAME = config["llm_model_name"]
 LLM_MODEL_TEMPERATURE = float(config["llm_model_temperature"])
 
 
 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)
 
-        # Initialize RAG chain
         self.kadiAPY_ragchain = KadiApyRagchain(llm, vectorstore)
 
-    def bot_kadi(self, chat_history):
-        user_query = chat_history[-1][0]   
+    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)
 
-        return chat_history  
+        return chat_history
+
 
 
 def add_text_to_chat_history(chat_history, user_input):