Update app.py

app.py

@@ -63,17 +63,22 @@ def main():
 
     uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
 
-    group_size = st.number_input("Group size (pages per group)", min_value=1, value=20)  # グループサイズのみ
+    group_size = st.number_input("Group size (pages per group)", min_value=1, value=20)
 
     if uploaded_file is not None:
         try:
             file_bytes = uploaded_file.read()
             reader = PdfReader(io.BytesIO(file_bytes))
             split_files = split_pdf_by_pages(reader)
-            merged_files = merge_pdfs_in_groups(split_files, group_size)  # グループサイズを渡す
+            merged_files = merge_pdfs_in_groups(split_files, group_size)
 
+            # 分割されたページのダウンロードリンク
+            st.subheader("Split Pages")  # サブヘッダーを追加
             for i, file_obj in split_files.items():
                 create_download_link(file_obj.getvalue(), f"page_{i+1}.pdf")
+
+            # グループ化されたページのダウンロードリンク
+            st.subheader("Merged Groups")  # サブヘッダーを追加
             for i, file_obj in merged_files.items():
                 create_download_link(file_obj.getvalue(), f"group_{i+1}.pdf")