From d116abc63e350b092c2a7f9e1bb9b54298e21b2d Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 6 Apr 2024 13:14:40 +0200
Subject: Wip streamlit

---
 rag/ui.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 rag/ui.py

(limited to 'rag/ui.py')

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()
-- 
cgit v1.2.3-70-g09d2