Update app.py

app.py

@@ -74,52 +74,65 @@ initialize()
 
 
 
+def bot_kadi(history):
+    user_query = history[-1][0]   
+    response = kadiAPY_bot.process_query(user_query)
+    history[-1] = (user_query, response)
 
-import streamlit as st
+    yield history  
+
+
+
+# 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, ""
+
+
+import gradio as gr
 
 def main():
-    st.set_page_config(page_title="KadiAPY - AI Coding-Assistant", layout="wide")
-
-    st.title("KadiAPY - AI Coding-Assistant")
-    st.markdown("AI assistant for KadiAPY based on RAG architecture powered by LLM")
-
-    tab1, tab2 = st.tabs(["KadiAPY - AI Assistant", "Try Asking"])
-
-    with tab1:
-        st.write("### Kadi Bot")
-        
-        if "chat_history" not in st.session_state:
-            st.session_state.chat_history = []
-        
-        # Chatbot
-        chatbot_placeholder = st.empty()
-        question = st.text_input("Question", placeholder="Type in your question and press Enter")
-        
-        if st.button("Submit"):
-            if question:
-                st.session_state.chat_history.append({"User": question})
-                response = bot_kadi(question)  # Replace `bot_kadi` with your response generation function
-                st.session_state.chat_history.append({"Kadi Bot": response})
-            
-        if st.button("Clear"):
-            st.session_state.chat_history = []
-        
-        for exchange in st.session_state.chat_history:
-            for speaker, text in exchange.items():
-                st.write(f"**{speaker}:** {text}")
-        
-    with tab2:
-        st.write("### Try Asking...")
-        examples = [
-            "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",
-        ]
-        for example in examples:
-            st.write(f"- {example}")
-
-def bot_kadi(question):
-    # Dummy function for the AI response, replace with your logic
-    return "This is a placeholder response from Kadi Bot."
+    with gr.Blocks() as demo:
+        gr.Markdown("## KadiAPY - AI Coding-Assistant")
+        gr.Markdown("AI assistant for KadiAPY based on RAG architecture powered by LLM")
+
+        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",
+                            "I need a method to upload a file to a record. The id of the record is 3",
+                        ],
+                        inputs=user_txt,
+                        outputs=chatbot,
+                        fn=add_text,
+                        label="Try asking...",
+                        cache_examples=False,
+                        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)
+
+    demo.launch() 
 
+    
 if __name__ == "__main__":
-    main()
+    main()