Spaces:
Sleeping
Sleeping
File size: 1,550 Bytes
4b41cfa a1c0614 4b41cfa 0402ac1 d59d547 4b41cfa 8da73e9 a1c0614 8da73e9 4b41cfa 13b91f3 a1c0614 bf2a56e 037c2d9 0402ac1 037c2d9 b441d00 037c2d9 a1c0614 13b91f3 a1c0614 d158285 4b41cfa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
import streamlit as st
from QA_Bot import QA_Bot
from PDF_Reader import PDF_4_QA
from PIL import Image
from pathlib import Path
# Streamlit app
def main():
st.session_state.disabled = False
st.session_state.visibility = "visible"
# Page config
st.set_page_config(page_title="Q&A ChatBot",
layout="wide"
)
st.sidebar.title("Upload PDF")
st.sidebar.write("Download Demo PDF file from Below....")
with open("docs/SamarthTandon_cv_2.pdf", "rb") as file:
btn = st.sidebar.download_button(
label="Download PDF",
data=file,
file_name="SamarthTandon_cv_2.pdf"
)
uploaded_file = st.sidebar.file_uploader("Choose a PDF file", type="pdf")
api_input = st.sidebar.text_input(
"Enter The API KEY 👇",
label_visibility=st.session_state.visibility,
disabled=st.session_state.disabled,
type="password"
)
if uploaded_file is not None:
save_folder = 'docs/'
save_path = Path(save_folder, uploaded_file.name)
print(save_path)
with open(save_path, mode='wb') as w:
w.write(uploaded_file.getvalue())
st.sidebar.success("File uploaded successfully.")
print(save_path)
vector_store = PDF_4_QA(save_path)
QA_Bot(vector_store,api_input)
# else:
# vector_store = PDF_4_QA("docs/SamarthTandon_cv_2.pdf")
# QA_Bot(vector_store,api_input)
if __name__ == '__main__':
main()
|