NEXAS commited on
Commit
0a26252
·
verified ·
1 Parent(s): b3a3ed2

Update src/pdf_up.py

Browse files
Files changed (1) hide show
  1. src/pdf_up.py +8 -11
src/pdf_up.py CHANGED
@@ -8,34 +8,31 @@ def process_uploaded_file():
8
  st.title("Upload File to Chat")
9
  uploaded_file = st.file_uploader("File upload", type="pdf")
10
  if uploaded_file:
 
11
  temp_dir = tempfile.mkdtemp()
12
  path = os.path.join(temp_dir, uploaded_file.name)
13
- #with open(path, "wb") as f:
14
- # f.write(uploaded_file.getvalue())
15
- print(path)
 
 
16
  st.write("Document uploaded successfully!")
 
 
17
  # Display the uploaded document
18
  st.write("Preview of the document:")
19
  st.write(uploaded_file)
20
 
21
  # Button to start parsing and vector database creation
22
  if st.button("Start Processing"):
23
- # Placeholder for processing logic
24
  st.write("Processing...")
25
-
26
- # Placeholder for progress bar
27
  with st.spinner('Processing...'):
28
  # Call your function to parse data and create vector database
29
  create_vector_database(path)
30
 
31
  st.success("Processing completed!")
32
-
33
- # Display success message
34
  st.write("Vector database created successfully!")
35
 
36
  # Show success image
37
  success_image = Image.open("success_image.jpg")
38
  st.image(success_image, caption="Success!", use_column_width=True)
39
-
40
- # Add a footer
41
- #st.text("Built with Streamlit")
 
8
  st.title("Upload File to Chat")
9
  uploaded_file = st.file_uploader("File upload", type="pdf")
10
  if uploaded_file:
11
+ # Create a temporary directory to save the uploaded file
12
  temp_dir = tempfile.mkdtemp()
13
  path = os.path.join(temp_dir, uploaded_file.name)
14
+
15
+ # Write the file to the temporary path
16
+ with open(path, "wb") as f:
17
+ f.write(uploaded_file.getvalue())
18
+
19
  st.write("Document uploaded successfully!")
20
+ st.write(f"File saved at: {path}") # Debugging aid
21
+
22
  # Display the uploaded document
23
  st.write("Preview of the document:")
24
  st.write(uploaded_file)
25
 
26
  # Button to start parsing and vector database creation
27
  if st.button("Start Processing"):
 
28
  st.write("Processing...")
 
 
29
  with st.spinner('Processing...'):
30
  # Call your function to parse data and create vector database
31
  create_vector_database(path)
32
 
33
  st.success("Processing completed!")
 
 
34
  st.write("Vector database created successfully!")
35
 
36
  # Show success image
37
  success_image = Image.open("success_image.jpg")
38
  st.image(success_image, caption="Success!", use_column_width=True)