Update app.py

app.py

@@ -280,14 +280,20 @@ def logout():
     st.session_state.auth_state["user"] = None
     st.session_state.auth_state["signed_in"] = False
 
+# Update the main function to include the new option
 # Update the main function to include the new option
 def main():
     st.sidebar.title("Options")
 
-    add_icon = Image.open(current_directory, "Add.png")
-    delete_icon = Image.open(current_directory, "delete.png")
-    recognize_icon = Image.open(current_directory, "Explore.png")
-    recognize_optimal_icon = Image.open(current_directory, "Explore+.png")
+    add_icon_path = os.path.join(current_directory, "Add.png")
+    delete_icon_path = os.path.join(current_directory, "delete.png")
+    recognize_icon_path = os.path.join(current_directory, "Explore.png")
+    recognize_optimal_icon_path = os.path.join(current_directory, "Explore+.png")
+
+    add_icon = Image.open(add_icon_path)
+    delete_icon = Image.open(delete_icon_path)
+    recognize_icon = Image.open(recognize_icon_path)
+    recognize_optimal_icon = Image.open(recognize_optimal_icon_path)
 
     option = st.sidebar.selectbox("Select Option", [add_icon, delete_icon, recognize_icon, recognize_optimal_icon], index=0, format_func=lambda x: "")
 
@@ -304,6 +310,7 @@ if __name__ == "__main__":
     authenticate_user_ui()
 
 
+
 #    add_icon_path = os.path.join
 #   delete_icon_path = os.path.join(current_directory, "delete.png")
 #  recognize_icon_path = os.path.join(current_directory, "Explore.png")