Update app.py

def remove_pdf(filename, state):
if filename not in state["files"]:
return f"{filename} not found.", state, list(state["files"].keys())

# Remove its chunks
chunks_to_remove = state["files"].pop(filename)

# Remove from global chunk list
state["chunks"] = [c for c in state["chunks"] if c not in chunks_to_remove]

# Rebuild FAISS
if state["chunks"]:
index, _ = build_vectorstore(state["chunks"])
state["index"] = index
else:
state["index"] = None

state["pdf_count"] -= 1

return (
f"Removed {filename}. Total chunks: {len(state['chunks'])}",
state,
list(state["files"].keys())
)

app.py

@@ -176,7 +176,30 @@ def load_pdf(files, state):
     except Exception as e:
         return f"Error processing PDF: {e}", state
 
+def remove_pdf(filename, state):
+    if filename not in state["files"]:
+        return f"{filename} not found.", state, list(state["files"].keys())
 
+    # Remove its chunks
+    chunks_to_remove = state["files"].pop(filename)
+
+    # Remove from global chunk list
+    state["chunks"] = [c for c in state["chunks"] if c not in chunks_to_remove]
+
+    # Rebuild FAISS
+    if state["chunks"]:
+        index, _ = build_vectorstore(state["chunks"])
+        state["index"] = index
+    else:
+        state["index"] = None
+
+    state["pdf_count"] -= 1
+
+    return (
+        f"Removed {filename}. Total chunks: {len(state['chunks'])}",
+        state,
+        list(state["files"].keys())
+    )
 
 def answer_question(question, state):