add Bayesian simulator button

app.py

@@ -1,17 +1,16 @@
 import streamlit as st
 from assist.chat import chat as chat_plugin
-from assist.heatmap import get_heatmap_figure
 
 st.title("RepoSage Chatbot Demo")
 
-# Tab or expander for heatmap
-with st.expander("🔢 View Token Similarity Heatmap"):
-    if st.button("Show Heatmap"):
-        fig = get_heatmap_figure("tensor.pt")
-        st.pyplot(fig)
-
-# Chat UI
+# 1) Change the label to make it obvious we're asking a question
 question = st.text_input("Ask RepoSage a question:", "")
-if st.button("Ask RepoSage"):
-    response = chat_plugin(question)
-    st.write(response)
+
+col1, col2 = st.columns(2)
+with col1:
+    if st.button("Ask Embedding RepoSage"):
+        st.write(embed_chat(question))
+
+with col2:
+    if st.button("Ask Bayesian RepoSage"):
+        st.write(bayes_chat(question))