import streamlit as st
from keybert import KeyBERT

# KeyBERT model
keybert_model = KeyBERT(model='all-MiniLM-L6-v2')

def get_bert_keywords(text, num_keywords):
    result = keybert_model.extract_keywords(text, keyphrase_ngram_range=(1, 1), top_n=num_keywords)
    keyberts = [keyword for keyword, score in result]
    return keyberts

# Streamlit app
st.title("Keyword Extraction with KeyBERT")
text = st.text_area("Enter your text here:", value='', height=200)

if st.button("Extract Keywords"):
    if text:
        keywords = get_bert_keywords(text, 5)
        st.write("Keywords:")
        st.write(", ".join(keywords))
    else:
        st.error("Please enter some text.")