refactor pages

pages/create_knowledge_box.py

@@ -1,44 +0,0 @@
-import streamlit as st
-from app import client, default_embedding_function, show_sidebar
-import pandas as pd
-from generate_kb import generate_knowledge_box_from_url
-from utils import get_chroma_client
-
-# Title of the app
-st.title("Create a knowledge box from CSV file")
-show_sidebar()
-
-# File uploader widget
-uploaded_file = st.file_uploader("Choose a CSV file", type=["csv"])
-df = None
-
-collection_name = st.text_input(label="empy collection name")
-if st.button("create empty knowledge box"):
-    client = get_chroma_client()
-    collection = client.create_collection(name=collection_name)
-    st.success("collection created")
-    st.write(collection)
-
-
-if uploaded_file is not None:
-    try:
-        df = pd.read_csv(uploaded_file)
-        st.write("DataFrame:")
-        st.write(df)
-    except Exception as e:
-        st.error(str(e))
-
-
-if uploaded_file is not None:
-    st.text("dont use spaces but underscores _ in your new name")
-    kb_name = st.text_input(label="new knowledge base name")
-    if st.button("Generate new knowledge box"):
-        urls = df.values.tolist()
-        res = generate_knowledge_box_from_url(
-            client=client,
-            urls=urls,
-            kb_name=kb_name,
-            embedding_fct=default_embedding_function,
-            chunk_size=2_000,
-        )
-        st.json(res)

pages/{delete_knowledge_box⚠️.py → create_or_delete_kb.py}

@@ -1,17 +1,30 @@
 import streamlit as st
 from retrieve_kb import get_current_knowledge_bases
 from app import client, show_sidebar
+from utils import get_chroma_client
 
 
 st.title("Delete knowledge Base ☠️")
 show_sidebar()
 
-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")
+kbs = get_current_knowledge_bases(client=client)
+kbs = (kb.name for kb in kbs)
+collection_name = st.selectbox("Select knowledge box to delete", kbs)
 if st.button("Delete Forever"):
     client.delete_collection(collection_name)
     st.success("Deleted")
+    st.experimental_rerun()
+
+
+# Title of the app
+st.title("Create a knowledge box")
+show_sidebar()
+
+collection_name = st.text_input(label="empy collection name")
+if st.button("create empty knowledge box"):
+    client = get_chroma_client()
+    collection = client.create_collection(name=collection_name)
+    st.success("collection created")
+    st.write(collection)
+    st.experimental_rerun()

pages/manage_knowledge_box.py

@@ -179,7 +179,7 @@ else:
 
 
 tab1, tab2, tab3, tab4, tab5 = st.tabs(
-    ["Remove", "Add URL", "Add CSV", "Add PDF", "Add Youtube"]
+    ["Remove", "Add URL", "Multiple URL", "Add PDF", "Add Youtube"]
 )
 
 # remove stuff tab
@@ -228,19 +228,6 @@ with tab2:
 with tab3:
     list_manager()
 
-    # st.header("Add csv to existing collection")
-    # uploaded_file = st.file_uploader("Choose a CSV file", type=["csv"])
-    # df = None
-    # if uploaded_file is not None:
-    #     try:
-    #         new_df = pd.read_csv(uploaded_file)
-    #         st.write("DataFrame:")
-    #         st.write(new_df)
-    #     except Exception as e:
-    #         st.error(str(e))
-    #     if st.button("add csv urls to collection"):
-    #         urls = new_df.values.tolist()
-    #         st.write(urls)
     if st.button("add csv urls to collection"):
         res = add_links_to_knowledge_base(
             client=client, kb_name=collection_name, urls=st.session_state["url_list"]
@@ -251,6 +238,10 @@ with tab3:
 # Add  PDF
 with tab4:
     st.header("Add pdf to existing collection")
+    st.write(
+        "Trick: if you want to add a Notion page, \
+        download the page as pdf, and load the pdf here together with the notion url"
+    )
     uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
     pdf_optional_link = st.text_input(
         "Insert a URL link you want to associate with the pdf"
@@ -286,12 +277,13 @@ with tab5:
 
     video_url = st.text_input("Youtube video url")
     st.text(
-        "Aggiungere il video puo impiegare un bel pò. Avvia e vatti a fare una canna"
+        "Aggiungere il video puo impiegare un bel pò. Avvia e vatti a fare una sigaretta"
     )
     if st.button("Add video"):
         # Create a temporary file
         # Write the uploaded PDF to the temporary file
         try:
+            st.video(video_url)
             download_and_transcribe_youtube(video_url)
             st.success("Video Added")
         except Exception as e: