Spaces:
Running
Running
File size: 2,625 Bytes
450bd79 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import streamlit as st
import os
import pandas as pd
import base64
from findUpdate import FindUpdate # Import the FindUpdate class
from tempfile import NamedTemporaryFile
class PDFDocumentProcessor:
def __init__(self):
self.uploaded_agreement = None
self.uploaded_template = None
self.find_update = FindUpdate() # Create an instance of FindUpdate class
def file_uploaders(self):
"""Function to handle file uploads."""
self.uploaded_agreement = st.file_uploader("Upload the PDF Agreement", type=['pdf'])
self.uploaded_template = st.file_uploader("Upload the PDF Template", type=['pdf'])
def process_files(self, agreement_path, template_path):
"""Main file processing logic."""
try:
# Use the find_update instance to call the processing function
result = self.find_update.main_processing_function(agreement_path, template_path)
st.success("Files successfully processed!")
# Convert the result dictionary to a DataFrame
df_changes = pd.DataFrame(result['changes'])
df_changes = df_changes[['section_number', 'page_number', 'actual', 'changed', 'analysis', 'type_of_change']]
# Display the DataFrame in the UI
st.dataframe(df_changes, height=600) # You can adjust height based on your needs
# Convert DataFrame to CSV for download
csv = df_changes.to_csv(index=False)
b64 = base64.b64encode(csv.encode()).decode() # some browsers need base64 encoding
# href = f'<a href="data:file/csv;base64,{b64}" download="document_changes.csv">Download CSV File</a>'
st.markdown(href, unsafe_allow_html=True)
except Exception as e:
st.error(f"Error processing files: {e}")
finally:
# Clean up temporary files after processing
os.remove(agreement_path)
os.remove(template_path)
def save_uploaded_files(self):
"""Save the uploaded files temporarily for processing."""
if self.uploaded_agreement and self.uploaded_template:
with NamedTemporaryFile(delete=False, suffix=".pdf", mode='wb') as temp_agreement:
temp_agreement.write(self.uploaded_agreement.read())
agreement_path = temp_agreement.name
with NamedTemporaryFile(delete=False, suffix=".pdf", mode='wb') as temp_template:
temp_template.write(self.uploaded_template.read())
template_path = temp_template.name
self.process_files(agreement_path, template_path)
|