up 1

.gitignore

@@ -0,0 +1,60 @@
+
+.DS_Store
+app.py
+background_remover/remove_bg.py
+db/.DS_Store
+db/data.json
+db/graphs/psd1.png
+db/pdf_reports/CSD Aker BP - 4-2-2024 - AkerBPSample1.pdf
+db/pdf_reports/CSD Aker BP - 4-2-2024 - AkerBPSample2.pdf
+db/pdf_reports/CSD Aker BP - 4-2-2024 - AkerBPSample5.pdf
+db/pdf_reports/Report 01:10:2023.pdf
+db/pdf_reports/report 28:09:2023.pdf
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 3950-3.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 4250-4.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 4550-5.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 4850-6.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5150-7.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5250-8.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5350-9.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5420-10.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5450-11.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5550-12.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5650-13.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5750-14.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 5950-15.csv
+db/reports/CSD REPSOL CRANCK - 1-7-2024 - 6050-16.csv
+functions/__init__.py
+functions/__pycache__/__init__.cpython-311.pyc
+functions/__pycache__/rtmp.cpython-311.pyc
+functions/__pycache__/utils.cpython-311.pyc
+functions/rtmp.py
+functions/utils.py
+main.py
+model/.DS_Store
+model/__init__.py
+model/__pycache__/__init__.cpython-311.pyc
+model/__pycache__/background_remover.cpython-311.pyc
+model/__pycache__/grain_analysis.cpython-311.pyc
+model/__pycache__/rock_image.cpython-311.pyc
+model/__pycache__/segmentation_model.cpython-311.pyc
+model/__pycache__/test_segmentation_model.cpython-311.pyc
+model/background_remover.py
+model/checkpoints/checkpoint
+model/checkpoints/seg_model.data-00000-of-00001
+model/checkpoints/seg_model.index
+model/grain_analysis.py
+model/image_quality_checker.py
+model/rock_image.py
+model/segmentation_model.py
+model/test_segmentation_model.py
+README.md
+requirements.txt
+roboflow/landingai_inference.py
+roboflow/roboflow_inference.py
+samples/.DS_Store
+samples/A003_20201229_103823_image.png
+samples/nobackground/XTO_2A_09-19-23_no_background.png
+samples/original/XTO_2A_09-19-23.jpg
+samples/samples.json
+samples/segmented/XTO_2A_09-19-23_segmented.png

README.md

@@ -10,4 +10,3 @@ pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,6 +0,0 @@
-import streamlit as st
-
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
-
-

views/__init__.py

@@ -0,0 +1 @@
+import streamlit as st

views/dashboard_view.py

@@ -0,0 +1,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)

views/sidebar.py

@@ -0,0 +1,53 @@
+import os 
+import streamlit as st
+from views.utils import * 
+from PIL import Image
+
+class Sidebar:
+    def __init__(self):
+        pass
+   
+    def choose_folder_view(self,ct):
+        with ct:
+            
+            #read image with pillow
+            current_dir = os.path.dirname(os.path.abspath(__file__))
+            image_path = os.path.join(current_dir, 'assets/drilldocs_logo.jpeg')
+            image = Image.open(image_path)
+            new_height = 200  # Desired height
+            aspect_ratio = image.width / image.height
+            new_width = int(new_height * aspect_ratio)
+            resized_image = image.resize((new_width, new_height))
+            ct.image(resized_image,use_column_width=False) 
+
+            ct.header('⚙️ Parameters')
+            ct.subheader('Choose the folder to store the images')
+
+    def file_selector_view(self,ct):
+        folder_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../..', 'uploads'))
+        folders = []
+        files = os.listdir(folder_path)
+        new_folder = '➕ new client'
+        for folder in files:
+            flags =['.streamlit','__pycache__']
+            if os.path.isdir(folder) and folder not in flags:
+                folders.append(folder)
+        
+        folders.append(new_folder)
+        selected_filename = ct.selectbox('📁 Select a folder for saving', folders, key="folder_selectbox")
+        if selected_filename==folders[-1]: #if we select "add new folder"
+            fn,fb=ct.columns(2)
+            folder_name = fn.text_input('Client name',placeholder="Client name", label_visibility='collapsed')
+            add_btn = fb.button('Add')
+            if add_btn:
+                try:
+                    on_click_btn(folder_name=folder_name)
+                    st.experimental_rerun()
+                except:
+                    ct.error("Error with folder name ")
+                
+        elif is_any_directory_not_empty(selected_filename):
+            ct.warning("This folder is not empty! it's content will be erased ")
+        
+        return os.path.join(folder_path, selected_filename)
+    

views/utils.py

@@ -0,0 +1,18 @@
+import os 
+import streamlit as st
+
+def on_click_btn(folder_name):
+    if not os.path.exists(folder_name):
+        os.makedirs(folder_name)
+    else:
+        st.error('Folder exists')
+
+def is_any_directory_not_empty(directories):
+    """Check if any of the specified directories is not empty."""
+    
+    if os.listdir(directories):
+        # Found a non-empty directory
+        return True
+    # All directories are empty
+    return False
+