Update app.py

app.py

@@ -46,6 +46,7 @@ def chat_bubble(message, is_user=True):
 
 # Function to load answer from the model
 def load_answer(question):
+    # Add the user question to sessionMessages
     st.session_state.sessionMessages.append(HumanMessage(content=question))
     assistant_answer = llm.invoke(st.session_state.sessionMessages)
     
@@ -56,14 +57,14 @@ def load_answer(question):
         st.session_state.sessionMessages.append(AIMessage(content=assistant_answer))
         return assistant_answer
 
-# Chat history display
+# Display the chat history
 for message in st.session_state.sessionMessages:
     if isinstance(message, HumanMessage):
         chat_bubble(message.content, is_user=True)
     elif isinstance(message, AIMessage):
         chat_bubble(message.content, is_user=False)
 
-# Input area
+# Input area for new prompt
 user_input = st.text_input("You: ", key="input", placeholder="Type your question here...")
 
 # Button for submission