Update app.py

app.py

@@ -43,33 +43,28 @@ initialize()
 
 
 
-def bot_kadi(history):
-    user_query = history[-1][0]   
-    response = kadiAPY_bot.process_query(user_query)
-    history[-1] = (user_query, response)
-
-    yield history  
-
 
+def bot_kadi(history, session_state):
+    user_query = history[-1][0]   
 
-# Gradio utils
-def check_input_text(text):
-    if not text:
-        gr.Warning("Please input a question.")
-        raise TypeError
-    return True
-
-def add_text(history, text):
-    history = history + [(text, None)]
-    yield history, ""
+    response = kadiAPY_bot.process_query({
+        "query": user_query,
+        "history": session_state["conversation"]  # Pass full conversation history
+    })
 
+    # Update the session history
+    history[-1] = (user_query, response)
+    session_state["conversation"].append({"query": user_query, "response": response})
 
-import gradio as gr
+    yield history
 
+# Gradio utils with session state
 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")
+        
+        session_state = gr.State({"conversation": []}) 
 
         with gr.Tab("KadiAPY - AI Assistant"):
             with gr.Row():
@@ -85,7 +80,7 @@ def main():
 
                     gr.Examples(
                         examples=[
-                            "Write me a python script with which can convert plain JSON to a Kadi4Mat-compatible extra metadata structure",
+                            "Write me a python script which can convert plain JSON to a Kadi4Mat-compatible extra metadata structure",
                             "I need a method to upload a file to a record. The id of the record is 3",
                         ],
                         inputs=user_txt,
@@ -96,12 +91,18 @@ def main():
                         examples_per_page=3,
                     )
 
-            user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
-            submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
-            clear_btn.click(lambda: None, None, chatbot, queue=False)
+            # Bind input and button to modified bot_kadi
+            user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(
+                bot_kadi, [chatbot, session_state], [chatbot]
+            )
+            submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(
+                bot_kadi, [chatbot, session_state], [chatbot]
+            )
+            clear_btn.click(lambda: None, None, chatbot, queue=False).then(
+                lambda: {"conversation": []}, None, session_state, queue=False  # Clear session state
+            )
 
-    demo.launch() 
+    demo.launch()
 
-    
 if __name__ == "__main__":
     main()