Spaces:
Sleeping
Sleeping
Commit
·
de8b3d2
1
Parent(s):
4d2ad6d
updated files display
Browse files- app.py +9 -7
- utils/config.py +1 -1
app.py
CHANGED
@@ -22,6 +22,7 @@ import streamlit_authenticator as stauth
|
|
22 |
import pickle
|
23 |
|
24 |
from streamlit_modal import Modal
|
|
|
25 |
|
26 |
|
27 |
|
@@ -39,10 +40,11 @@ DISABLE_FILE_UPLOAD = bool(os.getenv("DISABLE_FILE_UPLOAD"))
|
|
39 |
def show_documents_list(retrieved_documents):
|
40 |
data = []
|
41 |
for i, document in enumerate(retrieved_documents):
|
42 |
-
data.append([document.
|
43 |
-
df = pd.DataFrame(data, columns=['
|
44 |
-
df.drop_duplicates(subset=['Document
|
45 |
-
|
|
|
46 |
|
47 |
# Define a function to handle file uploads
|
48 |
def upload_files():
|
@@ -153,8 +155,8 @@ try:
|
|
153 |
|
154 |
set_initial_state()
|
155 |
|
156 |
-
modal = Modal("
|
157 |
-
open_modal = st.sidebar.button("
|
158 |
if open_modal:
|
159 |
modal.open()
|
160 |
|
@@ -177,7 +179,7 @@ try:
|
|
177 |
# upload_container.button('Upload Files', on_click=upload_document, args=())
|
178 |
|
179 |
# Button to reset the documents in DocumentStore
|
180 |
-
st.sidebar.button("Reset documents", on_click=reset_documents, args=())
|
181 |
|
182 |
if "question" not in st.session_state:
|
183 |
st.session_state.question = ""
|
|
|
22 |
import pickle
|
23 |
|
24 |
from streamlit_modal import Modal
|
25 |
+
import numpy as np
|
26 |
|
27 |
|
28 |
|
|
|
40 |
def show_documents_list(retrieved_documents):
|
41 |
data = []
|
42 |
for i, document in enumerate(retrieved_documents):
|
43 |
+
data.append([document.meta['name']])
|
44 |
+
df = pd.DataFrame(data, columns=['Uploaded Document Name'])
|
45 |
+
df.drop_duplicates(subset=['Uploaded Document Name'], inplace=True)
|
46 |
+
df.index = np.arange(1, len(df) + 1)
|
47 |
+
return df
|
48 |
|
49 |
# Define a function to handle file uploads
|
50 |
def upload_files():
|
|
|
155 |
|
156 |
set_initial_state()
|
157 |
|
158 |
+
modal = Modal("Manage Files", key="demo-modal")
|
159 |
+
open_modal = st.sidebar.button("Manage Files", use_container_width=True)
|
160 |
if open_modal:
|
161 |
modal.open()
|
162 |
|
|
|
179 |
# upload_container.button('Upload Files', on_click=upload_document, args=())
|
180 |
|
181 |
# Button to reset the documents in DocumentStore
|
182 |
+
st.sidebar.button("Reset documents", on_click=reset_documents, args=(), use_container_width=True)
|
183 |
|
184 |
if "question" not in st.session_state:
|
185 |
st.session_state.question = ""
|
utils/config.py
CHANGED
@@ -8,7 +8,7 @@ parser = argparse.ArgumentParser(description='This app lists animals')
|
|
8 |
|
9 |
document_store_choices = ('inmemory', 'weaviate', 'milvus', 'opensearch')
|
10 |
parser.add_argument('--store', choices=document_store_choices, default='inmemory', help='DocumentStore selection (default: %(default)s)')
|
11 |
-
parser.add_argument('--name', default="
|
12 |
|
13 |
model_configs = {
|
14 |
'EMBEDDING_MODEL': os.getenv("EMBEDDING_MODEL", "sentence-transformers/all-MiniLM-L12-v2"),
|
|
|
8 |
|
9 |
document_store_choices = ('inmemory', 'weaviate', 'milvus', 'opensearch')
|
10 |
parser.add_argument('--store', choices=document_store_choices, default='inmemory', help='DocumentStore selection (default: %(default)s)')
|
11 |
+
parser.add_argument('--name', default="Document Insights: Extractive & Generative Methods")
|
12 |
|
13 |
model_configs = {
|
14 |
'EMBEDDING_MODEL': os.getenv("EMBEDDING_MODEL", "sentence-transformers/all-MiniLM-L12-v2"),
|