diff options
Diffstat (limited to 'rag')
-rw-r--r-- | rag/ui.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/rag/ui.py b/rag/ui.py new file mode 100644 index 0000000..1e4dd64 --- /dev/null +++ b/rag/ui.py @@ -0,0 +1,20 @@ +import streamlit as st + + +# from loguru import logger as log +# from rag.rag import RAG + +def upload_pdfs(): + files = st.file_uploader( + "Choose pdfs to add to the knowledge base", + type="pdf", + accept_multiple_files=True, + ) + for file in files: + bytes = file.read() + st.write(bytes) + + +if __name__ == "__main__": + st.header("RAG-UI") + upload_pdfs() |