Improve OAuth login button visibility and error handling

src/Start_Page.py

@@ -27,8 +27,11 @@ handle_oauth_callback()
 login_required, login_error_message = require_hf_login()
 if login_required:
     st.error(login_error_message)
-    st.info("💡 **Tip:** Look for the login button in the top right corner of this page.")
+    st.info("💡 **Tip:** Click the button below to sign in with your Hugging Face account.")
+    st.divider()
     login_button()  # Show the login button
+    st.divider()
+    st.info("**Note:** You need to complete the OAuth flow within this app, even if you're already logged into Hugging Face in your browser.")
     st.stop()  # Stop execution - user must log in
 
 # User is logged in - show username (optional)

src/gui_elements/session_cache.py

@@ -34,30 +34,40 @@ def login_button():
     """Display a login button that redirects to HF OAuth."""
     if not os.environ.get("OAUTH_CLIENT_ID"):
         st.error("OAuth is not configured. Please enable hf_oauth: true in README.md")
+        st.info("The Space needs to be rebuilt after adding hf_oauth: true to README.md")
         return
     
-    cfg = oidc_config()
-    if not cfg:
-        return
-        
-    auth_endpoint = cfg.get("authorization_endpoint")
-    if not auth_endpoint:
-        st.error("Could not get authorization endpoint")
-        return
+    try:
+        cfg = oidc_config()
+        if not cfg:
+            st.error("Could not fetch OpenID configuration. Please try again later.")
+            return
+            
+        auth_endpoint = cfg.get("authorization_endpoint")
+        if not auth_endpoint:
+            st.error("Could not get authorization endpoint from OpenID configuration.")
+            return
 
-    state = secrets.token_urlsafe(24)
-    st.session_state["oauth_state"] = state
+        state = secrets.token_urlsafe(24)
+        st.session_state["oauth_state"] = state
 
-    redirect_uri = get_space_url() + "/"
-    params = {
-        "client_id": os.environ["OAUTH_CLIENT_ID"],
-        "redirect_uri": redirect_uri,
-        "response_type": "code",
-        "scope": os.getenv("OAUTH_SCOPES", "openid profile"),
-        "state": state,
-    }
-    url = auth_endpoint + "?" + urllib.parse.urlencode(params)
-    st.link_button("🔐 Sign in with Hugging Face", url, use_container_width=True)
+        redirect_uri = get_space_url() + "/"
+        if not redirect_uri:
+            st.warning("Could not determine Space URL. OAuth may not work correctly.")
+            redirect_uri = "https://" + os.getenv("SPACE_HOST", "") + "/"
+        
+        params = {
+            "client_id": os.environ["OAUTH_CLIENT_ID"],
+            "redirect_uri": redirect_uri,
+            "response_type": "code",
+            "scope": os.getenv("OAUTH_SCOPES", "openid profile"),
+            "state": state,
+        }
+        url = auth_endpoint + "?" + urllib.parse.urlencode(params)
+        st.link_button("🔐 Sign in with Hugging Face", url, use_container_width=True)
+    except Exception as e:
+        st.error(f"Error setting up OAuth login: {e}")
+        st.info("Please check the Space logs for more details.")
 
 def handle_oauth_callback():
     """Handle OAuth callback and exchange code for token."""