Update app.py
Browse files
app.py
CHANGED
@@ -238,11 +238,15 @@ def demo():
|
|
238 |
vector_db = gr.State()
|
239 |
qa_chain = gr.State()
|
240 |
collection_name = gr.State()
|
241 |
-
|
242 |
-
|
|
|
|
|
|
|
243 |
with gr.Row():
|
244 |
-
document = gr.Files(height=100, file_count="multiple",visible=True,
|
245 |
-
|
|
|
246 |
with gr.Row():
|
247 |
db_btn = gr.Radio(["ChromaDB"], label="Vector database type", value = "ChromaDB", type="index", info="Choose your vector database",visible=False)
|
248 |
with gr.Accordion("Advanced options - Document text splitter", open=False, visible=False):
|
|
|
238 |
vector_db = gr.State()
|
239 |
qa_chain = gr.State()
|
240 |
collection_name = gr.State()
|
241 |
+
pdf_directory = '/home/user/app/pdfs'
|
242 |
+
|
243 |
+
pdf_files = [f for f in os.listdir(pdf_directory) if f.endswith(".pdf")]
|
244 |
+
gr_files = [gr.File(os.path.join(pdf_directory, file)) for file in pdf_files]
|
245 |
+
|
246 |
with gr.Row():
|
247 |
+
# document = gr.Files(value = '/home/user/app/pdfs', height=100, file_count="multiple",visible=True,
|
248 |
+
# file_types=["pdf"], interactive=True, label="Upload your PDF documents (single or multiple)")
|
249 |
+
document = [gr.File(os.path.join(pdf_directory, file)) for file in pdf_files]
|
250 |
with gr.Row():
|
251 |
db_btn = gr.Radio(["ChromaDB"], label="Vector database type", value = "ChromaDB", type="index", info="Choose your vector database",visible=False)
|
252 |
with gr.Accordion("Advanced options - Document text splitter", open=False, visible=False):
|