Update app.py

app.py

@@ -1,3 +1,236 @@
+import os
+#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
-st.title("Hello World")
-print("Application started")
+from PyPDF2 import PdfMerger
+from google.cloud import vision
+from google.oauth2 import service_account
+import fitz
+from PIL import Image
+from io import BytesIO
+import tempfile
+import json
+from streamlit_sortables import sort_items
+import subprocess
+from tqdm import tqdm
+
+# Check if /usr/bin/pdfinfo exists
+if os.path.exists("/usr/bin/pdfinfo"):
+    print("pdfinfo exists at /usr/bin/pdfinfo")
+    # Check the file permissions
+    permissions = os.stat("/usr/bin/pdfinfo").st_mode
+    print(f"File permissions: {oct(permissions)}")
+else:
+    print("pdfinfo does not exist at /usr/bin/pdfinfo")
+
+
+# Load Google Cloud Vision credentials from secret
+credentials_json = os.getenv("GOOGLE_CREDENTIALS_JSON")
+if credentials_json:
+    credentials_dict = json.loads(credentials_json)
+    credentials = service_account.Credentials.from_service_account_info(credentials_dict)
+    client = vision.ImageAnnotatorClient(credentials=credentials)
+else:
+    client = None
+
+# Function to extract text using Google Cloud Vision
+def extract_text_with_google_vision(image_bytes):
+    """Extracts text using Google Cloud Vision."""
+    image = vision.Image(content=image_bytes)
+    response = client.document_text_detection(image=image)
+    if response.error.message:
+        raise Exception(f"Google Cloud Vision API Error: {response.error.message}")
+    return response.full_text_annotation.text if response.full_text_annotation else ""
+
+# Function to process PDF for transcription
+def process_pdf(file):
+    """Converts PDF pages to images and extracts text."""
+    text = ""
+    with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as temp_pdf:
+        temp_pdf.write(file.read())
+        temp_pdf_path = temp_pdf.name
+
+    try:
+        doc = fitz.open(temp_pdf_path)
+        for i in tqdm(range(len(doc)), desc="Processing pages"):
+            page = doc.load_page(i)
+            pix = page.get_pixmap()
+            image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+            image_bytes = BytesIO()
+            image.save(image_bytes, format="PNG")
+            image_bytes.seek(0)
+            try:
+                page_text = extract_text_with_google_vision(image_bytes.getvalue())
+                text += f"--- Page {i + 1} ---\n{page_text}\n\n"
+            except Exception as e:
+                st.error(f"Error on page {i + 1}: {e}")
+    finally:
+        os.remove(temp_pdf_path)
+
+    return text
+
+
+# Function to generate thumbnail from PDF
+def get_pdf_thumbnail(uploaded_file):
+    """Generates a thumbnail image of the first page of a PDF."""
+    uploaded_file.seek(0)
+    doc = fitz.open(stream=uploaded_file.read(), filetype="pdf")
+    page = doc.load_page(0)
+    pix = page.get_pixmap()
+    img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+    img.thumbnail((100, 140))
+    return img
+
+# Function to merge PDFs
+def merge_pdfs(reordered_files):
+    """Merges multiple PDFs into one."""
+    merger = PdfMerger()
+    for file in reordered_files:
+        file.seek(0)
+        merger.append(file)
+    output_filename = "combined_document.pdf"
+    with open(output_filename, "wb") as output_file:
+        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."""
+
+    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
+
+    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)
+
+# Sidebar Navigation
+st.sidebar.title("Tool Selector")
+selection = st.sidebar.radio("Choose a tool:", ["PDF Combiner", "PDF Transcriber", "PDF Signer"])
+
+# PDF Combiner with Preview and Reordering
+if selection == "PDF Combiner":
+    st.title("PDF Combiner with Preview & Reordering")
+    st.write("Upload individual PDF pages, visualize them, reorder, and merge into a single PDF.")
+
+    uploaded_files = st.file_uploader("Upload PDF pages", type="pdf", accept_multiple_files=True)
+
+    if uploaded_files:
+        # Generate thumbnails and filenames for each uploaded PDF
+        thumbnails = []
+        filenames = []
+
+        for file in uploaded_files:
+            thumbnails.append(get_pdf_thumbnail(file))
+            filenames.append(file.name)
+
+        # Display thumbnails with filenames for reordering
+        st.write("**Drag and drop to reorder the PDFs:**")
+        reordered_filenames = sort_items(filenames)
+
+        # Map the filenames back to the corresponding files
+        reordered_files = [uploaded_files[filenames.index(name)] for name in reordered_filenames]
+
+        # Display the thumbnails in the new order
+        st.write("**Preview of selected order:**")
+        cols = st.columns(len(reordered_files))
+        for idx, file in enumerate(reordered_files):
+            with cols[idx]:
+                st.image(get_pdf_thumbnail(file), caption=file.name, use_container_width=True)
+
+        # Merge PDFs in the specified order
+        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")
+            os.remove(output_file)
+
+# PDF Transcriber Tool
+elif selection == "PDF Transcriber":
+    st.title("PDF Transcriber Tool")
+    st.write("Upload a scanned PDF to transcribe the text.")
+    if not client:
+        st.error("Google Cloud credentials are not set. Please configure the secret.")
+    uploaded_file = st.file_uploader("Upload a PDF", type=["pdf"])
+    if uploaded_file and st.button("Transcribe PDF"):
+        with st.spinner("Processing..."):
+            pdf_text = process_pdf(uploaded_file)
+            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)
+
+
+# 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_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)