PDFToolBox

src/apps/pages/programs/StudyPrograms/PDFToolbox.py

@@ -15,21 +15,132 @@ def readPDF():
   else:
     st.warning("No text found in this page", icon="⚠️")
 
-# TODO: Implement mergePDF, splitPDF, rotatePDF, encryptPDF, decryptPDF functions
+def readPDF():
+    st.markdown("#### PDF Reader App")
+    file = st.file_uploader("Upload a PDF file", type=["pdf"])
+    if not file:
+        st.stop()
+    reader = PyPDF2.PdfReader(file)
+    numPage = st.number_input("From which page to start reading?", format="%d", min_value=1, max_value=len(reader.pages))
+    page = reader.pages[numPage-1]
+    text = page.extract_text()
+    if text:
+        st.write(text)
+    else:
+        st.warning("No text found on this page", icon="⚠️")
+
+def mergePDF():
+    st.markdown("#### PDF Merger")
+    uploaded_files = st.file_uploader("Upload PDF files to merge", type=["pdf"], accept_multiple_files=True)
+    if uploaded_files:
+        pdf_writer = PyPDF2.PdfWriter()
+        for file in uploaded_files:
+            pdf_reader = PyPDF2.PdfReader(file)
+            for page_num in range(len(pdf_reader.pages)):
+                page = pdf_reader.pages[page_num]
+                pdf_writer.add_page(page)
+        output_pdf = "merged_output.pdf"
+        with open(output_pdf, "wb") as f:
+            pdf_writer.write(f)
+        with open(output_pdf, "rb") as f:
+            st.download_button("Download Merged PDF", f, file_name="merged_output.pdf")
+
+def splitPDF():
+    st.markdown("#### PDF Splitter")
+    file = st.file_uploader("Upload a PDF file to split", type=["pdf"])
+    if file:
+        pdf_reader = PyPDF2.PdfReader(file)
+        start_page = st.number_input("Start page", min_value=1, max_value=len(pdf_reader.pages), value=1)
+        end_page = st.number_input("End page", min_value=start_page, max_value=len(pdf_reader.pages), value=len(pdf_reader.pages))
+        pdf_writer = PyPDF2.PdfWriter()
+        for i in range(start_page - 1, end_page):
+            pdf_writer.add_page(pdf_reader.pages[i])
+        output_pdf = "split_output.pdf"
+        with open(output_pdf, "wb") as f:
+            pdf_writer.write(f)
+        with open(output_pdf, "rb") as f:
+            st.download_button("Download Split PDF", f, file_name="split_output.pdf")
+
+def rotatePDF():
+    st.markdown("#### PDF Rotate Tool")
+    file = st.file_uploader("Upload a PDF file to rotate", type=["pdf"])
+
+    if not file:
+        st.warning("Please upload a PDF file", icon="⚠️")
+        return
+
+    reader = PyPDF2.PdfReader(file)
+    writer = PyPDF2.PdfWriter()
+
+    # Select rotation angle
+    angle = st.selectbox("Select rotation angle (clockwise)", [90, 180, 270])
+
+    # Rotate each page
+    for page_num in range(len(reader.pages)):
+        page = reader.pages[page_num]
+        page.rotate(angle)  # Use positive angles for clockwise rotation
+        writer.add_page(page)
+
+    # Save rotated PDF
+    output_file = "rotated_output.pdf"
+    with open(output_file, "wb") as f:
+        writer.write(f)
+
+    # Provide download link
+    with open(output_file, "rb") as f:
+        st.download_button("Download Rotated PDF", f, file_name=output_file)
+
+def encryptPDF():
+    st.markdown("#### PDF Encryptor")
+    file = st.file_uploader("Upload a PDF file to encrypt", type=["pdf"])
+    if file:
+        password = st.text_input("Enter password", type="password")
+        if password:
+            pdf_reader = PyPDF2.PdfReader(file)
+            pdf_writer = PyPDF2.PdfWriter()
+            for page in pdf_reader.pages:
+                pdf_writer.add_page(page)
+            pdf_writer.encrypt(password)
+            output_pdf = "encrypted_output.pdf"
+            with open(output_pdf, "wb") as f:
+                pdf_writer.write(f)
+            with open(output_pdf, "rb") as f:
+                st.download_button("Download Encrypted PDF", f, file_name="encrypted_output.pdf")
+
+def decryptPDF():
+    st.markdown("#### PDF Decryptor")
+    file = st.file_uploader("Upload an encrypted PDF file to decrypt", type=["pdf"])
+    if file:
+        password = st.text_input("Enter password to decrypt", type="password")
+        if password:
+            pdf_reader = PyPDF2.PdfReader(file)
+            if pdf_reader.decrypt(password):
+                pdf_writer = PyPDF2.PdfWriter()
+                for page in pdf_reader.pages:
+                    pdf_writer.add_page(page)
+                output_pdf = "decrypted_output.pdf"
+                with open(output_pdf, "wb") as f:
+                    pdf_writer.write(f)
+                with open(output_pdf, "rb") as f:
+                    st.download_button("Download Decrypted PDF", f, file_name="decrypted_output.pdf")
+            else:
+                st.error("Incorrect password", icon="🚫")
+
 def PDFToolbox():
-  st.title("PDF Reader & Editor App")
-  choice = st.selectbox("Choose an operation", [None, "Read PDF", "Merge PDF", "Split PDF", "Rotate PDF", "Encrypt PDF", "Decrypt PDF"])
-  if choice == "Read PDF":
-    readPDF()
-  elif choice == "Merge PDF":
-    st.info("Coming soon!", icon="🚧")
-  elif choice == "Split PDF":
-    st.info("Coming soon!", icon="🚧")
-  elif choice == "Rotate PDF":
-    st.info("Coming soon!", icon="🚧")
-  elif choice == "Encrypt PDF":
-    st.info("Coming soon!", icon="🚧")
-  elif choice == "Decrypt PDF":
-    st.info("Coming soon!", icon="🚧")
-  else:
-    st.warning("Invalid choice!", icon="⚠️")
+    st.title("PDF Reader & Editor App")
+    choice = st.selectbox("Choose an operation", [None, "Read PDF", "Merge PDF", "Split PDF", "Rotate PDF", "Encrypt PDF", "Decrypt PDF"])
+
+    if choice == "Read PDF":
+        readPDF()
+    elif choice == "Merge PDF":
+        mergePDF()
+    elif choice == "Split PDF":
+        splitPDF()
+    elif choice == "Rotate PDF":
+        rotatePDF()
+    elif choice == "Encrypt PDF":
+        encryptPDF()
+    elif choice == "Decrypt PDF":
+        decryptPDF()
+    else:
+        st.warning("Invalid choice!", icon="⚠️")