Update app.py

app.py

@@ -201,8 +201,8 @@ from langchain_nomic.embeddings import NomicEmbeddings
 import os
 
 llm_groq = ChatGroq(
-            model_name='llama3-70b-8192'
-    )
+    model_name='llama3-70b-8192'
+)
 
 # Initialize anonymizer
 anonymizer = PresidioReversibleAnonymizer(analyzed_fields=['PERSON', 'EMAIL_ADDRESS', 'PHONE_NUMBER', 'IBAN_CODE', 'CREDIT_CARD', 'CRYPTO', 'IP_ADDRESS', 'LOCATION', 'DATE_TIME', 'NRP', 'MEDICAL_LICENSE', 'URL'], faker_seed=18)
@@ -285,7 +285,6 @@ 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, 
@@ -313,7 +312,6 @@ async def on_chat_start():
     docsearch = await cl.make_async(Chroma.from_texts)(
         [anonymized_text], embeddings, metadatas=[{"source": "0-pl"}]
     )
-    # }
     
     # Initialize message history for conversation
     message_history = ChatMessageHistory()
@@ -340,6 +338,7 @@ async def on_chat_start():
     await msg.update()
     # Store the chain in user session
     cl.user_session.set("chain", chain)
+    cl.user_session.set("docsearch", docsearch)  # Store the docsearch in session
     cl.user_session.set("file_path", file.path)  # Store the file path in session
 
 
@@ -363,12 +362,12 @@ async def main(message: cl.Message):
 
 @cl.on_chat_end
 async def on_chat_end():
-    chain = cl.user_session.get("chain")
+    docsearch = cl.user_session.get("docsearch")
     file_path = cl.user_session.get("file_path")
 
-    if chain:
+    if docsearch:
         # Clear the vector store
-        chain.retriever.store.clear()
+        docsearch.delete()
 
     if file_path and os.path.exists(file_path):
         # Remove the uploaded file
@@ -378,4 +377,3 @@ async def on_chat_end():
     cl.user_session.clear()
 
     logging.info("User session ended, data cleared.")
-