Update app.py

app.py

@@ -51,6 +51,7 @@ if uploaded_file:
     num_splits = st.number_input("How many splits do you want?", min_value=1, max_value=total_pages, step=1, value=1)
 
     output_files_with_ranges = []
+    download_links = []
     for i in range(num_splits):
         st.write(f"### Split {i + 1}")
         start = st.number_input(f"Start page for split {i + 1}", min_value=1, max_value=total_pages, step=1, value=1, key=f"start_{i}")
@@ -64,3 +65,10 @@ if uploaded_file:
 
     if st.button("Split PDF"):
         split_pdf(input_pdf_path, output_files_with_ranges)
+
+        for output_file, _ in output_files_with_ranges:
+            with open(output_file, "rb") as f:
+                download_links.append((output_file, f.read()))
+
+        for filename, data in download_links:
+            st.download_button(label=f"Download {filename}", data=data, file_name=filename, mime="application/pdf")