Update app.py

app.py

@@ -31,8 +31,7 @@ def load_doc(list_file_path, chunk_size, chunk_overlap):
     return doc_splits
 
 # Create vector database
-def create_db(list_file_path, chunk_size, chunk_overlap, db_type):
-    splits = load_doc(list_file_path, chunk_size, chunk_overlap)
+def create_db(splits, collection_name, db_type):
     embedding = HuggingFaceEmbeddings()
     
     if db_type == "ChromaDB":
@@ -41,7 +40,7 @@ def create_db(list_file_path, chunk_size, chunk_overlap, db_type):
             documents=splits,
             embedding=embedding,
             client=new_client,
-            collection_name="default_collection",
+            collection_name=collection_name,
         )
     elif db_type == "FAISS":
         vectordb = FAISS.from_documents(
@@ -57,12 +56,12 @@ def create_db(list_file_path, chunk_size, chunk_overlap, db_type):
         vectordb = Milvus.from_documents(
             documents=splits,
             embedding=embedding,
-            collection_name="default_collection",
+            collection_name=collection_name,
         )
     else:
         raise ValueError(f"Unsupported vector database type: {db_type}")
     
-    return vectordb, "default_collection", "Vector database created successfully"
+    return vectordb
 
 # Initialize langchain LLM chain
 def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, initial_prompt, progress=gr.Progress()):
@@ -252,13 +251,13 @@ def demo():
                 clear_btn_no_doc = gr.ClearButton([msg_no_doc, chatbot_no_doc], value="Clear conversation")
 
         # Preprocessing events
-        db_btn.click(create_db, 
+        db_btn.click(initialize_database, 
             inputs=[document, slider_chunk_size, slider_chunk_overlap, db_type_radio], 
             outputs=[vector_db, collection_name, db_progress])
         set_prompt_btn.click(lambda prompt: gr.update(value=prompt), 
                              inputs=prompt_input, 
                              outputs=initial_prompt)
-        qachain_btn.click(initialize_llmchain, 
+        qachain_btn.click(initialize_LLM, 
             inputs=[llm_btn, slider_temperature, slider_maxtokens, slider_topk, vector_db, initial_prompt], 
             outputs=[qa_chain, llm_progress]).then(lambda:[None,"",0,"",0,"",0], 
             inputs=None,