Update app.py

app.py

@@ -2,6 +2,7 @@ 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():
@@ -14,6 +15,7 @@ def main():
 
     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",
@@ -22,7 +24,12 @@ def main():
         )
 
     uploaded_file = st.sidebar.file_uploader("Choose a PDF file", type="pdf")
-
+    save_folder = 'docs/'
+    save_path = Path(save_folder, File.name)
+    print(save_path)
+    with open(save_path, mode='wb') as w:
+        w.write(uploaded_file.getvalue())
+    
     api_input = st.sidebar.text_input(
         "Enter The API KEY 👇",
         label_visibility=st.session_state.visibility,
@@ -30,12 +37,13 @@ def main():
         type="password"
     )
     
-    # if uploaded_file is not None:
-    #     st.sidebar.success("File uploaded successfully.")
-    #     print(uploaded_file)
-    #     vector_store = PDF_4_QA(uploaded_file)
-    #     QA_Bot(vector_store,api_input)
-    # else:
+    if uploaded_file is not None:
+        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)