Spaces:
Sleeping
Sleeping
File size: 1,608 Bytes
7c4e3b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from utils.prepare_vectordb import PrepareVectorDB
from typing import List, Tuple
from utils.load_config import LoadConfig
APPCFG = LoadConfig()
class UploadFile:
"""
Utility class for handling file uploads and processing.
This class provides static methods for checking directories and processing uploaded files
to prepare a VectorDB.
"""
@staticmethod
def process_uploaded_files(files_dir: List, chatbot: List, rag_with_dropdown: str) -> Tuple:
"""
Process uploaded files to prepare a VectorDB.
Parameters:
files_dir (List): List of paths to the uploaded files.
chatbot: An instance of the chatbot for communication.
Returns:
Tuple: A tuple containing an empty string and the updated chatbot instance.
"""
if rag_with_dropdown == "Upload doc: Process for RAG":
prepare_vectordb_instance = PrepareVectorDB(data_directory=files_dir,
persist_directory=APPCFG.custom_persist_directory,
chunk_size=APPCFG.chunk_size,
chunk_overlap=APPCFG.chunk_overlap)
prepare_vectordb_instance.prepare_and_save_vectordb()
chatbot.append(
(" ", "Uploaded files are ready. Please ask your question"))
else:
chatbot.append(
(" ", "If you would like to upload a PDF, please select your desired action in 'rag_with' dropdown."))
return "", chatbot
|