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.")