RAG prepaired queries adde

pages/RAG.py

@@ -109,14 +109,20 @@ questions = [
     "Что такое t-test и для чего он применяется? расскажи на продвинутом уровне шаги"
 ]
 
-# Поле для выбора вопроса
-selected_question = st.selectbox("📝 Выберите ваш вопрос:", questions)
+# Виджет выбора между подготовленным вопросом и вводом собственного
+question_option = st.radio("Выберите способ ввода вопроса:", ("Выбрать из списка", "Ввести свой вопрос"))
+
+if question_option == "Выбрать из списка":
+    selected_question = st.selectbox("📝 Выберите ваш вопрос:", questions)
+    question = selected_question
+else:
+    question = st.text_input("📝 Введите ваш вопрос:", '')
 
 # Кнопка для запуска поиска и генерации ответа
 if st.button("🚀 Поиск и генерация ответа"):
-    if selected_question:
+    if question:
         # Генерация ответа на вопрос
-        answer, documents = answer_question(selected_question, embedding_retriever, client)
+        answer, documents = answer_question(question, embedding_retriever, client)
 
         if answer:
             # Оформление ответа
@@ -127,6 +133,9 @@ if st.button("🚀 Поиск и генерация ответа"):
 
         else:
             st.warning("⚠️ Не удалось получить ответ от модели.")
+    else:
+        st.warning("⚠️ Пожалуйста, введите запрос.")
+