import gradio as gr

# Временная функция для отладки
def mock_model_response(user_input):
    return f"Ответ на ваш запрос: {user_input}"

# Функция для обработки сообщений
def code_assistant(system_message, chat_history, user_input):
    # Добавляем системное сообщение в начало истории, если история пуста
    if not chat_history:
        chat_history.append({"role": "system", "content": system_message})
    
    # Используем временную функцию вместо модели
    response = mock_model_response(user_input)
    
    # Добавляем пользовательский ввод и ответ модели в историю чата
    chat_history.append({"role": "user", "content": user_input})
    chat_history.append({"role": "assistant", "content": response})
    
    return chat_history

# Создаем интерфейс
with gr.Blocks() as demo:
    gr.Markdown("## Чат-бот для помощи в кодировании")

    # Поле для системного сообщения
    system_message = gr.Textbox(placeholder="Введите системное сообщение...", label="Системное сообщение")

    # Компонент чата
    chatbot = gr.Chatbot(label="Чат с ботом", type='messages')

    # Поле ввода текста
    user_input = gr.Textbox(placeholder="Введите ваш вопрос или код здесь...", label="Ваш ввод")

    # Кнопка для отправки запроса
    submit_btn = gr.Button("Отправить")

    # Связываем кнопку с функцией обработки и обновления чата
    submit_btn.click(code_assistant, inputs=[system_message, chatbot, user_input], outputs=chatbot)

# Запуск интерфейса
demo.launch()