Rooni's picture
Update app.py
f4c1707 verified
raw
history blame
1.97 kB
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()