Update app.py

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
 import firebase_admin
-from firebase_admin import credentials, auth
+from firebase_admin import credentials, auth, db
 import os
 import json
+
 # Load Firebase credentials from Hugging Face Secrets
 firebase_creds = os.getenv("FIREBASE_CREDENTIALS")
 
@@ -10,10 +11,13 @@ if firebase_creds:
     firebase_creds = json.loads(firebase_creds)
 else:
     st.error("Firebase credentials not found. Please check your secrets.")
+
 # Initialize Firebase (only once)
 if not firebase_admin._apps:
     cred = credentials.Certificate(firebase_creds)  # Replace with your Firebase service account key
-    firebase_admin.initialize_app(cred)
+    firebase_admin.initialize_app(cred, {
+        'databaseURL': 'https://creative-623ef-default-rtdb.firebaseio.com/'  # Replace with your Firebase Realtime Database URL
+    })
 
 # Initialize session state
 if "logged_in" not in st.session_state:
@@ -51,6 +55,19 @@ def logout_callback():
     st.session_state.current_user = None
     st.info("Logged out successfully!")
 
+# Function to store text in Firebase Realtime Database
+def store_text_in_db(user_id, text):
+    try:
+        ref = db.reference(f'users/{user_id}/texts')
+        new_text_ref = ref.push()
+        new_text_ref.set({
+            'text': text,
+            'timestamp': firebase_admin.db.ServerValue.TIMESTAMP
+        })
+        st.success("Text saved successfully!")
+    except Exception as e:
+        st.error(f"Failed to save text: {e}")
+
 # Registration form
 def registration_form():
     with st.form("Registration"):
@@ -78,6 +95,8 @@ def main_app():
     # Display greeting in real-time
     if name:
         st.write(f"Hello {name}!")
+        # Store the entered text in Firebase Realtime Database
+        store_text_in_db(st.session_state.current_user, name)
 
     # Logout button
     if st.button("Logout", on_click=logout_callback):