Update app.py

app.py

@@ -78,7 +78,7 @@ def update_vectors(files, parser):
         data = load_document(file, parser)
         all_data.extend(data)
         total_chunks += len(data)
-        uploaded_documents.append({"name": file.name, "selected": True})
+        uploaded_documents = [{"name": file.name, "selected": True} for file in files]
     
     if os.path.exists("faiss_database"):
         database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
@@ -88,7 +88,11 @@ def update_vectors(files, parser):
     
     database.save_local("faiss_database")
     
-    return f"Vector store updated successfully. Processed {total_chunks} chunks from {len(files)} files using {parser}."
+     return f"Vector store updated successfully. Processed {total_chunks} chunks from {len(files)} files using {parser}.", gr.CheckboxGroup(
+        choices=[doc["name"] for doc in uploaded_documents],
+        value=[doc["name"] for doc in uploaded_documents if doc["selected"]],
+        label="Select documents to query"
+    )
 
 def generate_chunked_response(prompt, model, max_tokens=1000, num_calls=3, temperature=0.2, should_stop=False):
     print(f"Starting generate_chunked_response with {num_calls} calls")
@@ -469,11 +473,14 @@ with demo:
         file_input = gr.Files(label="Upload your PDF documents", file_types=[".pdf"])
         parser_dropdown = gr.Dropdown(choices=["pypdf", "llamaparse"], label="Select PDF Parser", value="llamaparse")
         update_button = gr.Button("Upload Document")
-        document_selector
     
     update_output = gr.Textbox(label="Update Status")
-    update_button.click(update_vectors, inputs=[file_input, parser_dropdown], outputs=update_output)
-    update_button.click(display_documents, outputs=document_selector)
+    document_selector = gr.CheckboxGroup(label="Select documents to query")
+    
+    # Update both the output text and the document selector
+    update_button.click(update_vectors, 
+                        inputs=[file_input, parser_dropdown], 
+                        outputs=[update_output, document_selector])
 
     gr.Markdown(
     """