summaryrefslogtreecommitdiff
path: root/rag/ui.py
blob: 1e4dd6456b084e050bcf5d0012cd2ef8dea6f2d8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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()