Update streamlit_app2.py

streamlit_app2.py

@@ -28,22 +28,42 @@ with st.sidebar:
 
     if process_button and uploaded_files:
         with st.spinner("Processing PDFs..."):
-            # Create temporary file paths to pass to your PDF Loader
             pdf_paths = []
+            original_names = [] # <--- Add this
             for uploaded_file in uploaded_files:
+                original_names.append(uploaded_file.name) # <--- Capture real name
                 with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
                     tmp.write(uploaded_file.getvalue())
                     pdf_paths.append(tmp.name)
             
-            # Use your existing process_documents method
-            st.session_state.rag_graph.process_documents(pdf_paths)
+            # Pass BOTH the paths and the original names
+            st.session_state.rag_graph.process_documents(
+                pdf_paths, 
+                original_names=original_names
+            )
             
-            # Clean up temp files
             for path in pdf_paths:
                 os.remove(path)
-                
             st.success("Documents Indexed Successfully!")
 
+    # if process_button and uploaded_files:
+    #     with st.spinner("Processing PDFs..."):
+    #         # Create temporary file paths to pass to your PDF Loader
+    #         pdf_paths = []
+    #         for uploaded_file in uploaded_files:
+    #             with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+    #                 tmp.write(uploaded_file.getvalue())
+    #                 pdf_paths.append(tmp.name)
+            
+    #         # Use your existing process_documents method
+    #         st.session_state.rag_graph.process_documents(pdf_paths)
+            
+    #         # Clean up temp files
+    #         for path in pdf_paths:
+    #             os.remove(path)
+                
+    #         st.success("Documents Indexed Successfully!")
+
 # --- Chat Interface ---
 # Display existing messages
 for message in st.session_state.messages: