wehub-resource-sync
60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
from llmware.library import Library
|
|
from llmware.retrieval import Query
|
|
|
|
import streamlit as st
|
|
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(0, os.getcwd())
|
|
|
|
from Utils import get_stored_files, get_stored_libraries
|
|
|
|
ACCOUNT_NAME = 'lecture_tool'
|
|
|
|
|
|
#
|
|
# Accesses transcription by concatenating text from each block corresponding to
|
|
# the specified filename in the specified library.
|
|
#
|
|
@st.cache_data(show_spinner=False)
|
|
def get_transcript(library_name, filename):
|
|
library = Library().load_library(library_name, account_name=ACCOUNT_NAME)
|
|
print('\nupdate: library card - ', library.get_library_card())
|
|
|
|
query = Query(library)
|
|
|
|
text = ''
|
|
for result in query.get_whole_library():
|
|
if result['file_source'] == filename:
|
|
text += result['text_search']
|
|
|
|
return text
|
|
|
|
|
|
#
|
|
# Main block for GUI logic.
|
|
#
|
|
if __name__ == '__main__':
|
|
st.title('View your transcripts')
|
|
|
|
st.write('### Prompt info')
|
|
|
|
library_name = st.selectbox(
|
|
'Select the library:',
|
|
tuple(get_stored_libraries())
|
|
)
|
|
|
|
if library_name:
|
|
filename = st.selectbox(
|
|
'Select the file:',
|
|
tuple(get_stored_files(library_name))
|
|
)
|
|
|
|
if (st.button('View')):
|
|
with st.spinner('Loading transcript... don\'t leave this page!'):
|
|
response = get_transcript(library_name, filename)
|
|
|
|
st.write('### Transcript')
|
|
st.write(response)
|