Update app.py

app.py

@@ -5,6 +5,7 @@ from streamlit_extras.colored_header import colored_header
 from streamlit_extras.add_vertical_space import add_vertical_space
 from streamlit_mic_recorder import speech_to_text
 from model_pipeline import ModelPipeLine
+from q_learning_chatbot import QLearningChatbot
 
 from gtts import gTTS
 from io import BytesIO
@@ -14,6 +15,14 @@ final_chain = mdl.create_final_chain()
 
 st.set_page_config(page_title="PeacePal")
 
+# Define states and actions
+states = [
+    "Negative",
+    "Moderately Negative",
+    "Neutral",
+    "Moderately Positive",
+    "Positive",
+]
 
 # Add logo to the sidebar
 #logo_path = os.path.join('images', 'logo.jpeg')