shimizukawa commited on
Commit
b6dd5cc
1 Parent(s): 240fe8a

Re: Update streamlit UI

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +2 -2
  3. config.py +4 -4
README.md CHANGED
@@ -14,7 +14,7 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
14
 
15
  # Required Environment variables
16
 
17
- - `INDEX_KEYS`: comma separated index names
18
  - `QDRANT_URL`: Qdrant API endpoint
19
  - `QDRANT_API_KEY`: Qdrant API Key
20
  - `OPENAI_API_KEY`: OpenAI API Key
 
14
 
15
  # Required Environment variables
16
 
17
+ - `INDEX_NAMES`: comma separated index names
18
  - `QDRANT_URL`: Qdrant API endpoint
19
  - `QDRANT_API_KEY`: Qdrant API Key
20
  - `OPENAI_API_KEY`: OpenAI API Key
app.py CHANGED
@@ -14,7 +14,7 @@ from langchain.chains import RetrievalQA
14
  from openai.error import InvalidRequestError
15
  from langchain.chat_models import ChatOpenAI
16
 
17
- from config import DB_CONFIG, INDEX_KEYS
18
  from models import BaseModel
19
 
20
 
@@ -203,7 +203,7 @@ def run_search(
203
  with st.form("my_form"):
204
  st.title("Document Search")
205
  query = st.text_area(label="query")
206
- index = st.selectbox(label="index", options=INDEX_KEYS)
207
 
208
  submit_col1, submit_col2 = st.columns(2)
209
  searched = submit_col1.form_submit_button("Search")
 
14
  from openai.error import InvalidRequestError
15
  from langchain.chat_models import ChatOpenAI
16
 
17
+ from config import DB_CONFIG, INDEX_NAMES
18
  from models import BaseModel
19
 
20
 
 
203
  with st.form("my_form"):
204
  st.title("Document Search")
205
  query = st.text_area(label="query")
206
+ index = st.selectbox(label="index", options=INDEX_NAMES)
207
 
208
  submit_col1, submit_col2 = st.columns(2)
209
  searched = submit_col1.form_submit_button("Search")
config.py CHANGED
@@ -18,18 +18,18 @@ def get_local_db_congin():
18
  return url, None, collection_name
19
 
20
 
21
- def get_index_keys():
22
  keys = [
23
  k for k in [
24
  k.strip().lower()
25
- for k in os.environ["INDEX_KEYS"].split(",")
26
  ]
27
  if k
28
  ]
29
  if not keys:
30
- keys = ["INDEX_KEYS is empty"]
31
  return keys
32
 
33
 
34
  DB_CONFIG = get_db_config() if SAAS else get_local_db_congin()
35
- INDEX_KEYS = get_index_keys()
 
18
  return url, None, collection_name
19
 
20
 
21
+ def get_index_names():
22
  keys = [
23
  k for k in [
24
  k.strip().lower()
25
+ for k in os.environ["INDEX_NAMES"].split(",")
26
  ]
27
  if k
28
  ]
29
  if not keys:
30
+ keys = ["INDEX_NAMES is empty"]
31
  return keys
32
 
33
 
34
  DB_CONFIG = get_db_config() if SAAS else get_local_db_congin()
35
+ INDEX_NAMES = get_index_names()