Update app.py

app.py

@@ -200,8 +200,8 @@ def run_chatbot(retriever, pc, llm):
     {context}
     </context>
 
-    Important: You cannot quote the context in the responses. If you do that, there will be a strict penalty for it.
-                                                  
+    Important: Don't start revealing context in your responses until its asked. First look at the question and then think if the context is needed to answer this or its a normal question, once you have judged then only answer the question.                                              
+
     Answer the following question:
 
     {question}""")
@@ -262,6 +262,19 @@ def process_and_chat():
     # Get current uploaded filenames
     current_uploaded_filenames = {file.name for file in uploaded_files} if uploaded_files else set()
 
+    # Check for removed files
+    files_to_remove = set(st.session_state.uploaded_files.keys()) - current_uploaded_filenames
+    if files_to_remove:
+        for file_name in files_to_remove:
+            # Remove file from session state
+            if file_name in st.session_state.uploaded_files:
+                # Delete the file from data directory
+                file_path = st.session_state.uploaded_files[file_name]["path"]
+                if os.path.exists(file_path):
+                    os.remove(file_path)
+                # Remove from session state
+                del st.session_state.uploaded_files[file_name]
+
     # Process newly uploaded files
     if uploaded_files:
         files_added = False
@@ -279,8 +292,13 @@ def process_and_chat():
     # Check for changes in files
     current_files = set(st.session_state.uploaded_files.keys())
 
-    # Process documents only if files have changed
-    if current_files != st.session_state.previous_files:
+    # If files have changed (added or removed), reset chat and process documents
+    if current_files != st.session_state.previous_files or files_to_remove:
+        # Reset chat state
+        st.session_state.chat_enabled = False
+        if "messages" in st.session_state:
+            del st.session_state.messages
+        
         st.session_state.previous_files = current_files
         
         if current_files:
@@ -291,7 +309,6 @@ def process_and_chat():
             st.session_state.processing = False
         else:
             st.warning('Please upload a file to continue')
-            st.session_state.chat_enabled = False
     
     # If files exist and chat is enabled, show chat interface
     if current_files and st.session_state.chat_enabled:
@@ -319,6 +336,8 @@ def process_and_chat():
             st.session_state.chat_enabled = False
             # Clear the previous files to force reprocessing
             st.session_state.previous_files = set()
+            if "messages" in st.session_state:
+                del st.session_state.messages
 
 # Call the main function
 process_and_chat()