5та спроба

app.py

@@ -13,20 +13,26 @@ st.title("Український Чат-бот")
 if "history" not in st.session_state:
     st.session_state.history = []
 
-user_input = st.text_input("Ви:", key="user_input_enter")
+if "user_input" not in st.session_state:
+    st.session_state.user_input = ""
 
 tokenizer, model = load_model()
 
 def send_message():
-    if user_input:
-        inputs = tokenizer(st.session_state.history + [user_input], return_tensors="pt", padding=True, truncation=True)
+    if st.session_state.user_input:
+        inputs = tokenizer(st.session_state.history + [st.session_state.user_input], return_tensors="pt", padding=True, truncation=True)
         with torch.no_grad():
             outputs = model.generate(**inputs, max_length=100)
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        st.session_state.history.extend([user_input, response])
-        st.session_state.user_input_enter = "" # clear the input field after sending
-        
-if st.button("Надіслати") or st.session_state.get("user_input_enter", "") != "":
+        st.session_state.history.extend([st.session_state.user_input, response])
+        st.session_state.user_input = "" # clear the input field after sending
+
+def update_user_input():
+    st.session_state.user_input = st.session_state.temp_user_input # update the user input on change
+
+st.text_input("Ви:", key="temp_user_input", on_change=update_user_input)
+
+if st.button("Надіслати"):
     send_message()
 
 if st.session_state.history: