Update app.py

app.py

@@ -103,9 +103,9 @@ def create_collection_name(filepath):
     return collection_name
 
 # Initialize database
-def initialize_database(directory_path, chunk_size, chunk_overlap, progress=gr.Progress()):
+def initialize_database(chunk_size, chunk_overlap, progress=gr.Progress()):
     progress(0.1, desc="Loading documents from directory...")
-    doc_splits, pdf_files = load_docs_from_directory(directory_path, chunk_size, chunk_overlap)
+    doc_splits, pdf_files = load_docs_from_directory(pdf_directory, chunk_size, chunk_overlap)
     collection_name = create_collection_name(pdf_files[0])
     progress(0.5, desc="Generating vector database...")
     vector_db = create_db(doc_splits, collection_name)
@@ -203,7 +203,7 @@ def demo():
             clear_btn = gr.ClearButton([msg, chatbot], value="Clear conversation")
 
         db_btn.click(initialize_database, \
-                     inputs=[pdf_directory, slider_chunk_size, slider_chunk_overlap], \
+                     inputs=[slider_chunk_size, slider_chunk_overlap], \
                      outputs=[vector_db, collection_name, db_progress])
         qachain_btn.click(initialize_LLM, \
                           inputs=[slider_temperature, slider_maxtokens, slider_topk, vector_db], \