Upload 7 files

.gitattributes

@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/ab.png filter=lfs diff=lfs merge=lfs -text
+docs/image/receipt_00001.png filter=lfs diff=lfs merge=lfs -text
+docs/image/receipt_00002.png filter=lfs diff=lfs merge=lfs -text
+docs/image/receipt_00003.png filter=lfs diff=lfs merge=lfs -text
+docs/image/w_invoice1.png filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,22 @@
+FROM python:3.11-slim
+
+WORKDIR /code
+
+COPY requirements.txt ./
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+RUN useradd -m -u 1000 user
+
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+COPY --chown=user config/config.toml $HOME/app/.streamlit/config.toml
+
+CMD ["streamlit", "run", "main.py", "--server.port=7860", "--server.address=0.0.0.0"]

config.py

@@ -0,0 +1,5 @@
+class Settings():
+    sparrow_key = ""
+
+
+settings = Settings()

main.py

@@ -0,0 +1,187 @@
+import streamlit as st
+from streamlit_option_menu import option_menu
+from tools.utilities import load_css
+import json
+
+from views.dashboard import Dashboard
+from views.data_annotation import DataAnnotation
+from views.model_training import ModelTraining
+from views.model_tuning import ModelTuning
+from views.data_inference import DataInference
+from views.setup import Setup
+from views.data_review import DataReview
+from views.about import About
+
+import streamlit_javascript as st_js
+
+st.set_page_config(
+    page_title="Sparrow",
+    page_icon="favicon.ico",
+    layout="wide"
+)
+
+load_css()
+
+
+class Model:
+    menuTitle = "Sparrow"
+    option1 = "Dashboard"
+    option2 = "Data Annotation"
+    option3 = "Model Training"
+    option4 = "Model Tuning"
+    option5 = "Inference"
+    option6 = "Data Review"
+    option7 = "Setup"
+    option8 = "About"
+
+    menuIcon = "menu-up"
+    icon1 = "speedometer"
+    icon2 = "activity"
+    icon3 = "motherboard"
+    icon4 = "graph-up-arrow"
+    icon5 = "journal-arrow-down"
+    icon6 = "droplet"
+    icon7 = "clipboard-data"
+    icon8 = "chat"
+
+
+def view(model):
+    with st.sidebar:
+        menuItem = option_menu(model.menuTitle,
+                               [model.option1, model.option2, model.option5, model.option6, model.option7, model.option8],
+                               icons=[model.icon1, model.icon2, model.icon5, model.icon6, model.icon7, model.icon8],
+                               menu_icon=model.menuIcon,
+                               default_index=0,
+                               styles={
+                                   "container": {"padding": "5!important", "background-color": "#fafafa"},
+                                   "icon": {"color": "black", "font-size": "25px"},
+                                   "nav-link": {"font-size": "16px", "text-align": "left", "margin": "0px",
+                                                "--hover-color": "#eee"},
+                                   "nav-link-selected": {"background-color": "#037ffc"},
+                               })
+
+    if menuItem == model.option1:
+        Dashboard().view(Dashboard.Model())
+        logout_widget()
+
+    if menuItem == model.option2:
+        if 'ui_width' not in st.session_state or 'device_type' not in st.session_state or 'device_width' not in st.session_state:
+            # Get UI width
+            ui_width = st_js.st_javascript("window.innerWidth", key="ui_width_comp")
+            device_width = st_js.st_javascript("window.screen.width", key="device_width_comp")
+
+            if ui_width > 0 and device_width > 0:
+                # Add 20% of current screen width to compensate for the sidebar
+                ui_width = round(ui_width + (20 * ui_width / 100))
+
+                if device_width > 768:
+                    device_type = 'desktop'
+                else:
+                    device_type = 'mobile'
+
+                st.session_state['ui_width'] = ui_width
+                st.session_state['device_type'] = device_type
+                st.session_state['device_width'] = device_width
+
+                st.experimental_rerun()
+        else:
+            DataAnnotation().view(DataAnnotation.Model(), st.session_state['ui_width'], st.session_state['device_type'],
+                                  st.session_state['device_width'])
+        logout_widget()
+
+    if menuItem == model.option3:
+        ModelTraining().view(ModelTraining.Model())
+        logout_widget()
+
+    if menuItem == model.option4:
+        ModelTuning().view(ModelTuning.Model())
+        logout_widget()
+
+    if menuItem == model.option5:
+        if 'ui_width' not in st.session_state or 'device_type' not in st.session_state or 'device_width' not in st.session_state:
+            # Get UI width
+            ui_width = st_js.st_javascript("window.innerWidth", key="ui_width_comp")
+            device_width = st_js.st_javascript("window.screen.width", key="device_width_comp")
+
+            if ui_width > 0 and device_width > 0:
+                # Add 20% of current screen width to compensate for the sidebar
+                ui_width = round(ui_width + (20 * ui_width / 100))
+
+                if device_width > 768:
+                    device_type = 'desktop'
+                else:
+                    device_type = 'mobile'
+
+                st.session_state['ui_width'] = ui_width
+                st.session_state['device_type'] = device_type
+                st.session_state['device_width'] = device_width
+
+                st.experimental_rerun()
+        else:
+            DataInference().view(DataInference.Model(), st.session_state['ui_width'], st.session_state['device_type'],
+                                 st.session_state['device_width'])
+
+        logout_widget()
+
+    if menuItem == model.option6:
+        if 'ui_width' not in st.session_state or 'device_type' not in st.session_state or 'device_width' not in st.session_state:
+            # Get UI width
+            ui_width = st_js.st_javascript("window.innerWidth", key="ui_width_comp")
+            device_width = st_js.st_javascript("window.screen.width", key="device_width_comp")
+
+            if ui_width > 0 and device_width > 0:
+                # Add 20% of current screen width to compensate for the sidebar
+                ui_width = round(ui_width + (20 * ui_width / 100))
+
+                if device_width > 768:
+                    device_type = 'desktop'
+                else:
+                    device_type = 'mobile'
+
+                st.session_state['ui_width'] = ui_width
+                st.session_state['device_type'] = device_type
+                st.session_state['device_width'] = device_width
+
+                st.experimental_rerun()
+        else:
+            DataReview().view(DataReview.Model(), st.session_state['ui_width'], st.session_state['device_type'],
+                              st.session_state['device_width'])
+
+        logout_widget()
+
+    if menuItem == model.option7:
+        Setup().view(Setup.Model())
+        logout_widget()
+
+    if menuItem == model.option8:
+        About().view(About.Model())
+        logout_widget()
+
+
+def logout_widget():
+    with st.sidebar:
+        st.markdown("---")
+        # st.write("User:", "John Doe")
+        st.write("Version:", "2.0.0")
+        # st.button("Logout")
+        # st.markdown("---")
+
+        if 'visitors' not in st.session_state:
+            with open("docs/visitors.json", "r") as f:
+                visitors_json = json.load(f)
+                visitors = visitors_json["meta"]["visitors"]
+
+            visitors += 1
+            visitors_json["meta"]["visitors"] = visitors
+
+            with open("docs/visitors.json", "w") as f:
+                json.dump(visitors_json, f)
+
+            st.session_state['visitors'] = visitors
+        else:
+            visitors = st.session_state['visitors']
+
+        st.write("Counter:", visitors)
+
+
+view(Model())

requirements.txt

@@ -0,0 +1,8 @@
+streamlit==1.22.0
+streamlit_option_menu
+streamlit-sparrow-labeling==0.1.1
+streamlit_nested_layout
+streamlit-javascript
+natsort
+streamlit-aggrid==0.3.3
+pandas<2.0