| #src/components/upload.py | |
| import streamlit as st | |
| from utils.database import insert_document | |
| def handle_document_upload(conn, backend): | |
| st.markdown("### 📤 Upload Documents") | |
| uploaded_files = st.file_uploader( | |
| "Upload PDF documents", | |
| type="pdf", | |
| accept_multiple_files=True | |
| ) | |
| if uploaded_files: | |
| new_files = [f for f in uploaded_files if f.name not in st.session_state.processed_files] | |
| if new_files: | |
| with st.spinner("Processing documents..."): | |
| all_texts, document_names, _ = backend.upload_and_parse_documents(new_files) | |
| if all_texts: | |
| try: | |
| with conn: | |
| for doc, doc_name in zip(all_texts, document_names): | |
| if insert_document(conn, doc_name, doc): | |
| st.session_state.processed_files.add(doc_name) | |
| st.success(f"Successfully uploaded {len(new_files)} new documents!") | |
| except Exception as e: | |
| st.error(f"Error saving documents to database: {e}") |