Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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().
|
| 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,
|
| 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"):
|