Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,22 +8,35 @@ from views import (
|
|
| 8 |
show_security_settings_page
|
| 9 |
)
|
| 10 |
from models import initialize_database
|
| 11 |
-
from
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
def main():
|
| 15 |
-
|
|
|
|
|
|
|
| 16 |
if 'app_state' not in st.session_state:
|
| 17 |
st.session_state.app_state = init_app()
|
| 18 |
initialize_database()
|
| 19 |
-
setup_localization()
|
| 20 |
|
| 21 |
-
#
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
show_login_page()
|
| 24 |
else:
|
| 25 |
-
# Route to appropriate page based on session state
|
| 26 |
-
page = st.session_state.get('page', 'dashboard')
|
| 27 |
pages = {
|
| 28 |
'dashboard': show_dashboard,
|
| 29 |
'add_account': show_add_account_page,
|
|
@@ -31,8 +44,9 @@ def main():
|
|
| 31 |
'security': show_security_settings_page
|
| 32 |
}
|
| 33 |
|
| 34 |
-
|
| 35 |
-
|
|
|
|
| 36 |
else:
|
| 37 |
st.error("Page not found")
|
| 38 |
|
|
|
|
| 8 |
show_security_settings_page
|
| 9 |
)
|
| 10 |
from models import initialize_database
|
| 11 |
+
from creator import Creator
|
| 12 |
+
import sqlite3
|
| 13 |
+
|
| 14 |
+
def initialize_session_state():
|
| 15 |
+
if 'initialized' not in st.session_state:
|
| 16 |
+
st.session_state.initialized = True
|
| 17 |
+
st.session_state.db = sqlite3.connect('digital_heir.db', check_same_thread=False)
|
| 18 |
+
st.session_state.creator = Creator.get_info()
|
| 19 |
+
if 'page' not in st.session_state:
|
| 20 |
+
st.session_state.page = 'dashboard'
|
| 21 |
+
if 'user' not in st.session_state:
|
| 22 |
+
st.session_state.user = None
|
| 23 |
|
| 24 |
def main():
|
| 25 |
+
initialize_session_state()
|
| 26 |
+
|
| 27 |
+
# Initialize core services only if not already present
|
| 28 |
if 'app_state' not in st.session_state:
|
| 29 |
st.session_state.app_state = init_app()
|
| 30 |
initialize_database()
|
|
|
|
| 31 |
|
| 32 |
+
# Display creator info in sidebar
|
| 33 |
+
with st.sidebar:
|
| 34 |
+
st.write(f"Created by: {st.session_state.creator['name']}")
|
| 35 |
+
st.write(f"Version: {st.session_state.creator['version']}")
|
| 36 |
+
|
| 37 |
+
if not st.session_state.user:
|
| 38 |
show_login_page()
|
| 39 |
else:
|
|
|
|
|
|
|
| 40 |
pages = {
|
| 41 |
'dashboard': show_dashboard,
|
| 42 |
'add_account': show_add_account_page,
|
|
|
|
| 44 |
'security': show_security_settings_page
|
| 45 |
}
|
| 46 |
|
| 47 |
+
current_page = st.session_state.get('page', 'dashboard')
|
| 48 |
+
if current_page in pages:
|
| 49 |
+
pages[current_page]()
|
| 50 |
else:
|
| 51 |
st.error("Page not found")
|
| 52 |
|