Samarth991 commited on
Commit
5536bb5
·
verified ·
1 Parent(s): ef6452f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -15,6 +15,7 @@ def main():
15
  )
16
 
17
  st.sidebar.title("Upload PDF")
 
18
  # st.sidebar.write("Download Demo PDF file from Below....")
19
 
20
  # with open("docs/SamarthTandon_cv_2.pdf", "rb") as file:
@@ -33,19 +34,22 @@ def main():
33
  type="password")
34
 
35
 
36
- if uploaded_file is not None and api_input is not None:
37
  save_folder = 'docs/'
38
  if not os.path.isdir(save_folder):
39
  os.makedirs(save_folder)
40
  save_path = Path(save_folder, uploaded_file.name)
41
- print(save_path)
 
42
  with open(save_path, mode='wb') as w:
43
  w.write(uploaded_file.getvalue())
44
 
45
  st.sidebar.success("File uploaded successfully",icon="✅")
 
 
 
 
46
 
47
- vector_store = PDF_4_QA(save_path)
48
- st.sidebar.success("Vector database created successfully",icon="✅")
49
  QA_Bot(vector_store,api_input)
50
 
51
  if __name__ == '__main__':
 
15
  )
16
 
17
  st.sidebar.title("Upload PDF")
18
+ api_input = None
19
  # st.sidebar.write("Download Demo PDF file from Below....")
20
 
21
  # with open("docs/SamarthTandon_cv_2.pdf", "rb") as file:
 
34
  type="password")
35
 
36
 
37
+ if uploaded_file is not None:
38
  save_folder = 'docs/'
39
  if not os.path.isdir(save_folder):
40
  os.makedirs(save_folder)
41
  save_path = Path(save_folder, uploaded_file.name)
42
+ print(save_path,api_input)
43
+
44
  with open(save_path, mode='wb') as w:
45
  w.write(uploaded_file.getvalue())
46
 
47
  st.sidebar.success("File uploaded successfully",icon="✅")
48
+
49
+ if api_input is not None:
50
+ vector_store = PDF_4_QA(save_path)
51
+ st.sidebar.success("Vector database created successfully",icon="✅")
52
 
 
 
53
  QA_Bot(vector_store,api_input)
54
 
55
  if __name__ == '__main__':