На цей раз ми створимо простий інтерфейс за допомогою Gradio, однієї з бібліотек, яка дозволяє нам створювати інтерактивні веб-додатки, що є дружніми до користувача та простими для впровадження.
Знайомство з Gradio: Що таке Gradio?
Gradio — це open-source бібліотека, розроблена HuggingFace. З її допомогою ми можемо легко і швидко створювати демо для чат-ботів на основі веб-додатків за допомогою вбудованої функції для спільного доступу. О, і до речі, Gradio є пакетом Python, тому ми будемо створювати модель додатка за допомогою мови Python.
Встановлення
Щоб почати використовувати Gradio, потрібно мати Python версії 3.10 або новішу. Потім ми можемо встановити Gradio за допомогою pip. Виконайте цей код у терміналі або командному рядку:
pip install --upgrade gradio
Створення простого чат-бота
Для початку ми створимо демо-додаток, який прийматиме введення від користувача та надаватиме відповіді на основі цього введення. Почнемо!
Крок 1: Імпортуємо Gradio
Перш за все, імпортуємо необхідну бібліотеку:
import gradio as gr
Крок 2: Опис логіки чат-бота
Тепер визначимо функцію, яка оброблятиме введення користувача та генеруватиме відповідь. Наприклад, ми використаємо набір базових правил:
def chatbot_response(user_input):
responses = {
"halo": "Привіт! Як я можу допомогти вам сьогодні?",
"apa kabar?": "Я лише програма, але дякую, що запитали!",
"selamat tinggal": "До побачення! Нехай ваш день буде чудовим!",
}
return responses.get(user_input.lower(), "Вибачте, я не розумію це.")
Крок 3: Створення інтерфейсу
Тепер створимо інтерфейс Gradio, що включатиме компоненти заголовка, опису, текстового введення та виходу в додатку:
iface = gr.Interface(fn=chatbot_response,
inputs="text",
outputs="text",
title="Простий чат-бот",
description="Наберіть повідомлення для спілкування з ботом.")
Крок 4: Запуск додатка
Час запустити додаток Gradio та подивитися, як працює наш чат-бот:
iface.launch()
Повний код
Ось повний код:
import gradio as gr
def chatbot_response(user_input):
responses = {
"halo": "Привіт! Як я можу допомогти вам сьогодні?",
"apa kabar?": "Я лише програма, але дякую, що запитали!",
"selamat tinggal": "До побачення! Нехай ваш день буде чудовим!",
}
return responses.get(user_input.lower(), "Вибачте, я не розумію це.")
iface = gr.Interface(fn=chatbot_response,
inputs="text",
outputs="text",
title="Простий чат-бот",
description="Наберіть повідомлення для спілкування з ботом.")
iface.launch()
Демо, що наведене нижче, відкриється у браузері на http://localhost:7860, якщо його запустити з файлу. Якщо ж ми запустимо його в notebook (наприклад, в цьому прикладі я запускаю його в Google Colab), демо з'явиться безпосередньо в notebook. Ось як це виглядатиме:
Підсумок
Створення user-interface чат-бота за допомогою Gradio — це простий процес, який дозволяє нам швидко створювати прототипи та демонструвати можливості чат-ботів або моделей machine learning, що ми створили. Лише кілька рядків коду — і ми маємо функціональний чат-бот, готовий до взаємодії з користувачами. У наступних статтях ми створимо більш складні додатки, наприклад, інтегруючи модель machine learning та додаючи мультимодальні компоненти (аналіз тексту, зображень, аудіо та відео). До зустрічі та happy coding!
Перекладено з: Membuat Antarmuka Chatbot Menggunakan Gradio (bagian 1)