Spaces:
Running
Running
cleanup
Browse files- streamlit_app.py +43 -24
streamlit_app.py
CHANGED
|
@@ -71,15 +71,35 @@ st.set_page_config(
|
|
| 71 |
}
|
| 72 |
)
|
| 73 |
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 83 |
|
| 84 |
|
| 85 |
def new_file():
|
|
@@ -231,9 +251,9 @@ with st.sidebar:
|
|
| 231 |
# is_api_key_provided = st.session_state['api_key']
|
| 232 |
|
| 233 |
st.button(
|
| 234 |
-
|
| 235 |
-
|
| 236 |
-
|
| 237 |
|
| 238 |
left_column, right_column = st.columns([1, 1])
|
| 239 |
|
|
@@ -322,15 +342,15 @@ with left_column:
|
|
| 322 |
left_column.markdown(get_pdf_display(st.session_state['binary']), unsafe_allow_html=True)
|
| 323 |
|
| 324 |
with right_column:
|
| 325 |
-
css = '''
|
| 326 |
-
|
| 327 |
-
|
| 328 |
-
|
| 329 |
-
|
| 330 |
-
|
| 331 |
-
|
| 332 |
-
|
| 333 |
-
st.markdown(css, unsafe_allow_html=True)
|
| 334 |
|
| 335 |
# st.markdown(
|
| 336 |
# """
|
|
@@ -341,7 +361,6 @@ with right_column:
|
|
| 341 |
# unsafe_allow_html=True,
|
| 342 |
# )
|
| 343 |
|
| 344 |
-
|
| 345 |
if st.session_state.loaded_embeddings and question and len(question) > 0 and st.session_state.doc_id:
|
| 346 |
for message in st.session_state.messages:
|
| 347 |
with st.chat_message(message["role"]):
|
|
@@ -365,8 +384,8 @@ with right_column:
|
|
| 365 |
elif mode == "LLM":
|
| 366 |
with st.spinner("Generating response..."):
|
| 367 |
_, text_response = st.session_state['rqa'][model].query_document(question, st.session_state.doc_id,
|
| 368 |
-
|
| 369 |
-
|
| 370 |
|
| 371 |
if not text_response:
|
| 372 |
st.error("Something went wrong. Contact Luca Foppiano (Foppiano.Luca@nims.co.jp) to report the issue.")
|
|
|
|
| 71 |
}
|
| 72 |
)
|
| 73 |
|
| 74 |
+
css_modify_left_column = '''
|
| 75 |
+
<style>
|
| 76 |
+
[data-testid="stHorizontalBlock"] > div:nth-child(1) {
|
| 77 |
+
overflow: hidden;
|
| 78 |
+
background-color: red;
|
| 79 |
+
height: 70vh;
|
| 80 |
+
}
|
| 81 |
+
</style>
|
| 82 |
+
'''
|
| 83 |
+
css_modify_right_column = '''
|
| 84 |
+
<style>
|
| 85 |
+
[data-testid="stHorizontalBlock"]> div:first-child {
|
| 86 |
+
background-color: red;
|
| 87 |
+
position: fixed
|
| 88 |
+
height: 70vh;
|
| 89 |
+
}
|
| 90 |
+
</style>
|
| 91 |
+
'''
|
| 92 |
+
css_disable_scrolling_container = '''
|
| 93 |
+
<style>
|
| 94 |
+
[data-testid="ScrollToBottomContainer"] {
|
| 95 |
+
overflow: hidden;
|
| 96 |
+
}
|
| 97 |
+
</style>
|
| 98 |
+
'''
|
| 99 |
+
|
| 100 |
+
|
| 101 |
+
# st.markdown(css_lock_column_fixed, unsafe_allow_html=True)
|
| 102 |
+
# st.markdown(css2, unsafe_allow_html=True)
|
| 103 |
|
| 104 |
|
| 105 |
def new_file():
|
|
|
|
| 251 |
# is_api_key_provided = st.session_state['api_key']
|
| 252 |
|
| 253 |
st.button(
|
| 254 |
+
'Reset chat memory.',
|
| 255 |
+
on_click=clear_memory(),
|
| 256 |
+
help="Clear the conversational memory. Currently implemented to retrain the 4 most recent messages.")
|
| 257 |
|
| 258 |
left_column, right_column = st.columns([1, 1])
|
| 259 |
|
|
|
|
| 342 |
left_column.markdown(get_pdf_display(st.session_state['binary']), unsafe_allow_html=True)
|
| 343 |
|
| 344 |
with right_column:
|
| 345 |
+
# css = '''
|
| 346 |
+
# <style>
|
| 347 |
+
# [data-testid="column"] {
|
| 348 |
+
# overflow: auto;
|
| 349 |
+
# height: 70vh;
|
| 350 |
+
# }
|
| 351 |
+
# </style>
|
| 352 |
+
# '''
|
| 353 |
+
# st.markdown(css, unsafe_allow_html=True)
|
| 354 |
|
| 355 |
# st.markdown(
|
| 356 |
# """
|
|
|
|
| 361 |
# unsafe_allow_html=True,
|
| 362 |
# )
|
| 363 |
|
|
|
|
| 364 |
if st.session_state.loaded_embeddings and question and len(question) > 0 and st.session_state.doc_id:
|
| 365 |
for message in st.session_state.messages:
|
| 366 |
with st.chat_message(message["role"]):
|
|
|
|
| 384 |
elif mode == "LLM":
|
| 385 |
with st.spinner("Generating response..."):
|
| 386 |
_, text_response = st.session_state['rqa'][model].query_document(question, st.session_state.doc_id,
|
| 387 |
+
context_size=context_size,
|
| 388 |
+
memory=st.session_state.memory)
|
| 389 |
|
| 390 |
if not text_response:
|
| 391 |
st.error("Something went wrong. Contact Luca Foppiano (Foppiano.Luca@nims.co.jp) to report the issue.")
|