adding stablity patch

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 from openai import OpenAI
 from params import params
 from database import get_client
-from add_data import start_troggin_off, create_client
 
-# CLIENT = get_client()
-CLIENT = None
+# from add_data import start_troggin_off, create_client
+
+CLIENT = get_client()
 APP_NAME: str = "Groove-GPT"
 history = []
 st.set_page_config(layout="wide")
@@ -13,13 +13,15 @@ st.set_page_config(layout="wide")
 # INFO
 st.title(APP_NAME)
 
+l_col, r_col = st.columns((3, 1))
 
-start_embedding = st.button("Hacker man")
-if start_embedding:
-    CLIENT = create_client()
-    start_troggin_off("./data", CLIENT)
+if "trigger" not in st.session_state:
+    st.session_state["trigger"] = False
+
+
+def on_enter():
+    st.session_state["trigger"] = True
 
-l_col, r_col = st.columns((3, 1))
 
 # param column
 with r_col:
@@ -34,10 +36,17 @@ with r_col:
 
 # input & response
 with l_col:
-    user_question: str = st.text_input("Enter your groovy questions here")
+    user_question: str = st.text_input(
+        "Enter your groovy questions here",
+        on_change=on_enter,
+    )
 
     # ON BUTTON CLICK
-    if submit_button & (access_key != "") & (user_question != ""):
+    if (
+        (submit_button | st.session_state["trigger"])
+        & (access_key != "")
+        & (user_question != "")
+    ):
         openai_client = OpenAI(api_key=access_key)
 
         with st.spinner("Loading..."):
@@ -69,7 +78,6 @@ with l_col:
         content = ""
         for i, chunk in enumerate(response):
             if chunk.choices[0].delta.content is not None:
-                # Append the chunk content to the string
                 content += chunk.choices[0].delta.content
 
                 text_placeholder.markdown(content)

database.py

@@ -2,7 +2,6 @@ import chromadb
 from chromadb.utils import embedding_functions
 
 def get_client():
-    # CONSTANTS
     client = chromadb.PersistentClient(path="./chromadb_linux/")
     MODEL_NAME: str = "mixedbread-ai/mxbai-embed-large-v1"  # ~ 0.5 gb
     COLLECTION_NAME: str = "scheme"

params.py

@@ -3,7 +3,7 @@ import streamlit as st
 
 def params():
     submit_button = st.button("Start Scheming")
-    
+
     # API Key
     access_key: str = st.text_input("Enter your gpt key here", type="password")
     st.markdown(