import streamlit as st
import io
from utils import mock_api_call, parse_analysis_response, validate_pdf, displayPDF
from styles import apply_custom_styles, show_factor_section, show_detailed_factors, show_factor_summary
from policy_analyser.analyse import analyse
def main():
# Apply custom styles
apply_custom_styles()
# Header
st.markdown("""
""", unsafe_allow_html=True)
# File upload section
st.markdown('', unsafe_allow_html=True)
uploaded_files = st.file_uploader("Choose policy PDF files", type="pdf", accept_multiple_files=True)
st.markdown('
', unsafe_allow_html=True)
if uploaded_files:
# Create tabs for different views
tab1, tab2 = st.tabs(["Summary View", "Detailed Comparison"])
# Store analysis results
all_analyses = []
# Process each uploaded file
for uploaded_file in uploaded_files:
# Read PDF content
pdf_bytes = uploaded_file.read()
# displayPDF(pdf_bytes)
# Validate PDF
if not validate_pdf(pdf_bytes):
st.error(f"Invalid PDF file: {uploaded_file.name}")
continue
# Show loading state
with st.spinner(f"Analyzing {uploaded_file.name}..."):
try:
# Make API call
response = analyse(pdf_bytes)
print(response)
summary = next(
(item for item in response if item.get("stage") == "ANALYSIS_SUMMARY"), None
)['response']
# Parse response
good_factors, average_factors, bad_factors = parse_analysis_response(response)
# Store results
all_analyses.append({
'name': uploaded_file.name,
'good_factors': good_factors,
'average_factors': average_factors,
'bad_factors': bad_factors
})
except Exception as e:
st.error(f"Error analyzing {uploaded_file.name}: {str(e)}")
# Summary View Tab
with tab1:
for idx, analysis in enumerate(all_analyses):
with st.expander(f"### Policy {idx + 1}: {analysis['name']}"):
with st.container():
cols = st.columns(2)
displayed = 0
for verdict in ['Good', 'Average', 'Bad']:
lst = [f.split(':')[0] for f in analysis[f'{verdict.lower()}_factors']]
if len(lst) > 0:
title = f'{verdict} Factors'
if verdict == 'Good':
title += '✅'
sentiment = 'Yay!'
elif verdict == 'Average':
title += '⚠️'
sentiment = 'Hmmmm'
else:
title += '❌'
sentiment = 'Meh'
cols = st.columns(2)
with st.container():
with cols[0]:
if displayed % 3 == 0:
show_factor_section(title, lst, verdict.lower())
else:
show_factor_summary(summary[verdict], verdict.lower(), sentiment)
with cols[1]:
if displayed % 3 == 0:
show_factor_summary(summary[verdict], verdict.lower(), sentiment)
else:
show_factor_section(title, lst, verdict.lower())
displayed += 1
st.markdown('-----')
# with st.container():
# suggestion = next((item for item in response if item.get("stage") == "SUGGEST"), None)['response']
# show_factor_summary(suggestion, 'Good', 'What we have ⛑️')
# Detailed Comparison Tab
with tab2:
if len(all_analyses) > 1:
# Create comparison matrix
factors_to_compare = set()
for analysis in all_analyses:
factors_to_compare.update(
[f.split(':')[0] for f in analysis['good_factors'] +
analysis['average_factors'] + analysis['bad_factors']]
)
# Create comparison table
st.markdown("### Policy Comparison Matrix")
comparison_data = []
for factor in sorted(factors_to_compare):
row = {'Factor': factor}
for idx, analysis in enumerate(all_analyses):
policy_name = f"Policy {idx + 1}"
verdict = 'Not Found'
for category in ['good_factors', 'average_factors', 'bad_factors']:
for item in analysis[category]:
if item.split(':')[0] == factor:
verdict = category.split('_')[0].title()
break
row[policy_name] = verdict
comparison_data.append(row)
# Display comparison table
st.table(comparison_data)
else:
st.info("Upload multiple policies to see comparison")
# Footer
st.markdown("""
Upload one or more insurance policy PDFs to get detailed analysis and comparison.
We support all major insurance providers.
""", unsafe_allow_html=True)
if __name__ == "__main__":
st.set_page_config(
page_title="Insurance Policy Analyzer",
page_icon="📋",
layout="wide"
)
main() 