Update app.py

app.py

@@ -131,11 +131,13 @@ def update_vectors(files, parser):
     save_document_metadata(metadata)
     logging.info("Document metadata saved")
     
-    return f"Vector store updated successfully. Processed {total_chunks} chunks from {len(files)} files using {parser}.", gr.CheckboxGroup(
-        choices=[doc["name"] for doc in metadata],
-        value=[doc["name"] for doc in metadata if doc["selected"]],
-        label="Select documents to query"
-    )
+    completion_message = f"Vector store updated successfully. Processed {total_chunks} chunks from {len(files)} files using {parser}."
+    logging.info(completion_message)
+    
+    updated_choices = [doc["name"] for doc in metadata]
+    updated_values = [doc["name"] for doc in metadata if doc["selected"]]
+    
+    return completion_message, gr.CheckboxGroup.update(choices=updated_choices, value=updated_values, label="Select documents to query")
 
 # Make sure to call this function at the start of your script
 
@@ -581,6 +583,7 @@ with demo:
         update_button = gr.Button("Upload Document")
     
     update_output = gr.Textbox(label="Update Status")
+    document_selector = gr.CheckboxGroup(label="Select documents to query")
     
     # Update both the output text and the document selector
     update_button.click(update_vectors,