Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import requests
-import os
 
 # Firebase Auth REST API to login
 def firebase_login(email, password):
@@ -21,42 +20,37 @@ def firebase_login(email, password):
 
 def main():
     st.set_page_config(page_title="Login", layout="centered")
-    
-    # Header
-    st.title("Login")
-    st.markdown("---")
 
     # Check login state
     if "logged_in" not in st.session_state:
         st.session_state.logged_in = False
 
-    # If already logged in, redirect to the main app (app2.py)
+    # Check if already logged in
     if st.session_state.logged_in:
-        st.success("You are already logged in.")
+        st.write("You are already logged in.")
         st.write("Redirecting to your dashboard...")
-        st.experimental_set_query_params(page="app2")  # Set URL query to indicate page switch
+
+        # Update URL with the page parameter to redirect to app2.py
+        st.experimental_set_query_params(page="app2")
+        return
 
     # Login form if not logged in
-    else:
-        email = st.text_input("Email")
-        password = st.text_input("Password", type="password")
-
-        if st.button("Login"):
-            user_id, id_token = firebase_login(email, password)
-            if user_id:
-                st.session_state.logged_in = True
-                st.session_state.email = email
-                st.session_state.user_id = user_id
-
-                # Store session state and redirect to app2.py
-                st.success("Successfully logged in!")
-                st.write("Redirecting to your dashboard...")
-                st.experimental_set_query_params(page="app2")  # Set URL query to indicate page switch
-
-                # Alternatively, you can manually run app2.py if necessary
-                os.system('streamlit run app2.py')
-            else:
-                st.error("Login failed. Please check your credentials.")
+    email = st.text_input("Email")
+    password = st.text_input("Password", type="password")
+
+    if st.button("Login"):
+        user_id, id_token = firebase_login(email, password)
+        if user_id:
+            st.session_state.logged_in = True
+            st.session_state.email = email
+            st.session_state.user_id = user_id
+
+            # Update the URL with the page query parameter
+            st.experimental_set_query_params(page="app2")
+
+            st.success("Login successful! Redirecting to your dashboard...")
+        else:
+            st.error("Login failed. Please check your credentials.")
 
 if __name__ == "__main__":
     main()