Update app.py

app.py

@@ -403,4 +403,11 @@ def main_app():
 
     # Logout button
     if st.button("Logout", on_click=logout_callback):
-        pass
+        pass
+
+# Render the appropriate screen based on login status
+if st.session_state.logged_in:
+    main_app()
+else:
+    registration_form()
+    login_form()