Updated the login page

app.py

@@ -92,6 +92,7 @@ def main():
     if "authenticated" not in st.session_state:
         st.session_state.authenticated = False
         st.session_state.username = None
+        st.session_state.page = "Login"
 
     def login_page():
         st.title("Login")
@@ -99,11 +100,13 @@ def main():
         password = st.text_input("Password", type="password", key="login_password")
         if st.button("Login"):
             if authenticate_user(username, password):
+                st.success(f"Welcome back, {username}!")
+                st.write("Cliok the Go to Settings to run the AI.")
                 st.session_state.authenticated = True
                 st.session_state.username = username
+                st.session_state.update(page="Main Page")
             else:
                 st.error("Invalid username or password.")
-
         st.write("Don't have an account? [Sign Up](#)")
 
     def sign_up_page():
@@ -117,7 +120,7 @@ def main():
                 st.error("Username already exists. Please choose another.")
 
     def settings_page():
-        st.title("Manage Your Settings")
+        st.title("Chat with Gemini 1.5 flash AI")
         username = st.session_state.username
         settings = load_settings(username)
 
@@ -125,28 +128,40 @@ def main():
         topic = st.text_area("Topic", value=settings.get("topic", ""), key="topic")
         instructions = st.text_area("Custom Instructions", value=settings.get("instructions", ""), key="instructions")
 
-        if st.button("Save Settings"):
-            settings = {"topic": topic, "instructions": instructions}
-            save_settings(username, settings)
-            st.success("Settings saved successfully!")
-
         if st.button("Generate Prompt"):
+            settings = {"topic": topic, "instructions": instructions}
+            save_settings(username, settings)            
             result = generate_prompt(topic, instructions)
             st.success(result)
 
     # Navigation between pages
-    st.sidebar.title("Navigation")
-    page = st.sidebar.radio("Go to", ["Login", "Sign Up", "Settings"])
-
-    if page == "Login":
+    if st.session_state.page == "Login":
         login_page()
-    elif page == "Sign Up":
+    elif st.session_state.page == "Sign Up":
         sign_up_page()
-    elif page == "Settings":
+    elif st.session_state.page == "Main Page":
         if st.session_state.authenticated:
             settings_page()
         else:
             st.error("You must log in to access this page.")
+            st.session_state.page = "Login"
+
+    # Sidebar for navigation
+    st.sidebar.title("Navigation")
+    if st.session_state.authenticated:
+        if st.sidebar.button("Go to Main Page"):
+            st.session_state.page = "Main Page"
+        if st.sidebar.button("Logout"):
+            st.session_state.authenticated = False
+            st.session_state.username = None
+            st.session_state.page = "Login"
+    else:
+        st.sidebar.radio(
+            "Go to",
+            ["Login", "Sign Up"],
+            key="sidebar_nav",
+            on_change=lambda: st.session_state.update(page=st.session_state.sidebar_nav)
+        )
 
 if __name__ == "__main__":
     main()