Update app.py

app.py

@@ -731,8 +731,6 @@ def handle_onedrive_integration(access_token):
         else:
             st.error("Failed to delete file.")
 
-
-
 def main():
     st.title("🦄 MS Graph API with AI & Cloud Integration for M365")
 
@@ -751,7 +749,17 @@ def main():
             st.sidebar.write(f"AI Capabilities: {details['ai_capabilities']}")
             st.sidebar.write(f"Graph Solution: {details['graph_solution']}")
 
-    # Handle menu navigation
+    # Handle query parameters via session_state
+    if 'section' not in st.session_state:
+        st.session_state['section'] = 'dashboard'  # Default section is the dashboard
+
+    selected_section = st.session_state['section']
+    
+    # Example of re-running based on interaction (you can change sections)
+    if st.button("Go to Landing Page"):
+        st.session_state['section'] = 'landing-page'
+
+    # Request scopes and authentication management
     request_scopes = BASE_SCOPES.copy()
     for product in selected_products:
         request_scopes.extend(PRODUCT_SCOPES[product]['scopes'])
@@ -767,7 +775,7 @@ def main():
         )
         st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(auth_url))
         
-        query_params = st.query_params
+        query_params = st.session_state.get('query_params', {})
         if 'code' in query_params:
             code = query_params.get('code')
             st.write('🔑 Authorization Code Obtained:', code[:10] + '...')
@@ -776,7 +784,8 @@ def main():
                 access_token = get_access_token(code)
                 st.session_state['access_token'] = access_token
                 st.success("Access token acquired successfully!")
-                st.rerun()
+                # No need for st.experimental_rerun; changing session state triggers rerun
+                st.session_state['section'] = 'dashboard'
             except Exception as e:
                 st.error(f"Error acquiring access token: {str(e)}")
                 st.stop()
@@ -787,9 +796,7 @@ def main():
         if user_info:
             st.sidebar.write(f"👋 Hello, {user_info.get('displayName', 'User')}!")
         
-        # Handle navigation based on the selected section from the sticky menu
-        selected_section = st.experimental_get_query_params().get('section', ['dashboard'])[0]
-        
+        # Handle navigation based on the selected section
         if selected_section == 'dashboard':
             dashboard(access_token)
         elif selected_section == 'landing-page':
@@ -807,6 +814,7 @@ def main():
         elif selected_section == 'filter-by':
             filter_by(access_token)
 
+
 if __name__ == "__main__":
     main()