Update app.py

app.py

@@ -94,11 +94,10 @@ with container:
             
 
             with response_container:
-                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)
+                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')