Update app.py

def load_pdf(files, state):
if not files:
return "No file uploaded.", state, list(state["files"].keys())

try:
for file in files:
filename = file.name

text = read_pdf(file)
new_chunks = chunk_text(text)

if not new_chunks:
continue

# Save chunks for this file
state["files"][filename] = new_chunks

# Add to global chunk list
state["chunks"].extend(new_chunks)

state["pdf_count"] += 1

# Rebuild FAISS
index, _ = build_vectorstore(state["chunks"])
state["index"] = index

return (
f"{state['pdf_count']} PDF(s) loaded. Total chunks: {len(state['chunks'])}",
state,
list(state["files"].keys())
)

except Exception as e:
return (
f"Error processing PDF: {e}",
state,
list(state["files"].keys())
)

app.py

@@ -147,34 +147,43 @@ def generate_answer(question, context):
 # ============================================================
 
 def load_pdf(files, state):
-    print("FILES RECEIVED:", files)
-    print("TYPE:", type(files))
-
     if not files:
-        return "No file uploaded.", state
+        return "No file uploaded.", state, list(state["files"].keys())
 
     try:
         for file in files:
+            filename = file.name
+
             text = read_pdf(file)
             new_chunks = chunk_text(text)
 
             if not new_chunks:
                 continue
 
+            # Save chunks for this file
+            state["files"][filename] = new_chunks
+
+            # Add to global chunk list
             state["chunks"].extend(new_chunks)
+
             state["pdf_count"] += 1
 
-        # Rebuild vectorstore with ALL chunks
+        # Rebuild FAISS
         index, _ = build_vectorstore(state["chunks"])
         state["index"] = index
 
         return (
             f"{state['pdf_count']} PDF(s) loaded. Total chunks: {len(state['chunks'])}",
             state,
+            list(state["files"].keys())
         )
 
     except Exception as e:
-        return f"Error processing PDF: {e}", state
+        return (
+            f"Error processing PDF: {e}",
+            state,
+            list(state["files"].keys())
+        )
 
 def remove_pdf(filename, state):
     if filename not in state["files"]: