Update app.py

app.py

@@ -122,10 +122,54 @@ def get_access_token2(code, code_verifier=None):
     except Exception as e:
         st.error(f"Exception in get_access_token: {str(e)}")
         raise
+        
+def clear_query_params():
+    # Get the current URL without query parameters
+    base_url = st.get_page_config().base_url_path
+    st.write(f'<meta http-equiv="refresh" content="0; url={base_url}">', unsafe_allow_html=True)
+    st.stop()
+    
+def process_query_params():
+    query_params = st.query_params
+    st.write("Debug: All query parameters:", query_params)
+
+    if 'error' in query_params:
+        error = query_params.get('error')
+        error_description = query_params.get('error_description', 'No description provided')
+        st.error(f"Authentication Error: {error}")
+        st.error(f"Error Description: {urllib.parse.unquote(error_description)}")
+        st.stop()
+
+    if 'code' in query_params:
+        code = query_params.get('code')
+        st.write('🔑 Authorization Code Obtained:', code[:10] + '...')
+        
+        try:
+            access_token = get_access_token(code)
+            st.session_state['access_token'] = access_token
+            st.success("Access token acquired successfully!")
+            # Redirect to clear query params
+            st.rerun()
+        except Exception as e:
+            st.error(f"Error acquiring access token: {str(e)}")
+            st.stop()
 
+    # Handle other query parameters as needed
+    if 'q' in query_params or 'query' in query_params:
+        query = query_params.get('q') or query_params.get('query')
+        if query:
+            # Process the query
+            pass
 
+    if 'action' in query_params:
+        action = query_params.get('action')
+        if action == 'show_message':
+            st.success("Showing a message because 'action=show_message' was found in the URL.")
+        elif action == 'clear':
+            # Instead of clearing query params, we'll redirect
+            st.rerun()
         
-def process_query_params():
+def process_query_params2():
     # ⚙️q= Run ArXiv search from query parameters
     try:
         query_params = st.query_params
@@ -198,20 +242,37 @@ def main():
     process_query_params()
 
     if 'access_token' not in st.session_state:
-        if 'code_verifier' not in st.session_state:
-            # Ensure the code_verifier is available
-            st.error("Code verifier missing. Please restart the login process.")
+        query_params = st.query_params
+        if 'code' in query_params:
+            code = query_params.get('code')
+            st.write('🔑 Authorization Code Obtained:', code[:10] + '...')
+            
+            try:
+                access_token = get_access_token(code)
+                st.session_state['access_token'] = access_token
+                st.success("Access token acquired successfully!")
+                # Instead of clearing query params, we'll redirect to remove them
+                st.rerun()
+            except Exception as e:
+                if "AADSTS70000" in str(e) and "code has expired" in str(e):
+                    st.error("The authorization code has expired. Please log in again.")
+                    st.session_state.pop('access_token', None)
+                    # Redirect to clear query params
+                    st.rerun()
+                else:
+                    st.error(f"Error acquiring access token: {str(e)}")
+                    st.stop()
+        else:
+            client_instance = get_msal_app()
+            authorization_url = client_instance.get_authorization_request_url(
+                scopes=SCOPES,
+                redirect_uri=REDIRECT_URI
+            )
+            st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(authorization_url))
             st.stop()
-
-        client_instance = get_msal_app()
-        auth_url = client_instance.get_authorization_request_url(
-            scopes=SCOPES,
-            redirect_uri=REDIRECT_URI,
-            code_challenge=code_challenge,
-            code_challenge_method="S256"
-        )
-        st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(auth_url))
-        st.stop()
+    else:
+        # Rest of your code for authenticated users
+        pass
 
 
     # 🚀 Sidebar navigation