Update app.py
Browse files
app.py
CHANGED
|
@@ -40,20 +40,19 @@ def process_upload(files):
|
|
| 40 |
"""
|
| 41 |
Accepts a list of uploaded file paths, saves them to a local folder,
|
| 42 |
loads them as documents, and builds the vector index and chat engine.
|
|
|
|
|
|
|
| 43 |
"""
|
| 44 |
upload_dir = "uploaded_files"
|
| 45 |
if not os.path.exists(upload_dir):
|
| 46 |
os.makedirs(upload_dir)
|
| 47 |
-
else:
|
| 48 |
-
# Clear any existing files in the folder.
|
| 49 |
-
for f in os.listdir(upload_dir):
|
| 50 |
-
os.remove(os.path.join(upload_dir, f))
|
| 51 |
|
| 52 |
-
#
|
| 53 |
for file_path in files:
|
| 54 |
file_name = os.path.basename(file_path)
|
| 55 |
dest = os.path.join(upload_dir, file_name)
|
| 56 |
-
|
|
|
|
| 57 |
|
| 58 |
# Load documents from the saved folder.
|
| 59 |
documents = SimpleDirectoryReader(upload_dir).load_data()
|
|
@@ -135,7 +134,7 @@ def gradio_interface():
|
|
| 135 |
label="Upload Files",
|
| 136 |
file_count="multiple",
|
| 137 |
file_types=[".pdf", ".csv", ".txt", ".xlsx", ".xls", ".doc", ".docx"],
|
| 138 |
-
type="filepath"
|
| 139 |
)
|
| 140 |
upload_status = gr.Textbox(label="Upload Status", interactive=False)
|
| 141 |
upload_button = gr.Button("Process Upload")
|
|
|
|
| 40 |
"""
|
| 41 |
Accepts a list of uploaded file paths, saves them to a local folder,
|
| 42 |
loads them as documents, and builds the vector index and chat engine.
|
| 43 |
+
This version accumulates files, so if you upload more files later,
|
| 44 |
+
they are added to the existing document set.
|
| 45 |
"""
|
| 46 |
upload_dir = "uploaded_files"
|
| 47 |
if not os.path.exists(upload_dir):
|
| 48 |
os.makedirs(upload_dir)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
+
# Copy new files into the folder without clearing existing ones.
|
| 51 |
for file_path in files:
|
| 52 |
file_name = os.path.basename(file_path)
|
| 53 |
dest = os.path.join(upload_dir, file_name)
|
| 54 |
+
if not os.path.exists(dest):
|
| 55 |
+
shutil.copy(file_path, dest)
|
| 56 |
|
| 57 |
# Load documents from the saved folder.
|
| 58 |
documents = SimpleDirectoryReader(upload_dir).load_data()
|
|
|
|
| 134 |
label="Upload Files",
|
| 135 |
file_count="multiple",
|
| 136 |
file_types=[".pdf", ".csv", ".txt", ".xlsx", ".xls", ".doc", ".docx"],
|
| 137 |
+
type="filepath"
|
| 138 |
)
|
| 139 |
upload_status = gr.Textbox(label="Upload Status", interactive=False)
|
| 140 |
upload_button = gr.Button("Process Upload")
|