File size: 4,792 Bytes
e7539e5 |
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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
import json
from PIL import Image
import os
import streamlit as st
import pandas as pd
class DashboardView():
def __init__(self):
self.current_path = os.path.dirname(os.path.abspath(__file__))
self.samples_path = os.path.join(self.current_path, '../samples')
json_path = os.path.join(self.samples_path, 'samples.json')
self.samples = json.loads(open(json_path).read())
def main_view(self,ct):
ct.title("Rocks segmentation")
uploaded_file = ct.file_uploader("Choose a file", type=["jpeg", "png", "jpg"])
if uploaded_file is not None:
file_contents = uploaded_file.read()
#two columns
col1, col2 = ct.columns(2)
with col1:
with ct.expander("original image"):
ct.image(file_contents, use_column_width=True)
with col2:
with ct.expander("no background image"):
self.nobackground_image_view(ct,uploaded_file.name)
col3, col4 = ct.columns(2)
with col3:
with ct.expander("masked image"):
self.masked_image_view(ct,uploaded_file.name)
with col4:
with ct.expander("segmented image"):
self.segmented_image_view(ct,uploaded_file.name)
return uploaded_file
def nobackground_image_view(self,ct,image_path):
# get current path
try:
masked_image_path = os.path.join(self.samples_path, self.samples[image_path]['no_background_path'])
masked_image = Image.open(masked_image_path)
ct.image(masked_image, use_column_width=True)
except:
ct.error("An error has occured...")
def masked_image_view(self,ct,image_path):
# get current path
try:
masked_image_path = os.path.join(self.samples_path, self.samples[image_path]['masked_path'])
masked_image = Image.open(masked_image_path)
ct.image(masked_image, use_column_width=True)
except:
ct.error("An error has occured...")
def segmented_image_view(self,ct,image_path):
# get current path
try:
segmented_image_path = os.path.join(self.samples_path, self.samples[image_path]['segmentation_path'])
segmented_image = Image.open(segmented_image_path)
ct.image(segmented_image, use_column_width=True)
except:
ct.error("An error has occured...")
def save_image_tojson_view(self,ct, data):
# get current path
try:
data_path = os.path.join(self.current_path, '../db/data.json')
with open(data_path, 'w') as f:
json.dump(data, f)
ct.success("The image has been saved")
except:
ct.error("An error has occured...")
def load_data_from_folder(self, ct):
# get current path
try:
data_path = os.path.join(self.current_path, '../db/reports')
data = os.listdir(data_path)
report = ct.selectbox('📁 Select a report', data, key="csv_key")
report_path = os.path.join(data_path, report)
df = pd.read_csv(report_path)
ct.dataframe(df)
except:
ct.error("An error has occured...")
def load_reports_from_folder(self, ct):
# get current path
try:
reports_path = os.path.join(self.current_path, '../db/pdf_reports')
reports = os.listdir(reports_path)
print(reports)
report = st.selectbox('📁 Select a report', reports, key="pdf_key")
report_path = os.path.join(reports_path, report)
with open(report_path, 'rb') as f:
pdf_data = f.read()
# with st.expander("PDF Viewer"):
# with open(report_path, 'rb') as f:
# pdf_data = f.read()
# ct.write(f'<iframe src="data:application/pdf;base64,{pdf_data.encode("base64")}" width="700" height="1000" type="application/pdf"></iframe>', unsafe_allow_html=True)
ct.download_button(
label=f"Download {report}",
data=pdf_data,
file_name=report_path,
mime="application/pdf"
)
except Exception as e:
ct.write(e)
ct.error("An error has occurred...")
def display_psd_chart(self,ct):
charts_path = os.path.join(self.current_path, '../db/graphs')
chart = os.path.join(charts_path, 'psd1.png')
ct.image(chart, use_column_width=False, width=400) |