4та спроба

app.py

@@ -13,26 +13,24 @@ st.title("Український Чат-бот")
 if "history" not in st.session_state:
     st.session_state.history = []
 
-user_input = st.text_input("Ви:", "")
+user_input = st.text_input("Ви:", key="user_input_enter")
 
 tokenizer, model = load_model()
 
-if st.button("Надіслати") or st.session_state.get("enter_pressed", False):
-    st.session_state.enter_pressed = False
+def send_message():
     if user_input:
-        inputs = tokenizer(st.session_state.history + [user_input], return_tensors="pt")
+        inputs = tokenizer(st.session_state.history + [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", "") != "":
+    send_message()
 
 if st.session_state.history:
     for i in range(0, len(st.session_state.history), 2):
         st.write(f"Ви: {st.session_state.history[i]}")
         if i + 1 < len(st.session_state.history):
-            st.write(f"Бот: {st.session_state.history[i+1]}")
-
-def set_enter_pressed():
-    st.session_state.enter_pressed = True
-
-st.text_input("Ви:", key="user_input_enter", on_change=set_enter_pressed)
+            st.write(f"Бот: {st.session_state.history[i+1]}")