Update app.py

app.py

@@ -17,7 +17,8 @@ from sentence_transformers import CrossEncoder
 import numpy as np
 import random
 import string
-import os
+import tempfile
+
 
 # User variables
 uploaded_file = st.sidebar.file_uploader("Upload your PDF", type="pdf")
@@ -52,16 +53,15 @@ elif st.session_state.error_message:
     st.sidebar.error(st.session_state.error_message)
 
 if uploaded_file:
-    # Save the uploaded file temporarily
-    temp_file_path = os.path.join("/tmp", uploaded_file.name)
-    with open(temp_file_path, "wb") as f:
-        f.write(uploaded_file.getbuffer())
+    # Create a temporary file
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
+        temp_file.write(uploaded_file.getbuffer())
+        temp_file_path = temp_file.name
 
     # Load the PDF using PyPDFLoader
     docs = PyPDFLoader(temp_file_path).load()
 
-    # Clean up the temporary file after loading
-    os.remove(temp_file_path)
+    # The temporary file will be automatically deleted when the application stops
 else:
     st.warning("Please upload a PDF file.")