Update app.py

app.py

@@ -109,7 +109,7 @@ for message in st.session_state.messages:
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     # Display user message in chat message container and Add user message to chat history
-    pos = len(st.session_state.messages)
+    pos = len(st.session_state.messages)+1
     with st.chat_message("user"):
         col1, col2 = st.columns([9,1])
         col1.markdown(prompt)
@@ -118,7 +118,7 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     
     # Display assistant response in chat message container
     assistant = ask_assistant_stream(model_links[selected_model], st.session_state.messages, temp_values, max_token_value)
-    pos = len(st.session_state.messages)
+    pos = len(st.session_state.messages)+1
     if "stream" in assistant: 
         with st.chat_message("assistant"):
             col1, col2 = st.columns([9,1])