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")
|