Upload folder using huggingface_hub

frontend/src/App.jsx

@@ -16,6 +16,12 @@ export default function App() {
   const [userName, setUserName] = useState('');
   const [authError, setAuthError] = useState('');
   const [initialPolicyId, setInitialPolicyId] = useState(null);
+  const [isAuthResolving, setIsAuthResolving] = useState(() => {
+    const hash = window.location.hash || '';
+    const hasOAuthFragment = hash.includes('access_token=') || hash.includes('refresh_token=');
+    const hasOAuthIntent = localStorage.getItem('oauth_redirect_intent') === 'google';
+    return hasOAuthFragment || hasOAuthIntent;
+  });
   const lastOAuthSyncKey = useRef('');
 
   const getRouteFromHash = (rawHash) => {
@@ -122,8 +128,13 @@ export default function App() {
 
       setUserName(nextName || '');
 
-      const currentState = window.location.hash.replace('#', '') || 'home';
-      if (currentState === 'login' || currentState === 'home') {
+      const currentHash = window.location.hash || '';
+      const currentState = getRouteFromHash(currentHash);
+      const fromOAuthCallback = currentHash.includes('access_token=') || currentHash.includes('refresh_token=');
+      const oauthIntent = localStorage.getItem('oauth_redirect_intent') === 'google';
+
+      if (currentState === 'login' || fromOAuthCallback || oauthIntent) {
+        localStorage.removeItem('oauth_redirect_intent');
         redirectToDashboard();
       }
     };
@@ -131,21 +142,28 @@ export default function App() {
     const {
       data: { subscription },
     } = supabase.auth.onAuthStateChange((event, session) => {
-      void syncSession(session, event);
+      void syncSession(session, event).finally(() => {
+        setIsAuthResolving(false);
+      });
     });
 
-    void supabase.auth.getSession().then(({ data }) => syncSession(data.session));
+    void supabase.auth.getSession()
+      .then(({ data }) => syncSession(data.session))
+      .finally(() => {
+        setIsAuthResolving(false);
+      });
 
     return () => subscription.unsubscribe();
   }, []);
 
   useEffect(() => {
     const hash = window.location.hash || '';
-    const hasOAuthFragment = hash.includes('access_token=');
-    const initialState = getRouteFromHash(hash);
+    const hasOAuthFragment = hash.includes('access_token=') || hash.includes('refresh_token=');
+    const hasOAuthIntent = localStorage.getItem('oauth_redirect_intent') === 'google';
+    const initialState = (hasOAuthFragment || hasOAuthIntent) ? 'login' : getRouteFromHash(hash);
 
     // Keep OAuth fragment untouched initially so Supabase can parse and persist session.
-    if (!hasOAuthFragment) {
+    if (!hasOAuthFragment && !hasOAuthIntent) {
       window.history.replaceState({ appState: initialState }, '', `#${initialState}`);
     }
 
@@ -170,6 +188,26 @@ export default function App() {
     navigateTo('dboard');
   };
 
+  if (isAuthResolving) {
+    return (
+      <div
+        style={{
+          minHeight: '100vh',
+          display: 'flex',
+          alignItems: 'center',
+          justifyContent: 'center',
+          background: isDark ? '#0c0908' : '#f9fafb',
+          color: isDark ? '#ecfeff' : '#111827',
+          fontFamily: "'DM Sans', sans-serif",
+          fontSize: 14,
+          letterSpacing: 0.2,
+        }}
+      >
+        Completing sign-in...
+      </div>
+    );
+  }
+
   // 1. DASHBOARD ROUTE
   if (appState === 'dboard' || appState === 'dashboard') {
     return (
@@ -178,8 +216,17 @@ export default function App() {
         toggleTheme={toggleTheme}
         userName={userName}
         initialPolicyId={initialPolicyId}
-        onLogout={() => {
+        onLogout={async () => {
+          localStorage.removeItem('token');
+          localStorage.removeItem('oauth_redirect_intent');
+          lastOAuthSyncKey.current = '';
+          setAuthError('');
           setUserName('');
+          try {
+            await supabase.auth.signOut();
+          } catch (err) {
+            console.error('Sign out error:', err);
+          }
           navigateTo('home');
         }}
         onTriggerUpload={() => navigateTo('upload')}

frontend/src/components/Auth/Login.jsx

@@ -765,6 +765,7 @@ export default function LoginPage({ onLoginSuccess, onGoogleSuccess, onSignupSuc
     setGoogleLoading(true);
 
     try {
+      localStorage.setItem('oauth_redirect_intent', 'google');
       const redirectTo = `${window.location.origin}${window.location.pathname}`;
       const { data, error: oauthError } = await supabase.auth.signInWithOAuth({
         provider: 'google',
@@ -781,8 +782,8 @@ export default function LoginPage({ onLoginSuccess, onGoogleSuccess, onSignupSuc
       if (!data?.url) throw new Error('Unable to start Google sign-in flow.');
 
       window.location.assign(data.url);
-      onGoogleSuccess?.();
     } catch (err) {
+      localStorage.removeItem('oauth_redirect_intent');
       triggerError(err.message || 'Google login failed. Please try again.');
       setGoogleLoading(false);
     }

frontend/src/components/Dashboard/Dboard.jsx

@@ -3,7 +3,7 @@ import { useState, useRef, useEffect, useCallback } from 'react';
 import {
   Upload, Clock, X, FileText, AlertTriangle, CheckCircle2,
   Shield, DollarSign, Calendar, Info, Trash2, Eye, FilePlus,
-  Home, Aperture, Sun, Moon, Menu, BadgeCheck, Banknote,
+  Home, Aperture, Sun, Moon, Menu, BadgeCheck, Banknote, LogOut,
   RefreshCw, Send, Loader2,
 } from 'lucide-react';
 import { fetchApi, streamApi } from '../../api';
@@ -144,7 +144,7 @@ function UploadZone({ T, dark, isDragging, uploading, uploadPct, onDragOver, onD
 }
 
 // ─────────────────────────────────────────────────────────────────────────────
-export default function Dboard({ file, isDark: _initDark, userName = 'My Account', initialPolicyId = null, onTriggerUpload }) {
+export default function Dboard({ file, isDark: _initDark, userName = 'My Account', initialPolicyId = null, onTriggerUpload, onLogout }) {
   // Always boot light — user can toggle via sidebar
   const [dark,            setDark]           = useState(false);
   const [sidebarOpen,     setSidebarOpen]    = useState(true);
@@ -160,6 +160,7 @@ export default function Dboard({ file, isDark: _initDark, userName = 'My Account
   const [chatOpen,        setChatOpen]       = useState(false);
   const [chatInput,       setChatInput]      = useState('');
   const [isTyping,        setIsTyping]       = useState(false);
+  const [isSigningOut,    setIsSigningOut]   = useState(false);
   const [chatDimensions,  setChatDimensions] = useState({ w: 360, h: 500 });
   const [chatMessagesMap, setChatMessagesMap] = useState({
     default: [makeWelcomeMsg(file?.name ?? null)],
@@ -457,6 +458,16 @@ export default function Dboard({ file, isDark: _initDark, userName = 'My Account
     }
   };
 
+  const handleSignOut = async () => {
+    if (!onLogout || isSigningOut) return;
+    setIsSigningOut(true);
+    try {
+      await onLogout();
+    } finally {
+      setIsSigningOut(false);
+    }
+  };
+
   // ════════════════════════ SUB-COMPONENTS ══════════════════════════════════
 
   const IrisAvatar = ({ size=40 }) => (
@@ -722,6 +733,24 @@ export default function Dboard({ file, isDark: _initDark, userName = 'My Account
         </nav>
 
         <div style={{ padding:'12px 8px', borderTop:`1px solid ${T.headerBorder}` }}>
+          <button onClick={handleSignOut}
+            disabled={!onLogout || isSigningOut}
+            style={{ width:'100%', display:'flex', alignItems:'center', gap:12, padding: sidebarOpen ? '10px 14px' : '10px', justifyContent: sidebarOpen ? 'flex-start' : 'center', borderRadius:12, cursor: !onLogout || isSigningOut ? 'not-allowed' : 'pointer', background:T.redBg, border:`1px solid ${T.redBrd}`, color:T.redClr, ...f, fontSize:13, fontWeight:600, transition:'all .2s', opacity: !onLogout || isSigningOut ? 0.6 : 1, marginBottom:8 }}
+            onMouseEnter={e => {
+              if (!isSigningOut && onLogout) {
+                e.currentTarget.style.filter = 'brightness(1.03)';
+                e.currentTarget.style.transform = 'scale(1.02)';
+              }
+            }}
+            onMouseLeave={e => {
+              e.currentTarget.style.filter = 'none';
+              e.currentTarget.style.transform = 'none';
+            }}
+          >
+            <LogOut size={15} style={{ flexShrink:0 }} />
+            {sidebarOpen && <span>{isSigningOut ? 'Signing Out...' : 'Sign Out'}</span>}
+          </button>
+
           <button onClick={() => setDark(v => !v)}
             style={{ width:'100%', display:'flex', alignItems:'center', gap:12, padding: sidebarOpen ? '10px 14px' : '10px', justifyContent: sidebarOpen ? 'flex-start' : 'center', borderRadius:12, cursor:'pointer', background:T.cardBg, border:`1px solid ${T.cardBorder}`, color:T.t3, ...f, fontSize:13, fontWeight:500, boxShadow:T.cardShadow, transition:'all .2s' }}
             onMouseEnter={e => { e.currentTarget.style.borderColor=T.acc; e.currentTarget.style.color=T.acc; e.currentTarget.style.transform='scale(1.02)'; }}