Update app.py

app.py

@@ -658,8 +658,8 @@ demo = gr.ChatInterface(
         gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[3]),
         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"),
-        gr.CheckboxGroup(label="Select documents to query", choices=[]),
-        gr.Checkbox(label="Use Web Search", value=True)
+        gr.Checkbox(label="Use Web Search", value=True),
+        gr.CheckboxGroup(label="Select documents to query", choices=[])
     ],
     title="AI-powered PDF Chat and Web Search Assistant",
     description="Chat with your PDFs or use web search to answer questions.",
@@ -700,10 +700,10 @@ demo = gr.ChatInterface(
     )
 )
 
+# Add file upload functionality
 # Add file upload functionality
 with demo:
     gr.Markdown("## Upload and Manage PDF Documents")
-
     with gr.Row():
         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")
@@ -712,21 +712,27 @@ with demo:
     
     update_output = gr.Textbox(label="Update Status")
     delete_button = gr.Button("Delete Selected Documents")
-
-
-    # In your Gradio interface setup
-    update_button.click(update_vectors, 
-                        inputs=[file_input, parser_dropdown], 
-                        outputs=[update_output, document_selector])    
+    
+    # Update both the output text and the document selector
+    update_button.click(
+        update_vectors, 
+        inputs=[file_input, parser_dropdown], 
+        outputs=[update_output, demo.additional_inputs[-1]]  # Use the CheckboxGroup from additional_inputs
+    )
+    
     # Add the refresh button functionality
-    refresh_button.click(refresh_documents, 
-                         inputs=[], 
-                         outputs=[document_selector])
+    refresh_button.click(
+        refresh_documents, 
+        inputs=[], 
+        outputs=[demo.additional_inputs[-1]]  # Use the CheckboxGroup from additional_inputs
+    )
     
     # Add the delete button functionality
-    delete_button.click(delete_documents,
-                        inputs=[document_selector],
-                        outputs=[update_output, document_selector])
+    delete_button.click(
+        delete_documents,
+        inputs=[demo.additional_inputs[-1]],  # Use the CheckboxGroup from additional_inputs
+        outputs=[update_output, demo.additional_inputs[-1]]
+    )
 
     gr.Markdown(
     """