import gradio as gr
import os


HF_TOKEN = os.getenv('HF_TOKEN')
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "Rick-bot-flags")

title = "Have Fun With RickBot"
description = """
<p>
<center>
The bot is trained on Rick and Morty dialogues Kaggle Dataset using DialoGPT.
<img src="https://gradio.app/assets/img/rick.gif">
</center>
</p>
"""
article = "<p style='text-align: center'><a href='https://medium.com/geekculture/discord-bot-using-dailogpt-and-huggingface-api-c71983422701' target='_blank'>Complete Tutorial</a></p><p style='text-align: center'><a href='https://dagshub.com/kingabzpro/DailoGPT-RickBot' target='_blank'>Project is Available at DAGsHub</a></p></center></p>"

gr.Interface.load("huggingface/kingabzpro/DialoGPT-small-Rick-Bot","textbox","chatbot",theme ="grass", title = title, flagging_callback=hf_writer,description = description, article = article).launch(enable_queue=True)  # customizes the input component