pop info in chat for doc upload

app.py

@@ -134,20 +134,29 @@ def process_documents(ignored_files: List[str] = []) -> List[Document]:
     print(f"Split into {len(texts)} chunks of text (max. {chunk_size} tokens each)")
     return texts
 
-def UpdateDb(file_path: str):
-    global vectordb_p
-    # pinecone.Index(index_name).delete(delete_all=True, namespace='')
-    # collection = vectordb_p.get()
-    # split_docs = process_documents([metadata['source'] for metadata in collection['metadatas']])
-    # split_docs = process_documents()
-    documents = load_single_document(file_path)
+def process_documents_2(ignored_files: List[str] = []) -> List[Document]:
+    """
+    Load documents and split in chunks
+    """
+    print(f"Loading documents from {source_directory}")
+    print("File Path to start processing:", file_path)
+    documents = load_documents(file_path, ignored_files)
     if not documents:
         print("No new documents to load")
         exit(0)
     print(f"Loaded {len(documents)} new documents from {source_directory}")
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
-    split_docs = text_splitter.split_documents(documents)
-    print(f"Split into {len(split_docs)} chunks of text (max. {chunk_size} tokens each)")
+    texts = text_splitter.split_documents(documents)
+    print(f"Split into {len(texts)} chunks of text (max. {chunk_size} tokens each)")
+    return texts
+
+def UpdateDb():
+    global vectordb_p
+    # pinecone.Index(index_name).delete(delete_all=True, namespace='')
+    # collection = vectordb_p.get()
+    # split_docs = process_documents([metadata['source'] for metadata in collection['metadatas']])
+    # split_docs = process_documents()
+    split_docs = process_documents_2()
     tt = len(split_docs)
     print(split_docs[tt-1])
     print(f"Creating embeddings. May take some minutes...")
@@ -316,18 +325,21 @@ def chathmi2(message, history):
 # chatbot = gr.Chatbot().style(color_map =("blue", "pink"))
 # chatbot = gr.Chatbot(color_map =("blue", "pink"))
 
-def upload_file(files):
-    print(files)
-    pass
+def func_upload_file(files):
+    file_path = files
+    print(file_path)
+    # UpdateDb()
+    return("Test Upload")
 
 with gr.Blocks() as demo:
-    gr.ChatInterface(
+    main = gr.ChatInterface(
         chathmi2,
-        title="STLA BABY - YOUR FRIENDLY GUIDE ",
-        description= "v0.2: Powered by MECH Core Team",
+        examples=["What is PDP?"],
+        title="STLA BABY - YOUR FRIENDLY GUIDE",
+        description= "v0.3: Powered by MECH Core Team",
     )
     upload_button = gr.UploadButton("Upload File", file_count="multiple")
-    upload_button.upload(upload_file, upload_button)
+    upload_button.upload(func_upload_file, upload_button, main)
 
 # demo = gr.Interface(
 #     chathmi,