Перенесено st.session_state.temp_user_input = "" всередину функції send_message, після того як повідомлення було успішно оброблено та додано до історії. Тепер очищення текстового поля відбувається через оновлення st.session_state.temp_user_input у контексті функції, викликаної дією користувача (натисканням кнопки або зміною тексту).

app.py

@@ -25,7 +25,8 @@ def send_message():
             outputs = model.generate(**inputs, max_length=100)
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         st.session_state.history.extend([st.session_state.user_input, response])
-        st.session_state.user_input = "" # clear the input field after sending
+        st.session_state.user_input = "" # clear the stored user input
+        st.session_state.temp_user_input = "" # clear the text input field
 
 def update_user_input():
     st.session_state.user_input = st.session_state.temp_user_input
@@ -35,8 +36,6 @@ st.text_input("Ви:", key="temp_user_input", on_change=update_user_input)
 if st.button("Надіслати") or st.session_state.get("user_input", "") != "":
     if st.session_state.get("user_input", "") != "":
         send_message()
-        # Reset the temporary input to avoid immediate resend
-        st.session_state.temp_user_input = ""
 
 if st.session_state.history:
     for i in range(0, len(st.session_state.history), 2):