fix

my-blog/static/js/ask-ai.js

@@ -16,11 +16,11 @@ function toggleAskAI() {
     }
 }
 
+// Global auth status
+let isOAuthAuthenticated = false;
+
 function checkAuthenticationStatus() {
-    const userSections = document.querySelectorAll('.user-section');
-    const isAuthenticated = userSections.length > 0;
-    
-    if (isAuthenticated) {
+    if (isOAuthAuthenticated) {
         // User is authenticated - show Ask AI UI
         document.getElementById('authCheck').style.display = 'none';
         document.getElementById('chatForm').style.display = 'block';
@@ -76,27 +76,26 @@ function askQuestion() {
     }
 }
 
+// Global user data
+let currentUser = null;
+
 function addUserMessage(question) {
     const chatHistory = document.getElementById('chatHistory');
-    const userSection = document.querySelector('.user-section');
     
     let userAvatar = '👤';
     let userDisplay = 'You';
     let userHandle = 'user';
     
-    if (userSection) {
-        const avatarImg = userSection.querySelector('.user-avatar');
-        const displayName = userSection.querySelector('.user-display-name');
-        const handle = userSection.querySelector('.user-handle');
-        
-        if (avatarImg && avatarImg.src) {
-            userAvatar = `<img src="${avatarImg.src}" alt="${displayName?.textContent || 'User'}" class="profile-avatar">`;
+    // Use currentUser data if available
+    if (currentUser) {
+        if (currentUser.avatar) {
+            userAvatar = `<img src="${currentUser.avatar}" alt="${currentUser.displayName || 'User'}" class="profile-avatar">`;
         }
-        if (displayName?.textContent) {
-            userDisplay = displayName.textContent;
+        if (currentUser.displayName) {
+            userDisplay = currentUser.displayName;
         }
-        if (handle?.textContent) {
-            userHandle = handle.textContent.replace('@', '');
+        if (currentUser.handle) {
+            userHandle = currentUser.handle;
         }
     }
     
@@ -253,6 +252,14 @@ function setupAskAIEventListeners() {
         handleAIResponse(event.detail);
     });
     
+    // Listen for OAuth authentication status updates
+    window.addEventListener('oauthAuthStatus', function(event) {
+        console.log('OAuth auth status updated:', event.detail);
+        isOAuthAuthenticated = event.detail.isAuthenticated;
+        currentUser = event.detail.user;
+        checkAuthenticationStatus();
+    });
+    
     // Listen for OAuth callback completion from iframe
     window.addEventListener('message', function(event) {
         if (event.data.type === 'oauth_success') {