docverifyrag / app.py
Carlos Salgado
add debugging messages, import os
80d18e0
raw
history blame
821 Bytes
import streamlit as st
import io
import os
import tempfile
from scripts import generate_metadata, ingest
st.title('DocVerifyRAG')
st.write('Anomaly detection for BIM document metadata')
uploaded_file = st.file_uploader("Choose a PDF file", type=["pdf","txt"])
if uploaded_file is not None:
try:
with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as tmp:
tmp.write(uploaded_file.read())
file_path = tmp.name
st.write(f'Created temporary file {file_path}')
docs = ingest(file_path)
metadata = generate_metadata(docs)
st.write('## Converted Text')
st.write(metadata)
# Clean up the temporary file
os.remove(file_path)
except Exception as e:
st.error(f'Error: {e}')