readability

app.py

@@ -4,6 +4,7 @@ from chromadb.utils import embedding_functions
 from sentence_transformers import SentenceTransformer
 from openai import OpenAI
 
+# CONSTANTS
 client = chromadb.PersistentClient(path="./chromadb/")
 MODEL_NAME: str = "mixedbread-ai/mxbai-embed-large-v1" # ~ 0.5 gb
 COLLECTION_NAME: str = "scheme"
@@ -16,25 +17,32 @@ DATA_AVAL: bool = schemer.count() > 0
 APP_NAME: str = "Groove-GPT"
 history = []
 
+# INFO 
 st.title(APP_NAME)
 st.header("What is Groovy-GPT?")
 st.write("Groovy-GPT is a RAG (Retrieval-Augmented Generation) model that uses ChromaDB to retrieve relevant documents and then uses OpenAI's models to generate a response.")
 st.write("The model is trained on the MIT Scheme textbook and a handful of Discrete Math and Paradigms related content that Professor Troeger posted")
 st.write("Data Avaliable: ", DATA_AVAL)
 
+# INPUTS
 user_question: str = st.text_area("Enter your groovy questions here")
+
 remember_chat_history = st.toggle("Remember This Chat's History")
+
 temperature = st.slider(label="Creativity of Model", min_value=0, max_value=2, value=1, step='float')
+
 st.markdown("High creativity will make it go crazy - keep it low")
 
 access_key: str = st.text_input("Enter your gpt key here", type="password")
 st.markdown("*For more information about how to get an access key, read [this article](https://platform.openai.com/api-keys). Make sure it has money in it ☠️*", unsafe_allow_html=True)
+
 gpt_type: str = st.selectbox(label="Choose GPT Type", options=["gpt-3.5-turbo", "gpt-3.5-turbo-1106", "gpt-3.5-turbo-0125", "gpt-4-32k-0613", "gpt-4-0613", "gpt-4-0125-preview"], index=0)
 st.markdown("*For more information about GPT types, read [this article](https://platform.openai.com/docs/models).*", unsafe_allow_html=True)
 
 st.divider()
 
-if st.button('Query Database') & (access_key != "") & (user_question != ""):
+# ON BUTTON CLICK
+if st.button('Start Scheming') & (access_key != "") & (user_question != ""):
     openai_client = OpenAI(api_key=access_key)
 
     with st.spinner('Loading...'):
@@ -60,5 +68,7 @@ if st.button('Query Database') & (access_key != "") & (user_question != ""):
         
     st.header("Prof Says ...")
     st.write(response.choices[0].message.content)
+else:
+    st.write("Please provide an input and (valid) API key")