sourabhzanwar commited on
Commit
de8b3d2
·
1 Parent(s): 4d2ad6d

updated files display

Browse files
Files changed (2) hide show
  1. app.py +9 -7
  2. 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.id, document.meta['name']])
43
- df = pd.DataFrame(data, columns=['Document ID', 'Uploaded Document Name'])
44
- df.drop_duplicates(subset=['Document ID'], inplace=True)
45
- return df[['Uploaded Document Name']]
 
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("Upload Files", key="demo-modal")
157
- open_modal = st.sidebar.button("Upload Files")
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="My Search App")
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"),