Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import re
 import PyPDF2
 from langchain_community.embeddings import OllamaEmbeddings
@@ -102,13 +103,17 @@ async def on_chat_start():
     while files is None:
         files = await cl.AskFileMessage(
             content="Please upload a pdf file to begin!",
-            # accept=["application/pdf"],
             accept=["application/pdf", "image/jpeg", "image/png", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"],
             max_size_mb=100,
             timeout=180, 
         ).send()
 
     file = files[0] # Get the first uploaded file
+
+    # Store file path in user session for later cleanup
+    uploaded_files = cl.user_session.get("uploaded_files", [])
+    uploaded_files.append(file.path)
+    cl.user_session.set("uploaded_files", uploaded_files)
     
     # Inform the user that processing has started
     msg = cl.Message(content=f"Processing `{file.name}`...")
@@ -179,6 +184,17 @@ async def on_chat_start():
     # Store the chain in user session
     cl.user_session.set("chain", chain)
 
+@cl.on_session_end
+async def on_session_end():
+    # Get the uploaded files from the user session
+    uploaded_files = cl.user_session.get("uploaded_files", [])
+    # Delete each file
+    for file_path in uploaded_files:
+        if os.path.exists(file_path):
+            os.remove(file_path)
+            logging.info(f"Deleted file: {file_path}")
+    # Clear the uploaded files list
+    cl.user_session.set("uploaded_files", [])
 
 @cl.on_message
 async def main(message: cl.Message):