errantanomie commited on
Commit
61edfd9
·
verified ·
1 Parent(s): a4a8662

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -100,9 +100,10 @@ def merge_pdfs(reordered_pages, uploaded_file):
100
  page = doc.load_page(page_data['page_index'])
101
  page.set_rotation(page_data['rotation'])
102
  temp_pdf_page = BytesIO()
103
- page.get_pixmap().writePNG(temp_pdf_page)
104
  temp_pdf_page.seek(0)
105
  merger.append(temp_pdf_page)
 
106
  output_filename = "combined_document.pdf"
107
  with open(output_filename, "wb") as output_file:
108
  merger.write(output_file)
@@ -117,6 +118,10 @@ def download_file(output_file, file_name, mime_type):
117
  mime=mime_type,
118
  )
119
 
 
 
 
 
120
  def generate_unique_filename(original_filename, suffix, file_type):
121
  """Generates a unique filename based on date, time, and original filename."""
122
  now = datetime.now()
@@ -241,7 +246,7 @@ if selection == "PDF Document Separator":
241
  with st.spinner("Creating PDF..."):
242
  output_file = merge_pdfs(st.session_state.target_pages, st.session_state.uploaded_pdf)
243
  st.success("PDF created successfully!")
244
- download_file(output_file, generate_unique_filename(st.session_state.pdf_filename, "", ".pdf"), "application/pdf")
245
  os.remove(output_file)
246
  st.text_input("Enter output transcription filename", key='transcribe_filename_input', value = st.session_state.transcribe_filename)
247
  if st.button("Transcribe PDF"):
 
100
  page = doc.load_page(page_data['page_index'])
101
  page.set_rotation(page_data['rotation'])
102
  temp_pdf_page = BytesIO()
103
+ page.get_pixmap(alpha=False).save(temp_pdf_page, "png")
104
  temp_pdf_page.seek(0)
105
  merger.append(temp_pdf_page)
106
+
107
  output_filename = "combined_document.pdf"
108
  with open(output_filename, "wb") as output_file:
109
  merger.write(output_file)
 
118
  mime=mime_type,
119
  )
120
 
121
+ def generate_user_filename(filename, file_type):
122
+ """Generates a filename based on user input and file type."""
123
+ return f"{filename}{file_type}"
124
+
125
  def generate_unique_filename(original_filename, suffix, file_type):
126
  """Generates a unique filename based on date, time, and original filename."""
127
  now = datetime.now()
 
246
  with st.spinner("Creating PDF..."):
247
  output_file = merge_pdfs(st.session_state.target_pages, st.session_state.uploaded_pdf)
248
  st.success("PDF created successfully!")
249
+ download_file(output_file, generate_user_filename(st.session_state.pdf_filename, ".pdf"), "application/pdf")
250
  os.remove(output_file)
251
  st.text_input("Enter output transcription filename", key='transcribe_filename_input', value = st.session_state.transcribe_filename)
252
  if st.button("Transcribe PDF"):