Update app.py

app.py

@@ -1,12 +1,87 @@
 import os
 import json
-
+import streamlit as st
+import firebase_admin
+from firebase_admin import credentials, auth
 # Retrieve the JSON string from the secret
 json_str = os.getenv("FIREBASE_CREDENTIALS")
 
-# Convert the string back to a JSON object
-if json_str:
-    firebase_creds = json.loads(json_str)
-    print("Firebase credentials loaded successfully!")
+# Initialize Firebase (only once)
+if not firebase_admin._apps:
+    cred = credentials.Certificate("path/to/your/serviceAccountKey.json")  # Replace with your Firebase service account key
+    firebase_admin.initialize_app(cred)
+
+# Initialize session state
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+if "current_user" not in st.session_state:
+    st.session_state.current_user = None
+
+# Callback for registration
+def register_callback():
+    email = st.session_state.reg_email
+    password = st.session_state.reg_password
+    try:
+        # Create a new user in Firebase
+        user = auth.create_user(email=email, password=password)
+        st.success("Registration successful! Please log in.")
+    except Exception as e:
+        st.error(f"Registration failed: {e}")
+
+# Callback for login
+def login_callback():
+    email = st.session_state.login_email
+    password = st.session_state.login_password
+    try:
+        # Authenticate user with Firebase
+        user = auth.get_user_by_email(email)
+        st.session_state.logged_in = True
+        st.session_state.current_user = user.uid  # Store the user UID in session state
+        st.success("Logged in successfully!")
+    except Exception as e:
+        st.error(f"Login failed: {e}")
+
+# Callback for logout
+def logout_callback():
+    st.session_state.logged_in = False
+    st.session_state.current_user = None
+    st.info("Logged out successfully!")
+
+# Registration form
+def registration_form():
+    with st.form("Registration"):
+        st.subheader("Register")
+        email = st.text_input("Email", key="reg_email")
+        password = st.text_input("Password (min 6 characters)", type="password", key="reg_password")
+        submit_button = st.form_submit_button("Register", on_click=register_callback)
+
+# Login form
+def login_form():
+    with st.form("Login"):
+        st.subheader("Login")
+        email = st.text_input("Email", key="login_email")
+        password = st.text_input("Password", type="password", key="login_password")
+        submit_button = st.form_submit_button("Login", on_click=login_callback)
+
+# Main app screen (after login)
+def main_app():
+    st.subheader(f"Welcome, {st.session_state.current_user}!")
+    st.write("Enter a name below to get a greeting.")
+
+    # Input field for name
+    name = st.text_input("Enter a name", key="name_input")
+
+    # Display greeting in real-time
+    if name:
+        st.write(f"Hello {name}!")
+
+    # Logout button
+    if st.button("Logout", on_click=logout_callback):
+        pass
+
+# Render the appropriate screen based on login status
+if st.session_state.logged_in:
+    main_app()
 else:
-    print("Failed to load Firebase credentials.")
+    registration_form()
+    login_form()