|
|
import gradio as gr |
|
|
from huggingface_hub import InferenceClient |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
client = InferenceClient("microsoft/phi-4") |
|
|
|
|
|
def respond(message, history): |
|
|
|
|
|
|
|
|
messages = [{"role": "system", "content": "You are a friendly kpop expert chatbot."}] |
|
|
|
|
|
if history: |
|
|
messages.extend(history) |
|
|
|
|
|
messages.append({"role": "user", "content": message}) |
|
|
|
|
|
response = client.chat_completion( |
|
|
messages, |
|
|
max_tokens=10000 |
|
|
) |
|
|
|
|
|
return response['choices'][0]['message']['content'].strip() |
|
|
|
|
|
chatbot = gr.ChatInterface(respond, type="messages") |
|
|
|
|
|
title = "Kpop chatbot" |
|
|
|
|
|
topics = """ |
|
|
<div class='big-paragraph'> |
|
|
👋 Hey Kpop fan! <br> |
|
|
Whether you’re new to the fandom or a longtime stan, I’m your ultimate Kpop buddy here to spill all the tea ☕️, trivia 🧠, and facts 📚 about your favorite groups and idols! 🌟<br><br> |
|
|
Want to find your perfect group or bias? Just ask me! 💜🎤 <br><br> |
|
|
Not only that, if you love an artist outside of Kpop, I can help you discover similar vibes and styles from the amazing world of Kpop too! 🎧✨<br><br> |
|
|
So whether you’re looking for your first bias or your next obsession, I’ve got you covered. <br> |
|
|
Let’s dive into the colorful, energetic, and ever-growing universe of Kpop together! 🚀💫 |
|
|
</div> |
|
|
""" |
|
|
disclaimer = "" |
|
|
|
|
|
css = """ |
|
|
.big-text { |
|
|
font-size: 100px; |
|
|
font-weight: bold; |
|
|
} |
|
|
.big-paragraph { |
|
|
font-size: 100px; |
|
|
line-height: 3; |
|
|
} |
|
|
""" |
|
|
|
|
|
|
|
|
with gr.Blocks() as chatbot: |
|
|
with gr.Row(): |
|
|
with gr.Column(scale=1): |
|
|
gr.Markdown(title) |
|
|
gr.Markdown(topics) |
|
|
with gr.Column(scale=2): |
|
|
gr.ChatInterface( |
|
|
fn=respond, |
|
|
type="messages" |
|
|
) |
|
|
with gr.Row(): |
|
|
gr.Markdown(disclaimer) |
|
|
|
|
|
chatbot.launch() |
