Update app.py

app.py

@@ -68,7 +68,11 @@ def get_embeddings():
 def update_vectors(files, parser):
     global uploaded_documents
     if not files:
-        return "Please upload at least one PDF file."
+        return "Please upload at least one PDF file.", 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"
+        )
     
     embed = get_embeddings()
     total_chunks = 0
@@ -78,7 +82,9 @@ def update_vectors(files, parser):
         data = load_document(file, parser)
         all_data.extend(data)
         total_chunks += len(data)
-        uploaded_documents = [{"name": file.name, "selected": True} for file in files]
+        # Append new documents instead of replacing
+        if not any(doc["name"] == file.name for doc in uploaded_documents):
+            uploaded_documents.append({"name": file.name, "selected": True})
     
     if os.path.exists("faiss_database"):
         database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
@@ -431,13 +437,14 @@ document_selector = gr.CheckboxGroup(label="Select documents to query")
 use_web_search = gr.Checkbox(label="Use Web Search", value=False)
 
 demo = gr.ChatInterface(
-       respond,
-       additional_inputs=[
-           gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[0]),
-           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
-       ],
+    respond,
+    additional_inputs=[
+        gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[0]),
+        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 the document selector to the chat interface
+    ],
     title="AI-powered Web Search and PDF Chat Assistant",
     description="Chat with your PDFs or use web search to answer questions.",
     theme=gr.themes.Soft(
@@ -478,7 +485,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")
     
     # Update both the output text and the document selector
     update_button.click(update_vectors,