summaryrefslogtreecommitdiff
path: root/rag/ui.py
diff options
context:
space:
mode:
Diffstat (limited to 'rag/ui.py')
-rw-r--r--rag/ui.py20
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()