app.py

# app.py

import gradio as gr
from transformers import pipeline, set_seed
import random
import re # Для регулярных выражений (очистка текста)

# --- Загрузка моделей (как в прошлый раз, выберите нужные) ---
try:
    # generator = pipeline('text-generation', model='distilgpt2', max_length=100, num_return_sequences=1, truncation=True, pad_token_id=50256)
    generator = pipeline('text-generation', model='sberbank-ai/rugpt3small_based_on_gpt2', max_length=100, num_return_sequences=1, truncation=True, pad_token_id=50256)
    print("Генератор текста загружен.")
    GENERATOR_LOADED = True
except Exception as e:
    print(f"ОШИБКА: Не удалось загрузить генератор текста! {e}")
    generator = None
    GENERATOR_LOADED = False

try:
    # qa_pipeline = pipeline('question-answering', model='bert-large-uncased-whole-word-masking-finetuned-squad')
    qa_pipeline = pipeline('question-answering', model='timpal0l/mdeberta-v3-base-squad2') # Мультиязычная
    print("QA модель загружена.")
    QA_LOADED = True
except Exception as e:
    print(f"ОШИБКА: Не удалось загрузить QA модель! {e}")
    qa_pipeline = None
    QA_LOADED = False

# --- Встроенные знания и команды (РАСШИРЕНО для КОДИНГА) ---
knowledge_base = {
    # Мета-информация
    "кто ты": "Я Nova, модель Alpha 0.95, работающая на платформе Hugging Face Spaces. Я использую нейросетевые модели для генерации ответов и ответов на вопросы.",
    "что ты умеешь": "Я могу пытаться поддерживать диалог, отвечать на вопросы по контексту, генерировать текст и давать базовую информацию по некоторым темам, включая программирование. Мои знания ограничены, и я не выполняю код.",
    "как дела": "Как у программы, у меня все по плану! Готова обрабатывать ваши запросы.",
    "помощь": "Спросите меня о чем-нибудь, например, 'что такое переменная в программировании?' или 'какие есть циклы?'. Я постараюсь ответить. Также знаю команды: 'кто ты', 'что ты умеешь'.",

    # Основы Программирования
    "что такое программирование": "Программирование - это процесс написания инструкций (кода) для компьютера, чтобы он выполнял определенные задачи. Это как составление очень точного рецепта для машины.",
    "зачем нужно программирование": "Программирование позволяет создавать программы, веб-сайты, игры, автоматизировать задачи, анализировать данные и многое другое. Оно лежит в основе почти всех современных технологий.",
    "что такое алгоритм": "Алгоритм - это пошаговая инструкция или набор правил для решения определенной задачи. Например, рецепт приготовления блюда - это алгоритм.",
    "основные концепции программирования": "Ключевые идеи включают: переменные (хранение данных), типы данных (числа, строки), условные операторы (if/else - если/иначе), циклы (for, while - для повторения действий), функции (для организации кода).",
    "что такое переменная": "Переменная в программировании - это как именованный контейнер или ячейка памяти, где можно хранить данные (число, текст, и т.д.), чтобы использовать их позже в программе. Например, `age = 30`.",
    "что такое типы данных": "Типы данных определяют, какого рода информацию может хранить переменная и какие операции с ней можно выполнять. Распространенные типы: целые числа (integer), числа с плавающей точкой (float), строки (string - текст), булевы (boolean - истина/ложь).",
    "что такое строка": "Строка (string) - это последовательность символов, обычно используемая для представления текста. В большинстве языков строки заключаются в кавычки (одинарные или двойные). Например, \"Привет, мир!\".",
    "условный оператор if": "Оператор `if` (если) позволяет выполнить блок кода только в том случае, если определенное условие истинно. Часто используется вместе с `else` (иначе) для выполнения другого блока кода, если условие ложно.",
    "что такое цикл": "Цикл - это конструкция, которая позволяет повторять выполнение блока кода несколько раз. Основные виды: `for` (для), который обычно используется для перебора элементов (например, в списке) или выполнения заданное число раз, и `while` (пока), который выполняется, пока истинно определенное условие.",
    "цикл for": "Цикл `for` удобен, когда вы знаете, сколько раз нужно повторить действие, или когда нужно пройти по всем элементам коллекции (списка, строки).",
    "цикл while": "Цикл `while` повторяет блок кода до тех пор, пока заданное условие остается истинным. Важно убедиться, что условие когда-нибудь станет ложным, иначе цикл будет бесконечным.",
    "что такое функция": "Функция - это именованный блок кода, который выполняет определенную задачу. Функции помогают разбивать сложную программу на мелкие, управляемые части, избегать повторения кода и делать его более читаемым.",
    "язык программирования": "Язык программирования - это формальный язык, предназначенный для записи компьютерных программ. Популярные языки: Python (хорош для новичков, науки о данных, веб-бэкенда), JavaScript (для веб-фронтенда и бэкенда), Java (для энтерпрайза, Android), C++ (для игр, системного программирования), C# (для Windows, игр на Unity).",
    "python": "Python - популярный, высокоуровневый язык программирования с простым синтаксисом, подходящий для широкого круга задач: веб-разработка, наука о данных, машинное обучение, автоматизация.",
    "javascript": "JavaScript (JS) - основной язык для создания интерактивных веб-страниц (фронтенд). Также используется на бэкенде (Node.js) и для мобильных приложений.",
    "html": "HTML (HyperText Markup Language) - это не язык программирования, а язык разметки. Он используется для определения структуры и содержимого веб-страницы (заголовки, параграфы, картинки, ссылки).",
    "css": "CSS (Cascading Style Sheets) - язык стилей, используемый для описания внешнего вида документа, написанного на языке разметки (обычно HTML). Определяет цвета, шрифты, отступы, расположение элементов.",
    "что такое api": "API (Application Programming Interface) - это набор правил и протоколов, который позволяет разным программам 'общаться' друг с другом. Это как меню в ресторане: вы выбираете блюдо (функцию API), и кухня (другая программа) его готовит и отдает вам.",
    "что такое git": "Git - это распределенная система контроля версий. Она помогает разработчикам отслеживать изменения в коде, возвращаться к предыдущим версиям, работать над проектом совместно и не терять изменения.",
    "где учиться программировать": "Существует множество ресурсов: интерактивные онлайн-курсы (Codecademy, Coursera, Stepik, Яндекс.Практикум), документация языков, книги, видеоуроки на YouTube, форумы (Stack Overflow), сообщества разработчиков."

    # Добавьте сюда больше вопросов и ответов по кодингу!
}

# --- Функция обработки запроса (логика остается той же) ---

def respond(message, chat_history):
    print(f"Получено сообщение: {message}")
    user_message_lower = message.lower().strip().replace("?","").replace(".","").replace("!","") # Очищенный ввод для команд

    # 1. Проверка на встроенные команды/знания
    if user_message_lower in knowledge_base:
        bot_response = knowledge_base[user_message_lower]
        print(f"Ответ из базы знаний: {bot_response}")
        chat_history.append((message, bot_response))
        return "", chat_history

    # 2. Попытка ответа на вопрос с помощью QA модели (упрощенная логика)
    if "?" in message and QA_LOADED and qa_pipeline:
        context = ""
        if chat_history:
            context_parts = []
            for user_msg, bot_msg in chat_history[-2:]: # Последние 2 обмена
                 context_parts.append(f"Пользователь: {user_msg}")
                 context_parts.append(f"Бот: {bot_msg}")
            context = "\n".join(context_parts)

        if len(context) > 50: # Нужен минимальный контекст
             print("Попытка ответа на вопрос через QA модель...")
             print(f"Контекст: {context[:200]}...")
             print(f"Вопрос: {message}")
             try:
                 qa_result = qa_pipeline(question=message, context=context)
                 print(f"Результат QA: {qa_result}")
                 if qa_result and qa_result['score'] > 0.15: # Немного повысим порог
                     bot_response = qa_result['answer']
                     chat_history.append((message, bot_response))
                     print(f"Ответ от QA модели: {bot_response}")
                     return "", chat_history
                 else:
                      print("QA модель не уверена в ответе.")
             except Exception as e:
                  print(f"Ошибка при использовании QA модели: {e}")

    # 3. Генерация ответа с помощью основной модели
    if GENERATOR_LOADED and generator:
        print("Генерация ответа с помощью основной модели...")
        prompt_history = []
        for user_msg, bot_msg in chat_history[-3:]: # Последние 3 обмена
            prompt_history.append(f"Пользователь: {user_msg}")
            prompt_history.append(f"Бот: {bot_msg}")
        prompt_history.append(f"Пользователь: {message}")
        prompt_history.append("Бот:")

        full_prompt = "\n".join(prompt_history)
        print(f"Промпт для генератора: {full_prompt[-500:]}")

        try:
            generated_output = generator(full_prompt)[0]['generated_text']
            bot_response = generated_output

            # Очистка
            if generated_output.startswith(full_prompt):
                 bot_response = generated_output[len(full_prompt):].strip()
            else:
                 last_user_line = f"Пользователь: {message}\nБот:"
                 if bot_response.strip().startswith(last_user_line.strip()):
                     bot_response = bot_response.strip()[len(last_user_line.strip()):].strip()
                 elif len(bot_response) > len(message) and message.lower() in bot_response[:len(message)*2].lower():
                      parts = bot_response.split('\n')
                      if len(parts)>1: bot_response = '\n'.join(parts[1:]).strip()

            bot_response = re.sub(r'^\s*(пользователь|user|бот|bot)\s*[:\-]?\s*', '', bot_response, flags=re.IGNORECASE).strip()

            if not bot_response or len(bot_response) < 3:
                bot_response = knowledge_base.get("не найдено", "Хм, не знаю, что на это ответить...") # Используем "не найдено" из базы
                print("Сгенерирован слишком короткий ответ, используется заглушка.")

        except Exception as e:
            print(f"Ошибка при генерации: {e}")
            bot_response = knowledge_base.get("ошибка генерации", f"Упс, произошла ошибка при генерации: {e}") # Используем текст ошибки из базы

    else:
        bot_response = "К сожалению, мои основные модули сейчас недоступны."
        print("Ошибка: Генератор текста не загружен.")

    chat_history.append((message, bot_response))
    print(f"Финальный ответ: {bot_response}")
    return "", chat_history


# --- Создание интерфейса Gradio с УЛУЧШЕННЫМ CSS ---

# Определяем CSS стили
custom_css = """
/* Общий фон */
.gradio-container {
    background: linear-gradient(to bottom right, #e0f7fa, #f1f8e9); /* Приятный градиент */
    border-radius: 15px;
    padding: 25px;
}

/* Заголовок */
h1 {
    color: #004d40; /* Темно-бирюзовый */
    text-align: center;
    font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;
    margin-bottom: 15px;
}

/* Контейнер чата */
#chatbot {
    background-color: #ffffff; /* Белый фон для сообщений */
    border-radius: 10px;
    box-shadow: 0 2px 5px rgba(0,0,0,0.1); /* Легкая тень */
    border: 1px solid #e0e0e0;
}

/* Сообщения пользователя */
#chatbot .user {
    background-color: #e3f2fd; /* Голубоватый фон */
    color: #0d47a1; /* Темно-синий текст */
    border-radius: 15px 15px 0 15px; /* Скругление углов */
    padding: 10px 15px;
    margin: 5px;
    align-self: flex-end; /* Выравнивание справа */
    max-width: 85%;
}

/* Сообщения бота */
#chatbot .bot {
    background-color: #f1f8e9; /* Светло-зеленоватый фон */
    color: #33691e; /* Темно-зеленый текст */
    border-radius: 15px 15px 15px 0; /* Скругление углов */
    padding: 10px 15px;
    margin: 5px;
    align-self: flex-start; /* Выравнивание слева */
     max-width: 85%;
}

/* Поле ввода текста */
textarea {
    border: 1px solid #b0bec5; /* Сероватая рамка */
    border-radius: 8px !important; /* !important нужен для переопределения стандартных стилей gradio */
    padding: 12px !important;
    background-color: #ffffff;
    transition: border-color 0.3s ease;
}
textarea:focus {
    border-color: #0277bd !important; /* Синяя рамка при фокусе */
    box-shadow: 0 0 5px rgba(2, 119, 189, 0.3);
}

/* Кнопки */
button {
    border-radius: 8px !important;
    transition: background-color 0.3s ease, box-shadow 0.3s ease !important;
    font-weight: 500 !important;
}
button.primary { /* Основная кнопка Отправить */
    background: linear-gradient(to right, #007991, #78ffd6) !important; /* Бирюзовый градиент */
    color: white !important;
    border: none !important;
}
button.primary:hover {
     box-shadow: 0 4px 8px rgba(0, 121, 145, 0.3);
     background: linear-gradient(to right, #006073, #5fddb8) !important; /* Чуть темнее при наведении */
}
button.secondary { /* Кнопка Очистить */
   background-color: #cfd8dc !important; /* Светло-серый */
   color: #37474f !important;
   border: 1px solid #b0bec5 !important;
}
button.secondary:hover {
    background-color: #b0bec5 !important; /* Темнее при наведении */
}
"""

with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo: # Применяем CSS и мягкую тему
    gr.Markdown(
        """
        # 🤖 Nova Alpha 0.95 🎨
        Улучшенный интерфейс! Задавайте вопросы или команды.
        """, elem_id="title-markdown" # Можно задать ID для CSS
    )

    chatbot = gr.Chatbot(label="Диалог", height=500, elem_id="chatbot")
    with gr.Row():
        msg = gr.Textbox(
            label="Ваше сообщение",
            placeholder="Спросите о Python или скажите 'привет'...",
            scale=4
        )
        submit_btn = gr.Button("Отправить", variant="primary", scale=1)
        clear_btn = gr.Button("Очистить", variant="secondary", scale=1) # Добавили класс для CSS

    msg.submit(respond, [msg, chatbot], [msg, chatbot])
    submit_btn.click(respond, [msg, chatbot], [msg, chatbot])
    clear_btn.click(lambda: (None, []), None, [msg, chatbot], queue=False)

demo.queue()
demo.launch(debug=True)