Update app.py

app.py

@@ -94,10 +94,11 @@ with container:
             
 
             with response_container:
-                for i in range(len(st.session_state['messages'])):
-                        if (i % 2) == 0:
-                            message(st.session_state['messages'][i], is_user=True, key=str(i) + '_user')
-                        else:
-                            message(st.session_state['messages'][i], key=str(i) + '_AI')
+                for i, message in enumerate(st.session_state['messages']):
+                    # Check if message is from user or AI
+                    if (i % 2) == 0:  # User messages
+                        st.text_area(label="", value=message, key=f"user_{i}", disabled=True)
+                    else:  # AI responses
+                        st.text_area(label="", value=message, key=f"ai_{i}", disabled=True)