Test / app.py
mohith96's picture
Update app.py
1daeb77 verified
raw
history blame contribute delete
827 Bytes
import streamlit as st
import tempfile
import os
@st.cache_resource
def process_pdfs(uploaded_files):
file_names = []
for uploaded_file in uploaded_files:
# Save uploaded file to a temp file
with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
tmp_file.write(uploaded_file.getvalue())
tmp_file_path = tmp_file.name
file_names.append(uploaded_file.name)
# Clean up immediately
os.remove(tmp_file_path)
return file_names
st.title("PDF Upload Test")
uploaded_files = st.file_uploader(
"Upload PDF files", type=["pdf"], accept_multiple_files=True
)
if uploaded_files:
st.info("Processing uploaded PDFs...")
file_names = process_pdfs(uploaded_files)
st.success(f"Uploaded {len(file_names)} file(s): {file_names}")