Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-os.environ["PATH"] += os.pathsep + "/usr/bin" + os.pathsep + "/usr/local/bin"
+#os.environ["PATH"] += os.pathsep + "/usr/bin" + os.pathsep + "/usr/local/bin" #Removed redundant path setting
 print(f"Current PATH: {os.environ['PATH']}")
 import streamlit as st
 from PyPDF2 import PdfMerger
@@ -24,13 +24,6 @@ else:
     print("pdfinfo does not exist at /usr/bin/pdfinfo")
 
 
-try:
-    output = subprocess.check_output(["/usr/bin/pdfinfo", "-version"])
-    print(f"Poppler version: {output.decode('utf-8')}")
-except FileNotFoundError:
-    print("Poppler (pdfinfo) is not installed or not found in PATH.")
-
-
 # Load Google Cloud Vision credentials from secret
 credentials_json = os.getenv("GOOGLE_CREDENTIALS_JSON")
 if credentials_json:
@@ -100,83 +93,33 @@ def merge_pdfs(reordered_files):
         merger.write(output_file)
     return output_filename
 
-
-def add_signature_to_pdf(pdf_file, signature_image, x, y):
-    """Adds a signature image to the PDF at the given coordinates."""
-
+def download_file(output_file, file_name, mime_type):
+    with open(output_file, "rb") as f:
+        st.download_button(
+            label="Download File",
+            data=f,
+            file_name=file_name,
+            mime=mime_type,
+        )
+
+def rotate_pdf(pdf_file, rotation_angle):
+    """Rotates all pages of a PDF by a specified angle."""
     with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as temp_pdf:
-        try:
-          temp_pdf.write(pdf_file.read())
-          temp_pdf_path = temp_pdf.name
-          print(f"temp file name: {temp_pdf_path}")
-        except Exception as e:
-          print(f"Error creating temp file: {e}")
-          return None
+      temp_pdf.write(pdf_file.read())
+      temp_pdf_path = temp_pdf.name
 
-    try:
-        doc = fitz.open(temp_pdf_path)
-        page = doc[0]  # Operate on the first page for simplicity
-        rect = fitz.Rect(x, y, x + 200, y + 100) # Size for the signature
-
-        img = Image.open(signature_image)
-        img_bytes = BytesIO()
-        img.save(img_bytes, format="PNG")
-        img_bytes.seek(0)
-        page.insert_image(rect, stream=img_bytes.read(), keep_proportion=True)
-        output_filename = "signed_document.pdf"
-        doc.save(output_filename)
-        os.remove(temp_pdf_path)
-        return output_filename
-    except Exception as e:
-      print(f"Error with pdf: {e}")
-      return None
-
-def pdf_signer_ui(pdf_file, signature_image):
-    """Handles the UI and logic for PDF signing."""
-    if st.button("Preview PDF"):
-      with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as temp_pdf:
-          try:
-              temp_pdf.write(pdf_file.read())
-              temp_pdf_path = temp_pdf.name
-              print(f"temp file name: {temp_pdf_path}")
-          except Exception as e:
-              print(f"Error creating temporary file: {e}")
-              st.error("Could not create temporary file. Please make sure your uploaded file is valid")
-              return
-
-          try:
-              doc = fitz.open(temp_pdf_path)
-              page = doc[0]
-              pix = page.get_pixmap()
-              img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-              clicked = st.image(img, use_container_width=True)
-              if clicked:
-                x = clicked.x
-                y = clicked.y
-                if st.button("Add Signature"):
-                  with st.spinner("Adding Signature..."):
-                    if pdf_file and signature_image:  # Check for file existence
-                        output_file = add_signature_to_pdf(pdf_file, signature_image, x, y)
-                        if output_file:
-                            st.success("Signature added successfully")
-                            with open(output_file, "rb") as f:
-                                st.download_button("Download Signed PDF", f, file_name=output_file, mime="application/pdf")
-                            os.remove(output_file)
-                        else:
-                            st.error("An Error occurred when creating the signed PDF")
-                    else:
-                        st.error("Please upload both a PDF file and a signature image")
-
-              os.remove(temp_pdf_path)
-
-          except Exception as e:
-            print(f"Error with pdf: {e}")
-            st.error(f"Could not open PDF file {e}")
-            os.remove(temp_pdf_path)
+    doc = fitz.open(temp_pdf_path)
+    for page in doc:
+        page.set_rotation(rotation_angle)
+    output_filename = "rotated_document.pdf"
+    doc.save(output_filename)
+    os.remove(temp_pdf_path)
+
+    return output_filename
 
 # Sidebar Navigation
 st.sidebar.title("Tool Selector")
-selection = st.sidebar.radio("Choose a tool:", ["PDF Combiner", "PDF Transcriber", "PDF Signer"])
+selection = st.sidebar.radio("Choose a tool:", ["PDF Combiner", "PDF Transcriber", "PDF Rotator"])
 
 # PDF Combiner with Preview and Reordering
 if selection == "PDF Combiner":
@@ -212,8 +155,7 @@ if selection == "PDF Combiner":
         if st.button("Merge PDFs"):
             output_file = merge_pdfs(reordered_files)
             st.success("PDF pages combined successfully!")
-            with open(output_file, "rb") as f:
-                st.download_button("Download Combined PDF", f, file_name=output_file, mime="application/pdf")
+            download_file(output_file, "combined_document.pdf", "application/pdf")
             os.remove(output_file)
 
 # PDF Transcriber Tool
@@ -229,15 +171,20 @@ elif selection == "PDF Transcriber":
             st.success("Text extraction complete!")
             st.text_area("Extracted Text", pdf_text, height=400)
             output_file_name = f"(T){os.path.splitext(uploaded_file.name)[0]}.txt"
-            st.download_button("Download Extracted Text", pdf_text, file_name=output_file_name)
+            download_file(output_file_name, f"(T){os.path.splitext(uploaded_file.name)[0]}.txt", "text/plain")
 
 
-# PDF Signer Tool
-elif selection == "PDF Signer":
-    st.title("PDF Signer")
-    st.write("Upload a PDF and place a signature on it by clicking the preview.")
+# PDF Rotator Tool
+elif selection == "PDF Rotator":
+    st.title("PDF Rotator")
+    st.write("Upload a PDF and rotate all pages by a specified angle.")
 
     pdf_file = st.file_uploader("Upload PDF", type=["pdf"])
-    signature_image = st.file_uploader("Upload Signature Image", type=["png", "jpg", "jpeg"])
-    if pdf_file and signature_image:
-        pdf_signer_ui(pdf_file, signature_image)
+    rotation_angle = st.selectbox("Select rotation angle:", [90, 180, 270])
+
+    if pdf_file and st.button("Rotate PDF"):
+        with st.spinner("Rotating PDF..."):
+            output_file = rotate_pdf(pdf_file, rotation_angle)
+            st.success("PDF rotated successfully!")
+            download_file(output_file, "rotated_document.pdf", "application/pdf")
+            os.remove(output_file)