Update app.py

app.py

@@ -23,50 +23,39 @@ 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"])
 
 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)
+    return KadiApyRagchain(llm, vectorstore)
 
-    kadiAPY_ragchain = KadiApyRagchain(llm, vectorstore)
-
-initialize()
-
-
-
-def bot_kadi(chat_history):
+def bot_kadi(chat_history, kadiAPY_ragchain):
     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
-
+#gradio utils
 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():
+    kadiAPY_ragchain = initialize()  # Initialize inside main()
+
     with gr.Blocks() as demo:
         gr.Markdown("## KadiAPY - AI Coding-Assistant")
         gr.Markdown("AI assistant for KadiAPY based on RAG architecture powered by LLM")
         
-        # Create a state for session management
         chat_history = gr.State([])
 
         with gr.Tab("KadiAPY - AI Assistant"):
@@ -74,13 +63,13 @@ def main():
                 with gr.Column(scale=10):
                     chatbot = gr.Chatbot([], elem_id="chatbot", label="Kadi Bot", bubble_full_width=False, show_copy_button=True, height=600)
                     user_txt = gr.Textbox(label="Question", placeholder="Type in your question and press Enter or click Submit")
-                    
+
                     with gr.Row():
                         with gr.Column(scale=1):
                             submit_btn = gr.Button("Submit", variant="primary")
                         with gr.Column(scale=1):
                             clear_btn = gr.Button("Clear", variant="stop")
-                    
+
                     gr.Examples(
                         examples=[
                             "Write me a python script with which can convert plain JSON to a Kadi4Mat-compatible extra metadata structure",
@@ -93,18 +82,22 @@ def main():
                         cache_examples=False,
                         examples_per_page=3,
                     )
-        
-        # 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(bot_kadi, [chat_history, kadiAPY_ragchain], [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, kadiAPY_ragchain], [chatbot])        
+
         clear_btn.click(
             reset_all, 
             None,  
             [chat_history, chatbot, user_txt], 
             queue=False
         )
+
     demo.launch()
 
 if __name__ == "__main__":