RagChatBot / app.py
Tanmay211998's picture
Update app.py
64403d4 verified
raw
history blame contribute delete
540 Bytes
import streamlit as st
from PDF_Reader import PDF_4_QA
# Streamlit app
def main():
# Page config
st.set_page_config(page_title="Rag Chatbot",
layout="wide"
)
st.sidebar.title("Upload PDF")
uploaded_file = st.sidebar.file_uploader("Choose a PDF file", type="pdf")
if uploaded_file is not None:
st.sidebar.success("File uploaded successfully.")
vector_store = PDF_4_QA(uploaded_file)
QA_Bot(vector_store)
if __name__ == '__main__':
main()