Update app.py

app.py

@@ -23,6 +23,10 @@ def reset_conversation():
     st.session_state.messages = []
     return None
 
+# Initialize session state for 'messages' if it doesn't exist
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
 # Define classification options
 classification_types = ["Sentiment Analysis", "Binary Classification", "Multi-Class Classification"]
 
@@ -129,3 +133,4 @@ if task == "Data Labeling":
 else:
     st.sidebar.write("This feature will allow you to generate new data. Coming soon!")
 
+