Update app.py

app.py

@@ -194,33 +194,48 @@ elif selection == "PDF Signer":
           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 # This is the return we were checking
+            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)
-        clicked = st.image(img, use_container_width=True)
-        os.remove(temp_pdf_path)
-
-        if clicked:
-          x = clicked.x
-          y = clicked.y
-          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 - Error 101")
-            else:
-              st.error("Please upload both a PDF file and a signature image")
+          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)
+          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