app.py

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

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

/* Заголовок */
h1 {
    color: #1a237e; /* Глубокий индиго */
    text-align: center;
    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; /* Читаемый шрифт */
    margin-bottom: 15px;
    font-weight: 600;
}
#title-markdown p { /* Стили для параграфа под заголовком */
     text-align: center;
     color: #546e7a; /* Серо-синий */
     margin-top: -10px;
     margin-bottom: 25px;
}


/* --- СТИЛИ ЧАТА --- */

/* Контейнер чата */
#chatbot {
    background-color: #eceff1; /* Очень светлый серо-голубой фон чата */
    border-radius: 12px; /* Немного другие скругления */
    border: 1px solid #cfd8dc; /* Граница контейнера */
    padding: 10px; /* Внутренние отступы */
    box-shadow: inset 0 1px 3px rgba(0,0,0,0.05); /* Легкая внутренняя тень */
}

/* Сообщения пользователя */
#chatbot .user-message .message-bubble-border {
     border: none !important; /* Убираем стандартную рамку gradio */
}
#chatbot .user-message .message-bubble {
    background: #007bff !important; /* Яркий синий фон */
    color: white !important; /* Белый текст */
    border-radius: 18px 18px 5px 18px !important; /* Асимметричное скругление */
    padding: 12px 18px !important;
    margin: 8px 5px 8px 0 !important; /* Отступы */
    align-self: flex-end !important;
    max-width: 80% !important;
    box-shadow: 0 2px 4px rgba(0, 123, 255, 0.2); /* Тень под цвет */
    word-wrap: break-word;
    text-align: left;
}


/* Сообщения бота */
#chatbot .bot-message .message-bubble-border {
    border: none !important; /* Убираем стандартную рамку gradio */
}
#chatbot .bot-message .message-bubble {
    background: #ffffff !important; /* Чисто белый фон для контраста */
    color: #212529 !important; /* Темный текст (почти черный) */
    border: 1px solid #dee2e6 !important; /* Легкая серая рамка */
    border-radius: 18px 18px 18px 5px !important; /* Асимметричное скругление */
    padding: 12px 18px !important;
    margin: 8px 0 8px 5px !important; /* Отступы */
    align-self: flex-start !important;
    max-width: 80% !important;
    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); /* Мягкая тень */
    word-wrap: break-word;
    text-align: left;
}

/* Блоки кода внутри сообщений бота */
#chatbot .bot-message .message-bubble pre {
    background-color: #f8f9fa; /* Фон для блока кода */
    border: 1px solid #e9ecef; /* Рамка блока кода */
    border-radius: 6px; /* Скругление */
    padding: 10px; /* Отступы внутри */
    margin: 8px 0 5px 0; /* Отступы снаружи */
    overflow-x: auto; /* Горизонтальная прокрутка если код длинный */
    word-wrap: normal; /* Предотвратить перенос слов в коде */
}
#chatbot .bot-message .message-bubble pre code {
    background-color: transparent !important; /* Убираем фон у самого тега code */
    color: #2d3748; /* Цвет текста кода */
    font-family: 'Consolas', 'Monaco', 'Courier New', monospace; /* Моноширинный шрифт */
    font-size: 0.9em; /* Размер шрифта кода */
    padding: 0; /* Убираем внутренние отступы тега code */
    white-space: pre; /* Сохраняем пробелы и переносы */
}

/* --- ОСТАЛЬНЫЕ ЭЛЕМЕНТЫ --- */

/* Поле ввода текста */
textarea {
    border: 1px solid #ced4da !important;
    border-radius: 10px !important;
    padding: 12px 15px !important;
    background-color: #ffffff;
    transition: border-color 0.3s ease, box-shadow 0.3s ease;
    font-size: 1rem; /* Слегка увеличим шрифт ввода */
}
textarea:focus {
    border-color: #80bdff !important;
    box-shadow: 0 0 0 0.2rem rgba(0, 123, 255, 0.25);
    outline: none; /* Убираем стандартный outline */
}

/* Кнопки */
button {
    border-radius: 10px !important;
    padding: 10px 15px !important; /* Паддинги для кнопок */
    transition: background-color 0.2s ease, transform 0.1s ease !important;
    font-weight: 500 !important;
    border: none !important; /* Убираем рамки по умолчанию */
}
button:active {
    transform: translateY(1px); /* Эффект нажатия */
}
button.primary { /* Кнопка "Отправить" */
    background-color: #007bff !important; /* Синий цвет */
    color: white !important;
}
button.primary:hover {
     background-color: #0056b3 !important; /* Темнее при наведении */
     box-shadow: 0 2px 5px rgba(0, 123, 255, 0.3);
}
button.secondary { /* Кнопка "Очистить" */
   background-color: #6c757d !important; /* Серый цвет */
   color: white !important;
}
button.secondary:hover {
    background-color: #5a6268 !important; /* Темнее при наведении */
     box-shadow: 0 2px 5px rgba(108, 117, 125, 0.3);
}
"""

# --- Python код остается тем же, что и в предыдущем ответе ---
# (начиная с import gradio as gr...)
# ... весь остальной Python код без изменений ...
import gradio as gr
from transformers import pipeline, set_seed
import random
import re # Для регулярных выражений (очистка текста)

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

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

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

    # --- НОВЫЕ ОТВЕТЫ на запросы КОДА ---
    "напиши код": "Я не могу писать произвольный код по запросу. Моя функция - давать информацию и объяснения. Возможно, вы имели в виду пример кода для какой-то задачи? Попробуйте спросить конкретнее, например: 'пример цикла for на python'.",
    "сгенерируй код": "Генерация сложного кода пока не входит в мои возможности. Я могу объяснить концепции или показать базовые синтаксические конструкции. Какой пример вас интересует?",
    "напиши скрипт": "К сожалению, я не пишу готовые скрипты. Опишите задачу, и я постараюсь объяснить, какие шаги или концепции программирования могут понадобиться для ее решения.",
    "код на питоне": "Пайтон - отличный язык! Уточните, какой именно аспект или пример кода на Python вас интересует? Например, 'как объявить переменную на python?' или 'пример функции на python'.",
    "пример кода": "Конечно! Какой пример кода вы хотели бы увидеть? Например, 'пример условного оператора if' или 'пример простого класса на python'.",
    "hello world на python": "Самый классический пример! Вот 'Hello, World!' на Python:\n```python\nprint(\"Hello, World!\")\n```",
    "пример цикла for на python": "Вот пример цикла `for`, который печатает числа от 0 до 4:\n```python\nfor i in range(5):\n    print(i)\n```",
    "пример функции на python": "Вот простая функция на Python, которая складывает два числа:\n```python\ndef add_numbers(a, b):\n    return a + b\n\nresult = add_numbers(5, 3)\nprint(result) # Выведет 8\n```",
    "как объявить переменную на python": "Переменная в Python объявляется простым присваиванием значения. Например:\n```python\nmy_variable = 10\nuser_name = \"Alice\"\nis_active = True\n```",

    # Основы Программирования (остальное из 0.95)
    "что такое программирование": "Программирование - это процесс написания инструкций (кода) для компьютера, чтобы он выполнял определенные задачи. Это как составление очень точного рецепта для машины.",
    "зачем нужно программирование": "Программирование позволяет создавать программы, веб-сайты, игры, автоматизировать задачи, анализировать данные и многое другое. Оно лежит в основе почти всех современных технологий.",
    "что такое алгоритм": "Алгоритм - это пошаговая инструкция или набор правил для решения определенной задачи. Например, рецепт приготовления блюда - это алгоритм.",
    "основные концепции программирования": "Ключевые идеи включают: переменные (хранение данных), типы данных (числа, строки), условные операторы (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 clean_text(text):
    if not isinstance(text, str): return ""
    text = text.lower().strip()
    text = re.sub(r"^[?!.,\s]+|[?!.,\s]+$", "", text)
    text = re.sub(r"\s+", " ", text)
    return text

def clean_generated_output(generated_text, prompt):
    if not generated_text or not isinstance(generated_text, str): return ""
    if not prompt or not isinstance(prompt, str): prompt = ""
    cleaned = generated_text
    prompt_clean = prompt.strip()
    if prompt_clean and cleaned.startswith(prompt_clean):
        cleaned = cleaned[len(prompt_clean):].strip()
    cleaned = re.sub(r'^\s*(ответ|бот|пользователь|user|bot)\s*[:\-]?\s*', '', cleaned, flags=re.IGNORECASE | re.MULTILINE).strip()
    last_lines = prompt_clean.split('\n')
    if len(last_lines) >= 2:
      last_user_line = last_lines[-2].strip()
      if last_user_line.lower().startswith(("пользователь:", "user:")):
        user_message_only = last_user_line.split(":", 1)[1].strip()
        if user_message_only and cleaned.lower().startswith(user_message_only.lower()):
             end_of_echo_pos = len(user_message_only)
             following_text = cleaned[end_of_echo_pos:].lstrip()
             if re.match(r'^(бот|bot)\s*[:\-]?\s*', following_text, flags=re.IGNORECASE):
                 following_text = re.sub(r'^(бот|bot)\s*[:\-]?\s*', '', following_text, flags=re.IGNORECASE).strip()
             cleaned = following_text
    cleaned = re.sub(r'^\s*(ответ|бот|пользователь|user|bot)\s*[:\-]?\s*', '', cleaned, flags=re.IGNORECASE | re.MULTILINE).strip()
    return cleaned

# --- Основная Функция Обработки (Alpha 0.96 Логика) ---
def respond(message, chat_history):
    print("-" * 30)
    print(f"ВХОД: '{message}'")
    user_message_clean = clean_text(message)
    bot_response = "" # Инициализация переменной

    # 1. Точное совпадение в knowledge_base
    if user_message_clean in knowledge_base:
        bot_response = knowledge_base[user_message_clean]
        print(f"Ответ [Точный]: {bot_response[:100]}...") # Логгируем начало ответа

    # 2. Запрос кода
    if not bot_response: # Только если не нашли точный ответ
        code_request_patterns = ["напиши код", "напиши скрипт", "сгенерируй код", "создай код", "write code", "generate code", "create script", "code example for", "пример кода для"]
        is_code_request = any(user_message_clean.startswith(pattern) for pattern in code_request_patterns)
        if is_code_request:
            bot_response = knowledge_base.get("напиши код", "Я не могу писать произвольный код, но могу дать объяснение или пример. Уточните ваш запрос.")
            print(f"Ответ [Запрос кода]: {bot_response}")

    # 3. QA модель
    if not bot_response and "?" in message and QA_LOADED and qa_pipeline:
        context = ""
        if chat_history:
            context_parts = []
            for user_msg, bot_msg in chat_history[-2:]:
                 context_parts.append(f"Пользователь: {user_msg}")
                 if bot_msg: context_parts.append(f"Бот: {bot_msg}")
            context = "\n".join(context_parts)
        if len(context) > 50:
             print("Попытка ответа через QA...")
             try:
                 qa_result = qa_pipeline(question=message, context=context)
                 print(f"Результат QA: score={qa_result.get('score', 0):.2f}, answer='{qa_result.get('answer', '')[:50]}...'")
                 if qa_result and qa_result['score'] > 0.25: # Уверенность > 25%
                     answer = qa_result['answer'].strip()
                     if answer:
                         bot_response = answer
                         print(f"Ответ [QA]: {bot_response[:100]}...")
                     else: print("QA модель дала пустой ответ.")
                 else: print("QA модель не уверена.")
             except Exception as e: print(f"Ошибка при использовании QA модели: {e}")

    # 4. Генерация
    if not bot_response and GENERATOR_LOADED and generator:
        print("Попытка генерации ответа...")
        prompt_history = []
        for user_msg, bot_msg in chat_history[-3:]:
            prompt_history.append(f"Пользователь: {user_msg}")
            if bot_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']
            print(f"Сырой ответ генератора: {generated_output}")
            cleaned_generated = clean_generated_output(generated_output, full_prompt)
            print(f"Очищенный ответ генератора: {cleaned_generated}")
            if cleaned_generated and len(cleaned_generated) > 5: # Минимальная длина для осмысленного ответа
                 bot_response = cleaned_generated
                 print(f"Ответ [Генерация]: {bot_response[:100]}...")
            else:
                print("Генерация не удалась или слишком короткая.")
        except Exception as e:
            print(f"Ошибка при генерации: {e}")
            bot_response = knowledge_base.get("ошибка генерации", "Произошла ошибка при попытке сгенерировать ответ.")

    # 5. Ответ по умолчанию (если ничего не сработало)
    if not bot_response:
        bot_response = knowledge_base.get("не найдено", "К сожалению, я пока не знаю, как на это ответить.")
        print("Ответ [Не найдено/Заглушка]")


    # Добавляем в историю и возвращаем
    chat_history.append((message, bot_response))
    return "", chat_history # Очищаем поле ввода


# --- Gradio Интерфейс ---
with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue=gr.themes.colors.blue, secondary_hue=gr.themes.colors.slate)) as demo: # Тема с цветами
    gr.Markdown("# 🤖 Nova Alpha 0.96 ✨ <br> Улучшенный интерфейс и знания о коде", elem_id="title-markdown")

    chatbot = gr.Chatbot(
        label="Диалог",
        height=550,
        elem_id="chatbot",
        bubble_full_width=False,
        avatar_images=(None, "https://img.icons8.com/plasticine/100/bot.png") # Аватарка для бота
        )

    with gr.Row():
        msg = gr.Textbox(
            label="Ваше сообщение",
            placeholder="Спросите о Python или скажите 'привет'...",
            scale=4,
            show_label=False,
            container=False # Убираем контейнер текстбокса для лучшего выравнивания
        )
        submit_btn = gr.Button("➤ Отправить", variant="primary", scale=1, min_width=120) # Иконка и мин. ширина
        clear_btn = gr.Button("🗑️ Очистить", variant="secondary", scale=1, min_width=120) # Иконка и мин. ширина

    # Обработчики
    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)