Added Credentials

.gitignore

@@ -0,0 +1 @@
+.vscode

app.py

@@ -2,6 +2,9 @@ import os
 import base64
 import streamlit as st
 from streamlit_option_menu import option_menu
+from streamlit_authenticator import Authenticate
+import yaml
+from yaml.loader import SafeLoader
 import pandas as pd
 from PIL import Image
 
@@ -16,11 +19,27 @@ from glob import glob
 
 MODEL_PATH = "pankaj-munde/FScout_v0.2"
 # image_dir = "./Data/images/"
-detr_preprocessor = AutoImageProcessor.from_pretrained(MODEL_PATH)
-detr_model = AutoModelForObjectDetection.from_pretrained(MODEL_PATH)
+detr_preprocessor = AutoImageProcessor.from_pretrained(MODEL_PATH, token=st.secrets["HF_TOKEN"])
+detr_model = AutoModelForObjectDetection.from_pretrained(MODEL_PATH, token=st.secrets["HF_TOKEN"])
 colors = [[236, 112, 99], [165, 105, 189], [ 225, 9, 232], [ 255, 38, 8 ], [ 247, 249, 249 ], [170, 183, 184 ],  [ 247, 249, 249 ],  [ 247, 249, 249 ]]
 
 
+# with open('./static/config.yaml') as file:
+#     config = yaml.load(file, Loader=SafeLoader)
+
+config = {"credentials": st.secrets["CREDENTIALS"], "cookie": st.secrets["COOKIE"]}
+
+authenticator = Authenticate(
+    config['credentials'],
+    config['cookie']['name'],
+    config['cookie']['key'],
+    config['cookie']['expiry_days'],
+    config['preauthorized']
+)
+
+
+name, authentication_status, username = authenticator.login('Login', 'main')
+
 # images_lst = os.listdir(image_dir)
 # images = []
 # for file in images_lst:
@@ -154,11 +173,6 @@ def image_checkup_v4(ipath, thresh, show_count):
     return [], img_with_box
 
 
-
-st.set_page_config(layout="wide")
-
-
-
 def add_logo(logo_path, width, height):
     """Read and return a resized logo"""
     logo = Image.open(logo_path)
@@ -167,62 +181,64 @@ def add_logo(logo_path, width, height):
 
 
 # st.write("<hr/>", unsafe_allow_html=True)
+if st.session_state["authentication_status"]:
+    with st.sidebar:
+        my_logo = add_logo(logo_path="./static//FarmGyan logo_1.png", width=50, height=60)
+        st.image(my_logo)
+        ucol, bcol = st.columns([3, 2])
+        ucol.write(f'## Welcome *{st.session_state["name"]}*')
+        with bcol:
+            authenticator.logout('Logout', 'main')
+        
+        st.write("<hr/>", unsafe_allow_html=True)
 
-with st.sidebar:
-    my_logo = add_logo(logo_path="./static/FarmGyan logo_1.png", width=50, height=60)
-    
-    st.image(my_logo)
-    
-    ucol, bcol = st.columns([3, 2])
-    # st.write("<hr/>", unsafe_allow_html=True)
-
-st.title(":seedling: FarmGyan | Insects Scouting")
-st.write("<hr/>", unsafe_allow_html=True)
-    
-st.write("## 🖖 Upload image for prediction")
-
-uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
-st.write("<hr/>", unsafe_allow_html=True)
-
-with st.spinner(text='In progress'):
-    st.sidebar.write("## ⚙️ Configurations") 
-    st.sidebar.write("<hr/>", unsafe_allow_html=True)
-
-    st.sidebar.write("#### Prediction Threshold") 
-    thresh = st.sidebar.slider("Threshold", 0.0, 1.0, 0.7, 0.1)
-    st.sidebar.write("#### Boxes Count") 
-    show_count = st.sidebar.checkbox("Show Count")
-    if uploaded_file is not None:
-        clicked = None
-        image = Image.open(uploaded_file).convert("RGB")
-        predicted_data, result_image = image_checkup_v4(image, thresh, show_count)
-        # print(predicted_data)
-
-        col, col1 = st.columns([2, 4])
-        feedback_submitted = False  # Initialize the flag
-        with col:
-            st.subheader("💯 Predicted Labels")
-            st.write(f"<h3>Total Count : {sum([d['Count'] for d in predicted_data])}</h3>", unsafe_allow_html=True)
-            for i, d in enumerate(predicted_data):
-                # Create HTML markup with style information
+    st.title(":seedling: FarmGyan | Insects Scouting")
+    st.write("<hr/>", unsafe_allow_html=True)
+        
+    st.write("## 🖖 Upload image for prediction")
+
+    uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
+    st.write("<hr/>", unsafe_allow_html=True)
+
+    with st.spinner(text='In progress'):
+        st.sidebar.write("## ⚙️ Configurations") 
+        st.sidebar.write("<hr/>", unsafe_allow_html=True)
+
+        st.sidebar.write("#### Prediction Threshold") 
+        thresh = st.sidebar.slider("Threshold", 0.0, 1.0, 0.7, 0.1)
+        st.sidebar.write("#### Boxes Count") 
+        show_count = st.sidebar.checkbox("Show Count")
+        if uploaded_file is not None:
+            clicked = None
+            image = Image.open(uploaded_file).convert("RGB")
+            predicted_data, result_image = image_checkup_v4(image, thresh, show_count)
+            # print(predicted_data)
+
+            col, col1 = st.columns([2, 4])
+            feedback_submitted = False  # Initialize the flag
+            with col:
+                st.subheader("💯 Predicted Labels")
+                st.write(f"<h3>Total Count : {sum([d['Count'] for d in predicted_data])}</h3>", unsafe_allow_html=True)
+                for i, d in enumerate(predicted_data):
+                    # Create HTML markup with style information
+                    
+                    html_string = f"""
+                    <div style="display: flex; align-items: center;">
+                        <b style="margin-right: 15px">{i + 1}. </b>
+                        <div style="background-color: rgb({d["Color"][0]}, {d["Color"][1]}, {d["Color"][2]}); width: 20px; height: 20px; border: 1px solid black; margin-right: 10px;"></div>
+                        <p style="margin-top: 15px"><b>{d["Insect"]} : {d["Count"]} </b></p>
+                    </div>
+                """
+
+                    st.markdown(html_string, unsafe_allow_html=True)
+
+                st.write("<hr/>", unsafe_allow_html=True)
                 
-                html_string = f"""
-                <div style="display: flex; align-items: center;">
-                    <b style="margin-right: 15px">{i + 1}. </b>
-                    <div style="background-color: rgb({d["Color"][0]}, {d["Color"][1]}, {d["Color"][2]}); width: 20px; height: 20px; border: 1px solid black; margin-right: 10px;"></div>
-                    <p style="margin-top: 15px"><b>{d["Insect"]} : {d["Count"]} </b></p>
-                </div>
-              """
-
-                st.markdown(html_string, unsafe_allow_html=True)
-
-            st.write("<hr/>", unsafe_allow_html=True)
-            
-        with col1:
-            st.subheader("🍀 Predicted Image")
-            st.write("<br/>", unsafe_allow_html=True)
-
-            st.image(result_image)
+            with col1:
+                st.subheader("🍀 Predicted Image")
+                st.write("<br/>", unsafe_allow_html=True)
+
+                st.image(result_image)