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