add upload

app/VectorStore/chroma-collections.parquet

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ab6e5b0679e953f06abf1c6e74ddfc5ffafc233169ce2d6e08b99c1337c5e0cc
-size 1036
+oid sha256:79cf0a7bde715ca62bcfb8bf4f9a737f550dc282abdbde3a3d861114be54c984
+size 967

app/load_vectors.py

@@ -49,13 +49,8 @@ def create_and_add(collection_name, sub_docs, model_name, metadata):
     vectorstore.persist()
 
     # Test Vectorstore
-    #vectorstore2 = Chroma(
-    #collection_name=collection_name,
-    #embedding_function=embeddings,
-    #client_settings=client_settings,
-    #persist_directory=persist_directory,
-    #)
-    #print( vectorstore2.similarity_search_with_score(query="What are AXAs green Goals?", k=4) )
+    vectorstore2 = load_vectorstore(model_name, collection_name, metadata = metadata)
+    print( vectorstore2.similarity_search_with_score(query="What are AXAs green Goals?", k=4) )
 
     return vectorstore
 

app/utils.py

@@ -45,9 +45,8 @@ def load_files():
         options=["Add Documents", "Start new collection"],
     )
 
-    collections = retrieve_collections()
-
     if option == "Add Documents":
+        collections = retrieve_collections()
         selected_collection = st.selectbox(
             'Add to exsisting collection or create a new one',
             collections )
@@ -56,6 +55,8 @@ def load_files():
             #retrieve_collections.clear()
             collections = retrieve_collections()
 
+        if selected_collection:
+            st.write("Selected Vectorstore:", selected_collection)
         option = st.radio(
             "",
             options=["Upload Files from Local", "Upload Files from Web"],
@@ -68,7 +69,10 @@ def load_files():
             if st.button('Upload'):
                 docs = load_from_file(uploaded_files)
                 sub_docs = load_and_split(docs, chunk_size=int(chunk_size))
-                create_and_add(selected_collection["name"], sub_docs, selected_collection['model_name'], {"model_name": selected_collection['model_name']})
+                print(sub_docs)
+                #create_and_add(selected_collection["name"], sub_docs, selected_collection['model_name'], {"model_name": selected_collection['model_name']})
+                create_and_add(selected_collection["name"], sub_docs, "hkunlp/instructor-large", metadata={"model_name": "hkunlp/instructor-large"})
+                
                 uploaded_files=None
                 st.write("Upload succesful")
         else: