Update app_v2.py

app_v2.py

@@ -139,7 +139,7 @@ def chat_interface():
 
     if user_input:
         # Add user message to chat history
-        st.session_state.chat_history.append({"role": "user", "content": user_input})
+        # st.session_state.chat_history.append({"role": "user", "content": user_input})
 
         # Prepare the prompt
         prompt = prepare_prompt(user_input, st.session_state.chat_history)
@@ -174,6 +174,7 @@ def chat_interface():
             print(prompt_data)
             stream = generate_response(watsonx_llm, prompt_data, params)
             response = st.write_stream(stream)
+            st.session_state.chat_history.append({"role": "user", "content": user_input})
 
         # Add AI response to chat history
         st.session_state.chat_history.append({"role": "Jimmy", "content": response})