Spaces:
Sleeping
Sleeping
| 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() |