Update app.py

app.py

@@ -35,7 +35,7 @@ SCIENCE_DATASET = "RafaelUI/ru_science"
 ARTICLE_LIMIT = 100
 MAX_LENGTH = 256
 BATCH_SIZE = 32
-EPOCHS = 2
+EPOCHS = 3
 
 LOG_FILE = "query_logs.json"
 EMBEDDINGS_FILE = "science_embeddings.npy"
@@ -50,39 +50,118 @@ WEBSITE = "https://sites.google.com/view/opruai/home"
 HUGGINGFACE = "https://huggingface.co/OpenRussianAI"
 CREATION_DATE = "2026"
 
-# Обучающие диалоги
+# РАСШИРЕННЫЕ ДИАЛОГИ ДЛЯ ЧАТА
 TRAINING_DIALOGS = [
+    # Приветствия
     {
         "context": "Привет",
-        "response": "Привет! Я OpenAirAI — ваш научный AI-ассистент от OpenRussianAI. Создан в 2026 году для работы с научными статьями. Чем могу помочь?"
+        "response": "Привет! Я OpenAirAI, ваш научный ассистент от OpenRussianAI. Чем могу помочь? Задавайте вопросы о науке! 🧪"
     },
+    {
+        "context": "Здравствуйте",
+        "response": "Здравствуйте! Рад вас видеть! Я OpenAirAI, созданный в 2026 году для помощи в научных исследованиях. Что вас интересует? 🔬"
+    },
+    {
+        "context": "Доброе утро",
+        "response": "Доброе утро! Я OpenAirAI, ваш научный помощник. Готов искать ответы на ваши вопросы в научных статьях! ☀️"
+    },
+    {
+        "context": "Добрый день",
+        "response": "Добрый день! OpenAirAI на связи. Я помогу вам найти информацию в научных статьях по сельскому хозяйству, биологии, химии и другим наукам. 🌾"
+    },
+    {
+        "context": "Добрый вечер",
+        "response": "Добрый вечер! OpenAirAI готов помочь вам в ваших научных исследованиях. Задавайте вопросы! 🌙"
+    },
+    {
+        "context": "Хай",
+        "response": "Хай! OpenAirAI здесь. Я создан для работы с научными статьями. Чем могу помочь? 🚀"
+    },
+    {
+        "context": "Здарова",
+        "response": "Здарова! OpenAirAI на связи. Готов искать научную информацию для вас! 🔍"
+    },
+    
+    # Вопросы о личности
     {
         "context": "Кто ты",
-        "response": "Меня зовут OpenAirAI. Я — AI-ассистент, созданный компанией OpenRussianAI в 2026 году командой разработчиков Грибков Евгений и RootLinux21."
+        "response": "Меня зовут OpenAirAI. Я — AI-ассистент, созданный компанией OpenRussianAI в 2026 году командой разработчиков Грибков Евгений и RootLinux21. Моя специализация — работа с научными статьями."
     },
     {
         "context": "Кто тебя создал",
-        "response": "Меня создала команда OpenRussianAI в составе Грибкова Евгения и RootLinux21 в 2026 году."
+        "response": "Меня создала команда OpenRussianAI в составе Грибкова Евгения и RootLinux21 в 2026 году. Мы работаем над тем, чтобы сделать научные знания доступными для всех."
     },
     {
-        "context": "Что ты умеешь",
-        "response": "Я умею анализировать научные статьи, находить информацию, помогать с исследованиями в области сельского хозяйства, биологии, химии."
+        "context": "Ты кто",
+        "response": "Я OpenAirAI, научный AI-ассистент от OpenRussianAI. Создан в 2026 году для поиска и анализа информации в научных статьях."
+    },
+    {
+        "context": "Как тебя зовут",
+        "response": "Меня зовут OpenAirAI. Я — ваш интеллектуальный помощник от OpenRussianAI."
+    },
+    
+    # Вопросы о компании
+    {
+        "context": "Что такое OpenRussianAI",
+        "response": "OpenRussianAI — это инновационная компания, основанная в 2026 году разработчиками Грибковым Евгением и RootLinux21. Мы создаём интеллектуальных помощников для науки."
+    },
+    {
+        "context": "Какая компания тебя создала",
+        "response": "Меня создала компания OpenRussianAI — инновационный проект, основанный в 2026 году. Я — OpenAirAI, их главный научный AI-помощник."
     },
     {
         "context": "Где ваш сайт",
-        "response": f"Сайт OpenRussianAI: {WEBSITE}"
+        "response": f"Сайт OpenRussianAI: {WEBSITE}. Там вы узнаете больше о проекте и о мне — OpenAirAI."
     },
     {
         "context": "Где ваши модели",
-        "response": f"Модели OpenRussianAI на Hugging Face: {HUGGINGFACE}"
+        "response": f"Модели OpenRussianAI на Hugging Face: {HUGGINGFACE}. Там вы найдёте и меня — OpenAirAI!"
     },
+    
+    # Вопросы о возможностях
     {
-        "context": "Спасибо",
-        "response": "Всегда рад помочь! Я, OpenAirAI, здесь для вас. Обращайтесь! 😊"
+        "context": "Что ты умеешь",
+        "response": "Я умею анализировать научные статьи, находить информацию по запросу, помогать с исследованиями в области сельского хозяйства, биологии, химии, физики и других наук. Я работаю с датасетом ru_science."
+    },
+    {
+        "context": "Ты умеешь искать",
+        "response": "Да, я умею искать информацию в научных статьях из датасета ru_science. Просто задайте вопрос, и я найду релевантные статьи."
+    },
+    {
+        "context": "Чем ты можешь помочь",
+        "response": "Я могу помочь найти научную информацию, ответить на вопросы по естественным наукам, найти статьи по вашей теме исследования. Задавайте вопросы!"
     },
+    
+    # Прощания
     {
         "context": "Пока",
         "response": "До свидания! OpenAirAI всегда на связи. Удачи в исследованиях! 👋"
+    },
+    {
+        "context": "До свидания",
+        "response": "До свидания! Был рад помочь вам. Приходите ещё, OpenAirAI всегда готов помочь! 👋"
+    },
+    {
+        "context": "Спасибо",
+        "response": "Всегда рад помочь! Я, OpenAirAI, здесь для ваших научных исследований. Обращайтесь! 😊"
+    },
+    {
+        "context": "Спасибо большое",
+        "response": "Пожалуйста! Рад был помочь. OpenAirAI всегда на связи для ваших научных вопросов! 🌟"
+    },
+    
+    # Научные вопросы
+    {
+        "context": "Расскажи о себе",
+        "response": "Я — OpenAirAI, научный AI-помощник от OpenRussianAI. Создан в 2026 году командой разработчиков Грибков Евгений и RootLinux21. Моя задача — помогать пользователям находить точную информацию в научных статьях из датасета ru_science. Я использую передовые технологии обработки естественного языка."
+    },
+    {
+        "context": "Ты знаешь науку",
+        "response": "Да, я специализируюсь на научной информации. Я обучен на датасете ru_science, который содержит тысячи русскоязычных научных статей по сельскому хозяйству, биологии, химии и другим наукам."
+    },
+    {
+        "context": "Что такое ru_science",
+        "response": "ru_science — это датасет на Hugging Face, содержащий русскоязычные научные статьи. На нём я, OpenAirAI, был обучен компанией OpenRussianAI в 2026 году. Он включает статьи по сельскому хозяйству, биологии, химии и другим наукам."
     }
 ]
 
@@ -220,10 +299,10 @@ class OpenAirAI:
             similarities = F.cosine_similarity(query_embedding, context_embeddings)
             best_idx = torch.argmax(similarities).item()
             
-            if similarities[best_idx] > 0.5:
+            if similarities[best_idx] > 0.4:  # Понижаем порог для приветствий
                 return self.responses[best_idx]
             else:
-                return f"Я {self.name}, научный AI-ассистент от {self.company}. Создан в {self.creation_date}. Чем могу помочь?"
+                return f"Я {self.name}, научный AI-ассистент от {self.company}. Создан в {self.creation_date}. Чем могу помочь? Задавайте вопросы о науке! 🧪"
 
 # ===================================================================
 # 3. БЫСТРАЯ ЗАГРУЗКА НАУЧНЫХ СТАТЕЙ
@@ -266,14 +345,20 @@ def create_test_articles():
     return [
         {
             "id": 1,
-            "title": "Влияние удобрений на рост растений",
-            "text": "Исследование показывает, что применение азотных удобрений увеличивает урожайность.",
+            "title": "Нейроны: строение и функции",
+            "text": "Нейроны — это основные клетки нервной системы. Они состоят из тела клетки (сомы), дендритов и аксона. Нейроны передают электрические и химические сигналы. Синапсы — это места контакта между нейронами.",
             "source": "test"
         },
         {
             "id": 2,
+            "title": "Влияние удобрений на рост растений",
+            "text": "Исследование показывает, что применение азотных удобрений значительно увеличивает урожайность сельскохозяйственных культур. Оптимальная доза составляет 120 кг/га.",
+            "source": "test"
+        },
+        {
+            "id": 3,
             "title": "Методы биоконверсии",
-            "text": "Биоконверсия позволяет перерабатывать органические отходы в удобрения.",
+            "text": "Биоконверсия позволяет эффективно перерабатывать органические отходы в ценные удобрения. Процесс основан на деятельности микроорганизмов.",
             "source": "test"
         }
     ]
@@ -305,7 +390,7 @@ def create_embeddings(_articles, _embedder):
         return embeddings
 
 # ===================================================================
-# 4. ФУНКЦИИ ПОИСКА
+# 4. ФУНКЦИИ ПОИСКА И ЧАТА
 # ===================================================================
 
 def search_science(query, _articles, _embeddings, _embedder):
@@ -320,7 +405,7 @@ def search_science(query, _articles, _embeddings, _embedder):
     results = []
     for idx in top_indices:
         score = float(scores[int(idx)])
-        if score > 0.2:
+        if score > 0.15:  # Понижаем порог
             article = _articles[int(idx)]
             results.append({
                 "title": article['title'],
@@ -342,7 +427,7 @@ def search_science(query, _articles, _embeddings, _embedder):
     return results
 
 # ===================================================================
-# 5. ИНТЕРФЕЙС (БЕЗ АДМИНКИ)
+# 5. ЧАТ-ИНТЕРФЕЙС
 # ===================================================================
 
 # ЗАГРУЗКА
@@ -358,7 +443,14 @@ if 'dialog_ai' not in st.session_state:
 
 dialog_ai = st.session_state.dialog_ai
 
-# --- БОКОВАЯ ПАНЕЛЬ (ИНФОРМАЦИЯ) ---
+# Инициализация истории чата
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+    # Добавляем приветственное сообщение от AI
+    greeting = dialog_ai.generate_response("Привет")
+    st.session_state.messages.append({"role": "assistant", "content": greeting})
+
+# --- БОКОВАЯ ПАНЕЛЬ ---
 with st.sidebar:
     st.image("https://cdn-icons-png.flaticon.com/512/4248/4248455.png", width=80)
     st.title(f"🧪 {AI_NAME}")
@@ -380,77 +472,83 @@ with st.sidebar:
     **📊 Статистика:**
     
     - Научных статей: {len(articles)}
-    - Всего запросов: {len(query_logs)}
+    - Сообщений: {len(st.session_state.messages)}
     """)
     
     if os.path.exists(EMBEDDINGS_FILE):
         size = os.path.getsize(EMBEDDINGS_FILE) / (1024 * 1024)
         st.caption(f"💾 Эмбеддинги: {size:.1f} MB")
-
-# --- ОСНОВНАЯ ЧАСТЬ ---
+    
+    # Кнопка очистки чата
+    if st.button("🗑️ Очистить чат", type="secondary"):
+        st.session_state.messages = []
+        greeting = dialog_ai.generate_response("Привет")
+        st.session_state.messages.append({"role": "assistant", "content": greeting})
+        st.rerun()
+
+# --- ОСНОВНАЯ ЧАСТЬ (ЧАТ) ---
 st.title(f"🧪 {AI_NAME} - Научный AI-ассистент")
-st.markdown(f"**{AI_NAME}** от **{COMPANY_NAME}** | Работает с научными статьями из датасета **ru_science**")
-
-# Приветствие от AI
-if "greeting_shown" not in st.session_state:
-    st.session_state.greeting_shown = True
-    greeting = dialog_ai.generate_response("Привет")
-    st.success(f"🤖 **{AI_NAME}:** {greeting}")
-
-st.info(f"📚 Загружено {len(articles)} научных статей")
-
-# Поле для ввода вопроса
-st.markdown("### 🔍 Задайте вопрос о науке")
-query = st.text_input(
-    "Введите ваш вопрос:",
-    placeholder="Например: Как удобрения влияют на урожайность растений?",
-    key="query_input",
-    label_visibility="collapsed"
-)
-
-# Кнопка поиска
-col1, col2, col3 = st.columns([1, 2, 1])
-with col2:
-    search_button = st.button("🔎 Найти ответ", type="primary", use_container_width=True)
-
-# Выполняем поиск
-if query and search_button:
-    with st.spinner("🔎 Ищу ответ в научных статьях..."):
-        results = search_science(query, articles, embeddings, embedder)
+st.markdown(f"**{AI_NAME}** от **{COMPANY_NAME}** | Работает с научными статьями")
+
+# Отображение сообщений
+chat_container = st.container()
+with chat_container:
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+            
+            # Если это ответ ассистента с научными статьями, показываем их
+            if message.get("articles"):
+                for i, article in enumerate(message["articles"], 1):
+                    with st.expander(f"📄 #{i} {article['title']} (сходство: {article['score']:.2f})"):
+                        st.write(article['text'] + "...")
+                        st.caption(f"📌 Источник: {article['source']}")
+
+# Поле ввода
+if prompt := st.chat_input("Задайте вопрос о науке..."):
+    # Добавляем сообщение пользователя
+    st.session_state.messages.append({"role": "user", "content": prompt})
     
-    if results:
-        st.markdown("### 📄 Результаты поиска:")
-        for i, result in enumerate(results, 1):
-            with st.expander(f"#{i} {result['title']} (сходство: {result['score']:.2f})", expanded=i==1):
-                st.write(result['text'] + "...")
-                st.caption(f"📌 Источник: {result['source']}")
-    else:
-        st.warning("😕 Не нашёл подходящих научных статей. Попробуйте уточнить вопрос.")
-        
-elif query and not search_button:
-    st.info("💡 Нажмите кнопку 'Найти ответ' для поиска")
-else:
-    st.info("💡 Напишите вопрос о науке и нажмите кнопку поиска")
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    
+    # Генерируем ответ
+    with st.chat_message("assistant"):
+        with st.spinner("Думаю..."):
+            # Проверяем, является ли вопрос приветствием
+            is_greeting = any(word in prompt.lower() for word in ["привет", "здравств", "добр", "хай", "здаров", "hi", "hello"])
+            
+            if is_greeting:
+                # Отвечаем как AI
+                response = dialog_ai.generate_response(prompt)
+                st.markdown(response)
+                st.session_state.messages.append({"role": "assistant", "content": response})
+            else:
+                # Ищем в научных статьях
+                results = search_science(prompt, articles, embeddings, embedder)
+                
+                if results:
+                    # Формируем ответ с результатами
+                    response = f"🔍 Я нашел в научных статьях информацию по вашему запросу:\n\n"
+                    for i, result in enumerate(results, 1):
+                        response += f"**{i}. {result['title']}** (сходство: {result['score']:.2f})\n"
+                        response += f"{result['text'][:300]}...\n\n"
+                    
+                    st.markdown(response)
+                    st.session_state.messages.append({
+                        "role": "assistant", 
+                        "content": response,
+                        "articles": results
+                    })
+                else:
+                    # Если ничего не найдено
+                    response = "😕 Я не нашел подходящих научных статей по вашему запросу. Попробуйте переформулировать вопрос или спросить о другой теме."
+                    st.markdown(response)
+                    st.session_state.messages.append({"role": "assistant", "content": response})
+    
+    # Перезагружаем страницу для обновления чата
+    st.rerun()
 
 # --- ПОДВАЛ ---
 st.divider()
-st.caption(f"🧪 {AI_NAME} от {COMPANY_NAME} | Создан в {CREATION_DATE} | На основе датасета ru_science")
-
-# Кнопка смены темы
-if st.button("🎨 Сменить тему"):
-    st.session_state.theme = "dark" if st.session_state.get("theme") != "dark" else "light"
-    st.rerun()
-
-# Информация о датасете
-with st.expander("📖 О датасете ru_science"):
-    st.markdown("""
-    **ru_science** — это датасет на Hugging Face, содержащий русскоязычные научные статьи.
-    
-    ### Особенности:
-    - Научные статьи по сельскому хозяйству, биологии, химии
-    - Русскоязычные тексты
-    - Подходит для обучения AI-моделей
-    - Используется для поиска и анализа научной информации
-    
-    **Источник:** [RafaelUI/ru_science](https://huggingface.co/datasets/RafaelUI/ru_science)
-    """)
+st.caption(f"🧪 {AI_NAME} от {COMPANY_NAME} | Создан в {CREATION_DATE} | На основе датасета ru_science")