Wip streamlit

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()