Update app.py

app.py

@@ -120,6 +120,61 @@ def add_signature_to_pdf(pdf_file, signature_image, x, y):
       print(f"Error with pdf: {e}")
       return None
 
+def pdf_signer_ui(pdf_file, signature_image):
+    """Handles the UI and logic for PDF signing."""
+    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)
+            st.image(img, use_container_width=True)
+            if st.button("Add Signature"):
+                x = st.session_state.get("x", 50)
+                y = st.session_state.get("y", 50)
+                if 'clicked_image' not in st.session_state:
+                    st.session_state.clicked_image = False
+                if st.session_state.clicked_image:
+                    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")
+
+            else:
+                clicked = st.image(img, use_container_width=True, key="clicked_image", on_click=set_session, )
+                if clicked:
+                    st.session_state.x = clicked.x
+                    st.session_state.y = clicked.y
+                    st.session_state.clicked_image = True
+                    st.experimental_rerun()
+            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)
+
+def set_session():
+    st.session_state.clicked_image = True
+
+
 # Sidebar Navigation
 st.sidebar.title("Tool Selector")
 selection = st.sidebar.radio("Choose a tool:", ["PDF Combiner", "PDF Transcriber", "PDF Signer"])
@@ -186,56 +241,4 @@ elif selection == "PDF Signer":
     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:
-      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)
-          st.image(img, use_container_width=True)
-
-
-          if st.button("Add Signature"):
-                x = st.session_state.get("x", 50)
-                y = st.session_state.get("y", 50)
-
-                if 'clicked_image' not in st.session_state:
-                    st.session_state.clicked_image = False
-                if st.session_state.clicked_image:
-                    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")
-          else:
-            clicked = st.image(img, use_container_width=True, key="clicked_image", on_click = set_session, )
-            if clicked:
-               st.session_state.x = clicked.x
-               st.session_state.y = clicked.y
-               st.session_state.clicked_image = True
-               st.experimental_rerun()
-
-          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)
-
-def set_session():
-    st.session_state.clicked_image = True
+        pdf_signer_ui(pdf_file, signature_image)