import gradio as gr
from sentence_transformers import SentenceTransformer

# Load model BGEM3
model = SentenceTransformer("BAAI/bge-m3")

def embed_text(text):
    if not text.strip():
        return "⚠️ Please enter some text!"
    embedding = model.encode(text).tolist()
    return embedding

# Gradio UI
with gr.Blocks() as demo:
    gr.Markdown("## 🔎 e5 Embedding Demo")
    inp = gr.Textbox(label="Nhập câu hoặc đoạn văn")
    out = gr.JSON(label="Vector (10 giá trị đầu)")
    btn = gr.Button("Generate Embedding")

    btn.click(embed_text, inputs=inp, outputs=out)

demo.launch()