Update app.py

app.py

@@ -228,18 +228,47 @@ def image_to_image(image, k=5):
 
 st.title("#ditaduranuncamais Data Explorer")
 
-# Check if the directory exists
-if not os.path.exists(model_dir):
-    download_models()
+def check_password():
+    """Returns `True` if the user had the correct password."""
+
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        if st.session_state["password"] == st.secrets["password"]:
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # don't store password
+        else:
+            st.session_state["password_correct"] = False
+
+    if "password_correct" not in st.session_state:
+        # First run, show input for password.
+        st.text_input(
+            "Password", type="password", on_change=password_entered, key="password"
+        )
+        return False
+    elif not st.session_state["password_correct"]:
+        # Password not correct, show input + error.
+        st.text_input(
+            "Password", type="password", on_change=password_entered, key="password"
+        )
+        st.error("😕 Password incorrect")
+        return False
+    else:
+        # Password correct.
+        return True
+
+if not check_password():
+    st.stop()
 
 dataset = load_dataset()
 df = load_dataframe(dataset)
 image_model = load_img_model()
 text_model = load_txt_model()
 
-tab1, tab2, tab3 = st.tabs(["Data exploration", "Semantic search", "Stats"])
+menu_options = ["Data exploration", "Semantic search", "Stats"]
+st.sidebar.markdown('## Menu')
+selected_menu_option = st.sidebar.selectbox("Menu", menu_options)
 
-with tab1:
+if selected_menu_option == "Data exploration":
     st.dataframe(
         data=filter_dataframe(df),
     # use_container_width=True,
@@ -254,7 +283,7 @@ with tab1:
         hide_index=True,
     )
 
-with tab2:
+elif selected_menu_option == "Semantic search":
     tabs = ["Text to Text", "Text to Image", "Image to Image", "Image to Text"]
     selected_tab = st.radio("Select a search type", tabs)
 
@@ -343,7 +372,7 @@ with tab2:
                     hide_index=True,
                 )   
             
-with tab3:
+elif selected_menu_option == "Stats":
     st.markdown("### Time Series Analysis")
     # Dropdown to select variables
     variable = st.selectbox('Select Variable', ['Followers at Posting', 'Total Interactions', 'Likes', 'Comments'])