import streamlit as st
from retrieve_kb import get_current_knowledge_bases
from main import client


st.title("Delete knowledge Base ☠️")

st.title("Get knowledge boxes")
if st.button("Get current knowledge bases"):
    kbs = get_current_knowledge_bases(client=client)
    st.json(kbs)

collection_name = st.text_input(label="collection name")
if st.button("Delete Forever"):
    client.delete_collection(collection_name)
    st.success("Deleted")