Update app.py

app.py

@@ -139,8 +139,5 @@ if prompt := st.text_input(f"Hi, I'm {selected_model}, let's chat"):
         st.session_state.messages.append(("assistant", assistant_response))
 
 # Display chat messages from history
-for role, message in st.session_state.messages:
-    if role == "user":
-        st.text_input("You:", message, disabled=True)
-    elif role == "assistant":
-        st.text_area("Assistant:", message, disabled=True)
+    # Add assistant response to chat history
+    st.session_state.messages.append(("assistant", assistant_response))