Update app.py

app.py

@@ -19,15 +19,20 @@ st.title("Чат с GPT")
 if 'messages' not in st.session_state:
     st.session_state.messages = []
 
+# Функция для отображения сообщений
+def display_messages():
+    for msg in st.session_state.messages:
+        if msg['role'] == 'user':
+            st.markdown(f"<div style='text-align: right; color: blue;'><b>Вы:</b> {msg['content']}</div>", unsafe_allow_html=True)
+        else:
+            st.markdown(f"<div style='text-align: left; color: green;'><b>GPT:</b> {msg['content']}</div>", unsafe_allow_html=True)
+
 # Отображение предыдущих сообщений
-for msg in st.session_state.messages:
-    if msg['role'] == 'user':
-        st.markdown(f"<div style='text-align: right; color: blue;'>Вы: {msg['content']}</div>", unsafe_allow_html=True)
-    else:
-        st.markdown(f"<div style='text-align: left; color: green;'>GPT: {msg['content']}</div>", unsafe_allow_html=True)
+st.header("История сообщений")
+display_messages()
 
 # Поле ввода для нового сообщения
-user_input = st.text_input("Введите ваше сообщение:")
+user_input = st.text_input("Введите ваше сообщение:", key="input")
 
 # Обработка отправки сообщения
 if st.button("Отправить"):
@@ -41,7 +46,11 @@ if st.button("Отправить"):
         # Сохранение ответа GPT
         st.session_state.messages.append({"role": "assistant", "content": gpt_response})
 
+        # Очистка поля ввода
+        st.session_state.input = ""
+
 # Кнопка для сброса чата
 if st.button("Сбросить чат"):
     st.session_state.messages = []
+    st.session_state.input = ""