Update app.py

app.py

@@ -22,29 +22,24 @@ def load_lottieurl(url: str):
 # Load the Lottie animation
 lottie_hello = load_lottieurl("https://lottie.host/d29edbdd-af94-402e-bb4c-b314f17897e6/n5SKHOkbYQ.json")
 
-# Function to display the login form in the main area, not sidebar
-def login():
-    login_container = st.empty()
-    with login_container.container():
-        st.header("Enter Password to Continue")
-        password = st.text_input("Password", type="password", key="password_input")
-        if st.button("Submit"):
-            if password == APP_PASSWORD:
-                st.session_state["authenticated"] = True
-                login_container.empty()
-                st.success("Authentication successful!")
-            else:
-                st.error("Incorrect password")
-
-# Main app logic
-if "authenticated" not in st.session_state:
-    st.session_state["authenticated"] = False
-
-if not st.session_state["authenticated"]:
-    login()
+# Initialize session state
+if 'authenticated' not in st.session_state:
+    st.session_state['authenticated'] = False
 
-if st.session_state["authenticated"]:
-    # Rest of your app code here (unchanged)
+# Function to handle login
+def login():
+    st.header("Enter Password to Continue")
+    password = st.text_input("Password", type="password", key="password_input")
+    if st.button("Submit"):
+        if password == APP_PASSWORD:
+            st.session_state['authenticated'] = True
+            st.success("Authentication successful!")
+            st.rerun()
+        else:
+            st.error("Incorrect password")
+
+# Main app content
+def main_content():
     # Custom styles with the background image and updated shadow effect
     st.markdown(
         """
@@ -150,4 +145,10 @@ if st.session_state["authenticated"]:
         height=150,
         width=150,
         key=None,
-    )
+    )
+
+# Main app logic
+if not st.session_state['authenticated']:
+    login()
+else:
+    main_content()