refresh button

app.py

@@ -39,15 +39,16 @@ def get_persist_directory(file_name):
 
 
 def process_file(file, chunk_size, chunk_overlap):
+    docs = split_file(file, chunk_size, chunk_overlap)
+    embeddings = load_embeddings()
+
     file_name, _ = os.path.splitext(os.path.basename(file.name))
     persist_directory = get_persist_directory(file_name)
-    print("persist directory", persist_directory)
 
-    docs = split_file(file, chunk_size, chunk_overlap)
-    embeddings = load_embeddings()
+    print("initializing vector store...", persist_directory)
     vectordb = Chroma.from_documents(documents=docs, embedding=embeddings,
                                      collection_name=file_name, persist_directory=persist_directory)
-    print(vectordb._client.list_collections())
+    print("persisting...", vectordb._client.list_collections())
     vectordb.persist()
     return 'Done!', gr.Dropdown.update(choices=get_vector_dbs(), value=file_name)
 
@@ -97,6 +98,11 @@ def create_qa_chain(collection_name, temperature, max_length):
         llm=llm, retriever=vectordb.as_retriever(), memory=memory)
 
 
+def refresh_collection():
+    choices = get_vector_dbs()
+    return gr.Dropdown.update(choices=choices, value=choices[0] if choices else None)
+
+
 def submit_message(bot_history, text):
     bot_history = bot_history + [(text, None)]
     return bot_history, ""
@@ -130,16 +136,21 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Bot"):
         with gr.Row():
-            with gr.Column(scale=0.5):
-                choices = get_vector_dbs()
-                collection = gr.Dropdown(
-                    choices, value=choices[0] if choices else None, label="Document")
+            with gr.Column():
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        choices = get_vector_dbs()
+                        collection = gr.Dropdown(
+                            choices, value=choices[0] if choices else None, label="Document", allow_custom_value=True)
+                    with gr.Column():
+                        refresh = gr.Button("Refresh")
+
                 temperature = gr.Slider(
                     0.0, 1.0, value=0.5, step=0.05, label="Temperature")
                 max_length = gr.Slider(
                     20, 1000, value=100, step=10, label="Max Length")
 
-            with gr.Column(scale=0.5):
+            with gr.Column():
                 chatbot = gr.Chatbot([], elem_id="chatbot").style(height=550)
                 message = gr.Textbox(
                     show_label=False, placeholder="Ask me anything!")
@@ -156,6 +167,7 @@ with gr.Blocks() as demo:
     temperature.change(create_qa_chain, [collection, temperature, max_length])
     max_length.change(create_qa_chain, [collection, temperature, max_length])
 
+    refresh.click(refresh_collection, None, collection)
     message.submit(submit_message, [chatbot, message], [chatbot, message]).then(
         bot, chatbot, chatbot
     )