Update app.py

app.py

@@ -41,26 +41,34 @@ def authenticate_user(username, password):
     conn.close()
     return user
 
-# Main application
-def main():
-    st.title("SimplifAI")
-
-    # Initialize database
-    create_user_table()
-
-    # Manage session state
+def initialize_session_state():
     if "authenticated" not in st.session_state:
         st.session_state.authenticated = False
     if "username" not in st.session_state:
         st.session_state.username = None
     if "page" not in st.session_state:
         st.session_state.page = "login"
+    if "current_project" not in st.session_state:
+        st.session_state.current_project = None
+    if "project_uploaded" not in st.session_state:
+        st.session_state.project_uploaded = False
+
+def main():
+    st.title("SimplifAI")
+
+    # Initialize session state
+    initialize_session_state()
+
+    # Initialize database
+    create_user_table()
 
     # Page routing logic
     if st.session_state.page == "login":
         login_page()
     elif st.session_state.page == "workspace":
         workspace_page()
+    elif st.session_state.page == "project_view":
+        project_view_page()
 
 def login_page():
     st.subheader("Please Log In or Register to Continue")