Update app.py

app.py

@@ -45,8 +45,6 @@ llama_parser = LlamaParse(
     language="en",
 )
 
-uploaded_documents = []
-
 def load_document(file: NamedTemporaryFile, parser: str = "llamaparse") -> List[Document]:
     """Loads and splits the document into pages."""
     if parser == "pypdf":
@@ -410,7 +408,8 @@ css = """
 /* Add your custom CSS here */
 """
 
-# Define the checkbox outside the demo block
+uploaded_documents = []
+
 def display_documents():
     return gr.CheckboxGroup(
         choices=[doc["name"] for doc in uploaded_documents],
@@ -418,6 +417,7 @@ def display_documents():
         label="Select documents to query"
     )
 
+# Define the checkbox outside the demo block
 document_selector = gr.CheckboxGroup(label="Select documents to query")
 
 use_web_search = gr.Checkbox(label="Use Web Search", value=False)
@@ -429,7 +429,7 @@ demo = gr.ChatInterface(
         gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
         gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
         use_web_search,
-        document_selector  # Add this line to include the document selector
+        document_selector
     ],
     title="AI-powered Web Search and PDF Chat Assistant",
     description="Chat with your PDFs or use web search to answer questions.",
@@ -451,7 +451,6 @@ demo = gr.ChatInterface(
         color_accent_soft_dark="transparent",
         code_background_fill_dark="#140b0b"
     ),
-    
     css=css,
     examples=[
         ["Tell me about the contents of the uploaded PDFs."],
@@ -473,17 +472,18 @@ with demo:
     
     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)
 
-    
     gr.Markdown(
     """
     ## How to use
     1. Upload PDF documents using the file input at the top.
     2. Select the PDF parser (pypdf or llamaparse) and click "Upload Document" to update the vector store.
-    3. Ask questions in the chat interface. 
-    4. Toggle "Use Web Search" to switch between PDF chat and web search.
-    5. Adjust Temperature and Number of API Calls to fine-tune the response generation.
-    6. Use the provided examples or ask your own questions.
+    3. Select the documents you want to query using the checkboxes.
+    4. Ask questions in the chat interface. 
+    5. Toggle "Use Web Search" to switch between PDF chat and web search.
+    6. Adjust Temperature and Number of API Calls to fine-tune the response generation.
+    7. Use the provided examples or ask your own questions.
     """
     )