Update app.py

app.py

@@ -28,7 +28,9 @@ input_text = st.chat_input("Enter your message:")
 if input_text:
     # Add the user's message to the conversation history
     session_state.conversation_history.append(input_text)
-    st.chat_message("user", input_text)
+
+    # Display the user's message
+    st.write("**User:**", input_text)
 
     # Create conversation history string
     history_string = "\n".join(session_state.conversation_history)
@@ -44,4 +46,6 @@ if input_text:
 
     # Add the model's response to the conversation history
     session_state.conversation_history.append(response)
-    st.chat_message("assistant", response)
+
+    # Display the model's response
+    st.write("**Assistant:**", response)