Spaces:
Build error
Build error
| import streamlit as st | |
| from utils.logger import setup_logger | |
| import pandas as pd | |
| from PIL import Image | |
| import os | |
| logger = setup_logger(__name__) | |
| def prune_passport_for_display(analysis_results_for_id): | |
| data_to_display = {} | |
| data_to_display["document_category"] = "identity_verification_document" | |
| data_to_display["document_type"] = "passport" | |
| data_to_display["passport_number"] = analysis_results_for_id.get( | |
| "passport_number", None) | |
| data_to_display["full_name"] = analysis_results_for_id.get( | |
| "full_name", None) | |
| data_to_display["date_of_birth"] = analysis_results_for_id.get( | |
| "date_of_birth", None) | |
| data_to_display["nationality"] = analysis_results_for_id.get( | |
| "nationality", None) | |
| data_to_display["date_of_issue"] = analysis_results_for_id.get( | |
| "date_of_issue", None) | |
| data_to_display["date_of_expiry"] = analysis_results_for_id.get( | |
| "date_of_expiry", None) | |
| data_to_display["sex"] = analysis_results_for_id.get( | |
| "sex", None) | |
| data_to_display["address"] = analysis_results_for_id.get( | |
| "address", None) | |
| return data_to_display | |
| def display_passport(extracted_files, analysis_results_pruned): | |
| col1, col2 = st.columns([2, 3]) | |
| logger.info(f"file_path while displaying: {extracted_files}") | |
| st.markdown("---") | |
| with col1: | |
| if len(extracted_files) > 1: | |
| st.image(extracted_files, caption=[os.path.basename( | |
| img) for img in extracted_files], use_container_width=True) | |
| else: | |
| image = Image.open(extracted_files[0]) | |
| st.image(image, caption=os.path.basename( | |
| extracted_files[0])) # , | |
| # use_container_width=True) | |
| logger.info( | |
| f"analysis_results_pruned : {analysis_results_pruned}") | |
| with col2: | |
| simple_df = pd.DataFrame.from_dict( | |
| analysis_results_pruned, | |
| orient='index', columns=['Value']).reset_index() | |
| simple_df.columns = ['Key', 'Value'] | |
| simple_df = simple_df.fillna(value="Missing") | |
| simple_df.index += 1 | |
| st.dataframe(simple_df, use_container_width=True) | |
| # logger.info(f"simple_df: {simple_df}") | |