Update app.py

app.py

@@ -15,6 +15,8 @@ from functionforDownloadButtons import download_button
 import fitz
 import pandas as pd
 import base64
+import tempfile
+import os
 
 # set page size wide and theme
 st.set_page_config(layout='wide', initial_sidebar_state='collapsed',)
@@ -162,14 +164,12 @@ def app2():
                 else:
                     print("No Entities Extracted!!!")
                 
-                          
-                document.save(uploaded_file.name.replace(".pdf", "_annot.pdf"))   
-
-                #final_df.to_csv(uploaded_file.replace(".pdf", "_df.csv"))
-                #return final_df
+                temp_dir = tempfile.TemporaryDirectory()
+                document.save(tempfile.gettempdir()+"\\annott.pdf")          
+            
 
                 with c2:
-                    with open(uploaded_file.name.replace(".pdf", "_annot.pdf"),"rb") as f:
+                    with open((tempfile.gettempdir()+"\\annott.pdf"),"rb") as f:
                         base64_pdf = base64.b64encode(f.read()).decode('utf-8')
                     pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="800" height="800" type="application/pdf"></iframe>'
                     st.markdown(pdf_display, unsafe_allow_html=True)