Update app.py

app.py

@@ -140,6 +140,14 @@ def update_vectors(files, parser):
 # Make sure to call this function at the start of your script
 load_uploaded_documents()
 
+def initialize_document_selector():
+    metadata = load_document_metadata()
+    return gr.CheckboxGroup(
+        choices=[doc["name"] for doc in metadata],
+        value=[doc["name"] for doc in metadata if doc["selected"]],
+        label="Select documents to query"
+    )
+
 def generate_chunked_response(prompt, model, max_tokens=10000, num_calls=3, temperature=0.2, should_stop=False):
     print(f"Starting generate_chunked_response with {num_calls} calls")
     full_response = ""
@@ -574,10 +582,6 @@ with demo:
         update_button = gr.Button("Upload Document")
     
     update_output = gr.Textbox(label="Update Status")
-    document_selector = gr.CheckboxGroup(label="Select documents to query")
-    
-    # Initialize document selector
-    document_selector = initialize_document_selector()
     
     # Update both the output text and the document selector
     update_button.click(update_vectors,