Update app.py

app.py

@@ -54,25 +54,28 @@ def add_text_to_chatbot(chat_history, user_input):
     
     chat_history = chat_history + [(user_input, None)]
     return chat_history, ""
-
+    
 
 def 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"):
             with gr.Row():
                 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",
@@ -85,11 +88,20 @@ def main():
                         cache_examples=False,
                         examples_per_page=3,
                     )
-
-        submit_btn.click(add_text_to_chatbot, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot],[chatbot])
-        clear_btn.click(lambda: ([], ""), None, [chatbot, user_txt])
+        
+        # Use the state to persist chat history between interactions
+        submit_btn.click(add_text_to_chatbot, [chat_history, user_txt], [chat_history, user_txt])\
+                  .then(bot_kadi, [chat_history], [chatbot])
+        
+        clear_btn.click(lambda: ([], ""), None, [chat_history, chatbot, user_txt])
 
     demo.launch()
 
+if __name__ == "__main__":
+    main()
+
+
+
+
 if __name__ == "__main__":
     main()