Merge branch 'release-v2.1.4' into feature/task-activities-by-user

worklenz-frontend/src/App.tsx

@@ -5,6 +5,7 @@ import i18next from 'i18next';
 
 // Components
 import ThemeWrapper from './features/theme/ThemeWrapper';
+import ModuleErrorBoundary from './components/ModuleErrorBoundary';
 
 // Routes
 import router from './app/routes';
@@ -13,6 +14,7 @@ import router from './app/routes';
 import { useAppSelector } from './hooks/useAppSelector';
 import { initMixpanel } from './utils/mixpanelInit';
 import { initializeCsrfToken } from './api/api-client';
+import CacheCleanup from './utils/cache-cleanup';
 
 // Types & Constants
 import { Language } from './features/i18n/localesSlice';
@@ -113,6 +115,56 @@ const App: React.FC = memo(() => {
     };
   }, []);
 
+  // Global error handlers for module loading issues
+  useEffect(() => {
+    const handleUnhandledRejection = (event: PromiseRejectionEvent) => {
+      const error = event.reason;
+      
+      // Check if this is a module loading error
+      if (
+        error?.message?.includes('Failed to fetch dynamically imported module') ||
+        error?.message?.includes('Loading chunk') ||
+        error?.name === 'ChunkLoadError'
+      ) {
+        console.error('Unhandled module loading error:', error);
+        event.preventDefault(); // Prevent default browser error handling
+        
+        // Clear caches and reload
+        CacheCleanup.clearAllCaches()
+          .then(() => CacheCleanup.forceReload('/auth/login'))
+          .catch(() => window.location.reload());
+      }
+    };
+
+    const handleError = (event: ErrorEvent) => {
+      const error = event.error;
+      
+      // Check if this is a module loading error
+      if (
+        error?.message?.includes('Failed to fetch dynamically imported module') ||
+        error?.message?.includes('Loading chunk') ||
+        error?.name === 'ChunkLoadError'
+      ) {
+        console.error('Global module loading error:', error);
+        event.preventDefault(); // Prevent default browser error handling
+        
+        // Clear caches and reload
+        CacheCleanup.clearAllCaches()
+          .then(() => CacheCleanup.forceReload('/auth/login'))
+          .catch(() => window.location.reload());
+      }
+    };
+
+    // Add global error handlers
+    window.addEventListener('unhandledrejection', handleUnhandledRejection);
+    window.addEventListener('error', handleError);
+
+    return () => {
+      window.removeEventListener('unhandledrejection', handleUnhandledRejection);
+      window.removeEventListener('error', handleError);
+    };
+  }, []);
+
   // Register service worker
   useEffect(() => {
     registerSW({
@@ -150,12 +202,14 @@ const App: React.FC = memo(() => {
   return (
     <Suspense fallback={<SuspenseFallback />}>
       <ThemeWrapper>
-        <RouterProvider
-          router={router}
-          future={{
-            v7_startTransition: true,
-          }}
-        />
+        <ModuleErrorBoundary>
+          <RouterProvider
+            router={router}
+            future={{
+              v7_startTransition: true,
+            }}
+          />
+        </ModuleErrorBoundary>
       </ThemeWrapper>
     </Suspense>
   );

worklenz-frontend/src/api/reporting/reporting.timesheet.api.service.updated.ts

@@ -0,0 +1,92 @@
+import { API_BASE_URL } from '@/shared/constants';
+import { toQueryString } from '@/utils/toQueryString';
+import apiClient from '../api-client';
+import { IServerResponse } from '@/types/common.types';
+import { IAllocationViewModel } from '@/types/reporting/reporting-allocation.types';
+import {
+  IProjectLogsBreakdown,
+  IRPTTimeMember,
+  IRPTTimeProject,
+  ITimeLogBreakdownReq,
+} from '@/types/reporting/reporting.types';
+
+const rootUrl = `${API_BASE_URL}/reporting`;
+
+// Helper function to get user's timezone
+const getUserTimezone = () => {
+  return Intl.DateTimeFormat().resolvedOptions().timeZone;
+};
+
+export const reportingTimesheetApiService = {
+  getTimeSheetData: async (
+    body = {},
+    archived = false
+  ): Promise<IServerResponse<IAllocationViewModel>> => {
+    const q = toQueryString({ archived });
+    const bodyWithTimezone = {
+      ...body,
+      timezone: getUserTimezone()
+    };
+    const response = await apiClient.post(`${rootUrl}/allocation/${q}`, bodyWithTimezone);
+    return response.data;
+  },
+
+  getAllocationProjects: async (body = {}) => {
+    const bodyWithTimezone = {
+      ...body,
+      timezone: getUserTimezone()
+    };
+    const response = await apiClient.post(`${rootUrl}/allocation/allocation-projects`, bodyWithTimezone);
+    return response.data;
+  },
+
+  getProjectTimeSheets: async (
+    body = {},
+    archived = false
+  ): Promise<IServerResponse<IRPTTimeProject[]>> => {
+    const q = toQueryString({ archived });
+    const bodyWithTimezone = {
+      ...body,
+      timezone: getUserTimezone()
+    };
+    const response = await apiClient.post(`${rootUrl}/time-reports/projects/${q}`, bodyWithTimezone);
+    return response.data;
+  },
+
+  getMemberTimeSheets: async (
+    body = {},
+    archived = false
+  ): Promise<IServerResponse<IRPTTimeMember[]>> => {
+    const q = toQueryString({ archived });
+    const bodyWithTimezone = {
+      ...body,
+      timezone: getUserTimezone()
+    };
+    const response = await apiClient.post(`${rootUrl}/time-reports/members/${q}`, bodyWithTimezone);
+    return response.data;
+  },
+
+  getProjectTimeLogs: async (
+    body: ITimeLogBreakdownReq
+  ): Promise<IServerResponse<IProjectLogsBreakdown[]>> => {
+    const bodyWithTimezone = {
+      ...body,
+      timezone: getUserTimezone()
+    };
+    const response = await apiClient.post(`${rootUrl}/project-timelogs`, bodyWithTimezone);
+    return response.data;
+  },
+
+  getProjectEstimatedVsActual: async (
+    body = {},
+    archived = false
+  ): Promise<IServerResponse<IRPTTimeProject[]>> => {
+    const q = toQueryString({ archived });
+    const bodyWithTimezone = {
+      ...body,
+      timezone: getUserTimezone()
+    };
+    const response = await apiClient.post(`${rootUrl}/time-reports/estimated-vs-actual${q}`, bodyWithTimezone);
+    return response.data;
+  },
+};

worklenz-frontend/src/api/survey/survey.api.service.ts

@@ -0,0 +1,27 @@
+import { IServerResponse } from '@/types/common.types';
+import { ISurvey, ISurveySubmissionRequest, ISurveyResponse } from '@/types/account-setup/survey.types';
+import apiClient from '../api-client';
+
+const API_BASE_URL = '/api/v1';
+
+export const surveyApiService = {
+  async getAccountSetupSurvey(): Promise<IServerResponse<ISurvey>> {
+    const response = await apiClient.get<IServerResponse<ISurvey>>(`${API_BASE_URL}/surveys/account-setup`);
+    return response.data;
+  },
+
+  async submitSurveyResponse(data: ISurveySubmissionRequest): Promise<IServerResponse<{ response_id: string }>> {
+    const response = await apiClient.post<IServerResponse<{ response_id: string }>>(`${API_BASE_URL}/surveys/responses`, data);
+    return response.data;
+  },
+
+  async getUserSurveyResponse(surveyId: string): Promise<IServerResponse<ISurveyResponse>> {
+    const response = await apiClient.get<IServerResponse<ISurveyResponse>>(`${API_BASE_URL}/surveys/responses/${surveyId}`);
+    return response.data;
+  },
+
+  async checkAccountSetupSurveyStatus(): Promise<IServerResponse<{ is_completed: boolean; completed_at?: string }>> {
+    const response = await apiClient.get<IServerResponse<{ is_completed: boolean; completed_at?: string }>>(`${API_BASE_URL}/surveys/account-setup/status`);
+    return response.data;
+  }
+};

worklenz-frontend/src/api/tasks/task-time-logs.api.service.ts

@@ -2,6 +2,7 @@ import { API_BASE_URL } from '@/shared/constants';
 import apiClient from '../api-client';
 import { IServerResponse } from '@/types/common.types';
 import { ITaskLogViewModel } from '@/types/tasks/task-log-view.types';
+import { getUserSession } from '@/utils/session-helper';
 
 const rootUrl = `${API_BASE_URL}/task-time-log`;
 
@@ -17,7 +18,11 @@ export interface IRunningTimer {
 
 export const taskTimeLogsApiService = {
   getByTask: async (id: string): Promise<IServerResponse<ITaskLogViewModel[]>> => {
-    const response = await apiClient.get(`${rootUrl}/task/${id}`);
+    const session = getUserSession();
+    const timezone = session?.timezone_name || 'UTC';
+    const response = await apiClient.get(`${rootUrl}/task/${id}`, {
+      params: { time_zone_name: timezone }
+    });
     return response.data;
   },
 

worklenz-frontend/src/app/routes/auth-routes.tsx

@@ -4,12 +4,12 @@ import { Navigate } from 'react-router-dom';
 import { SuspenseFallback } from '@/components/suspense-fallback/suspense-fallback';
 
 // Lazy load auth page components for better code splitting
-const LoginPage = lazy(() => import('@/pages/auth/login-page'));
-const SignupPage = lazy(() => import('@/pages/auth/signup-page'));
-const ForgotPasswordPage = lazy(() => import('@/pages/auth/forgot-password-page'));
-const LoggingOutPage = lazy(() => import('@/pages/auth/logging-out'));
-const AuthenticatingPage = lazy(() => import('@/pages/auth/authenticating'));
-const VerifyResetEmailPage = lazy(() => import('@/pages/auth/verify-reset-email'));
+const LoginPage = lazy(() => import('@/pages/auth/LoginPage'));
+const SignupPage = lazy(() => import('@/pages/auth/SignupPage'));
+const ForgotPasswordPage = lazy(() => import('@/pages/auth/ForgotPasswordPage'));
+const LoggingOutPage = lazy(() => import('@/pages/auth/LoggingOutPage'));
+const AuthenticatingPage = lazy(() => import('@/pages/auth/AuthenticatingPage'));
+const VerifyResetEmailPage = lazy(() => import('@/pages/auth/VerifyResetEmailPage'));
 
 const authRoutes = [
   {

worklenz-frontend/src/app/routes/index.tsx

@@ -90,6 +90,23 @@ export const SetupGuard = memo(({ children }: GuardProps) => {
 
 SetupGuard.displayName = 'SetupGuard';
 
+// Combined guard for routes that require both authentication and setup completion
+export const AuthAndSetupGuard = memo(({ children }: GuardProps) => {
+  const { isAuthenticated, isSetupComplete, location } = useAuthStatus();
+
+  if (!isAuthenticated) {
+    return <Navigate to="/auth" state={{ from: location }} replace />;
+  }
+
+  if (!isSetupComplete) {
+    return <Navigate to="/worklenz/setup" />;
+  }
+
+  return <>{children}</>;
+});
+
+AuthAndSetupGuard.displayName = 'AuthAndSetupGuard';
+
 // Optimized route wrapping function with Suspense boundaries
 const wrapRoutes = (
   routes: RouteObject[],
@@ -171,9 +188,11 @@ StaticLicenseExpired.displayName = 'StaticLicenseExpired';
 // Create route arrays (moved outside of useMemo to avoid hook violations)
 const publicRoutes = [...rootRoutes, ...authRoutes, notFoundRoute];
 
-const protectedMainRoutes = wrapRoutes(mainRoutes, AuthGuard);
+// Apply combined guard to main routes that require both auth and setup completion
+const protectedMainRoutes = wrapRoutes(mainRoutes, AuthAndSetupGuard);
 const adminRoutes = wrapRoutes(reportingRoutes, AdminGuard);
-const setupRoutes = wrapRoutes([accountSetupRoute], SetupGuard);
+// Setup route should be accessible without setup completion, only requires authentication
+const setupRoutes = wrapRoutes([accountSetupRoute], AuthGuard);
 
 // License expiry check function
 const withLicenseExpiryCheck = (routes: RouteObject[]): RouteObject[] => {

worklenz-frontend/src/app/routes/main-routes.tsx

@@ -17,6 +17,7 @@ const ProjectTemplateEditView = lazy(
 const LicenseExpired = lazy(() => import('@/pages/license-expired/license-expired'));
 const ProjectView = lazy(() => import('@/pages/projects/projectView/project-view'));
 const Unauthorized = lazy(() => import('@/pages/unauthorized/unauthorized'));
+const GanttDemoPage = lazy(() => import('@/pages/GanttDemoPage'));
 
 // Define AdminGuard component with defensive programming
 const AdminGuard = ({ children }: { children: React.ReactNode }) => {
@@ -106,6 +107,14 @@ const mainRoutes: RouteObject[] = [
           </Suspense>
         ),
       },
+      {
+        path: 'gantt-demo',
+        element: (
+          <Suspense fallback={<SuspenseFallback />}>
+            <GanttDemoPage />
+          </Suspense>
+        ),
+      },
       ...settingsRoutes,
       ...adminCenterRoutes,
     ],

worklenz-frontend/src/components/AssigneeSelector.tsx

@@ -1,7 +1,7 @@
 import React, { useState, useRef, useEffect, useMemo, useCallback } from 'react';
 import { createPortal } from 'react-dom';
 import { useSelector } from 'react-redux';
-import { PlusOutlined, UserAddOutlined } from '@ant-design/icons';
+import { PlusOutlined, UserAddOutlined } from '@/shared/antd-imports';
 import { RootState } from '@/app/store';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
 import { ITeamMembersViewModel } from '@/types/teamMembers/teamMembersViewModel.types';
@@ -11,8 +11,10 @@ import { useAuthService } from '@/hooks/useAuth';
 import { Avatar, Button, Checkbox } from '@/components';
 import { sortTeamMembers } from '@/utils/sort-team-members';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
-import { toggleProjectMemberDrawer } from '@/features/projects/singleProject/members/projectMembersSlice';
+import { setIsFromAssigner, toggleProjectMemberDrawer } from '@/features/projects/singleProject/members/projectMembersSlice';
 import { updateEnhancedKanbanTaskAssignees } from '@/features/enhanced-kanban/enhanced-kanban.slice';
+import useIsProjectManager from '@/hooks/useIsProjectManager';
+import { useAuthStatus } from '@/hooks/useAuthStatus';
 
 interface AssigneeSelectorProps {
   task: IProjectTask;
@@ -21,9 +23,9 @@ interface AssigneeSelectorProps {
   kanbanMode?: boolean;
 }
 
-const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({ 
-  task, 
-  groupId = null, 
+const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
+  task,
+  groupId = null,
   isDarkMode = false,
   kanbanMode = false
 }) => {
@@ -42,6 +44,8 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
   const currentSession = useAuthService().getCurrentSession();
   const { socket } = useSocket();
   const dispatch = useAppDispatch();
+  const { isAdmin } = useAuthStatus();
+  const isProjectManager = useIsProjectManager();
 
   const filteredMembers = useMemo(() => {
     return teamMembers?.data?.filter(member =>
@@ -64,7 +68,7 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
   useEffect(() => {
     const handleClickOutside = (event: MouseEvent) => {
       if (dropdownRef.current && !dropdownRef.current.contains(event.target as Node) &&
-          buttonRef.current && !buttonRef.current.contains(event.target as Node)) {
+        buttonRef.current && !buttonRef.current.contains(event.target as Node)) {
         setIsOpen(false);
       }
     };
@@ -74,10 +78,10 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
         // Check if the button is still visible in the viewport
         if (buttonRef.current) {
           const rect = buttonRef.current.getBoundingClientRect();
-          const isVisible = rect.top >= 0 && rect.left >= 0 && 
-                           rect.bottom <= window.innerHeight && 
-                           rect.right <= window.innerWidth;
-          
+          const isVisible = rect.top >= 0 && rect.left >= 0 &&
+            rect.bottom <= window.innerHeight &&
+            rect.right <= window.innerWidth;
+
           if (isVisible) {
             updateDropdownPosition();
           } else {
@@ -98,7 +102,7 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
       document.addEventListener('mousedown', handleClickOutside);
       window.addEventListener('scroll', handleScroll, true);
       window.addEventListener('resize', handleResize);
-      
+
       return () => {
         document.removeEventListener('mousedown', handleClickOutside);
         window.removeEventListener('scroll', handleScroll, true);
@@ -113,10 +117,10 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
   const handleDropdownToggle = (e: React.MouseEvent) => {
     e.preventDefault();
     e.stopPropagation();
-    
+
     if (!isOpen) {
       updateDropdownPosition();
-      
+
       // Prepare team members data when opening
       const assignees = task?.assignees?.map(assignee => assignee.team_member_id);
       const membersData = (members?.data || []).map(member => ({
@@ -125,7 +129,7 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
       }));
       const sortedMembers = sortTeamMembers(membersData);
       setTeamMembers({ data: sortedMembers });
-      
+
       setIsOpen(true);
       // Focus search input after opening
       setTimeout(() => {
@@ -160,8 +164,8 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
     // Update local team members state for dropdown UI
     setTeamMembers(prev => ({
       ...prev,
-      data: (prev.data || []).map(member => 
-        member.id === memberId 
+      data: (prev.data || []).map(member =>
+        member.id === memberId
           ? { ...member, selected: checked }
           : member
       )
@@ -198,14 +202,15 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
   const checkMemberSelected = (memberId: string) => {
     if (!memberId) return false;
     // Use optimistic assignees if available, otherwise fall back to task assignees
-    const assignees = optimisticAssignees.length > 0 
-      ? optimisticAssignees 
+    const assignees = optimisticAssignees.length > 0
+      ? optimisticAssignees
       : task?.assignees?.map(assignee => assignee.team_member_id) || [];
     return assignees.includes(memberId);
   };
 
   const handleInviteProjectMemberDrawer = () => {
     setIsOpen(false); // Close the assignee dropdown first
+    dispatch(setIsFromAssigner(true));
     dispatch(toggleProjectMemberDrawer()); // Then open the invite drawer
   };
 
@@ -217,12 +222,12 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
         className={`
           w-5 h-5 rounded-full border border-dashed flex items-center justify-center
           transition-colors duration-200
-          ${isOpen 
-            ? isDarkMode 
-              ? 'border-blue-500 bg-blue-900/20 text-blue-400' 
+          ${isOpen
+            ? isDarkMode
+              ? 'border-blue-500 bg-blue-900/20 text-blue-400'
               : 'border-blue-500 bg-blue-50 text-blue-600'
-            : isDarkMode 
-              ? 'border-gray-600 hover:border-gray-500 hover:bg-gray-800 text-gray-400' 
+            : isDarkMode
+              ? 'border-gray-600 hover:border-gray-500 hover:bg-gray-800 text-gray-400'
               : 'border-gray-300 hover:border-gray-400 hover:bg-gray-100 text-gray-600'
           }
         `}
@@ -236,8 +241,8 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
           onClick={e => e.stopPropagation()}
           className={`
             fixed z-[99999] w-72 rounded-md shadow-lg border
-            ${isDarkMode 
-              ? 'bg-gray-800 border-gray-600' 
+            ${isDarkMode
+              ? 'bg-gray-800 border-gray-600'
               : 'bg-white border-gray-200'
             }
           `}
@@ -273,10 +278,10 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
                   key={member.id}
                   className={`
                     flex items-center gap-2 p-2 cursor-pointer transition-colors
-                    ${member.pending_invitation 
-                      ? 'opacity-50 cursor-not-allowed' 
-                      : isDarkMode 
-                        ? 'hover:bg-gray-700' 
+                    ${member.pending_invitation
+                      ? 'opacity-50 cursor-not-allowed'
+                      : isDarkMode
+                        ? 'hover:bg-gray-700'
                         : 'hover:bg-gray-50'
                     }
                   `}
@@ -301,23 +306,21 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
                       />
                     </span>
                     {pendingChanges.has(member.id || '') && (
-                      <div className={`absolute inset-0 flex items-center justify-center ${
-                        isDarkMode ? 'bg-gray-800/50' : 'bg-white/50'
-                      }`}>
-                        <div className={`w-3 h-3 border border-t-transparent rounded-full animate-spin ${
-                          isDarkMode ? 'border-blue-400' : 'border-blue-600'
-                        }`} />
+                      <div className={`absolute inset-0 flex items-center justify-center ${isDarkMode ? 'bg-gray-800/50' : 'bg-white/50'
+                        }`}>
+                        <div className={`w-3 h-3 border border-t-transparent rounded-full animate-spin ${isDarkMode ? 'border-blue-400' : 'border-blue-600'
+                          }`} />
                       </div>
                     )}
                   </div>
-                  
+
                   <Avatar
                     src={member.avatar_url}
                     name={member.name || ''}
                     size={24}
                     isDarkMode={isDarkMode}
                   />
-                  
+
                   <div className="flex-1 min-w-0">
                     <div className={`text-xs font-medium truncate ${isDarkMode ? 'text-gray-100' : 'text-gray-900'}`}>
                       {member.name}
@@ -339,22 +342,26 @@ const AssigneeSelector: React.FC<AssigneeSelectorProps> = ({
           </div>
 
           {/* Footer */}
-          <div className={`p-2 border-t ${isDarkMode ? 'border-gray-600' : 'border-gray-200'}`}>
-            <button
-              className={`
-                w-full flex items-center justify-center gap-1 px-2 py-1 text-xs rounded
-                transition-colors
-                ${isDarkMode
-                  ? 'text-blue-400 hover:bg-gray-700'
-                  : 'text-blue-600 hover:bg-blue-50'
-                }
-              `}
-              onClick={handleInviteProjectMemberDrawer}
-            >
-              <UserAddOutlined />
-              Invite member
-            </button>
-          </div>
+
+          {(isAdmin || isProjectManager) && (
+            <div className={`p-2 border-t ${isDarkMode ? 'border-gray-600' : 'border-gray-200'}`}>
+              <button
+                className={`
+                  w-full flex items-center justify-center gap-1 px-2 py-1 text-xs rounded
+                  transition-colors
+                  ${isDarkMode
+                    ? 'text-blue-400 hover:bg-gray-700'
+                    : 'text-blue-600 hover:bg-blue-50'
+                  }
+                `}
+                onClick={handleInviteProjectMemberDrawer}
+              >
+                <UserAddOutlined />
+                Invite member
+              </button>
+            </div>
+          )}
+
         </div>,
         document.body
       )}

worklenz-frontend/src/components/AuthPageHeader.tsx

@@ -1,4 +1,4 @@
-import { Flex, Typography } from 'antd';
+import { Flex, Typography } from '@/shared/antd-imports';
 import logo from '@/assets/images/worklenz-light-mode.png';
 import logoDark from '@/assets/images/worklenz-dark-mode.png';
 import { useAppSelector } from '@/hooks/useAppSelector';

worklenz-frontend/src/components/CustomColordLabel.tsx

@@ -1,5 +1,5 @@
 import React from 'react';
-import { Tooltip } from 'antd';
+import { Tooltip } from '@/shared/antd-imports';
 import { Label } from '@/types/task-management.types';
 import { ITaskLabel } from '@/types/tasks/taskLabel.types';
 
@@ -39,7 +39,7 @@ const CustomColordLabel = React.forwardRef<HTMLSpanElement, CustomColordLabelPro
       <Tooltip title={label.name}>
         <span
           ref={ref}
-          className="inline-flex items-center px-2 py-0.5 rounded-sm text-xs font-medium shrink-0 max-w-[120px]"
+          className="inline-flex items-center px-1.5 py-0.5 rounded text-xs font-medium shrink-0 max-w-[100px]"
           style={{ 
             backgroundColor,
             color: textColor,

worklenz-frontend/src/components/CustomNumberLabel.tsx

@@ -1,5 +1,5 @@
 import React from 'react';
-import { Tooltip } from 'antd';
+import { Tooltip } from '@/shared/antd-imports';
 import { NumbersColorMap } from '@/shared/constants';
 
 interface CustomNumberLabelProps {
@@ -21,8 +21,8 @@ const CustomNumberLabel = React.forwardRef<HTMLSpanElement, CustomNumberLabelPro
       <Tooltip title={labelList.join(', ')}>
         <span
           ref={ref}
-          className="inline-flex items-center px-2 py-0.5 rounded text-xs font-medium text-white cursor-help"
-          style={{ backgroundColor }}
+          className="inline-flex items-center px-1.5 py-0.5 rounded text-xs font-medium cursor-help"
+          style={{ backgroundColor, color: 'white' }}
         >
           {namesString}
         </span>

worklenz-frontend/src/components/CustomSearchbar.tsx

@@ -1,5 +1,5 @@
-import { SearchOutlined } from '@ant-design/icons';
-import { Input } from 'antd';
+import { SearchOutlined } from '@/shared/antd-imports';
+import { Input } from '@/shared/antd-imports';
 
 type CustomSearchbarProps = {
   placeholderText: string;

worklenz-frontend/src/components/CustomTableTitle.tsx

@@ -1,6 +1,6 @@
-import { Flex, Tooltip, Typography } from 'antd';
+import { Flex, Tooltip, Typography } from '@/shared/antd-imports';
 import { colors } from '../styles/colors';
-import { ExclamationCircleOutlined } from '@ant-design/icons';
+import { ExclamationCircleOutlined } from '@/shared/antd-imports';
 
 // this custom table title used when the typography font weigh 500 needed
 const CustomTableTitle = ({ title, tooltip }: { title: string; tooltip?: string | null }) => {

worklenz-frontend/src/components/EmptyListPlaceholder.tsx

@@ -1,17 +1,24 @@
-import { Empty, Typography } from 'antd';
+import { Empty, Typography } from '@/shared/antd-imports';
 import React from 'react';
+import { useTranslation } from 'react-i18next';
 
 type EmptyListPlaceholderProps = {
   imageSrc?: string;
   imageHeight?: number;
-  text: string;
+  text?: string;
+  textKey?: string;
+  i18nNs?: string;
 };
 
 const EmptyListPlaceholder = ({
   imageSrc = 'https://s3.us-west-2.amazonaws.com/worklenz.com/assets/empty-box.webp',
   imageHeight = 60,
   text,
+  textKey,
+  i18nNs = 'task-list-table',
 }: EmptyListPlaceholderProps) => {
+  const { t } = useTranslation(i18nNs);
+  const description = textKey ? t(textKey) : text;
   return (
     <Empty
       image={imageSrc}
@@ -22,7 +29,7 @@ const EmptyListPlaceholder = ({
         alignItems: 'center',
         marginBlockStart: 24,
       }}
-      description={<Typography.Text type="secondary">{text}</Typography.Text>}
+      description={<Typography.Text type="secondary">{description}</Typography.Text>}
     />
   );
 };

worklenz-frontend/src/components/ErrorBoundary.tsx

@@ -1,5 +1,5 @@
 import React from 'react';
-import { Button, Result } from 'antd';
+import { Button, Result } from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 import { useNavigate } from 'react-router-dom';
 import logger from '@/utils/errorLogger';

worklenz-frontend/src/components/LabelsSelector.tsx

@@ -1,7 +1,7 @@
 import React, { useState, useRef, useEffect, useMemo, useCallback } from 'react';
 import { createPortal } from 'react-dom';
 import { useSelector } from 'react-redux';
-import { PlusOutlined, TagOutlined } from '@ant-design/icons';
+import { PlusOutlined, TagOutlined } from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 import { RootState } from '@/app/store';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';

worklenz-frontend/src/components/ModuleErrorBoundary.tsx

@@ -0,0 +1,110 @@
+import React, { Component, ErrorInfo, ReactNode } from 'react';
+import { Button, Result } from '@/shared/antd-imports';
+import CacheCleanup from '@/utils/cache-cleanup';
+
+interface Props {
+  children: ReactNode;
+}
+
+interface State {
+  hasError: boolean;
+  error?: Error;
+}
+
+class ModuleErrorBoundary extends Component<Props, State> {
+  constructor(props: Props) {
+    super(props);
+    this.state = { hasError: false };
+  }
+
+  static getDerivedStateFromError(error: Error): State {
+    // Check if this is a module loading error
+    const isModuleError = 
+      error.message.includes('Failed to fetch dynamically imported module') ||
+      error.message.includes('Loading chunk') ||
+      error.message.includes('Loading CSS chunk') ||
+      error.name === 'ChunkLoadError';
+
+    if (isModuleError) {
+      return { hasError: true, error };
+    }
+
+    // For other errors, let them bubble up
+    return { hasError: false };
+  }
+
+  componentDidCatch(error: Error, errorInfo: ErrorInfo) {
+    console.error('Module Error Boundary caught an error:', error, errorInfo);
+    
+    // If this is a module loading error, clear caches and reload
+    if (this.state.hasError) {
+      this.handleModuleError();
+    }
+  }
+
+  private async handleModuleError() {
+    try {
+      console.log('Handling module loading error - clearing caches...');
+      
+      // Clear all caches
+      await CacheCleanup.clearAllCaches();
+      
+      // Force reload to login page
+      CacheCleanup.forceReload('/auth/login');
+    } catch (cacheError) {
+      console.error('Failed to handle module error:', cacheError);
+      // Fallback: just reload the page
+      window.location.reload();
+    }
+  }
+
+  private handleRetry = async () => {
+    try {
+      await CacheCleanup.clearAllCaches();
+      CacheCleanup.forceReload('/auth/login');
+    } catch (error) {
+      console.error('Retry failed:', error);
+      window.location.reload();
+    }
+  };
+
+  render() {
+    if (this.state.hasError) {
+      return (
+        <div style={{ 
+          display: 'flex', 
+          justifyContent: 'center', 
+          alignItems: 'center', 
+          minHeight: '100vh',
+          padding: '20px'
+        }}>
+          <Result
+            status="error"
+            title="Module Loading Error"
+            subTitle="There was an issue loading the application. This usually happens after updates or during logout."
+            extra={[
+              <Button 
+                type="primary" 
+                key="retry" 
+                onClick={this.handleRetry}
+                loading={false}
+              >
+                Retry
+              </Button>,
+              <Button 
+                key="reload" 
+                onClick={() => window.location.reload()}
+              >
+                Reload Page
+              </Button>
+            ]}
+          />
+        </div>
+      );
+    }
+
+    return this.props.children;
+  }
+}
+
+export default ModuleErrorBoundary; 

worklenz-frontend/src/components/PinRouteToNavbarButton.tsx

@@ -1,7 +1,7 @@
 import React, { useState } from 'react';
 import { getJSONFromLocalStorage, saveJSONToLocalStorage } from '../utils/localStorageFunctions';
-import { Button, ConfigProvider, Tooltip } from 'antd';
-import { PushpinFilled, PushpinOutlined } from '@ant-design/icons';
+import { Button, ConfigProvider, Tooltip } from '@/shared/antd-imports';
+import { PushpinFilled, PushpinOutlined } from '@/shared/antd-imports';
 import { colors } from '../styles/colors';
 import { navRoutes, NavRoutesType } from '../features/navbar/navRoutes';
 

worklenz-frontend/src/components/PreferenceSelector.tsx

@@ -1,5 +1,5 @@
-import { FloatButton, Space, Tooltip } from 'antd';
-import { FormatPainterOutlined } from '@ant-design/icons';
+import { FloatButton, Space, Tooltip } from '@/shared/antd-imports';
+import { FormatPainterOutlined } from '@/shared/antd-imports';
 // import LanguageSelector from '../features/i18n/language-selector';
 // import ThemeSelector from '../features/theme/ThemeSelector';
 

worklenz-frontend/src/components/account-setup/admin-center-common.css

@@ -1,19 +0,0 @@
-@media (max-width: 1000px) {
-  .step-content,
-  .step-form,
-  .create-first-task-form,
-  .setup-action-buttons,
-  .invite-members-form {
-    width: 400px !important;
-  }
-}
-
-@media (max-width: 500px) {
-  .step-content,
-  .step-form,
-  .create-first-task-form,
-  .setup-action-buttons,
-  .invite-members-form {
-    width: 200px !important;
-  }
-}

worklenz-frontend/src/components/account-setup/members-step.tsx

@@ -1,16 +1,15 @@
-import React, { useEffect, useRef } from 'react';
-import { Form, Input, Button, List, Alert, message, InputRef } from 'antd';
-import { CloseCircleOutlined, MailOutlined, PlusOutlined } from '@ant-design/icons';
+import React, { useEffect, useRef, useState } from 'react';
+import { Form, Input, Button, Typography, Card, Avatar, Tag, Alert, Space, Dropdown, MenuProps } from '@/shared/antd-imports';
+import { CloseCircleOutlined, MailOutlined, PlusOutlined, UserOutlined, CheckCircleOutlined, ExclamationCircleOutlined, GlobalOutlined } from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
-import { Typography } from 'antd';
-import { setTeamMembers, setTasks } from '@/features/account-setup/account-setup.slice';
+import { setTeamMembers } from '@/features/account-setup/account-setup.slice';
 import { useDispatch, useSelector } from 'react-redux';
 import { RootState } from '@/app/store';
 import { validateEmail } from '@/utils/validateEmail';
 import { sanitizeInput } from '@/utils/sanitizeInput';
-import { Rule } from 'antd/es/form';
+import { setLanguage } from '@/features/i18n/localesSlice';
 
-const { Title } = Typography;
+const { Title, Paragraph, Text } = Typography;
 
 interface Email {
   id: number;
@@ -20,163 +19,233 @@ interface Email {
 interface MembersStepProps {
   isDarkMode: boolean;
   styles: any;
+  token?: any;
 }
 
-const MembersStep: React.FC<MembersStepProps> = ({ isDarkMode, styles }) => {
-  const { t } = useTranslation('account-setup');
+const getEmailSuggestions = (orgName?: string) => {
+  if (!orgName) return [];
+  const cleanOrgName = orgName.toLowerCase().replace(/[^a-z0-9]/g, '');
+  return [`info@${cleanOrgName}.com`, `team@${cleanOrgName}.com`, `hello@${cleanOrgName}.com`, `contact@${cleanOrgName}.com`];
+};
+
+const getRoleSuggestions = (t: any) => [
+  { role: 'Designer', icon: '🎨', description: t('roleSuggestions.designer') },
+  { role: 'Developer', icon: '💻', description: t('roleSuggestions.developer') },
+  { role: 'Project Manager', icon: '📊', description: t('roleSuggestions.projectManager') },
+  { role: 'Marketing', icon: '📢', description: t('roleSuggestions.marketing') },
+  { role: 'Sales', icon: '💼', description: t('roleSuggestions.sales') },
+  { role: 'Operations', icon: '⚙️', description: t('roleSuggestions.operations') }
+];
+
+const MembersStep: React.FC<MembersStepProps> = ({ isDarkMode, styles, token }) => {
+  const { t, i18n } = useTranslation('account-setup');
   const { teamMembers, organizationName } = useSelector(
     (state: RootState) => state.accountSetupReducer
   );
-  const inputRefs = useRef<(InputRef | null)[]>([]);
+  const { language } = useSelector((state: RootState) => state.localesReducer);
+  const inputRefs = useRef<(HTMLInputElement | null)[]>([]);
   const dispatch = useDispatch();
-  const [form] = Form.useForm();
+  const [focusedIndex, setFocusedIndex] = useState<number | null>(null);
+  const [showSuggestions, setShowSuggestions] = useState(false);
+  const [validatedEmails, setValidatedEmails] = useState<Set<number>>(new Set());
+
+  const emailSuggestions = getEmailSuggestions(organizationName);
 
   const addEmail = () => {
-    if (teamMembers.length == 5) return;
-
+    if (teamMembers.length >= 5) return;
     const newId = teamMembers.length > 0 ? Math.max(...teamMembers.map(t => t.id)) + 1 : 0;
     dispatch(setTeamMembers([...teamMembers, { id: newId, value: '' }]));
-    setTimeout(() => {
-      inputRefs.current[newId]?.focus();
-    }, 0);
+    setTimeout(() => inputRefs.current[teamMembers.length]?.focus(), 100);
   };
 
   const removeEmail = (id: number) => {
-    if (teamMembers.length > 1) {
-      dispatch(setTeamMembers(teamMembers.filter(teamMember => teamMember.id !== id)));
-    }
+    if (teamMembers.length > 1) dispatch(setTeamMembers(teamMembers.filter(teamMember => teamMember.id !== id)));
   };
 
   const updateEmail = (id: number, value: string) => {
     const sanitizedValue = sanitizeInput(value);
-    dispatch(
-      setTeamMembers(
-        teamMembers.map(teamMember =>
-          teamMember.id === id ? { ...teamMember, value: sanitizedValue } : teamMember
-        )
-      )
-    );
+    dispatch(setTeamMembers(teamMembers.map(teamMember => teamMember.id === id ? { ...teamMember, value: sanitizedValue } : teamMember)));
   };
 
-  const handleKeyPress = (e: React.KeyboardEvent<HTMLInputElement>) => {
-    const input = e.currentTarget as HTMLInputElement;
-    if (!input.value.trim()) return;
-    e.preventDefault();
-    addEmail();
+  const handleKeyPress = (e: React.KeyboardEvent<HTMLInputElement>, index: number) => {
+    if (e.key === 'Enter') {
+      const input = e.currentTarget as HTMLInputElement;
+      if (input.value.trim() && validateEmail(input.value.trim())) {
+        e.preventDefault();
+        if (index === teamMembers.length - 1 && teamMembers.length < 5) addEmail();
+        else if (index < teamMembers.length - 1) inputRefs.current[index + 1]?.focus();
+      }
+    }
   };
 
-  // Function to set ref that doesn't return anything (void)
-  const setInputRef = (index: number) => (el: InputRef | null) => {
-    inputRefs.current[index] = el;
+  const handleSuggestionClick = (suggestion: string) => {
+    const emptyEmailIndex = teamMembers.findIndex(member => !member.value.trim());
+    if (emptyEmailIndex !== -1) {
+      updateEmail(teamMembers[emptyEmailIndex].id, suggestion);
+    } else if (teamMembers.length < 5) {
+      const newId = teamMembers.length > 0 ? Math.max(...teamMembers.map(t => t.id)) + 1 : 0;
+      dispatch(setTeamMembers([...teamMembers, { id: newId, value: suggestion }]));
+    }
+    setShowSuggestions(false);
   };
 
   useEffect(() => {
-    setTimeout(() => {
-      inputRefs.current[teamMembers.length - 1]?.focus();
-      // Set initial form values
-      const initialValues: Record<string, string> = {};
-      teamMembers.forEach(teamMember => {
-        initialValues[`email-${teamMember.id}`] = teamMember.value;
-      });
-      form.setFieldsValue(initialValues);
-    }, 200);
+    setTimeout(() => inputRefs.current[0]?.focus(), 200);
   }, []);
 
-  const formRules = {
-    email: [
-      {
-        validator: async (_: any, value: string) => {
-          if (!value) return;
-          if (!validateEmail(value)) {
-            throw new Error(t('invalidEmail'));
-          }
-        },
-      },
-    ],
+  const getEmailStatus = (email: string, memberId: number) => {
+    if (!email.trim()) return 'empty';
+    if (!validatedEmails.has(memberId)) return 'empty';
+    return validateEmail(email) ? 'valid' : 'invalid';
   };
 
+  const handleBlur = (memberId: number, email: string) => {
+    setFocusedIndex(null);
+    if (email.trim()) setValidatedEmails(prev => new Set(prev).add(memberId));
+  };
+
+  const languages = [
+    { key: 'en', label: t('languages.en'), flag: '🇺🇸' },
+    { key: 'es', label: t('languages.es'), flag: '🇪🇸' },
+    { key: 'pt', label: t('languages.pt'), flag: '🇵🇹' },
+    { key: 'de', label: t('languages.de'), flag: '🇩🇪' },
+    { key: 'alb', label: t('languages.alb'), flag: '🇦🇱' },
+    { key: 'zh', label: t('languages.zh'), flag: '🇨🇳' }
+  ];
+
+  const handleLanguageChange = (languageKey: string) => {
+    dispatch(setLanguage(languageKey));
+    i18n.changeLanguage(languageKey);
+  };
+
+  const currentLanguage = languages.find(lang => lang.key === language) || languages[0];
+  const languageMenuItems: MenuProps['items'] = languages.map(lang => ({ key: lang.key, label: <div className="flex items-center space-x-2"><span>{lang.flag}</span><span>{lang.label}</span></div>, onClick: () => handleLanguageChange(lang.key) }));
+
   return (
-    <Form
-      form={form}
-      className="invite-members-form"
-      style={{
-        minHeight: '300px',
-        width: '600px',
-        paddingBottom: '1rem',
-        marginBottom: '3rem',
-        marginTop: '3rem',
-        display: 'flex',
-        flexDirection: 'column',
-      }}
-    >
-      <Form.Item>
-        <Title level={2} style={{ marginBottom: '1rem' }}>
-          {t('step3Title')} "<mark>{organizationName}</mark>".
+    <div className="w-full members-step">
+      {/* Header */}
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
+          {t('membersStepTitle')}
         </Title>
-      </Form.Item>
-      <Form.Item
-        layout="vertical"
-        rules={[{ required: true }]}
-        label={
-          <span className="font-medium">
-            {t('step3InputLabel')}&nbsp; <MailOutlined /> {t('maxMembers')}
-          </span>
-        }
-      >
-        <List
-          dataSource={teamMembers}
-          bordered={false}
-          itemLayout="vertical"
-          renderItem={(teamMember, index) => (
-            <List.Item key={teamMember.id}>
-              <div className="invite-members-form" style={{ display: 'flex', width: '600px' }}>
-                <Form.Item
-                  rules={formRules.email as Rule[]}
-                  className="w-full"
-                  validateTrigger={['onChange', 'onBlur']}
-                  name={`email-${teamMember.id}`}
-                >
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('membersStepDescription', { organizationName })}
+        </Paragraph>
+      </div>
+
+      {/* Team Members List */}
+      <div className="mb-6">
+        <div className="space-y-3">
+          {teamMembers.map((teamMember, index) => {
+            const emailStatus = getEmailStatus(teamMember.value, teamMember.id);
+            return (
+              <div 
+                key={teamMember.id}
+                className={`flex items-center space-x-3 p-3 rounded-lg border transition-all duration-200 ${
+                  focusedIndex === index ? 'border-2' : ''
+                }`}
+                style={{ 
+                  borderColor: focusedIndex === index ? token?.colorPrimary : 
+                               emailStatus === 'invalid' ? token?.colorError : token?.colorBorder,
+                  backgroundColor: token?.colorBgContainer 
+                }}
+              >
+                <Avatar 
+                  size={32}
+                  style={{ 
+                    backgroundColor: emailStatus === 'valid' ? token?.colorSuccess : 
+                                    emailStatus === 'invalid' ? token?.colorError : token?.colorBorderSecondary,
+                    color: '#fff'
+                  }}
+                  icon={
+                    emailStatus === 'valid' ? <CheckCircleOutlined /> :
+                    emailStatus === 'invalid' ? <ExclamationCircleOutlined /> :
+                    <UserOutlined />
+                  }
+                />
+                
+                <div className="flex-1">
                   <Input
-                    placeholder={t('emailPlaceholder')}
+                    placeholder={t('memberPlaceholder', { index: index + 1 })}
                     value={teamMember.value}
                     onChange={e => updateEmail(teamMember.id, e.target.value)}
-                    onPressEnter={handleKeyPress}
-                    ref={setInputRef(index)}
-                    status={teamMember.value && !validateEmail(teamMember.value) ? 'error' : ''}
-                    id={`member-${index}`}
+                    onKeyPress={e => handleKeyPress(e, index)}
+                    onFocus={() => setFocusedIndex(index)}
+                    onBlur={() => handleBlur(teamMember.id, teamMember.value)}
+                    ref={el => inputRefs.current[index] = el}
+                    className="border-0 shadow-none"
+                    style={{
+                      backgroundColor: 'transparent',
+                      color: token?.colorText
+                    }}
+                    prefix={<MailOutlined style={{ color: token?.colorTextTertiary }} />}
+                    status={emailStatus === 'invalid' ? 'error' : undefined}
+                    suffix={
+                      emailStatus === 'valid' ? (
+                        <CheckCircleOutlined style={{ color: token?.colorSuccess }} />
+                      ) : emailStatus === 'invalid' ? (
+                        <ExclamationCircleOutlined style={{ color: token?.colorError }} />
+                      ) : null
+                    }
                   />
-                </Form.Item>
-                <Button
-                  className="custom-close-button"
-                  style={{ marginLeft: '48px' }}
-                  type="text"
-                  icon={<CloseCircleOutlined />}
-                  disabled={teamMembers.length === 1}
-                  onClick={() => removeEmail(teamMember.id)}
-                />
+                  {emailStatus === 'invalid' && (
+                    <Text type="danger" className="text-xs mt-1 block">
+                      {t('invalidEmail')}
+                    </Text>
+                  )}
+                  {emailStatus === 'valid' && (
+                    <Text type="success" className="text-xs mt-1 block">
+                      {t('validEmailAddress')}
+                    </Text>
+                  )}
+                </div>
+
+                {teamMembers.length > 1 && (
+                  <Button
+                    type="text"
+                    size="small"
+                    icon={<CloseCircleOutlined />}
+                    onClick={() => removeEmail(teamMember.id)}
+                    style={{ color: token?.colorTextTertiary }}
+                  />
+                )}
               </div>
-            </List.Item>
-          )}
-        />
-        <Button
-          type="dashed"
-          icon={<PlusOutlined />}
-          onClick={addEmail}
-          style={{ marginTop: '16px' }}
-          disabled={teamMembers.length == 5}
-        >
-          {t('tasksStepAddAnother')}
-        </Button>
-        <div
+            );
+          })}
+        </div>
+
+        {/* Add Member Button */}
+        {teamMembers.length < 5 && (
+          <Button
+            type="dashed"
+            icon={<PlusOutlined />}
+            onClick={addEmail}
+            className="w-full mt-4 h-12 text-base"
+            style={{ 
+              borderColor: token?.colorBorder,
+              color: token?.colorTextSecondary 
+            }}
+          >
+            {t('addAnotherTeamMember', { current: teamMembers.length, max: 5 })}
+          </Button>
+        )}
+      </div>
+
+      {/* Skip Option */}
+      <div className="mb-6">
+        <Alert
+          message={t('canInviteLater')}
+          description={t('skipStepDescription')}
+          type="info"
+          showIcon
           style={{
-            marginTop: '24px',
-            display: 'flex',
-            justifyContent: 'space-between',
+            backgroundColor: token?.colorInfoBg,
+            borderColor: token?.colorInfoBorder
           }}
-        ></div>
-      </Form.Item>
-    </Form>
+        />
+      </div>
+    </div>
   );
 };
 
-export default MembersStep;
+export default MembersStep;

worklenz-frontend/src/components/account-setup/organization-step.tsx

@@ -1,31 +1,40 @@
-import React, { useEffect, useRef } from 'react';
-import { Form, Input, InputRef, Typography } from 'antd';
+import React, { useEffect, useRef, useState } from 'react';
+import { Form, Input, InputRef, Typography, Card, Tooltip } from '@/shared/antd-imports';
 import { useDispatch, useSelector } from 'react-redux';
 import { useTranslation } from 'react-i18next';
 import { setOrganizationName } from '@/features/account-setup/account-setup.slice';
 import { RootState } from '@/app/store';
 import { sanitizeInput } from '@/utils/sanitizeInput';
-import './admin-center-common.css';
 
-const { Title } = Typography;
+const { Title, Paragraph, Text } = Typography;
 
 interface Props {
   onEnter: () => void;
   styles: any;
   organizationNamePlaceholder: string;
+  organizationNameInitialValue?: string;
+  isDarkMode: boolean;
+  token?: any;
 }
 
 export const OrganizationStep: React.FC<Props> = ({
   onEnter,
   styles,
   organizationNamePlaceholder,
+  organizationNameInitialValue,
+  isDarkMode,
+  token,
 }) => {
   const { t } = useTranslation('account-setup');
   const dispatch = useDispatch();
   const { organizationName } = useSelector((state: RootState) => state.accountSetupReducer);
   const inputRef = useRef<InputRef>(null);
 
+  // Autofill organization name if not already set
   useEffect(() => {
+    if (!organizationName && organizationNameInitialValue) {
+      dispatch(setOrganizationName(organizationNameInitialValue));
+    }
     setTimeout(() => inputRef.current?.focus(), 300);
   }, []);
 
@@ -40,25 +49,85 @@ export const OrganizationStep: React.FC<Props> = ({
   };
 
   return (
-    <Form className="step-form" style={styles.form}>
-      <Form.Item>
-        <Title level={2} style={{ marginBottom: '1rem' }}>
-          {t('organizationStepTitle')}
+    <div className="w-full organization-step">
+      {/* Header */}
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
+          {t('organizationStepWelcome')}
         </Title>
-      </Form.Item>
-      <Form.Item
-        layout="vertical"
-        rules={[{ required: true }]}
-        label={<span style={styles.label}>{t('organizationStepLabel')}</span>}
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('organizationStepDescription')}
+        </Paragraph>
+      </div>
+
+      {/* Main Form Card */}
+      <div className="mb-6">
+        <Card 
+          className="border-2 hover:shadow-md transition-all duration-200"
+          style={{ 
+            borderColor: token?.colorPrimary,
+            backgroundColor: token?.colorBgContainer 
+          }}
+        >
+          <Form.Item 
+            className="mb-4"
+            label={
+              <div className="flex items-center space-x-2">
+                <span className="font-medium text-base" style={{ color: token?.colorText }}>
+                  {t('organizationStepLabel')}
+                </span>
+                <Tooltip title={t('organizationStepTooltip')}>
+                  <span 
+                    className="text-sm cursor-help"
+                    style={{ color: token?.colorTextTertiary }}
+                  >
+                    ⓘ
+                  </span>
+                </Tooltip>
+              </div>
+            }
+          >
+            <Input
+              placeholder={organizationNamePlaceholder || t('organizationStepPlaceholder')}
+              value={organizationName}
+              onChange={handleOrgNameChange}
+              onPressEnter={onPressEnter}
+              ref={inputRef}
+              className="text-base"
+            />
+          </Form.Item>
+
+          {/* Character Count and Validation */}
+          <div className="flex justify-between items-center text-sm">
+            <Text type="secondary">
+              {organizationName.length}/50 {t('organizationStepCharacters')}
+            </Text>
+            {organizationName.length > 0 && (
+              <div className="flex items-center space-x-1">
+                {organizationName.length >= 2 ? (
+                  <span style={{ color: token?.colorSuccess }}>✓ {t('organizationStepGoodLength')}</span>
+                ) : (
+                  <span style={{ color: token?.colorWarning }}>⚠ {t('organizationStepTooShort')}</span>
+                )}
+              </div>
+            )}
+          </div>
+        </Card>
+      </div>
+
+      {/* Footer Note */}
+      <div 
+        className="text-center p-4 rounded-lg"
+        style={{ 
+          backgroundColor: token?.colorInfoBg,
+          borderColor: token?.colorInfoBorder,
+          border: '1px solid'
+        }}
       >
-        <Input
-          placeholder={organizationNamePlaceholder}
-          value={organizationName}
-          onChange={handleOrgNameChange}
-          onPressEnter={onPressEnter}
-          ref={inputRef}
-        />
-      </Form.Item>
-    </Form>
+        <Text type="secondary" className="text-sm">
+          🔒 {t('organizationStepPrivacyNote')}
+        </Text>
+      </div>
+    </div>
   );
-};
+};

worklenz-frontend/src/components/account-setup/project-step.tsx

@@ -1,9 +1,9 @@
 import React, { startTransition, useEffect, useRef, useState } from 'react';
-import { useDispatch, useSelector } from 'react-redux';
+import { useSelector } from 'react-redux';
 import { useTranslation } from 'react-i18next';
 import { useNavigate } from 'react-router-dom';
 
-import { Button, Drawer, Form, Input, InputRef, Select, Typography } from 'antd';
+import { Button, Drawer, Form, Input, InputRef, Typography, Card, Row, Col, Tag, Tooltip, Spin, Alert } from '@/shared/antd-imports';
 import TemplateDrawer from '../common/template-drawer/template-drawer';
 
 import { RootState } from '@/app/store';
@@ -13,23 +13,56 @@ import { sanitizeInput } from '@/utils/sanitizeInput';
 import { projectTemplatesApiService } from '@/api/project-templates/project-templates.api.service';
 import logger from '@/utils/errorLogger';
 
-import { IAccountSetupRequest } from '@/types/project-templates/project-templates.types';
+import { IAccountSetupRequest, IWorklenzTemplate, IProjectTemplate } from '@/types/project-templates/project-templates.types';
 
 import { evt_account_setup_template_complete } from '@/shared/worklenz-analytics-events';
 import { useMixpanelTracking } from '@/hooks/useMixpanelTracking';
 import { createPortal } from 'react-dom';
+import { useAppDispatch } from '@/hooks/useAppDispatch';
+import { verifyAuthentication } from '@/features/auth/authSlice';
+import { setUser } from '@/features/user/userSlice';
+import { setSession } from '@/utils/session-helper';
+import { IAuthorizeResponse } from '@/types/auth/login.types';
 
-const { Title } = Typography;
+const { Title, Paragraph, Text } = Typography;
 
 interface Props {
   onEnter: () => void;
   styles: any;
   isDarkMode: boolean;
+  token?: any;
 }
 
-export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = false }) => {
+// Default icon mapping for templates (fallback if no image_url)
+const getTemplateIcon = (name?: string) => {
+  if (!name) return '📁';
+  const lowercaseName = name.toLowerCase();
+  if (lowercaseName.includes('software') || lowercaseName.includes('development')) return '💻';
+  if (lowercaseName.includes('marketing') || lowercaseName.includes('campaign')) return '📢';
+  if (lowercaseName.includes('construction') || lowercaseName.includes('building')) return '🏗️';
+  if (lowercaseName.includes('startup') || lowercaseName.includes('launch')) return '🚀';
+  if (lowercaseName.includes('design') || lowercaseName.includes('creative')) return '🎨';
+  if (lowercaseName.includes('education') || lowercaseName.includes('learning')) return '📚';
+  if (lowercaseName.includes('event') || lowercaseName.includes('planning')) return '📅';
+  if (lowercaseName.includes('retail') || lowercaseName.includes('sales')) return '🛍️';
+  return '📁';
+};
+
+const getProjectSuggestions = (orgType?: string) => {
+  const suggestions: Record<string, string[]> = {
+    'freelancer': ['Client Website', 'Logo Design', 'Content Writing', 'App Development'],
+    'startup': ['MVP Development', 'Product Launch', 'Marketing Campaign', 'Investor Pitch'],
+    'small_medium_business': ['Q1 Sales Initiative', 'Website Redesign', 'Process Improvement', 'Team Training'],
+    'agency': ['Client Campaign', 'Brand Strategy', 'Website Project', 'Creative Brief'],
+    'enterprise': ['Digital Transformation', 'System Migration', 'Annual Planning', 'Department Initiative'],
+    'other': ['New Project', 'Team Initiative', 'Q1 Goals', 'Special Project']
+  };
+  return suggestions[orgType || 'other'] || suggestions['other'];
+};
+
+export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = false, token }) => {
   const { t } = useTranslation('account-setup');
-  const dispatch = useDispatch();
+  const dispatch = useAppDispatch();
   const navigate = useNavigate();
   const { trackMixpanelEvent } = useMixpanelTracking();
 
@@ -37,13 +70,58 @@ export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = fal
 
   useEffect(() => {
     setTimeout(() => inputRef.current?.focus(), 200);
+    fetchTemplates();
   }, []);
 
-  const { projectName, templateId, organizationName } = useSelector(
+  const fetchTemplates = async () => {
+    try {
+      setLoadingTemplates(true);
+      setTemplateError(null);
+      
+      // Fetch list of available templates
+      const templatesResponse = await projectTemplatesApiService.getWorklenzTemplates();
+      
+      if (templatesResponse.done && templatesResponse.body) {
+        // Fetch detailed information for first 4 templates for preview
+        const templateDetails = await Promise.all(
+          templatesResponse.body.slice(0, 4).map(async (template) => {
+            if (template.id) {
+              try {
+                const detailResponse = await projectTemplatesApiService.getByTemplateId(template.id);
+                return detailResponse.done ? detailResponse.body : null;
+              } catch (error) {
+                logger.error(`Failed to fetch template details for ${template.id}`, error);
+                return null;
+              }
+            }
+            return null;
+          })
+        );
+        
+        // Filter out null results and set templates
+        const validTemplates = templateDetails.filter((template): template is IProjectTemplate => template !== null);
+        setTemplates(validTemplates);
+      }
+    } catch (error) {
+      logger.error('Failed to fetch templates', error);
+      setTemplateError('Failed to load templates');
+    } finally {
+      setLoadingTemplates(false);
+    }
+  };
+
+
+  const { projectName, templateId, organizationName, surveyData } = useSelector(
     (state: RootState) => state.accountSetupReducer
   );
   const [open, setOpen] = useState(false);
   const [creatingFromTemplate, setCreatingFromTemplate] = useState(false);
+  const [selectedTemplate, setSelectedTemplate] = useState<string | null>(templateId || null);
+  const [templates, setTemplates] = useState<IProjectTemplate[]>([]);
+  const [loadingTemplates, setLoadingTemplates] = useState(true);
+  const [templateError, setTemplateError] = useState<string | null>(null);
+
+  const projectSuggestions = getProjectSuggestions(surveyData.organization_type);
 
   const handleTemplateSelected = (templateId: string) => {
     if (!templateId) return;
@@ -69,6 +147,15 @@ export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = fal
       if (res.done && res.body.id) {
         toggleTemplateSelector(false);
         trackMixpanelEvent(evt_account_setup_template_complete);
+        try {
+          const authResponse = await dispatch(verifyAuthentication()).unwrap() as IAuthorizeResponse;
+          if (authResponse?.authenticated && authResponse?.user) {
+            setSession(authResponse.user);
+            dispatch(setUser(authResponse.user));
+          }
+        } catch (error) {
+          logger.error('Failed to refresh user session after template setup completion', error);
+        }
         navigate(`/worklenz/projects/${res.body.id}?tab=tasks-list&pinned_tab=tasks-list`);
       }
     } catch (error) {
@@ -77,8 +164,7 @@ export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = fal
   };
 
   const onPressEnter = () => {
-    if (!projectName.trim()) return;
-    onEnter();
+    if (projectName.trim()) onEnter();
   };
 
   const handleProjectNameChange = (e: React.ChangeEvent<HTMLInputElement>) => {
@@ -86,43 +172,205 @@ export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = fal
     dispatch(setProjectName(sanitizedValue));
   };
 
+  const handleProjectNameFocus = () => {
+    if (templateId) {
+      dispatch(setTemplateId(null));
+      setSelectedTemplate(null);
+    }
+  };
+
+  const handleSuggestionClick = (suggestion: string) => {
+    dispatch(setProjectName(suggestion));
+    inputRef.current?.focus();
+  };
+
   return (
-    <div>
-      <Form className="step-form" style={styles.form}>
-        <Form.Item>
-          <Title level={2} style={{ marginBottom: '1rem' }}>
-            {t('projectStepTitle')}
-          </Title>
-        </Form.Item>
-        <Form.Item
-          layout="vertical"
-          rules={[{ required: true }]}
-          label={<span style={styles.label}>{t('projectStepLabel')}</span>}
-        >
-          <Input
-            placeholder={t('projectStepPlaceholder')}
-            value={projectName}
-            onChange={handleProjectNameChange}
-            onPressEnter={onPressEnter}
-            ref={inputRef}
-          />
-        </Form.Item>
-      </Form>
-      <div style={{ position: 'relative' }}>
-        <Title level={4} className={isDarkMode ? 'vert-text-dark' : 'vert-text'}>
-          {t('or')}
+    <div className="w-full project-step">
+      {/* Header */}
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
+          {t('projectStepHeader')}
         </Title>
-        <div className={isDarkMode ? 'vert-line-dark' : 'vert-line'} />
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('projectStepSubheader')}
+        </Paragraph>
       </div>
 
-      <div className="flex justify-center">
-        <Button onClick={() => toggleTemplateSelector(true)} type="primary">
-          {t('templateButton')}
-        </Button>
+      {/* Project Name Section */}
+      <div className="mb-8">
+        <Card 
+          className={`border-2 hover:shadow-md transition-all duration-200 ${
+            templateId ? 'opacity-50' : ''
+          }`}
+          style={{ 
+            borderColor: templateId ? token?.colorBorder : token?.colorPrimary,
+            backgroundColor: token?.colorBgContainer 
+          }}
+        >
+          <div className="mb-4">
+            <div className="flex items-center justify-between">
+              <Text strong className="text-lg" style={{ color: token?.colorText }}>
+                {t('startFromScratch')}
+              </Text>
+              {templateId && (
+                <Text type="secondary" className="text-sm">
+                  {t('templateSelected')}
+                </Text>
+              )}
+            </div>
+          </div>
+          
+          <Form.Item 
+            className="mb-4"
+            label={<span className="font-medium" style={{ color: token?.colorText }}>{t('projectStepLabel')}</span>}
+          >
+            <Input
+              size="large"
+              placeholder={projectSuggestions[0] || t('projectStepPlaceholder')}
+              value={projectName}
+              onChange={handleProjectNameChange}
+              onPressEnter={onPressEnter}
+              onFocus={handleProjectNameFocus}
+              ref={inputRef}
+              className="text-base"
+              style={{ backgroundColor: token?.colorBgContainer, borderColor: token?.colorBorder, color: token?.colorText }}
+            />
+          </Form.Item>
+
+          <div>
+            <Text type="secondary" className="text-sm">{t('quickSuggestions')}</Text>
+            <div className="mt-2 flex flex-wrap gap-2">
+              {projectSuggestions.map((suggestion, index) => (
+                <button key={index} onClick={() => handleSuggestionClick(suggestion)} className="px-3 py-1 rounded-full text-sm border project-suggestion-button" style={{ backgroundColor: token?.colorBgContainer, borderColor: token?.colorBorder, color: token?.colorTextSecondary }}>
+                  {suggestion}
+                </button>
+              ))}
+            </div>
+          </div>
+        </Card>
       </div>
+
+      <div className="relative my-8">
+        <div className="absolute inset-0 flex items-center" style={{ color: token?.colorTextQuaternary }}>
+          <div className="w-full border-t" style={{ borderColor: token?.colorBorder }}></div>
+        </div>
+        <div className="relative flex justify-center">
+          <span className="px-4 text-sm font-medium" style={{ backgroundColor: token?.colorBgLayout, color: token?.colorTextSecondary }}>{t('orText')}</span>
+        </div>
+      </div>
+
+      <div>
+        <div className="text-center mb-6">
+          <Title level={4} className="mb-2" style={{ color: token?.colorText }}>{t('startWithTemplate')}</Title>
+          <Text type="secondary">
+            {t('templateHeadStart')}
+          </Text>
+        </div>
+
+        {/* Template Preview Cards */}
+        <div className="mb-6">
+          {loadingTemplates ? (
+            <div className="text-center py-12">
+              <Spin size="large" />
+              <div className="mt-4">
+                <Text type="secondary">Loading templates...</Text>
+              </div>
+            </div>
+          ) : templateError ? (
+            <Alert
+              message="Failed to load templates"
+              description={templateError}
+              type="error"
+              showIcon
+              action={
+                <Button size="small" onClick={fetchTemplates}>
+                  Retry
+                </Button>
+              }
+            />
+          ) : (
+            <Row gutter={[16, 16]}>
+              {templates.map((template) => (
+                <Col xs={24} sm={12} key={template.id}>
+                  <Card
+                    hoverable
+                    className={`h-full template-preview-card ${
+                      selectedTemplate === template.id ? 'selected border-2' : ''
+                    }`}
+                    style={{ 
+                      borderColor: selectedTemplate === template.id ? token?.colorPrimary : token?.colorBorder,
+                      backgroundColor: token?.colorBgContainer
+                    }}
+                    onClick={() => {
+                      setSelectedTemplate(template.id || null);
+                      dispatch(setTemplateId(template.id || ''));
+                    }}
+                  >
+                    <div className="flex items-start space-x-3">
+                      {template.image_url ? (
+                        <img 
+                          src={template.image_url} 
+                          alt={template.name} 
+                          className="w-12 h-12 object-cover rounded"
+                          onError={(e) => {
+                            // Fallback to icon if image fails to load
+                            e.currentTarget.style.display = 'none';
+                            if (e.currentTarget.nextSibling) {
+                              (e.currentTarget.nextSibling as HTMLElement).style.display = 'block';
+                            }
+                          }}
+                        />
+                      ) : null}
+                      <span 
+                        className="text-3xl" 
+                        style={{ display: template.image_url ? 'none' : 'block' }}
+                      >
+                        {getTemplateIcon(template.name)}
+                      </span>
+                      <div className="flex-1">
+                        <Text strong className="block mb-2" style={{ color: token?.colorText }}>
+                          {template.name || 'Untitled Template'}
+                        </Text>
+                        <div className="flex flex-wrap gap-1">
+                          {template.phases?.slice(0, 3).map((phase, index) => (
+                            <Tag key={index} color={phase.color_code || 'blue'} className="text-xs">
+                              {phase.name}
+                            </Tag>
+                          ))}
+                          {(template.phases?.length || 0) > 3 && (
+                            <Tag className="text-xs">+{(template.phases?.length || 0) - 3} more</Tag>
+                          )}
+                        </div>
+                      </div>
+                    </div>
+                  </Card>
+                </Col>
+              ))}
+            </Row>
+          )}
+        </div>
+
+        <div className="text-center">
+          <Button type="primary" size="large" icon={<span className="mr-2">🎨</span>} onClick={() => toggleTemplateSelector(true)} className="min-w-[200px]">{t('browseAllTemplates')}</Button>
+          <div className="mt-2">
+            <Text type="secondary" className="text-sm">{t('templatesAvailable')}</Text>
+          </div>
+        </div>
+      </div>
+
+      {/* Template Drawer */}
       {createPortal(
         <Drawer
-          title={t('templateDrawerTitle')}
+          title={
+            <div>
+              <Title level={4} style={{ marginBottom: 0 }}>
+                {t('templateDrawerTitle')}
+              </Title>
+              <Text type="secondary">
+                {t('chooseTemplate')}
+              </Text>
+            </div>
+          }
           width={1000}
           onClose={() => toggleTemplateSelector(false)}
           open={open}
@@ -135,11 +383,13 @@ export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = fal
                 type="primary"
                 onClick={() => createFromTemplate()}
                 loading={creatingFromTemplate}
+                disabled={!templateId}
               >
-                {t('create')}
+{t('createProject')}
               </Button>
             </div>
           }
+          style={{ backgroundColor: token?.colorBgLayout }}
         >
           <TemplateDrawer
             showBothTabs={false}
@@ -152,4 +402,4 @@ export const ProjectStep: React.FC<Props> = ({ onEnter, styles, isDarkMode = fal
       )}
     </div>
   );
-};
+};

worklenz-frontend/src/components/account-setup/survey-step.tsx

@@ -0,0 +1,374 @@
+import React from 'react';
+import { Form, Input, Typography, Button, Progress, Space } from '@/shared/antd-imports';
+import { useDispatch, useSelector } from 'react-redux';
+import { useTranslation } from 'react-i18next';
+import { setSurveyData, setSurveySubStep } from '@/features/account-setup/account-setup.slice';
+import { RootState } from '@/app/store';
+import { 
+  OrganizationType, 
+  UserRole, 
+  UseCase, 
+  HowHeardAbout,
+  IAccountSetupSurveyData 
+} from '@/types/account-setup/survey.types';
+
+const { Title, Paragraph } = Typography;
+const { TextArea } = Input;
+
+interface Props {
+  onEnter: () => void;
+  styles: any;
+  isDarkMode: boolean;
+  token?: any;
+  isModal?: boolean; // New prop to indicate if used in modal context
+}
+
+interface SurveyPageProps {
+  styles: any;
+  isDarkMode: boolean;
+  token?: any;
+  surveyData: IAccountSetupSurveyData;
+  handleSurveyDataChange: (field: keyof IAccountSetupSurveyData, value: any) => void;
+  handleUseCaseToggle?: (value: UseCase) => void;
+  isModal?: boolean;
+}
+
+// Page 1: About You
+const AboutYouPage: React.FC<SurveyPageProps> = ({ styles, token, surveyData, handleSurveyDataChange }) => {
+  const { t } = useTranslation('account-setup');
+  
+  const organizationTypeOptions: { value: OrganizationType; label: string; icon?: string }[] = [
+    { value: 'freelancer', label: t('organizationTypeFreelancer'), icon: '👤' },
+    { value: 'startup', label: t('organizationTypeStartup'), icon: '🚀' },
+    { value: 'small_medium_business', label: t('organizationTypeSmallMediumBusiness'), icon: '🏢' },
+    { value: 'agency', label: t('organizationTypeAgency'), icon: '🎯' },
+    { value: 'enterprise', label: t('organizationTypeEnterprise'), icon: '🏛️' },
+    { value: 'other', label: t('organizationTypeOther'), icon: '📋' },
+  ];
+
+  const userRoleOptions: { value: UserRole; label: string; icon?: string }[] = [
+    { value: 'founder_ceo', label: t('userRoleFounderCeo'), icon: '👔' },
+    { value: 'project_manager', label: t('userRoleProjectManager'), icon: '📊' },
+    { value: 'software_developer', label: t('userRoleSoftwareDeveloper'), icon: '💻' },
+    { value: 'designer', label: t('userRoleDesigner'), icon: '🎨' },
+    { value: 'operations', label: t('userRoleOperations'), icon: '⚙️' },
+    { value: 'other', label: t('userRoleOther'), icon: '✋' },
+  ];
+
+  return (
+    <div className="w-full">
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
+          {t('aboutYouStepTitle')}
+        </Title>
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('aboutYouStepDescription')}
+        </Paragraph>
+      </div>
+
+      {/* Organization Type */}
+      <Form.Item className="mb-8">
+        <label className="block font-medium text-base mb-4" style={{ color: token?.colorText }}>
+          {t('orgTypeQuestion')}
+        </label>
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-1">
+          {organizationTypeOptions.map((option) => {
+            const isSelected = surveyData.organization_type === option.value;
+            return (
+              <button
+                key={option.value}
+                onClick={() => handleSurveyDataChange('organization_type', option.value)}
+                className={`p-2 rounded border transition-all duration-200 text-left hover:shadow-sm ${isSelected ? 'border-blue-500 bg-blue-50 dark:bg-blue-900/20' : 'border-gray-200 dark:border-gray-700 hover:border-gray-300 dark:hover:border-gray-600'}`}
+                style={{
+                  backgroundColor: isSelected ? undefined : token?.colorBgContainer,
+                  borderColor: isSelected ? undefined : token?.colorBorder,
+                }}
+              >
+                <div className="flex items-center space-x-2">
+                  <div className={`w-3 h-3 rounded-full border flex items-center justify-center ${isSelected ? 'border-blue-500 bg-blue-500' : 'border-gray-300 dark:border-gray-600'}`}>
+                    {isSelected && <div className="w-1.5 h-1.5 bg-white rounded-full"></div>}
+                  </div>
+                  <span className="text-base">{option.icon}</span>
+                  <span 
+                    className={`font-medium text-xs ${isSelected ? 'text-blue-600 dark:text-blue-400' : ''}`}
+                    style={{ color: isSelected ? undefined : token?.colorText }}
+                  >
+                    {option.label}
+                  </span>
+                </div>
+              </button>
+            );
+          })}
+        </div>
+      </Form.Item>
+
+      {/* User Role */}
+      <Form.Item className="mb-4">
+        <label className="block font-medium text-base mb-4" style={{ color: token?.colorText }}>
+          {t('userRoleQuestion')}
+        </label>
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-1">
+          {userRoleOptions.map((option) => {
+            const isSelected = surveyData.user_role === option.value;
+            return (
+              <button
+                key={option.value}
+                onClick={() => handleSurveyDataChange('user_role', option.value)}
+                className={`p-2 rounded border transition-all duration-200 text-left hover:shadow-sm ${isSelected ? 'border-blue-500 bg-blue-50 dark:bg-blue-900/20' : 'border-gray-200 dark:border-gray-700 hover:border-gray-300 dark:hover:border-gray-600'}`}
+                style={{
+                  backgroundColor: isSelected ? undefined : token?.colorBgContainer,
+                  borderColor: isSelected ? undefined : token?.colorBorder,
+                }}
+              >
+                <div className="flex items-center space-x-2">
+                  <div className={`w-3 h-3 rounded-full border flex items-center justify-center ${isSelected ? 'border-blue-500 bg-blue-500' : 'border-gray-300 dark:border-gray-600'}`}>
+                    {isSelected && <div className="w-1.5 h-1.5 bg-white rounded-full"></div>}
+                  </div>
+                  <span className="text-base">{option.icon}</span>
+                  <span 
+                    className={`font-medium text-xs ${isSelected ? 'text-blue-600 dark:text-blue-400' : ''}`}
+                    style={{ color: isSelected ? undefined : token?.colorText }}
+                  >
+                    {option.label}
+                  </span>
+                </div>
+              </button>
+            );
+          })}
+        </div>
+      </Form.Item>
+    </div>
+  );
+};
+
+// Page 2: Your Needs
+const YourNeedsPage: React.FC<SurveyPageProps> = ({ styles, token, surveyData, handleSurveyDataChange, handleUseCaseToggle }) => {
+  const { t } = useTranslation('account-setup');
+  
+  const useCaseOptions: { value: UseCase; label: string; description: string }[] = [
+    { value: 'task_management', label: t('mainUseCasesTaskManagement'), description: 'Organize and track tasks' },
+    { value: 'team_collaboration', label: t('mainUseCasesTeamCollaboration'), description: 'Work together seamlessly' },
+    { value: 'resource_planning', label: t('mainUseCasesResourcePlanning'), description: 'Manage time and resources' },
+    { value: 'client_communication', label: t('mainUseCasesClientCommunication'), description: 'Stay connected with clients' },
+    { value: 'time_tracking', label: t('mainUseCasesTimeTracking'), description: 'Monitor project hours' },
+    { value: 'other', label: t('mainUseCasesOther'), description: 'Something else' },
+  ];
+
+  const onUseCaseClick = (value: UseCase) => {
+    if (handleUseCaseToggle) {
+      handleUseCaseToggle(value);
+    } else {
+      const currentUseCases = surveyData.main_use_cases || [];
+      const isSelected = currentUseCases.includes(value);
+      const newUseCases = isSelected ? currentUseCases.filter(useCase => useCase !== value) : [...currentUseCases, value];
+      handleSurveyDataChange('main_use_cases', newUseCases);
+    }
+  };
+
+  return (
+    <div className="w-full">
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
+          {t('yourNeedsStepTitle')}
+        </Title>
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('yourNeedsStepDescription')}
+        </Paragraph>
+      </div>
+
+      {/* Main Use Cases */}
+      <Form.Item className="mb-8">
+        <label className="block font-medium text-base mb-4" style={{ color: token?.colorText }}>
+          {t('yourNeedsQuestion')}
+        </label>
+        <div className="grid grid-cols-1 gap-1">
+          {useCaseOptions.map((option) => {
+            const isSelected = (surveyData.main_use_cases || []).includes(option.value);
+            return (
+              <button
+                key={option.value}
+                onClick={() => onUseCaseClick(option.value)}
+                className={`p-2 rounded border transition-all duration-200 text-left hover:shadow-sm ${isSelected ? 'border-blue-500 bg-blue-50 dark:bg-blue-900/20' : 'border-gray-200 dark:border-gray-700 hover:border-gray-300 dark:hover:border-gray-600'}`}
+                style={{
+                  backgroundColor: isSelected ? undefined : token?.colorBgContainer,
+                  borderColor: isSelected ? undefined : token?.colorBorder,
+                }}
+              >
+                <div className="flex items-center space-x-2">
+                  <div className={`w-3 h-3 rounded border flex items-center justify-center ${isSelected ? 'border-blue-500 bg-blue-500' : 'border-gray-300 dark:border-gray-600'}`}>
+                    {isSelected && (
+                      <svg width="10" height="10" fill="white" viewBox="0 0 20 20">
+                        <path fillRule="evenodd" d="M16.707 5.293a1 1 0 010 1.414l-8 8a1 1 0 01-1.414 0l-4-4a1 1 0 011.414-1.414L8 12.586l7.293-7.293a1 1 0 011.414 0z" clipRule="evenodd" />
+                      </svg>
+                    )}
+                  </div>
+                  <div className="flex-1">
+                    <span className={`font-medium text-xs ${isSelected ? 'text-blue-600 dark:text-blue-400' : ''}`} style={{ color: isSelected ? undefined : token?.colorText }}>{option.label}</span>
+                    <span className="text-xs ml-2" style={{ color: token?.colorTextSecondary }}>- {option.description}</span>
+                  </div>
+                </div>
+              </button>
+            );
+          })}
+        </div>
+        {surveyData.main_use_cases && surveyData.main_use_cases.length > 0 && (
+          <p className="mt-3 text-sm" style={{ color: token?.colorTextSecondary }}>
+            {surveyData.main_use_cases.length} {t('selected')}
+          </p>
+        )}
+      </Form.Item>
+
+      {/* Previous Tools */}
+      <Form.Item className="mb-4">
+        <label className="block font-medium text-base mb-2" style={{ color: token?.colorText }}>
+          {t('previousToolsLabel')}
+        </label>
+        <TextArea
+          placeholder="e.g., Asana, Trello, Jira, Monday.com, etc."
+          value={surveyData.previous_tools || ''}
+          onChange={(e) => handleSurveyDataChange('previous_tools', e.target.value)}
+          autoSize={{ minRows: 3, maxRows: 5 }}
+          className="text-base"
+          style={{ backgroundColor: token?.colorBgContainer, borderColor: token?.colorBorder, color: token?.colorText }}
+        />
+      </Form.Item>
+    </div>
+  );
+};
+
+// Page 3: Discovery
+const DiscoveryPage: React.FC<SurveyPageProps> = ({ styles, token, surveyData, handleSurveyDataChange, isModal }) => {
+  const { t } = useTranslation('account-setup');
+  
+  const howHeardAboutOptions: { value: HowHeardAbout; label: string; icon: string }[] = [
+    { value: 'google_search', label: t('howHeardAboutGoogleSearch'), icon: '🔍' },
+    { value: 'twitter', label: t('howHeardAboutTwitter'), icon: '🐦' },
+    { value: 'linkedin', label: t('howHeardAboutLinkedin'), icon: '💼' },
+    { value: 'friend_colleague', label: t('howHeardAboutFriendColleague'), icon: '👥' },
+    { value: 'blog_article', label: t('howHeardAboutBlogArticle'), icon: '📰' },
+    { value: 'other', label: t('howHeardAboutOther'), icon: '💡' },
+  ];
+
+  return (
+    <div className="w-full">
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
+          {t('discoveryTitle')}
+        </Title>
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('discoveryDescription')}
+        </Paragraph>
+      </div>
+
+      {/* How Heard About */}
+      <Form.Item className="mb-8">
+        <label className="block font-medium text-base mb-4" style={{ color: token?.colorText }}>
+          {t('discoveryQuestion')}
+        </label>
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-1">
+          {howHeardAboutOptions.map((option) => {
+            const isSelected = surveyData.how_heard_about === option.value;
+            return (
+              <button
+                key={option.value}
+                onClick={() => handleSurveyDataChange('how_heard_about', option.value)}
+                className={`p-2 rounded border transition-all duration-200 hover:shadow-sm ${isSelected ? 'border-blue-500 bg-blue-50 dark:bg-blue-900/20' : 'border-gray-200 dark:border-gray-700 hover:border-gray-300 dark:hover:border-gray-600'}`}
+                style={{
+                  backgroundColor: isSelected ? undefined : token?.colorBgContainer,
+                  borderColor: isSelected ? undefined : token?.colorBorder,
+                }}
+              >
+                <div className="flex items-center space-x-2">
+                  <div className={`w-3 h-3 rounded-full border flex items-center justify-center ${isSelected ? 'border-blue-500 bg-blue-500' : 'border-gray-300 dark:border-gray-600'}`}>
+                    {isSelected && <div className="w-1.5 h-1.5 bg-white rounded-full"></div>}
+                  </div>
+                  <span className="text-base">{option.icon}</span>
+                  <span className={`font-medium text-xs ${isSelected ? 'text-blue-600 dark:text-blue-400' : ''}`} style={{ color: isSelected ? undefined : token?.colorText }}>{option.label}</span>
+                </div>
+              </button>
+            );
+          })}
+        </div>
+      </Form.Item>
+
+      <div className="mt-12 p-1.5 rounded-lg text-center" style={{ backgroundColor: token?.colorSuccessBg, borderColor: token?.colorSuccessBorder, border: '1px solid' }}>
+        <div className="text-4xl mb-3">🎉</div>
+        <Title level={4} style={{ color: token?.colorText, marginBottom: 8 }}>
+          {isModal ? t('surveyCompleteTitle') : t('allSetTitle')}
+        </Title>
+        <Paragraph style={{ color: token?.colorTextSecondary, marginBottom: 0 }}>
+          {isModal ? t('surveyCompleteDescription') : t('allSetDescription')}
+        </Paragraph>
+      </div>
+    </div>
+  );
+};
+
+export const SurveyStep: React.FC<Props> = ({ onEnter, styles, isDarkMode, token, isModal = false }) => {
+  const { t } = useTranslation('account-setup');
+  const dispatch = useDispatch();
+  const { surveyData, surveySubStep } = useSelector((state: RootState) => state.accountSetupReducer);
+
+  const handleSurveyDataChange = (field: keyof IAccountSetupSurveyData, value: any) => {
+    dispatch(setSurveyData({ [field]: value }));
+  };
+
+  React.useEffect(() => {
+    const handleKeyPress = (e: KeyboardEvent) => {
+      if (e.key === 'Enter') {
+        const isValid = (surveySubStep === 0 && surveyData.organization_type && surveyData.user_role) || (surveySubStep === 1 && surveyData.main_use_cases && surveyData.main_use_cases.length > 0) || (surveySubStep === 2 && surveyData.how_heard_about);
+        if (isValid && surveySubStep < 2) {
+          dispatch(setSurveySubStep(surveySubStep + 1));
+        } else if (isValid && surveySubStep === 2) {
+          onEnter();
+        }
+      }
+    };
+    window.addEventListener('keypress', handleKeyPress);
+    return () => window.removeEventListener('keypress', handleKeyPress);
+  }, [surveySubStep, surveyData, dispatch, onEnter]);
+
+  const handleUseCaseToggle = (value: UseCase) => {
+    const currentUseCases = surveyData.main_use_cases || [];
+    const isSelected = currentUseCases.includes(value);
+    const newUseCases = isSelected ? currentUseCases.filter(useCase => useCase !== value) : [...currentUseCases, value];
+    handleSurveyDataChange('main_use_cases', newUseCases);
+  };
+
+  const getSubStepTitle = () => {
+    switch (surveySubStep) {
+      case 0: return t('aboutYouStepName');
+      case 1: return t('yourNeedsStepName');
+      case 2: return t('discoveryStepName');
+      default: return '';
+    }
+  };
+
+  const surveyPages = [
+    <AboutYouPage key="about-you" styles={styles} isDarkMode={isDarkMode} token={token} surveyData={surveyData} handleSurveyDataChange={handleSurveyDataChange} isModal={isModal} />,
+    <YourNeedsPage key="your-needs" styles={styles} isDarkMode={isDarkMode} token={token} surveyData={surveyData} handleSurveyDataChange={handleSurveyDataChange} handleUseCaseToggle={handleUseCaseToggle} isModal={isModal} />,
+    <DiscoveryPage key="discovery" styles={styles} isDarkMode={isDarkMode} token={token} surveyData={surveyData} handleSurveyDataChange={handleSurveyDataChange} isModal={isModal} />
+  ];
+
+  React.useEffect(() => {
+    dispatch(setSurveySubStep(0));
+  }, []);
+
+  return (
+    <div className="w-full">
+      {/* Progress Indicator */}
+      <div className="mb-8">
+        <div className="flex items-center justify-between mb-2">
+          <span className="text-sm font-medium" style={{ color: token?.colorTextSecondary }}>Step {surveySubStep + 1} of 3: {getSubStepTitle()}</span>
+          <span className="text-sm" style={{ color: token?.colorTextSecondary }}>{Math.round(((surveySubStep + 1) / 3) * 100)}%</span>
+        </div>
+        <Progress percent={Math.round(((surveySubStep + 1) / 3) * 100)} showInfo={false} strokeColor={token?.colorPrimary} className="mb-0" />
+      </div>
+
+      {/* Current Page Content */}
+      <div className="min-h-[400px] flex flex-col survey-page-transition" key={surveySubStep}>
+        {surveyPages[surveySubStep]}
+      </div>
+    </div>
+  );
+};

worklenz-frontend/src/components/account-setup/tasks-step.tsx

@@ -1,134 +1,130 @@
-import React, { useEffect, useRef } from 'react';
-import { Form, Input, Button, Typography, List, InputRef } from 'antd';
-import { PlusOutlined, DeleteOutlined, CloseCircleOutlined } from '@ant-design/icons';
+import React, { useEffect, useRef, useState } from 'react';
+import { Input, Button, Typography, Card } from '@/shared/antd-imports';
+import { PlusOutlined, DeleteOutlined, CloseCircleOutlined, CheckCircleOutlined } from '@/shared/antd-imports';
 import { useDispatch, useSelector } from 'react-redux';
 import { useTranslation } from 'react-i18next';
 import { RootState } from '@/app/store';
 import { setTasks } from '@/features/account-setup/account-setup.slice';
 import { sanitizeInput } from '@/utils/sanitizeInput';
 
-const { Title } = Typography;
+const { Title, Paragraph, Text } = Typography;
 
 interface Props {
   onEnter: () => void;
   styles: any;
   isDarkMode: boolean;
+  token?: any;
 }
 
-export const TasksStep: React.FC<Props> = ({ onEnter, styles, isDarkMode }) => {
+
+export const TasksStep: React.FC<Props> = ({ onEnter, styles, isDarkMode, token }) => {
   const { t } = useTranslation('account-setup');
   const dispatch = useDispatch();
-  const { tasks, projectName } = useSelector((state: RootState) => state.accountSetupReducer);
-  const inputRefs = useRef<(InputRef | null)[]>([]);
+  const { tasks, projectName, surveyData } = useSelector((state: RootState) => state.accountSetupReducer);
+  const inputRefs = useRef<(HTMLInputElement | null)[]>([]);
+  const [showSuggestions, setShowSuggestions] = useState(false);
+  const [focusedIndex, setFocusedIndex] = useState<number | null>(null);
 
   const addTask = () => {
-    if (tasks.length == 5) return;
-
+    if (tasks.length >= 5) return;
     const newId = tasks.length > 0 ? Math.max(...tasks.map(t => t.id)) + 1 : 0;
     dispatch(setTasks([...tasks, { id: newId, value: '' }]));
-    setTimeout(() => {
-      inputRefs.current[newId]?.focus();
-    }, 0);
+    setTimeout(() => inputRefs.current[tasks.length]?.focus(), 100);
   };
 
   const removeTask = (id: number) => {
-    if (tasks.length > 1) {
-      dispatch(setTasks(tasks.filter(task => task.id !== id)));
-    }
+    if (tasks.length > 1) dispatch(setTasks(tasks.filter(task => task.id !== id)));
   };
 
   const updateTask = (id: number, value: string) => {
     const sanitizedValue = sanitizeInput(value);
-    dispatch(
-      setTasks(tasks.map(task => (task.id === id ? { ...task, value: sanitizedValue } : task)))
-    );
+    dispatch(setTasks(tasks.map(task => (task.id === id ? { ...task, value: sanitizedValue } : task))));
   };
 
-  const handleKeyPress = (e: React.KeyboardEvent<HTMLInputElement>) => {
-    const input = e.currentTarget as HTMLInputElement;
-    if (!input.value.trim()) return;
-    e.preventDefault();
-    addTask();
+  const handleKeyPress = (e: React.KeyboardEvent<HTMLInputElement>, index: number) => {
+    if (e.key === 'Enter') {
+      const input = e.currentTarget as HTMLInputElement;
+      if (input.value.trim()) {
+        e.preventDefault();
+        if (index === tasks.length - 1 && tasks.length < 5) addTask();
+        else if (index < tasks.length - 1) inputRefs.current[index + 1]?.focus();
+      }
+    }
+  };
+
+  const handleSuggestionClick = (suggestion: string) => {
+    const emptyTaskIndex = tasks.findIndex(task => !task.value.trim());
+    if (emptyTaskIndex !== -1) {
+      updateTask(tasks[emptyTaskIndex].id, suggestion);
+    } else if (tasks.length < 5) {
+      const newId = tasks.length > 0 ? Math.max(...tasks.map(t => t.id)) + 1 : 0;
+      dispatch(setTasks([...tasks, { id: newId, value: suggestion }]));
+    }
+    setShowSuggestions(false);
   };
 
   useEffect(() => {
     setTimeout(() => inputRefs.current[0]?.focus(), 200);
   }, []);
 
-  // Function to set ref that doesn't return anything (void)
-  const setInputRef = (index: number) => (el: InputRef | null) => {
-    inputRefs.current[index] = el;
-  };
 
   return (
-    <Form
-      className="create-first-task-form"
-      style={{
-        minHeight: '300px',
-        width: '600px',
-        paddingBottom: '1rem',
-        marginBottom: '3rem',
-        marginTop: '3rem',
-        display: 'flex',
-        flexDirection: 'column',
-      }}
-    >
-      <Form.Item>
-        <Title level={2} style={{ marginBottom: '1rem' }}>
+    <div className="w-full tasks-step">
+      {/* Header */}
+      <div className="text-center mb-8">
+        <Title level={3} className="mb-2" style={{ color: token?.colorText }}>
           {t('tasksStepTitle')}
         </Title>
-      </Form.Item>
-      <Form.Item
-        layout="vertical"
-        rules={[{ required: true }]}
-        label={
-          <span className="font-medium">
-            {t('tasksStepLabel')} "<mark>{projectName}</mark>". {t('maxTasks')}
-          </span>
-        }
-      >
-        <List
-          dataSource={tasks}
-          bordered={false}
-          renderItem={(task, index) => (
-            <List.Item key={task.id}>
-              <div style={{ display: 'flex', width: '600px' }}>
-                <Input
-                  placeholder="Your Task"
-                  value={task.value}
-                  onChange={e => updateTask(task.id, e.target.value)}
-                  onPressEnter={handleKeyPress}
-                  ref={setInputRef(index)}
-                />
-                <Button
-                  className="custom-close-button"
-                  style={{ marginLeft: '48px' }}
-                  type="text"
-                  icon={<CloseCircleOutlined />}
-                  disabled={tasks.length === 1}
-                  onClick={() => removeTask(task.id)}
-                />
+        <Paragraph className="text-base" style={{ color: token?.colorTextSecondary }}>
+          {t('tasksStepDescription', { projectName })}
+        </Paragraph>
+      </div>
+
+
+      {/* Tasks List */}
+      <div className="mb-6">
+        <div className="space-y-4">
+          {tasks.map((task, index) => (
+            <Card 
+              key={task.id}
+              className={`task-item-card transition-all duration-200 ${
+                focusedIndex === index ? 'border-2' : ''
+              }`}
+              style={{ 
+                borderColor: focusedIndex === index ? token?.colorPrimary : token?.colorBorder,
+                backgroundColor: token?.colorBgContainer 
+              }}
+            >
+              <div className="flex items-center space-x-3">
+                <div className="flex items-center justify-center w-8 h-8 rounded-full text-sm font-medium" style={{ backgroundColor: task.value.trim() ? token?.colorSuccess : token?.colorBorderSecondary, color: task.value.trim() ? '#fff' : token?.colorTextSecondary }}>
+                  {task.value.trim() ? <CheckCircleOutlined /> : index + 1}
+                </div>
+                
+                <div className="flex-1">
+                  <Input
+                    placeholder={t('taskPlaceholder', { index: index + 1 })}
+                    value={task.value}
+                    onChange={e => updateTask(task.id, e.target.value)}
+                    onKeyPress={e => handleKeyPress(e, index)}
+                    onFocus={() => setFocusedIndex(index)}
+                    onBlur={() => setFocusedIndex(null)}
+                    ref={(el) => { inputRefs.current[index] = el as any; }}
+                    className="text-base border-0 shadow-none task-input"
+                    style={{ backgroundColor: 'transparent', color: token?.colorText }}
+                  />
+                </div>
+
+                {tasks.length > 1 && <Button type="text" icon={<CloseCircleOutlined />} onClick={() => removeTask(task.id)} className="text-gray-400 hover:text-red-500" style={{ color: token?.colorTextTertiary }} />}
               </div>
-            </List.Item>
-          )}
-        />
-        <Button
-          type="dashed"
-          icon={<PlusOutlined />}
-          onClick={addTask}
-          disabled={tasks.length == 5}
-          style={{ marginTop: '16px' }}
-        >
-          {t('tasksStepAddAnother')}
-        </Button>
-        <div
-          style={{
-            marginTop: '24px',
-            display: 'flex',
-            justifyContent: 'space-between',
-          }}
-        ></div>
-      </Form.Item>
-    </Form>
+            </Card>
+          ))}
+        </div>
+
+        {tasks.length < 5 && (
+          <Button type="dashed" icon={<PlusOutlined />} onClick={addTask} className="w-full mt-4 h-12 text-base" style={{ borderColor: token?.colorBorder, color: token?.colorTextSecondary }}>{t('addAnotherTask', { current: tasks.length, max: 5 })}</Button>
+        )}
+      </div>
+
+    </div>
   );
-};
+};

worklenz-frontend/src/components/add-members-dropdown-v2/add-members-dropdown.tsx

@@ -1,6 +1,6 @@
 import React, { useState } from 'react';
-import { Avatar, Button, Checkbox, Dropdown, Input, Menu, Typography } from 'antd';
-import { UserAddOutlined, UsergroupAddOutlined } from '@ant-design/icons';
+import { Avatar, Button, Checkbox, Dropdown, Input, Menu, Typography } from '@/shared/antd-imports';
+import { UserAddOutlined, UsergroupAddOutlined } from '@/shared/antd-imports';
 import './add-members-dropdown.css';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { AvatarNamesMap } from '@/shared/constants';

worklenz-frontend/src/components/add-members-dropdown/add-members-dropdown.tsx

@@ -1,6 +1,6 @@
 import React, { useState } from 'react';
-import { Avatar, Button, Checkbox, Dropdown, Input, Menu, Typography } from 'antd';
-import { PlusOutlined, UsergroupAddOutlined } from '@ant-design/icons';
+import { Avatar, Button, Checkbox, Dropdown, Input, Menu, Typography } from '@/shared/antd-imports';
+import { PlusOutlined, UsergroupAddOutlined } from '@/shared/antd-imports';
 import './add-members-dropdown.css';
 import { AvatarNamesMap } from '../../shared/constants';
 import { useAppDispatch } from '@/hooks/useAppDispatch';

worklenz-frontend/src/components/admin-center/billing/billing-tables/charges-table.tsx

@@ -2,7 +2,7 @@ import { adminCenterApiService } from '@/api/admin-center/admin-center.api.servi
 import { IBillingCharge, IBillingChargesResponse } from '@/types/admin-center/admin-center.types';
 import logger from '@/utils/errorLogger';
 import { formatDate } from '@/utils/timeUtils';
-import { Table, TableProps, Tag } from 'antd';
+import { Table, TableProps, Tag } from '@/shared/antd-imports';
 import React, { useEffect, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
@@ -27,19 +27,19 @@ const ChargesTable: React.FC = () => {
 
   const columns: TableProps<IBillingCharge>['columns'] = [
     {
-      title: t('description'),
+      title: t('description') as string,
       key: 'name',
       dataIndex: 'name',
     },
     {
-      title: t('billingPeriod'),
+      title: t('billingPeriod') as string,
       key: 'billingPeriod',
       render: record => {
         return `${formatDate(new Date(record.start_date))} - ${formatDate(new Date(record.end_date))}`;
       },
     },
     {
-      title: t('billStatus'),
+      title: t('billStatus') as string,
       key: 'status',
       dataIndex: 'status',
       render: (_, record) => {
@@ -55,7 +55,7 @@ const ChargesTable: React.FC = () => {
       },
     },
     {
-      title: t('perUserValue'),
+      title: t('perUserValue') as string,
       key: 'perUserValue',
       dataIndex: 'perUserValue',
       render: (_, record) => (
@@ -65,12 +65,12 @@ const ChargesTable: React.FC = () => {
       ),
     },
     {
-      title: t('users'),
+      title: t('users') as string,
       key: 'quantity',
       dataIndex: 'quantity',
     },
     {
-      title: t('amount'),
+      title: t('amount') as string,
       key: 'amount',
       dataIndex: 'amount',
       render: (_, record) => (

worklenz-frontend/src/components/admin-center/billing/billing-tables/invoices-table.tsx

@@ -2,8 +2,8 @@ import { adminCenterApiService } from '@/api/admin-center/admin-center.api.servi
 import { IBillingTransaction } from '@/types/admin-center/admin-center.types';
 import logger from '@/utils/errorLogger';
 import { formatDate } from '@/utils/timeUtils';
-import { ContainerOutlined } from '@ant-design/icons';
-import { Button, Table, TableProps, Tag, Tooltip } from 'antd';
+import { ContainerOutlined } from '@/shared/antd-imports';
+import { Button, Table, TableProps, Tag, Tooltip } from '@/shared/antd-imports';
 import React, { useEffect, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 

worklenz-frontend/src/components/admin-center/billing/current-bill.tsx

@@ -1,7 +1,7 @@
-import { Card, Col, Row, Tooltip, Typography } from 'antd';
-import React, { useEffect } from 'react';
+import { Card, Col, Row, Tooltip } from '@/shared/antd-imports';
+import React, { useEffect, useMemo, useCallback } from 'react';
 import './current-bill.css';
-import { InfoCircleTwoTone } from '@ant-design/icons';
+import { InfoCircleTwoTone } from '@/shared/antd-imports';
 import ChargesTable from './billing-tables/charges-table';
 import InvoicesTable from './billing-tables/invoices-table';
 
@@ -20,7 +20,7 @@ import AccountStorage from './account-storage/account-storage';
 import { useAuthService } from '@/hooks/useAuth';
 import { ISUBSCRIPTION_TYPE } from '@/shared/constants';
 
-const CurrentBill: React.FC = () => {
+const CurrentBill: React.FC = React.memo(() => {
   const dispatch = useAppDispatch();
   const { t } = useTranslation('admin-center/current-bill');
   const themeMode = useAppSelector(state => state.themeReducer.mode);
@@ -32,70 +32,90 @@ const CurrentBill: React.FC = () => {
     dispatch(fetchFreePlanSettings());
   }, [dispatch]);
 
-  const titleStyle = {
-    color: themeMode === 'dark' ? '#ffffffd9' : '#000000d9',
-    fontWeight: 500,
-    fontSize: '16px',
-    display: 'flex',
-    gap: '4px',
-  };
-
-  const renderMobileView = () => (
-    <div>
-      <Col span={24}>
-        <CurrentPlanDetails />
-      </Col>
-
-      <Col span={24} style={{ marginTop: '1.5rem' }}>
-        <AccountStorage themeMode={themeMode} />
-      </Col>
-    </div>
+  const titleStyle = useMemo(
+    () => ({
+      color: themeMode === 'dark' ? '#ffffffd9' : '#000000d9',
+      fontWeight: 500,
+      fontSize: '16px',
+      display: 'flex',
+      gap: '4px',
+    }),
+    [themeMode]
   );
 
-  const renderChargesAndInvoices = () => (
-    <>
-      <div style={{ marginTop: '1.5rem' }}>
-        <Card
-          title={
-            <span style={titleStyle}>
-              <span>{t('charges')}</span>
-              <Tooltip title={t('tooltip')}>
-                <InfoCircleTwoTone />
-              </Tooltip>
-            </span>
-          }
-          style={{ marginTop: '16px' }}
-        >
-          <ChargesTable />
-        </Card>
-      </div>
+  const cardStyle = useMemo(() => ({ marginTop: '16px' }), []);
+  const colStyle = useMemo(() => ({ marginTop: '1.5rem' }), []);
+  const tabletColStyle = useMemo(() => ({ paddingRight: '10px' }), []);
+  const tabletColStyleRight = useMemo(() => ({ paddingLeft: '10px' }), []);
 
-      <div style={{ marginTop: '1.5rem' }}>
-        <Card title={<span style={titleStyle}>{t('invoices')}</span>} style={{ marginTop: '16px' }}>
-          <InvoicesTable />
-        </Card>
+  const renderMobileView = useCallback(
+    () => (
+      <div>
+        <Col span={24}>
+          <CurrentPlanDetails />
+        </Col>
+
+        <Col span={24} style={colStyle}>
+          <AccountStorage themeMode={themeMode} />
+        </Col>
       </div>
-    </>
+    ),
+    [colStyle, themeMode]
+  );
+
+  const renderChargesAndInvoices = useCallback(
+    () => (
+      <>
+        <div style={colStyle}>
+          <Card
+            title={
+              <span style={titleStyle}>
+                <span>{t('charges')}</span>
+                <Tooltip title={t('tooltip')}>
+                  <InfoCircleTwoTone />
+                </Tooltip>
+              </span>
+            }
+            style={cardStyle}
+          >
+            <ChargesTable />
+          </Card>
+        </div>
+
+        <div style={colStyle}>
+          <Card title={<span style={titleStyle}>{t('invoices')}</span>} style={cardStyle}>
+            <InvoicesTable />
+          </Card>
+        </div>
+      </>
+    ),
+    [colStyle, titleStyle, cardStyle, t]
+  );
+
+  const shouldShowChargesAndInvoices = useMemo(
+    () => currentSession?.subscription_type === ISUBSCRIPTION_TYPE.PADDLE,
+    [currentSession?.subscription_type]
   );
 
   return (
     <div style={{ width: '100%' }} className="current-billing">
       {isTablet ? (
         <Row>
-          <Col span={16} style={{ paddingRight: '10px' }}>
+          <Col span={16} style={tabletColStyle}>
             <CurrentPlanDetails />
           </Col>
-          <Col span={8} style={{ paddingLeft: '10px' }}>
+          <Col span={8} style={tabletColStyleRight}>
             <AccountStorage themeMode={themeMode} />
           </Col>
         </Row>
       ) : (
         renderMobileView()
       )}
-      {currentSession?.subscription_type === ISUBSCRIPTION_TYPE.PADDLE &&
-        renderChargesAndInvoices()}
+      {shouldShowChargesAndInvoices && renderChargesAndInvoices()}
     </div>
   );
-};
+});
+
+CurrentBill.displayName = 'CurrentBill';
 
 export default CurrentBill;

worklenz-frontend/src/components/admin-center/billing/current-plan-details/current-plan-details.tsx

@@ -1,4 +1,4 @@
-import React, { useState } from 'react';
+import React, { useState, useMemo, useCallback, useEffect } from 'react';
 import { adminCenterApiService } from '@/api/admin-center/admin-center.api.service';
 import {
   evt_billing_pause_plan,
@@ -17,10 +17,9 @@ import {
   Typography,
   Statistic,
   Select,
-  Form,
   Row,
   Col,
-} from 'antd/es';
+} from '@/shared/antd-imports';
 import RedeemCodeDrawer from '../drawers/redeem-code-drawer/redeem-code-drawer';
 import {
   fetchBillingInfo,
@@ -30,7 +29,7 @@ import {
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { useTranslation } from 'react-i18next';
-import { WarningTwoTone, PlusOutlined } from '@ant-design/icons';
+import { WarningTwoTone, PlusOutlined } from '@/shared/antd-imports';
 import { calculateTimeGap } from '@/utils/calculate-time-gap';
 import { formatDate } from '@/utils/timeUtils';
 import UpgradePlansLKR from '../drawers/upgrade-plans-lkr/upgrade-plans-lkr';
@@ -38,6 +37,21 @@ import UpgradePlans from '../drawers/upgrade-plans/upgrade-plans';
 import { ISUBSCRIPTION_TYPE, SUBSCRIPTION_STATUS } from '@/shared/constants';
 import { billingApiService } from '@/api/admin-center/billing.api.service';
 
+type SubscriptionAction = 'pause' | 'resume';
+type SeatOption = { label: string; value: number | string };
+
+const SEAT_COUNT_LIMIT = '100+';
+const BILLING_DELAY_MS = 8000;
+const LTD_USER_LIMIT = 50;
+const BUTTON_STYLE = {
+  backgroundColor: '#1890ff',
+  borderColor: '#1890ff',
+};
+const STATISTIC_VALUE_STYLE = {
+  fontSize: '24px',
+  fontWeight: 'bold' as const,
+};
+
 const CurrentPlanDetails = () => {
   const dispatch = useAppDispatch();
   const { t } = useTranslation('admin-center/current-bill');
@@ -47,7 +61,7 @@ const CurrentPlanDetails = () => {
   const [cancellingPlan, setCancellingPlan] = useState(false);
   const [addingSeats, setAddingSeats] = useState(false);
   const [isMoreSeatsModalVisible, setIsMoreSeatsModalVisible] = useState(false);
-  const [selectedSeatCount, setSelectedSeatCount] = useState<number | string>(5);
+  const [selectedSeatCount, setSelectedSeatCount] = useState<number | string>(1);
 
   const themeMode = useAppSelector(state => state.themeReducer.mode);
   const { loadingBillingInfo, billingInfo, freePlanSettings, isUpgradeModalOpen } = useAppSelector(
@@ -55,14 +69,16 @@ const CurrentPlanDetails = () => {
   );
 
   const browserTimeZone = Intl.DateTimeFormat().resolvedOptions().timeZone;
+  
+  const seatCountOptions: SeatOption[] = useMemo(() => {
+    const options: SeatOption[] = [
+      1, 2, 3, 4, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90,
+    ].map(value => ({ label: value.toString(), value }));
+    options.push({ label: SEAT_COUNT_LIMIT, value: SEAT_COUNT_LIMIT });
+    return options;
+  }, []);
 
-  type SeatOption = { label: string; value: number | string };
-  const seatCountOptions: SeatOption[] = [
-    1, 2, 3, 4, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90,
-  ].map(value => ({ label: value.toString(), value }));
-  seatCountOptions.push({ label: '100+', value: '100+' });
-
-  const handleSubscriptionAction = async (action: 'pause' | 'resume') => {
+  const handleSubscriptionAction = useCallback(async (action: SubscriptionAction) => {
     const isResume = action === 'resume';
     const setLoadingState = isResume ? setCancellingPlan : setPausingPlan;
     const apiMethod = isResume
@@ -78,21 +94,21 @@ const CurrentPlanDetails = () => {
           setLoadingState(false);
           dispatch(fetchBillingInfo());
           trackMixpanelEvent(eventType);
-        }, 8000);
-        return; // Exit function to prevent finally block from executing
+        }, BILLING_DELAY_MS);
+        return;
       }
     } catch (error) {
       logger.error(`Error ${action}ing subscription`, error);
-      setLoadingState(false); // Only set to false on error
+      setLoadingState(false);
     }
-  };
+  }, [dispatch, trackMixpanelEvent]);
 
-  const handleAddMoreSeats = () => {
+  const handleAddMoreSeats = useCallback(() => {
     setIsMoreSeatsModalVisible(true);
-  };
+  }, []);
 
-  const handlePurchaseMoreSeats = async () => {
-    if (selectedSeatCount.toString() === '100+' || !billingInfo?.total_seats) return;
+  const handlePurchaseMoreSeats = useCallback(async () => {
+    if (selectedSeatCount.toString() === SEAT_COUNT_LIMIT || !billingInfo?.total_seats) return;
 
     try {
       setAddingSeats(true);
@@ -108,51 +124,75 @@ const CurrentPlanDetails = () => {
     } finally {
       setAddingSeats(false);
     }
-  };
+  }, [selectedSeatCount, billingInfo?.total_seats, dispatch, trackMixpanelEvent]);
 
-  const calculateRemainingSeats = () => {
+  const calculateRemainingSeats = useMemo(() => {
     if (billingInfo?.total_seats && billingInfo?.total_used) {
       return billingInfo.total_seats - billingInfo.total_used;
     }
     return 0;
-  };
+  }, [billingInfo?.total_seats, billingInfo?.total_used]);
 
-  const checkSubscriptionStatus = (allowedStatuses: any[]) => {
+  // Calculate intelligent default for seat selection based on current usage
+  const getDefaultSeatCount = useMemo(() => {
+    const currentUsed = billingInfo?.total_used || 0;
+    const availableSeats = calculateRemainingSeats;
+    
+    // If only 1 user and no available seats, suggest 1 additional seat
+    if (currentUsed === 1 && availableSeats === 0) {
+      return 1;
+    }
+    
+    // If they have some users but running low on seats, suggest enough for current users
+    if (availableSeats < currentUsed && currentUsed > 0) {
+      return Math.max(1, currentUsed - availableSeats);
+    }
+    
+    // Default fallback
+    return Math.max(1, Math.min(5, currentUsed || 1));
+  }, [billingInfo?.total_used, calculateRemainingSeats]);
+
+  // Update selected seat count when billing info changes
+  useEffect(() => {
+    setSelectedSeatCount(getDefaultSeatCount);
+  }, [getDefaultSeatCount]);
+
+  const checkSubscriptionStatus = useCallback((allowedStatuses: string[]) => {
     if (!billingInfo?.status || billingInfo.is_ltd_user) return false;
     return allowedStatuses.includes(billingInfo.status);
-  };
+  }, [billingInfo?.status, billingInfo?.is_ltd_user]);
 
-  const shouldShowRedeemButton = () => {
+  const shouldShowRedeemButton = useMemo(() => {
     if (billingInfo?.trial_in_progress) return true;
-    return billingInfo?.ltd_users ? billingInfo.ltd_users < 50 : false;
-  };
+    return billingInfo?.ltd_users ? billingInfo.ltd_users < LTD_USER_LIMIT : false;
+  }, [billingInfo?.trial_in_progress, billingInfo?.ltd_users]);
 
-  const showChangeButton = () => {
+  const showChangeButton = useMemo(() => {
     return checkSubscriptionStatus([SUBSCRIPTION_STATUS.ACTIVE, SUBSCRIPTION_STATUS.PASTDUE]);
-  };
+  }, [checkSubscriptionStatus]);
 
-  const showPausePlanButton = () => {
+  const showPausePlanButton = useMemo(() => {
     return checkSubscriptionStatus([SUBSCRIPTION_STATUS.ACTIVE, SUBSCRIPTION_STATUS.PASTDUE]);
-  };
+  }, [checkSubscriptionStatus]);
 
-  const showResumePlanButton = () => {
+  const showResumePlanButton = useMemo(() => {
     return checkSubscriptionStatus([SUBSCRIPTION_STATUS.PAUSED]);
-  };
+  }, [checkSubscriptionStatus]);
 
-  const shouldShowAddSeats = () => {
+  const shouldShowAddSeats = useMemo(() => {
     if (!billingInfo) return false;
     return (
       billingInfo.subscription_type === ISUBSCRIPTION_TYPE.PADDLE &&
       billingInfo.status === SUBSCRIPTION_STATUS.ACTIVE
     );
-  };
+  }, [billingInfo]);
 
-  const renderExtra = () => {
+  const renderExtra = useCallback(() => {
     if (!billingInfo || billingInfo.is_custom) return null;
 
     return (
       <Space>
-        {showPausePlanButton() && (
+        {showPausePlanButton && (
           <Button
             type="link"
             danger
@@ -163,7 +203,7 @@ const CurrentPlanDetails = () => {
           </Button>
         )}
 
-        {showResumePlanButton() && (
+        {showResumePlanButton && (
           <Button
             type="primary"
             loading={cancellingPlan}
@@ -179,7 +219,7 @@ const CurrentPlanDetails = () => {
           </Button>
         )}
 
-        {showChangeButton() && (
+        {showChangeButton && (
           <Button
             type="primary"
             loading={pausingPlan || cancellingPlan}
@@ -190,9 +230,19 @@ const CurrentPlanDetails = () => {
         )}
       </Space>
     );
-  };
+  }, [
+    billingInfo,
+    showPausePlanButton,
+    showResumePlanButton,
+    showChangeButton,
+    pausingPlan,
+    cancellingPlan,
+    handleSubscriptionAction,
+    dispatch,
+    t,
+  ]);
 
-  const renderLtdDetails = () => {
+  const renderLtdDetails = useCallback(() => {
     if (!billingInfo || billingInfo.is_custom) return null;
     return (
       <Flex vertical>
@@ -200,41 +250,41 @@ const CurrentPlanDetails = () => {
         <Typography.Text>{t('ltdUsers', { ltd_users: billingInfo.ltd_users })}</Typography.Text>
       </Flex>
     );
-  };
+  }, [billingInfo, t]);
 
-  const renderTrialDetails = () => {
-    const checkIfTrialExpired = () => {
-      if (!billingInfo?.trial_expire_date) return false;
-      const today = new Date();
-      today.setHours(0, 0, 0, 0); // Set to start of day for comparison
-      const trialExpireDate = new Date(billingInfo.trial_expire_date);
-      trialExpireDate.setHours(0, 0, 0, 0); // Set to start of day for comparison
-      return today > trialExpireDate;
-    };
+  const checkIfTrialExpired = useCallback(() => {
+    if (!billingInfo?.trial_expire_date) return false;
+    const today = new Date();
+    today.setHours(0, 0, 0, 0);
+    const trialExpireDate = new Date(billingInfo.trial_expire_date);
+    trialExpireDate.setHours(0, 0, 0, 0);
+    return today > trialExpireDate;
+  }, [billingInfo?.trial_expire_date]);
 
-    const getExpirationMessage = (expireDate: string) => {
-      const today = new Date();
-      today.setHours(0, 0, 0, 0); // Set to start of day for comparison
+  const getExpirationMessage = useCallback((expireDate: string) => {
+    const today = new Date();
+    today.setHours(0, 0, 0, 0);
 
-      const tomorrow = new Date(today);
-      tomorrow.setDate(tomorrow.getDate() + 1);
+    const tomorrow = new Date(today);
+    tomorrow.setDate(tomorrow.getDate() + 1);
 
-      const expDate = new Date(expireDate);
-      expDate.setHours(0, 0, 0, 0); // Set to start of day for comparison
+    const expDate = new Date(expireDate);
+    expDate.setHours(0, 0, 0, 0);
 
-      if (expDate.getTime() === today.getTime()) {
-        return t('expirestoday', 'today');
-      } else if (expDate.getTime() === tomorrow.getTime()) {
-        return t('expirestomorrow', 'tomorrow');
-      } else if (expDate < today) {
-        const diffTime = Math.abs(today.getTime() - expDate.getTime());
-        const diffDays = Math.ceil(diffTime / (1000 * 60 * 60 * 24));
-        return t('expiredDaysAgo', '{{days}} days ago', { days: diffDays });
-      } else {
-        return calculateTimeGap(expireDate);
-      }
-    };
+    if (expDate.getTime() === today.getTime()) {
+      return t('expirestoday', 'today');
+    } else if (expDate.getTime() === tomorrow.getTime()) {
+      return t('expirestomorrow', 'tomorrow');
+    } else if (expDate < today) {
+      const diffTime = Math.abs(today.getTime() - expDate.getTime());
+      const diffDays = Math.ceil(diffTime / (1000 * 60 * 60 * 24));
+      return t('expiredDaysAgo', '{{days}} days ago', { days: diffDays });
+    } else {
+      return calculateTimeGap(expireDate);
+    }
+  }, [t]);
 
+  const renderTrialDetails = useCallback(() => {
     const isExpired = checkIfTrialExpired();
     const trialExpireDate = billingInfo?.trial_expire_date || '';
 
@@ -257,9 +307,9 @@ const CurrentPlanDetails = () => {
         </Tooltip>
       </Flex>
     );
-  };
+  }, [billingInfo?.trial_expire_date, checkIfTrialExpired, getExpirationMessage, t]);
 
-  const renderFreePlan = () => (
+  const renderFreePlan = useCallback(() => (
     <Flex vertical>
       <Typography.Text strong>{t('freePlan')}</Typography.Text>
       <Typography.Text>
@@ -271,9 +321,9 @@ const CurrentPlanDetails = () => {
         <br />- {freePlanSettings?.free_tier_storage} MB {t('storage')}
       </Typography.Text>
     </Flex>
-  );
+  ), [freePlanSettings, t]);
 
-  const renderPaddleSubscriptionInfo = () => {
+  const renderPaddleSubscriptionInfo = useCallback(() => {
     return (
       <Flex vertical>
         <Typography.Text strong>{billingInfo?.plan_name}</Typography.Text>
@@ -287,14 +337,14 @@ const CurrentPlanDetails = () => {
           </Typography.Text>
         </Flex>
 
-        {shouldShowAddSeats() && billingInfo?.total_seats && (
+        {shouldShowAddSeats && billingInfo?.total_seats && (
           <div style={{ marginTop: '16px' }}>
             <Row gutter={16} align="middle">
               <Col span={6}>
                 <Statistic
-                  title={t('totalSeats')}
+                  title={t('totalSeats') as string}
                   value={billingInfo.total_seats}
-                  valueStyle={{ fontSize: '24px', fontWeight: 'bold' }}
+                  valueStyle={STATISTIC_VALUE_STYLE}
                 />
               </Col>
               <Col span={8}>
@@ -302,16 +352,16 @@ const CurrentPlanDetails = () => {
                   type="primary"
                   icon={<PlusOutlined />}
                   onClick={handleAddMoreSeats}
-                  style={{ backgroundColor: '#1890ff', borderColor: '#1890ff' }}
+                  style={BUTTON_STYLE}
                 >
                   {t('addMoreSeats')}
                 </Button>
               </Col>
               <Col span={6}>
                 <Statistic
-                  title={t('availableSeats')}
-                  value={calculateRemainingSeats()}
-                  valueStyle={{ fontSize: '24px', fontWeight: 'bold' }}
+                  title={t('availableSeats') as string}
+                  value={calculateRemainingSeats}
+                  valueStyle={STATISTIC_VALUE_STYLE}
                 />
               </Col>
             </Row>
@@ -319,17 +369,17 @@ const CurrentPlanDetails = () => {
         )}
       </Flex>
     );
-  };
+  }, [billingInfo, shouldShowAddSeats, handleAddMoreSeats, calculateRemainingSeats, t]);
 
-  const renderCreditSubscriptionInfo = () => {
+  const renderCreditSubscriptionInfo = useCallback(() => {
     return (
       <Flex vertical>
         <Typography.Text strong>{t('creditPlan', 'Credit Plan')}</Typography.Text>
       </Flex>
     );
-  };
+  }, [t]);
 
-  const renderCustomSubscriptionInfo = () => {
+  const renderCustomSubscriptionInfo = useCallback(() => {
     return (
       <Flex vertical>
         <Typography.Text strong>{t('customPlan', 'Custom Plan')}</Typography.Text>
@@ -340,7 +390,36 @@ const CurrentPlanDetails = () => {
         </Typography.Text>
       </Flex>
     );
-  };
+  }, [billingInfo?.valid_till_date, t]);
+
+  const renderSubscriptionContent = useCallback(() => {
+    if (!billingInfo) return null;
+
+    switch (billingInfo.subscription_type) {
+      case ISUBSCRIPTION_TYPE.LIFE_TIME_DEAL:
+        return renderLtdDetails();
+      case ISUBSCRIPTION_TYPE.TRIAL:
+        return renderTrialDetails();
+      case ISUBSCRIPTION_TYPE.FREE:
+        return renderFreePlan();
+      case ISUBSCRIPTION_TYPE.PADDLE:
+        return renderPaddleSubscriptionInfo();
+      case ISUBSCRIPTION_TYPE.CREDIT:
+        return renderCreditSubscriptionInfo();
+      case ISUBSCRIPTION_TYPE.CUSTOM:
+        return renderCustomSubscriptionInfo();
+      default:
+        return null;
+    }
+  }, [
+    billingInfo,
+    renderLtdDetails,
+    renderTrialDetails,
+    renderFreePlan,
+    renderPaddleSubscriptionInfo,
+    renderCreditSubscriptionInfo,
+    renderCustomSubscriptionInfo,
+  ]);
 
   return (
     <Card
@@ -361,19 +440,10 @@ const CurrentPlanDetails = () => {
     >
       <Flex vertical>
         <div style={{ marginBottom: '14px' }}>
-          {billingInfo?.subscription_type === ISUBSCRIPTION_TYPE.LIFE_TIME_DEAL &&
-            renderLtdDetails()}
-          {billingInfo?.subscription_type === ISUBSCRIPTION_TYPE.TRIAL && renderTrialDetails()}
-          {billingInfo?.subscription_type === ISUBSCRIPTION_TYPE.FREE && renderFreePlan()}
-          {billingInfo?.subscription_type === ISUBSCRIPTION_TYPE.PADDLE &&
-            renderPaddleSubscriptionInfo()}
-          {billingInfo?.subscription_type === ISUBSCRIPTION_TYPE.CREDIT &&
-            renderCreditSubscriptionInfo()}
-          {billingInfo?.subscription_type === ISUBSCRIPTION_TYPE.CUSTOM &&
-            renderCustomSubscriptionInfo()}
+          {renderSubscriptionContent()}
         </div>
 
-        {shouldShowRedeemButton() && (
+        {shouldShowRedeemButton && (
           <>
             <Button
               type="link"
@@ -408,17 +478,28 @@ const CurrentPlanDetails = () => {
             <Typography.Paragraph
               style={{ fontSize: '16px', margin: '0 0 16px 0', fontWeight: 500 }}
             >
-              {t('purchaseSeatsText', "To continue, you'll need to purchase additional seats.")}
+              {billingInfo?.total_used === 1 
+                ? t('purchaseSeatsTextSingle', "Add more seats to invite team members to your workspace.")
+                : t('purchaseSeatsText', "To continue, you'll need to purchase additional seats.")
+              }
             </Typography.Paragraph>
 
             <Typography.Paragraph style={{ margin: '0 0 16px 0' }}>
               {t('currentSeatsText', 'You currently have {{seats}} seats available.', {
                 seats: billingInfo?.total_seats,
               })}
+              {billingInfo?.total_used === 1 && (
+                <span style={{ color: '#666', marginLeft: '8px' }}>
+                  ({t('singleUserNote', 'Currently used by 1 team member')})
+                </span>
+              )}
             </Typography.Paragraph>
 
             <Typography.Paragraph style={{ margin: '0 0 24px 0' }}>
-              {t('selectSeatsText', 'Please select the number of additional seats to purchase.')}
+              {billingInfo?.total_used === 1
+                ? t('selectSeatsTextSingle', 'Select how many additional seats you need for new team members.')
+                : t('selectSeatsText', 'Please select the number of additional seats to purchase.')
+              }
             </Typography.Paragraph>
 
             <div style={{ marginBottom: '24px' }}>
@@ -430,18 +511,18 @@ const CurrentPlanDetails = () => {
                 options={seatCountOptions}
                 style={{ width: '300px' }}
               />
+
             </div>
 
             <Flex justify="end">
-              {selectedSeatCount.toString() !== '100+' ? (
+              {selectedSeatCount.toString() !== SEAT_COUNT_LIMIT ? (
                 <Button
                   type="primary"
                   loading={addingSeats}
                   onClick={handlePurchaseMoreSeats}
                   style={{
                     minWidth: '100px',
-                    backgroundColor: '#1890ff',
-                    borderColor: '#1890ff',
+                    ...BUTTON_STYLE,
                     borderRadius: '2px',
                   }}
                 >

worklenz-frontend/src/components/admin-center/billing/drawers/redeem-code-drawer/redeem-code-drawer.tsx

@@ -1,4 +1,4 @@
-import { Button, Drawer, Form, Input, notification, Typography } from 'antd';
+import { Button, Drawer, Form, Input, notification, Typography } from '@/shared/antd-imports';
 import React, { useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { useAppSelector } from '@/hooks/useAppSelector';

worklenz-frontend/src/components/admin-center/billing/drawers/upgrade-plans-lkr/upgrade-plans-lkr.tsx

@@ -1,7 +1,7 @@
-import { Button, Card, Col, Form, Input, notification, Row, Tag, Typography } from 'antd';
+import { Button, Card, Col, Form, Input, notification, Row, Tag, Typography } from '@/shared/antd-imports';
 import React, { useState } from 'react';
 import './upgrade-plans-lkr.css';
-import { CheckCircleFilled } from '@ant-design/icons';
+import { CheckCircleFilled } from '@/shared/antd-imports';
 import { RootState } from '@/app/store';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/admin-center/billing/drawers/upgrade-plans/upgrade-plans.css

@@ -0,0 +1,44 @@
+.upgrade-plans-responsive {
+  padding: 24px;
+}
+
+.upgrade-plans-row-responsive {
+  width: 100%;
+  flex-wrap: wrap;
+}
+
+@media (max-width: 900px) {
+  .upgrade-plans-row-responsive .ant-col {
+    flex: 0 0 100%;
+    max-width: 100%;
+    margin-bottom: 16px;
+  }
+  .upgrade-plans-row-responsive .ant-card {
+    width: 100%;
+    min-width: 0;
+  }
+}
+
+@media (max-width: 600px) {
+  .upgrade-plans-responsive {
+    padding: 8px;
+  }
+  .upgrade-plans-row-responsive .ant-col {
+    padding: 0 !important;
+    margin-bottom: 12px;
+  }
+  .upgrade-plans-row-responsive .ant-card {
+    width: 100%;
+    min-width: 0;
+    box-sizing: border-box;
+  }
+  .upgrade-plans-responsive .ant-typography,
+  .upgrade-plans-responsive .ant-btn,
+  .upgrade-plans-responsive .ant-form-item {
+    width: 100%;
+    text-align: center;
+  }
+  .upgrade-plans-responsive .ant-btn {
+    margin-top: 12px;
+  }
+} 

worklenz-frontend/src/components/admin-center/billing/drawers/upgrade-plans/upgrade-plans.tsx

@@ -11,7 +11,7 @@ import {
   Tooltip,
   Typography,
   message,
-} from 'antd/es';
+} from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 
 import { adminCenterApiService } from '@/api/admin-center/admin-center.api.service';
@@ -22,7 +22,7 @@ import {
 import logger from '@/utils/errorLogger';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { IPaddlePlans, SUBSCRIPTION_STATUS } from '@/shared/constants';
-import { CheckCircleFilled, InfoCircleOutlined } from '@ant-design/icons';
+import { CheckCircleFilled, InfoCircleOutlined } from '@/shared/antd-imports';
 import { useAuthService } from '@/hooks/useAuth';
 import { fetchBillingInfo, toggleUpgradeModal } from '@/features/admin-center/admin-center.slice';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
@@ -30,6 +30,7 @@ import { billingApiService } from '@/api/admin-center/billing.api.service';
 import { authApiService } from '@/api/auth/auth.api.service';
 import { setUser } from '@/features/user/userSlice';
 import { setSession } from '@/utils/session-helper';
+import './upgrade-plans.css';
 
 // Extend Window interface to include Paddle
 declare global {
@@ -65,20 +66,24 @@ const UpgradePlans = () => {
   const [paddleError, setPaddleError] = useState<string | null>(null);
 
   const populateSeatCountOptions = (currentSeats: number) => {
-    if (!currentSeats) return [];
+    if (typeof currentSeats !== 'number') return [];
 
     const step = 5;
     const maxSeats = 90;
-    const minValue = Math.min(currentSeats + 1);
-    const rangeStart = Math.ceil(minValue / step) * step;
-    const range = Array.from(
-      { length: Math.floor((maxSeats - rangeStart) / step) + 1 },
-      (_, i) => rangeStart + i * step
-    );
+    const minValue = currentSeats;
+    const options: { value: number; disabled: boolean }[] = [];
 
-    return currentSeats < step
-      ? [...Array.from({ length: rangeStart - minValue }, (_, i) => minValue + i), ...range]
-      : range;
+    // Always show 1-5, but disable if less than minValue
+    for (let i = 1; i <= 5; i++) {
+      options.push({ value: i, disabled: i < minValue });
+    }
+
+    // Show all multiples of 5 from 10 to maxSeats
+    for (let i = 10; i <= maxSeats; i += step) {
+      options.push({ value: i, disabled: i < minValue });
+    }
+
+    return options;
   };
 
   const fetchPricingPlans = async () => {
@@ -339,24 +344,25 @@ const UpgradePlans = () => {
   }, []);
 
   return (
-    <div>
+    <div className="upgrade-plans-responsive">
       <Flex justify="center" align="center">
         <Typography.Title level={2}>
           {billingInfo?.status === SUBSCRIPTION_STATUS.TRIALING
-            ? t('selectPlan')
-            : t('changeSubscriptionPlan')}
+            ? t('selectPlan', 'Select Plan')
+            : t('changeSubscriptionPlan', 'Change Subscription Plan')}
         </Typography.Title>
       </Flex>
 
       <Flex justify="center" align="center">
         <Form form={form}>
-          <Form.Item name="seatCount" label={t('noOfSeats')}>
+          <Form.Item name="seatCount" label={t('noOfSeats', 'Number of Seats')}>
             <Select
               style={{ width: 100 }}
               value={selectedSeatCount}
               options={seatCountOptions.map(option => ({
-                value: option,
-                text: option.toString(),
+                value: option.value,
+                label: option.value.toString(),
+                disabled: option.disabled,
               }))}
               onChange={setSelectedSeatCount}
             />
@@ -365,31 +371,31 @@ const UpgradePlans = () => {
       </Flex>
 
       <Flex>
-        <Row className="w-full">
+        <Row className="w-full upgrade-plans-row-responsive">
           {/* Free Plan */}
           <Col span={8} style={{ padding: '0 4px' }}>
             <Card
               style={{ ...isSelected(paddlePlans.FREE), height: '100%' }}
               hoverable
-              title={<span style={cardStyles.title}>{t('freePlan')}</span>}
+              title={<span style={cardStyles.title}>{t('freePlan', 'Free Plan')}</span>}
               onClick={() => setSelectedCard(paddlePlans.FREE)}
             >
               <div style={cardStyles.priceContainer}>
                 <Flex justify="space-between" align="center">
                   <Typography.Title level={1}>$ 0.00</Typography.Title>
-                  <Typography.Text>{t('freeForever')}</Typography.Text>
+                  <Typography.Text>{t('freeForever', 'Free Forever')}</Typography.Text>
                 </Flex>
                 <Flex justify="center" align="center">
                   <Typography.Text strong style={{ fontSize: '16px' }}>
-                    {t('bestForPersonalUse')}
+                    {t('bestForPersonalUse', 'Best for Personal Use')}
                   </Typography.Text>
                 </Flex>
               </div>
 
               <div style={cardStyles.featureList}>
-                {renderFeature(`${plans.free_tier_storage} ${t('storage')}`)}
-                {renderFeature(`${plans.projects_limit} ${t('projects')}`)}
-                {renderFeature(`${plans.team_member_limit} ${t('teamMembers')}`)}
+                {renderFeature(`${plans.free_tier_storage} ${t('storage', 'Storage')}`)}
+                {renderFeature(`${plans.projects_limit} ${t('projects', 'Projects')}`)}
+                {renderFeature(`${plans.team_member_limit} ${t('teamMembers', 'Team Members')}`)}
               </div>
             </Card>
           </Col>
@@ -401,9 +407,9 @@ const UpgradePlans = () => {
               hoverable
               title={
                 <span style={cardStyles.title}>
-                  {t('annualPlan')}{' '}
+                  {t('annualPlan', 'Annual Plan')}{' '}
                   <Tag color="volcano" style={{ lineHeight: '21px' }}>
-                    {t('tag')}
+                    {t('tag', 'Popular')}
                   </Tag>
                 </span>
               }
@@ -429,16 +435,16 @@ const UpgradePlans = () => {
                   </Typography.Text>
                 </Flex>
                 <Flex justify="center" align="center">
-                  <Typography.Text>{t('billedAnnually')}</Typography.Text>
+                  <Typography.Text>{t('billedAnnually', 'Billed Annually')}</Typography.Text>
                 </Flex>
               </div>
 
               <div style={cardStyles.featureList} className="mt-4">
-                {renderFeature(t('startupText01'))}
-                {renderFeature(t('startupText02'))}
-                {renderFeature(t('startupText03'))}
-                {renderFeature(t('startupText04'))}
-                {renderFeature(t('startupText05'))}
+                {renderFeature(t('startupText01', 'Unlimited Projects'))}
+                {renderFeature(t('startupText02', 'Unlimited Team Members'))}
+                {renderFeature(t('startupText03', 'Unlimited Storage'))}
+                {renderFeature(t('startupText04', 'Priority Support'))}
+                {renderFeature(t('startupText05', 'Advanced Analytics'))}
               </div>
             </Card>
           </Col>
@@ -448,7 +454,7 @@ const UpgradePlans = () => {
             <Card
               style={{ ...isSelected(paddlePlans.MONTHLY), height: '100%' }}
               hoverable
-              title={<span style={cardStyles.title}>{t('monthlyPlan')}</span>}
+              title={<span style={cardStyles.title}>{t('monthlyPlan', 'Monthly Plan')}</span>}
               onClick={() => setSelectedCard(paddlePlans.MONTHLY)}
             >
               <div style={cardStyles.priceContainer}>
@@ -469,16 +475,16 @@ const UpgradePlans = () => {
                   </Typography.Text>
                 </Flex>
                 <Flex justify="center" align="center">
-                  <Typography.Text>{t('billedMonthly')}</Typography.Text>
+                  <Typography.Text>{t('billedMonthly', 'Billed Monthly')}</Typography.Text>
                 </Flex>
               </div>
 
               <div style={cardStyles.featureList}>
-                {renderFeature(t('startupText01'))}
-                {renderFeature(t('startupText02'))}
-                {renderFeature(t('startupText03'))}
-                {renderFeature(t('startupText04'))}
-                {renderFeature(t('startupText05'))}
+                {renderFeature(t('startupText01', 'Unlimited Projects'))}
+                {renderFeature(t('startupText02', 'Unlimited Team Members'))}
+                {renderFeature(t('startupText03', 'Unlimited Storage'))}
+                {renderFeature(t('startupText04', 'Priority Support'))}
+                {renderFeature(t('startupText05', 'Advanced Analytics'))}
               </div>
             </Card>
           </Col>
@@ -509,8 +515,8 @@ const UpgradePlans = () => {
             disabled={billingInfo?.plan_id === plans.annual_plan_id}
           >
             {billingInfo?.status === SUBSCRIPTION_STATUS.ACTIVE
-              ? t('changeToPlan', { plan: t('annualPlan') })
-              : t('continueWith', { plan: t('annualPlan') })}
+              ? t('changeToPlan', 'Change to {{plan}}', { plan: t('annualPlan', 'Annual Plan') })
+              : t('continueWith', 'Continue with {{plan}}', { plan: t('annualPlan', 'Annual Plan') })}
           </Button>
         )}
         {selectedPlan === paddlePlans.MONTHLY && (
@@ -522,8 +528,8 @@ const UpgradePlans = () => {
             disabled={billingInfo?.plan_id === plans.monthly_plan_id}
           >
             {billingInfo?.status === SUBSCRIPTION_STATUS.ACTIVE
-              ? t('changeToPlan', { plan: t('monthlyPlan') })
-              : t('continueWith', { plan: t('monthlyPlan') })}
+              ? t('changeToPlan', 'Change to {{plan}}', { plan: t('monthlyPlan', 'Monthly Plan') })
+              : t('continueWith', 'Continue with {{plan}}', { plan: t('monthlyPlan', 'Monthly Plan') })}
           </Button>
         )}
       </Row>

worklenz-frontend/src/components/admin-center/configuration/configuration.tsx

@@ -1,5 +1,5 @@
-import { Button, Card, Col, Divider, Form, Input, notification, Row, Select } from 'antd';
-import React, { useEffect, useState } from 'react';
+import { Button, Card, Col, Divider, Form, Input, Row, Select } from '@/shared/antd-imports';
+import React, { useEffect, useState, useMemo, useCallback } from 'react';
 import { RootState } from '../../../app/store';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { IBillingConfigurationCountry } from '@/types/admin-center/country.types';
@@ -7,14 +7,15 @@ import { adminCenterApiService } from '@/api/admin-center/admin-center.api.servi
 import { IBillingConfiguration } from '@/types/admin-center/admin-center.types';
 import logger from '@/utils/errorLogger';
 
-const Configuration: React.FC = () => {
+const Configuration: React.FC = React.memo(() => {
   const themeMode = useAppSelector((state: RootState) => state.themeReducer.mode);
 
   const [countries, setCountries] = useState<IBillingConfigurationCountry[]>([]);
   const [configuration, setConfiguration] = useState<IBillingConfiguration>();
   const [loading, setLoading] = useState(false);
   const [form] = Form.useForm();
-  const fetchCountries = async () => {
+
+  const fetchCountries = useCallback(async () => {
     try {
       const res = await adminCenterApiService.getCountries();
       if (res.done) {
@@ -23,61 +24,85 @@ const Configuration: React.FC = () => {
     } catch (error) {
       logger.error('Error fetching countries:', error);
     }
-  };
+  }, []);
 
-  const fetchConfiguration = async () => {
+  const fetchConfiguration = useCallback(async () => {
     const res = await adminCenterApiService.getBillingConfiguration();
     if (res.done) {
       setConfiguration(res.body);
       form.setFieldsValue(res.body);
     }
-  };
+  }, [form]);
 
   useEffect(() => {
     fetchCountries();
     fetchConfiguration();
-  }, []);
+  }, [fetchCountries, fetchConfiguration]);
 
-  const handleSave = async (values: any) => {
-    try {
-      setLoading(true);
-      const res = await adminCenterApiService.updateBillingConfiguration(values);
-      if (res.done) {
-        fetchConfiguration();
+  const handleSave = useCallback(
+    async (values: any) => {
+      try {
+        setLoading(true);
+        const res = await adminCenterApiService.updateBillingConfiguration(values);
+        if (res.done) {
+          fetchConfiguration();
+        }
+      } catch (error) {
+        logger.error('Error updating configuration:', error);
+      } finally {
+        setLoading(false);
       }
-    } catch (error) {
-      logger.error('Error updating configuration:', error);
-    } finally {
-      setLoading(false);
-    }
-  };
+    },
+    [fetchConfiguration]
+  );
 
-  const countryOptions = countries.map(country => ({
-    label: country.name,
-    value: country.id,
-  }));
+  const countryOptions = useMemo(
+    () =>
+      countries.map(country => ({
+        label: country.name,
+        value: country.id,
+      })),
+    [countries]
+  );
+
+  const titleStyle = useMemo(
+    () => ({
+      color: `${themeMode === 'dark' ? '#ffffffd9' : '#000000d9'}`,
+      fontWeight: 500,
+      fontSize: '16px',
+      display: 'flex',
+      gap: '4px',
+    }),
+    [themeMode]
+  );
+
+  const dividerTitleStyle = useMemo(
+    () => ({
+      color: `${themeMode === 'dark' ? '#ffffffd9' : '#000000d9'}`,
+      fontWeight: 600,
+      fontSize: '16px',
+      display: 'flex',
+      gap: '4px',
+    }),
+    [themeMode]
+  );
+
+  const cardStyle = useMemo(() => ({ marginTop: '16px' }), []);
+  const colStyle = useMemo(() => ({ padding: '0 12px', height: '86px' }), []);
+  const dividerStyle = useMemo(() => ({ margin: '16px 0' }), []);
+  const buttonColStyle = useMemo(() => ({ paddingLeft: '12px' }), []);
+
+  const handlePhoneInput = useCallback((e: React.FormEvent<HTMLInputElement>) => {
+    const input = e.target as HTMLInputElement;
+    input.value = input.value.replace(/[^0-9]/g, '');
+  }, []);
 
   return (
     <div>
-      <Card
-        title={
-          <span
-            style={{
-              color: `${themeMode === 'dark' ? '#ffffffd9' : '#000000d9'}`,
-              fontWeight: 500,
-              fontSize: '16px',
-              display: 'flex',
-              gap: '4px',
-            }}
-          >
-            Billing Details
-          </span>
-        }
-        style={{ marginTop: '16px' }}
-      >
+      <Card title={<span style={titleStyle}>Billing Details</span>} style={cardStyle}>
         <Form form={form} initialValues={configuration} onFinish={handleSave}>
-          <Row>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+          <Row gutter={[0, 0]}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item
                 name="name"
                 label="Name"
@@ -88,10 +113,10 @@ const Configuration: React.FC = () => {
                   },
                 ]}
               >
-                <Input placeholder="Name" />
+                <Input placeholder="Name" disabled />
               </Form.Item>
             </Col>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item
                 name="email"
                 label="Email Address"
@@ -102,10 +127,10 @@ const Configuration: React.FC = () => {
                   },
                 ]}
               >
-                <Input placeholder="Name" disabled />
+                <Input placeholder="Email Address" disabled />
               </Form.Item>
             </Col>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item
                 name="phone"
                 label="Contact Number"
@@ -117,58 +142,34 @@ const Configuration: React.FC = () => {
                   },
                 ]}
               >
-                <Input
-                  placeholder="Phone Number"
-                  maxLength={10}
-                  onInput={e => {
-                    const input = e.target as HTMLInputElement; // Type assertion to access 'value'
-                    input.value = input.value.replace(/[^0-9]/g, ''); // Restrict non-numeric input
-                  }}
-                />
+                <Input placeholder="Phone Number" maxLength={10} onInput={handlePhoneInput} />
               </Form.Item>
             </Col>
           </Row>
 
-          <Divider orientation="left" style={{ margin: '16px 0' }}>
-            <span
-              style={{
-                color: `${themeMode === 'dark' ? '#ffffffd9' : '#000000d9'}`,
-                fontWeight: 600,
-                fontSize: '16px',
-                display: 'flex',
-                gap: '4px',
-              }}
-            >
-              Company Details
-            </span>
+          <Divider orientation="left" style={{ ...dividerStyle, fontSize: '14px' }}>
+            <span style={dividerTitleStyle}>Company Details</span>
           </Divider>
 
-          <Row>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+          <Row gutter={[0, 0]}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="company_name" label="Company Name" layout="vertical">
                 <Input placeholder="Company Name" />
               </Form.Item>
             </Col>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="address_line_1" label="Address Line 01" layout="vertical">
                 <Input placeholder="Address Line 01" />
               </Form.Item>
             </Col>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="address_line_2" label="Address Line 02" layout="vertical">
                 <Input placeholder="Address Line 02" />
               </Form.Item>
             </Col>
           </Row>
-          <Row>
-            <Col
-              span={8}
-              style={{
-                padding: '0 12px',
-                height: '86px',
-                scrollbarColor: 'red',
-              }}
-            >
+          <Row gutter={[0, 0]}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="country" label="Country" layout="vertical">
                 <Select
                   dropdownStyle={{ maxHeight: 256, overflow: 'auto' }}
@@ -181,28 +182,28 @@ const Configuration: React.FC = () => {
                 />
               </Form.Item>
             </Col>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="city" label="City" layout="vertical">
                 <Input placeholder="City" />
               </Form.Item>
             </Col>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="state" label="State" layout="vertical">
                 <Input placeholder="State" />
               </Form.Item>
             </Col>
           </Row>
-          <Row>
-            <Col span={8} style={{ padding: '0 12px', height: '86px' }}>
+          <Row gutter={[0, 0]}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={colStyle}>
               <Form.Item name="postal_code" label="Postal Code" layout="vertical">
                 <Input placeholder="Postal Code" />
               </Form.Item>
             </Col>
           </Row>
           <Row>
-            <Col style={{ paddingLeft: '12px' }}>
+            <Col xs={24} sm={24} md={8} lg={8} xl={8} style={{ ...buttonColStyle, marginTop: 8 }}>
               <Form.Item>
-                <Button type="primary" htmlType="submit">
+                <Button type="primary" htmlType="submit" loading={loading} block>
                   Save
                 </Button>
               </Form.Item>
@@ -212,6 +213,8 @@ const Configuration: React.FC = () => {
       </Card>
     </div>
   );
-};
+});
+
+Configuration.displayName = 'Configuration';
 
 export default Configuration;

worklenz-frontend/src/components/admin-center/overview/organization-admins-table/organization-admins-table.tsx

@@ -1,4 +1,4 @@
-import { Table, TableProps, Typography } from 'antd';
+import { Table, TableProps, Typography } from '@/shared/antd-imports';
 import React, { useMemo } from 'react';
 import { IOrganizationAdmin } from '@/types/admin-center/admin-center.types';
 

worklenz-frontend/src/components/admin-center/overview/organization-name/organization-name.tsx

@@ -1,9 +1,8 @@
 import { adminCenterApiService } from '@/api/admin-center/admin-center.api.service';
 import logger from '@/utils/errorLogger';
-import { EnterOutlined, EditOutlined } from '@ant-design/icons';
-import { Card, Button, Tooltip, Typography } from 'antd';
+import { EnterOutlined, EditOutlined } from '@/shared/antd-imports';
+import { Card, Button, Tooltip, Typography } from '@/shared/antd-imports';
 import TextArea from 'antd/es/input/TextArea';
-import Paragraph from 'antd/es/typography/Paragraph';
 import { TFunction } from 'i18next';
 import { useState, useEffect } from 'react';
 

worklenz-frontend/src/components/admin-center/overview/organization-owner/organization-owner.tsx

@@ -1,8 +1,8 @@
 import { adminCenterApiService } from '@/api/admin-center/admin-center.api.service';
 import { IOrganization } from '@/types/admin-center/admin-center.types';
 import logger from '@/utils/errorLogger';
-import { MailOutlined, PhoneOutlined, EditOutlined } from '@ant-design/icons';
-import { Card, Tooltip, Input, Button, Typography, InputRef } from 'antd';
+import { MailOutlined, PhoneOutlined, EditOutlined } from '@/shared/antd-imports';
+import { Card, Tooltip, Input, Button, Typography, InputRef } from '@/shared/antd-imports';
 import { TFunction } from 'i18next';
 import { useEffect, useRef, useState } from 'react';
 

worklenz-frontend/src/components/admin-center/teams/add-team-drawer/add-team-drawer.tsx

@@ -1,5 +1,5 @@
 import React, { useRef, useState } from 'react';
-import { Button, Drawer, Form, Input, InputRef, Typography } from 'antd';
+import { Button, Drawer, Form, Input, InputRef, Typography } from '@/shared/antd-imports';
 import { fetchTeams } from '@features/teams/teamSlice';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/admin-center/teams/settings-drawer/settings-drawer.tsx

@@ -11,7 +11,7 @@ import {
   TableProps,
   Typography,
   Tooltip,
-} from 'antd';
+} from '@/shared/antd-imports';
 import React, { useState } from 'react';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import './settings-drawer.css';

worklenz-frontend/src/components/admin-center/teams/teams-table/teams-table.tsx

@@ -5,8 +5,8 @@ import { toggleSettingDrawer, deleteTeam, fetchTeams } from '@/features/teams/te
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { IOrganizationTeam } from '@/types/admin-center/admin-center.types';
 import logger from '@/utils/errorLogger';
-import { SettingOutlined, DeleteOutlined } from '@ant-design/icons';
-import { Badge, Button, Card, Popconfirm, Table, TableProps, Tooltip, Typography } from 'antd';
+import { SettingOutlined, DeleteOutlined } from '@/shared/antd-imports';
+import { Badge, Button, Card, Popconfirm, Table, TableProps, Tooltip, Typography } from '@/shared/antd-imports';
 import { TFunction } from 'i18next';
 import { useState } from 'react';
 import { useMediaQuery } from 'react-responsive';

worklenz-frontend/src/components/advanced-gantt/AdvancedGanttChart.tsx

@@ -0,0 +1,612 @@
+import React, { useReducer, useMemo, useCallback, useRef, useEffect, useState } from 'react';
+import { 
+  GanttTask, 
+  ColumnConfig, 
+  TimelineConfig, 
+  VirtualScrollConfig,
+  ZoomLevel,
+  GanttState,
+  GanttAction,
+  AdvancedGanttProps,
+  SelectionState,
+  GanttViewState,
+  DragState
+} from '../../types/advanced-gantt.types';
+import GanttGrid from './GanttGrid';
+import DraggableTaskBar from './DraggableTaskBar';
+import TimelineMarkers, { holidayPresets, workingDayPresets } from './TimelineMarkers';
+import VirtualScrollContainer, { VirtualTimeline } from './VirtualScrollContainer';
+import { 
+  usePerformanceMonitoring, 
+  useTaskCalculations, 
+  useDateCalculations,
+  useDebounce,
+  useThrottle
+} from '../../utils/gantt-performance';
+import { useAppSelector } from '../../hooks/useAppSelector';
+import { themeWiseColor } from '../../utils/themeWiseColor';
+
+// Default configurations
+const defaultColumns: ColumnConfig[] = [
+  { 
+    field: 'name', 
+    title: 'Task Name', 
+    width: 250, 
+    minWidth: 150, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'text'
+  },
+  { 
+    field: 'startDate', 
+    title: 'Start Date', 
+    width: 120, 
+    minWidth: 100, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'date'
+  },
+  { 
+    field: 'endDate', 
+    title: 'End Date', 
+    width: 120, 
+    minWidth: 100, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'date'
+  },
+  { 
+    field: 'duration', 
+    title: 'Duration', 
+    width: 80, 
+    minWidth: 60, 
+    resizable: true, 
+    sortable: false, 
+    fixed: true 
+  },
+  { 
+    field: 'progress', 
+    title: 'Progress', 
+    width: 100, 
+    minWidth: 80, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'number'
+  },
+];
+
+const defaultTimelineConfig: TimelineConfig = {
+  topTier: { unit: 'month', format: 'MMM yyyy', height: 30 },
+  bottomTier: { unit: 'day', format: 'dd', height: 25 },
+  showWeekends: true,
+  showNonWorkingDays: true,
+  holidays: holidayPresets.US,
+  workingDays: workingDayPresets.standard,
+  workingHours: { start: 9, end: 17 },
+  dayWidth: 30,
+};
+
+const defaultVirtualScrollConfig: VirtualScrollConfig = {
+  enableRowVirtualization: true,
+  enableTimelineVirtualization: true,
+  bufferSize: 10,
+  itemHeight: 40,
+  overscan: 5,
+};
+
+const defaultZoomLevels: ZoomLevel[] = [
+  { 
+    name: 'Year', 
+    dayWidth: 2, 
+    scale: 0.1,
+    topTier: { unit: 'year', format: 'yyyy' },
+    bottomTier: { unit: 'month', format: 'MMM' }
+  },
+  { 
+    name: 'Month', 
+    dayWidth: 8, 
+    scale: 0.5,
+    topTier: { unit: 'month', format: 'MMM yyyy' },
+    bottomTier: { unit: 'week', format: 'w' }
+  },
+  { 
+    name: 'Week', 
+    dayWidth: 25, 
+    scale: 1,
+    topTier: { unit: 'week', format: 'MMM dd' },
+    bottomTier: { unit: 'day', format: 'dd' }
+  },
+  { 
+    name: 'Day', 
+    dayWidth: 50, 
+    scale: 2,
+    topTier: { unit: 'day', format: 'MMM dd' },
+    bottomTier: { unit: 'hour', format: 'HH' }
+  },
+];
+
+// Gantt state reducer
+function ganttReducer(state: GanttState, action: GanttAction): GanttState {
+  switch (action.type) {
+    case 'SET_TASKS':
+      return { ...state, tasks: action.payload };
+    
+    case 'UPDATE_TASK':
+      return {
+        ...state,
+        tasks: state.tasks.map(task =>
+          task.id === action.payload.id
+            ? { ...task, ...action.payload.updates }
+            : task
+        ),
+      };
+    
+    case 'ADD_TASK':
+      return { ...state, tasks: [...state.tasks, action.payload] };
+    
+    case 'DELETE_TASK':
+      return {
+        ...state,
+        tasks: state.tasks.filter(task => task.id !== action.payload),
+      };
+    
+    case 'SET_SELECTION':
+      return {
+        ...state,
+        selectionState: { ...state.selectionState, selectedTasks: action.payload },
+      };
+    
+    case 'SET_DRAG_STATE':
+      return { ...state, dragState: action.payload };
+    
+    case 'SET_ZOOM_LEVEL':
+      const newZoomLevel = Math.max(0, Math.min(state.zoomLevels.length - 1, action.payload));
+      return {
+        ...state,
+        viewState: { ...state.viewState, zoomLevel: newZoomLevel },
+        timelineConfig: {
+          ...state.timelineConfig,
+          dayWidth: state.zoomLevels[newZoomLevel].dayWidth,
+          topTier: state.zoomLevels[newZoomLevel].topTier,
+          bottomTier: state.zoomLevels[newZoomLevel].bottomTier,
+        },
+      };
+    
+    case 'SET_SCROLL_POSITION':
+      return {
+        ...state,
+        viewState: { ...state.viewState, scrollPosition: action.payload },
+      };
+    
+    case 'SET_SPLITTER_POSITION':
+      return {
+        ...state,
+        viewState: { ...state.viewState, splitterPosition: action.payload },
+      };
+    
+    case 'TOGGLE_TASK_EXPANSION':
+      return {
+        ...state,
+        tasks: state.tasks.map(task =>
+          task.id === action.payload
+            ? { ...task, isExpanded: !task.isExpanded }
+            : task
+        ),
+      };
+    
+    case 'SET_VIEW_STATE':
+      return {
+        ...state,
+        viewState: { ...state.viewState, ...action.payload },
+      };
+    
+    case 'UPDATE_COLUMN_WIDTH':
+      return {
+        ...state,
+        columns: state.columns.map(col =>
+          col.field === action.payload.field
+            ? { ...col, width: action.payload.width }
+            : col
+        ),
+      };
+    
+    default:
+      return state;
+  }
+}
+
+const AdvancedGanttChart: React.FC<AdvancedGanttProps> = ({
+  tasks: initialTasks,
+  columns = defaultColumns,
+  timelineConfig = {},
+  virtualScrollConfig = {},
+  zoomLevels = defaultZoomLevels,
+  initialViewState = {},
+  initialSelection = [],
+  onTaskUpdate,
+  onTaskCreate,
+  onTaskDelete,
+  onTaskMove,
+  onTaskResize,
+  onProgressChange,
+  onSelectionChange,
+  onColumnResize,
+  onDependencyCreate,
+  onDependencyDelete,
+  className = '',
+  style = {},
+  theme = 'auto',
+  enableDragDrop = true,
+  enableResize = true,
+  enableProgressEdit = true,
+  enableInlineEdit = true,
+  enableVirtualScrolling = true,
+  enableDebouncing = true,
+  debounceDelay = 300,
+  maxVisibleTasks = 1000,
+}) => {
+  const containerRef = useRef<HTMLDivElement>(null);
+  const [containerSize, setContainerSize] = useState({ width: 0, height: 0 });
+  const themeMode = useAppSelector(state => state.themeReducer.mode);
+  const { startMeasure, endMeasure, metrics } = usePerformanceMonitoring();
+  const { getDaysBetween } = useDateCalculations();
+
+  // Initialize state
+  const initialState: GanttState = {
+    tasks: initialTasks,
+    columns,
+    timelineConfig: { ...defaultTimelineConfig, ...timelineConfig },
+    virtualScrollConfig: { ...defaultVirtualScrollConfig, ...virtualScrollConfig },
+    dragState: null,
+    selectionState: {
+      selectedTasks: initialSelection,
+      selectedRows: [],
+      focusedTask: undefined,
+    },
+    viewState: {
+      zoomLevel: 2, // Week view by default
+      scrollPosition: { x: 0, y: 0 },
+      viewportSize: { width: 0, height: 0 },
+      splitterPosition: 40, // 40% for grid, 60% for timeline
+      showCriticalPath: false,
+      showBaseline: false,
+      showProgress: true,
+      showDependencies: true,
+      autoSchedule: false,
+      readOnly: false,
+      ...initialViewState,
+    },
+    zoomLevels,
+    performanceMetrics: {
+      renderTime: 0,
+      taskCount: initialTasks.length,
+      visibleTaskCount: 0,
+    },
+  };
+
+  const [state, dispatch] = useReducer(ganttReducer, initialState);
+  const { taskMap, parentChildMap, totalTasks } = useTaskCalculations(state.tasks);
+
+  // Calculate project timeline bounds
+  const projectBounds = useMemo(() => {
+    if (state.tasks.length === 0) {
+      const today = new Date();
+      return {
+        start: new Date(today.getFullYear(), today.getMonth(), 1),
+        end: new Date(today.getFullYear(), today.getMonth() + 3, 0),
+      };
+    }
+
+    const startDates = state.tasks.map(task => task.startDate);
+    const endDates = state.tasks.map(task => task.endDate);
+    const minStart = new Date(Math.min(...startDates.map(d => d.getTime())));
+    const maxEnd = new Date(Math.max(...endDates.map(d => d.getTime())));
+
+    // Add some padding
+    minStart.setDate(minStart.getDate() - 7);
+    maxEnd.setDate(maxEnd.getDate() + 7);
+
+    return { start: minStart, end: maxEnd };
+  }, [state.tasks]);
+
+  // Debounced event handlers
+  const debouncedTaskUpdate = useDebounce(
+    useCallback((taskId: string, updates: Partial<GanttTask>) => {
+      dispatch({ type: 'UPDATE_TASK', payload: { id: taskId, updates } });
+      onTaskUpdate?.(taskId, updates);
+    }, [onTaskUpdate]),
+    enableDebouncing ? debounceDelay : 0
+  );
+
+  const debouncedTaskMove = useDebounce(
+    useCallback((taskId: string, newDates: { start: Date; end: Date }) => {
+      dispatch({ type: 'UPDATE_TASK', payload: { 
+        id: taskId, 
+        updates: { startDate: newDates.start, endDate: newDates.end } 
+      }});
+      onTaskMove?.(taskId, newDates);
+    }, [onTaskMove]),
+    enableDebouncing ? debounceDelay : 0
+  );
+
+  const debouncedProgressChange = useDebounce(
+    useCallback((taskId: string, progress: number) => {
+      dispatch({ type: 'UPDATE_TASK', payload: { id: taskId, updates: { progress } }});
+      onProgressChange?.(taskId, progress);
+    }, [onProgressChange]),
+    enableDebouncing ? debounceDelay : 0
+  );
+
+  // Throttled scroll handler
+  const throttledScrollHandler = useThrottle(
+    useCallback((scrollLeft: number, scrollTop: number) => {
+      dispatch({ type: 'SET_SCROLL_POSITION', payload: { x: scrollLeft, y: scrollTop } });
+    }, []),
+    16 // 60fps
+  );
+
+  // Container size observer
+  useEffect(() => {
+    const observer = new ResizeObserver((entries) => {
+      for (const entry of entries) {
+        const { width, height } = entry.contentRect;
+        setContainerSize({ width, height });
+        dispatch({ 
+          type: 'SET_VIEW_STATE', 
+          payload: { viewportSize: { width, height } } 
+        });
+      }
+    });
+
+    if (containerRef.current) {
+      observer.observe(containerRef.current);
+    }
+
+    return () => observer.disconnect();
+  }, []);
+
+  // Calculate grid and timeline dimensions
+  const gridWidth = useMemo(() => {
+    return Math.floor(containerSize.width * (state.viewState.splitterPosition / 100));
+  }, [containerSize.width, state.viewState.splitterPosition]);
+
+  const timelineWidth = useMemo(() => {
+    return containerSize.width - gridWidth;
+  }, [containerSize.width, gridWidth]);
+
+  // Handle zoom changes
+  const handleZoomChange = useCallback((direction: 'in' | 'out') => {
+    const currentZoom = state.viewState.zoomLevel;
+    const newZoom = direction === 'in' 
+      ? Math.min(state.zoomLevels.length - 1, currentZoom + 1)
+      : Math.max(0, currentZoom - 1);
+    
+    dispatch({ type: 'SET_ZOOM_LEVEL', payload: newZoom });
+  }, [state.viewState.zoomLevel, state.zoomLevels.length]);
+
+  // Theme-aware colors
+  const colors = useMemo(() => ({
+    background: themeWiseColor('#ffffff', '#1f2937', themeMode),
+    border: themeWiseColor('#e5e7eb', '#4b5563', themeMode),
+    timelineBackground: themeWiseColor('#f8f9fa', '#374151', themeMode),
+  }), [themeMode]);
+
+  // Render timeline header
+  const renderTimelineHeader = () => {
+    const currentZoom = state.zoomLevels[state.viewState.zoomLevel];
+    const totalDays = getDaysBetween(projectBounds.start, projectBounds.end);
+    const totalWidth = totalDays * state.timelineConfig.dayWidth;
+
+    return (
+      <div className="timeline-header border-b" style={{ 
+        height: (currentZoom.topTier.height || 30) + (currentZoom.bottomTier.height || 25),
+        backgroundColor: colors.timelineBackground,
+        borderColor: colors.border,
+      }}>
+        <VirtualTimeline
+          startDate={projectBounds.start}
+          endDate={projectBounds.end}
+          dayWidth={state.timelineConfig.dayWidth}
+          containerWidth={timelineWidth}
+          containerHeight={(currentZoom.topTier.height || 30) + (currentZoom.bottomTier.height || 25)}
+          onScroll={throttledScrollHandler}
+        >
+          {(date, index, style) => (
+            <div className="timeline-cell flex flex-col border-r text-xs text-center" style={{
+              ...style,
+              borderColor: colors.border,
+            }}>
+              <div className="top-tier border-b px-1 py-1" style={{
+                height: currentZoom.topTier.height || 30,
+                borderColor: colors.border,
+              }}>
+                {formatDateForUnit(date, currentZoom.topTier.unit)}
+              </div>
+              <div className="bottom-tier px-1 py-1" style={{
+                height: currentZoom.bottomTier.height || 25,
+              }}>
+                {formatDateForUnit(date, currentZoom.bottomTier.unit)}
+              </div>
+            </div>
+          )}
+        </VirtualTimeline>
+      </div>
+    );
+  };
+
+  // Render timeline content
+  const renderTimelineContent = () => {
+    const headerHeight = (state.zoomLevels[state.viewState.zoomLevel].topTier.height || 30) + 
+                        (state.zoomLevels[state.viewState.zoomLevel].bottomTier.height || 25);
+    const contentHeight = containerSize.height - headerHeight;
+
+    return (
+      <div className="timeline-content relative" style={{ height: contentHeight }}>
+        {/* Timeline markers (weekends, holidays, etc.) */}
+        <TimelineMarkers
+          startDate={projectBounds.start}
+          endDate={projectBounds.end}
+          dayWidth={state.timelineConfig.dayWidth}
+          containerHeight={contentHeight}
+          timelineConfig={state.timelineConfig}
+          holidays={state.timelineConfig.holidays}
+          showWeekends={state.timelineConfig.showWeekends}
+          showHolidays={true}
+          showToday={true}
+        />
+
+        {/* Task bars */}
+        <VirtualScrollContainer
+          items={state.tasks}
+          itemHeight={state.virtualScrollConfig.itemHeight}
+          containerHeight={contentHeight}
+          containerWidth={timelineWidth}
+          overscan={state.virtualScrollConfig.overscan}
+          onScroll={throttledScrollHandler}
+        >
+          {(task, index, style) => (
+            <DraggableTaskBar
+              key={task.id}
+              task={task}
+              timelineStart={projectBounds.start}
+              dayWidth={state.timelineConfig.dayWidth}
+              rowHeight={state.virtualScrollConfig.itemHeight}
+              index={index}
+              onTaskMove={debouncedTaskMove}
+              onTaskResize={debouncedTaskMove}
+              onProgressChange={debouncedProgressChange}
+              enableDragDrop={enableDragDrop}
+              enableResize={enableResize}
+              enableProgressEdit={enableProgressEdit}
+              readOnly={state.viewState.readOnly}
+            />
+          )}
+        </VirtualScrollContainer>
+      </div>
+    );
+  };
+
+  // Render toolbar
+  const renderToolbar = () => (
+    <div className="gantt-toolbar flex items-center justify-between p-2 border-b bg-gray-50 dark:bg-gray-800" style={{
+      borderColor: colors.border,
+    }}>
+      <div className="toolbar-left flex items-center space-x-2">
+        <button
+          onClick={() => handleZoomChange('out')}
+          disabled={state.viewState.zoomLevel === 0}
+          className="px-2 py-1 text-sm border rounded hover:bg-gray-100 dark:hover:bg-gray-700 disabled:opacity-50"
+        >
+          Zoom Out
+        </button>
+        <span className="text-sm text-gray-600 dark:text-gray-400">
+          {state.zoomLevels[state.viewState.zoomLevel].name}
+        </span>
+        <button
+          onClick={() => handleZoomChange('in')}
+          disabled={state.viewState.zoomLevel === state.zoomLevels.length - 1}
+          className="px-2 py-1 text-sm border rounded hover:bg-gray-100 dark:hover:bg-gray-700 disabled:opacity-50"
+        >
+          Zoom In
+        </button>
+      </div>
+      
+      <div className="toolbar-right flex items-center space-x-2 text-xs text-gray-500">
+        <span>Tasks: {state.tasks.length}</span>
+        <span>•</span>
+        <span>Render: {Math.round(metrics.renderTime)}ms</span>
+      </div>
+    </div>
+  );
+
+  // Performance monitoring
+  useEffect(() => {
+    startMeasure('render');
+    return () => endMeasure('render');
+  });
+
+  return (
+    <div
+      ref={containerRef}
+      className={`advanced-gantt-chart flex flex-col ${className}`}
+      style={{
+        height: '100%',
+        backgroundColor: colors.background,
+        ...style,
+      }}
+    >
+      {/* Toolbar */}
+      {renderToolbar()}
+
+      {/* Main content */}
+      <div className="gantt-content flex flex-1 overflow-hidden">
+        {/* Grid */}
+        <div className="gantt-grid-container" style={{ width: gridWidth }}>
+          <GanttGrid
+            tasks={state.tasks}
+            columns={state.columns}
+            rowHeight={state.virtualScrollConfig.itemHeight}
+            containerHeight={containerSize.height - 50} // Subtract toolbar height
+            selection={state.selectionState}
+            enableInlineEdit={enableInlineEdit}
+            onTaskClick={(task) => {
+              // Handle task selection
+              const newSelection = { ...state.selectionState, selectedTasks: [task.id] };
+              dispatch({ type: 'SET_SELECTION', payload: [task.id] });
+              onSelectionChange?.(newSelection);
+            }}
+            onTaskExpand={(taskId) => {
+              dispatch({ type: 'TOGGLE_TASK_EXPANSION', payload: taskId });
+            }}
+            onColumnResize={(field, width) => {
+              dispatch({ type: 'UPDATE_COLUMN_WIDTH', payload: { field, width } });
+              onColumnResize?.(field, width);
+            }}
+            onTaskUpdate={debouncedTaskUpdate}
+          />
+        </div>
+
+        {/* Timeline */}
+        <div className="gantt-timeline-container border-l" style={{ 
+          width: timelineWidth,
+          borderColor: colors.border,
+        }}>
+          {renderTimelineHeader()}
+          {renderTimelineContent()}
+        </div>
+      </div>
+    </div>
+  );
+};
+
+// Helper function to format dates based on unit
+function formatDateForUnit(date: Date, unit: string): string {
+  switch (unit) {
+    case 'year':
+      return date.getFullYear().toString();
+    case 'month':
+      return date.toLocaleDateString('en-US', { month: 'short', year: 'numeric' });
+    case 'week':
+      return `W${getWeekNumber(date)}`;
+    case 'day':
+      return date.getDate().toString();
+    case 'hour':
+      return date.getHours().toString().padStart(2, '0');
+    default:
+      return '';
+  }
+}
+
+function getWeekNumber(date: Date): number {
+  const d = new Date(Date.UTC(date.getFullYear(), date.getMonth(), date.getDate()));
+  const dayNum = d.getUTCDay() || 7;
+  d.setUTCDate(d.getUTCDate() + 4 - dayNum);
+  const yearStart = new Date(Date.UTC(d.getUTCFullYear(), 0, 1));
+  return Math.ceil((((d.getTime() - yearStart.getTime()) / 86400000) + 1) / 7);
+}
+
+export default AdvancedGanttChart;

worklenz-frontend/src/components/advanced-gantt/AdvancedGanttDemo.tsx

@@ -0,0 +1,668 @@
+import React, { useState, useMemo } from 'react';
+import { Button, Space, message, Card } from 'antd';
+import AdvancedGanttChart from './AdvancedGanttChart';
+import { GanttTask, ColumnConfig } from '../../types/advanced-gantt.types';
+import { useAppSelector } from '../../hooks/useAppSelector';
+import { holidayPresets, workingDayPresets } from './TimelineMarkers';
+
+// Enhanced sample data with more realistic project structure
+const generateSampleTasks = (): GanttTask[] => {
+  const baseDate = new Date(2024, 11, 1); // December 1, 2024
+  
+  return [
+    // Project Phase 1: Planning & Design
+    {
+      id: 'project-1',
+      name: '🚀 Web Platform Development',
+      startDate: new Date(2024, 11, 1),
+      endDate: new Date(2025, 2, 31),
+      progress: 45,
+      type: 'project',
+      status: 'in-progress',
+      priority: 'high',
+      color: '#1890ff',
+      hasChildren: true,
+      isExpanded: true,
+      level: 0,
+    },
+    {
+      id: 'planning-phase',
+      name: '📋 Planning & Analysis Phase',
+      startDate: new Date(2024, 11, 1),
+      endDate: new Date(2024, 11, 20),
+      progress: 85,
+      type: 'project',
+      status: 'in-progress',
+      priority: 'high',
+      parent: 'project-1',
+      color: '#52c41a',
+      hasChildren: true,
+      isExpanded: true,
+      level: 1,
+    },
+    {
+      id: 'requirements-analysis',
+      name: 'Requirements Gathering & Analysis',
+      startDate: new Date(2024, 11, 1),
+      endDate: new Date(2024, 11, 8),
+      progress: 100,
+      type: 'task',
+      status: 'completed',
+      priority: 'high',
+      parent: 'planning-phase',
+      assignee: {
+        id: 'user-1',
+        name: 'Alice Johnson',
+        avatar: 'https://ui-avatars.com/api/?name=Alice+Johnson&background=1890ff&color=fff',
+      },
+      tags: ['research', 'documentation'],
+      level: 2,
+    },
+    {
+      id: 'technical-architecture',
+      name: 'Technical Architecture Design',
+      startDate: new Date(2024, 11, 8),
+      endDate: new Date(2024, 11, 18),
+      progress: 75,
+      type: 'task',
+      status: 'in-progress',
+      priority: 'high',
+      parent: 'planning-phase',
+      assignee: {
+        id: 'user-2',
+        name: 'Bob Smith',
+        avatar: 'https://ui-avatars.com/api/?name=Bob+Smith&background=52c41a&color=fff',
+      },
+      dependencies: ['requirements-analysis'],
+      tags: ['architecture', 'design'],
+      level: 2,
+    },
+    {
+      id: 'ui-ux-design',
+      name: 'UI/UX Design & Prototyping',
+      startDate: new Date(2024, 11, 10),
+      endDate: new Date(2024, 11, 20),
+      progress: 60,
+      type: 'task',
+      status: 'in-progress',
+      priority: 'medium',
+      parent: 'planning-phase',
+      assignee: {
+        id: 'user-3',
+        name: 'Carol Davis',
+        avatar: 'https://ui-avatars.com/api/?name=Carol+Davis&background=faad14&color=fff',
+      },
+      dependencies: ['requirements-analysis'],
+      tags: ['design', 'prototype'],
+      level: 2,
+    },
+    {
+      id: 'milestone-planning-complete',
+      name: '🎯 Planning Phase Complete',
+      startDate: new Date(2024, 11, 20),
+      endDate: new Date(2024, 11, 20),
+      progress: 0,
+      type: 'milestone',
+      status: 'not-started',
+      priority: 'critical',
+      parent: 'planning-phase',
+      dependencies: ['technical-architecture', 'ui-ux-design'],
+      level: 2,
+    },
+
+    // Development Phase
+    {
+      id: 'development-phase',
+      name: '⚡ Development Phase',
+      startDate: new Date(2024, 11, 21),
+      endDate: new Date(2025, 1, 28),
+      progress: 35,
+      type: 'project',
+      status: 'in-progress',
+      priority: 'high',
+      parent: 'project-1',
+      color: '#722ed1',
+      hasChildren: true,
+      isExpanded: true,
+      level: 1,
+    },
+    {
+      id: 'backend-development',
+      name: 'Backend API Development',
+      startDate: new Date(2024, 11, 21),
+      endDate: new Date(2025, 1, 15),
+      progress: 45,
+      type: 'task',
+      status: 'in-progress',
+      priority: 'high',
+      parent: 'development-phase',
+      assignee: {
+        id: 'user-4',
+        name: 'David Wilson',
+        avatar: 'https://ui-avatars.com/api/?name=David+Wilson&background=722ed1&color=fff',
+      },
+      dependencies: ['milestone-planning-complete'],
+      tags: ['backend', 'api'],
+      level: 2,
+    },
+    {
+      id: 'frontend-development',
+      name: 'Frontend React Application',
+      startDate: new Date(2025, 0, 5),
+      endDate: new Date(2025, 1, 25),
+      progress: 25,
+      type: 'task',
+      status: 'in-progress',
+      priority: 'high',
+      parent: 'development-phase',
+      assignee: {
+        id: 'user-5',
+        name: 'Eva Brown',
+        avatar: 'https://ui-avatars.com/api/?name=Eva+Brown&background=ff7a45&color=fff',
+      },
+      dependencies: ['backend-development'],
+      tags: ['frontend', 'react'],
+      level: 2,
+    },
+    {
+      id: 'database-setup',
+      name: 'Database Schema & Migration',
+      startDate: new Date(2024, 11, 21),
+      endDate: new Date(2025, 0, 10),
+      progress: 80,
+      type: 'task',
+      status: 'in-progress',
+      priority: 'medium',
+      parent: 'development-phase',
+      assignee: {
+        id: 'user-6',
+        name: 'Frank Miller',
+        avatar: 'https://ui-avatars.com/api/?name=Frank+Miller&background=13c2c2&color=fff',
+      },
+      dependencies: ['milestone-planning-complete'],
+      tags: ['database', 'migration'],
+      level: 2,
+    },
+
+    // Testing Phase
+    {
+      id: 'testing-phase',
+      name: '🧪 Testing & QA Phase',
+      startDate: new Date(2025, 2, 1),
+      endDate: new Date(2025, 2, 20),
+      progress: 0,
+      type: 'project',
+      status: 'not-started',
+      priority: 'high',
+      parent: 'project-1',
+      color: '#fa8c16',
+      hasChildren: true,
+      isExpanded: false,
+      level: 1,
+    },
+    {
+      id: 'unit-testing',
+      name: 'Unit Testing Implementation',
+      startDate: new Date(2025, 2, 1),
+      endDate: new Date(2025, 2, 10),
+      progress: 0,
+      type: 'task',
+      status: 'not-started',
+      priority: 'high',
+      parent: 'testing-phase',
+      assignee: {
+        id: 'user-7',
+        name: 'Grace Lee',
+        avatar: 'https://ui-avatars.com/api/?name=Grace+Lee&background=fa8c16&color=fff',
+      },
+      dependencies: ['frontend-development'],
+      tags: ['testing', 'unit'],
+      level: 2,
+    },
+    {
+      id: 'integration-testing',
+      name: 'Integration & E2E Testing',
+      startDate: new Date(2025, 2, 8),
+      endDate: new Date(2025, 2, 18),
+      progress: 0,
+      type: 'task',
+      status: 'not-started',
+      priority: 'high',
+      parent: 'testing-phase',
+      assignee: {
+        id: 'user-8',
+        name: 'Henry Clark',
+        avatar: 'https://ui-avatars.com/api/?name=Henry+Clark&background=eb2f96&color=fff',
+      },
+      dependencies: ['unit-testing'],
+      tags: ['testing', 'integration'],
+      level: 2,
+    },
+    {
+      id: 'milestone-beta-ready',
+      name: '🎯 Beta Release Ready',
+      startDate: new Date(2025, 2, 20),
+      endDate: new Date(2025, 2, 20),
+      progress: 0,
+      type: 'milestone',
+      status: 'not-started',
+      priority: 'critical',
+      parent: 'testing-phase',
+      dependencies: ['integration-testing'],
+      level: 2,
+    },
+
+    // Deployment Phase
+    {
+      id: 'deployment-phase',
+      name: '🚀 Deployment & Launch',
+      startDate: new Date(2025, 2, 21),
+      endDate: new Date(2025, 2, 31),
+      progress: 0,
+      type: 'project',
+      status: 'not-started',
+      priority: 'critical',
+      parent: 'project-1',
+      color: '#f5222d',
+      hasChildren: true,
+      isExpanded: false,
+      level: 1,
+    },
+    {
+      id: 'production-deployment',
+      name: 'Production Environment Setup',
+      startDate: new Date(2025, 2, 21),
+      endDate: new Date(2025, 2, 25),
+      progress: 0,
+      type: 'task',
+      status: 'not-started',
+      priority: 'critical',
+      parent: 'deployment-phase',
+      assignee: {
+        id: 'user-9',
+        name: 'Ivy Taylor',
+        avatar: 'https://ui-avatars.com/api/?name=Ivy+Taylor&background=f5222d&color=fff',
+      },
+      dependencies: ['milestone-beta-ready'],
+      tags: ['deployment', 'production'],
+      level: 2,
+    },
+    {
+      id: 'go-live',
+      name: 'Go Live & Monitoring',
+      startDate: new Date(2025, 2, 26),
+      endDate: new Date(2025, 2, 31),
+      progress: 0,
+      type: 'task',
+      status: 'not-started',
+      priority: 'critical',
+      parent: 'deployment-phase',
+      assignee: {
+        id: 'user-10',
+        name: 'Jack Anderson',
+        avatar: 'https://ui-avatars.com/api/?name=Jack+Anderson&background=2f54eb&color=fff',
+      },
+      dependencies: ['production-deployment'],
+      tags: ['launch', 'monitoring'],
+      level: 2,
+    },
+    {
+      id: 'milestone-project-complete',
+      name: '🎉 Project Launch Complete',
+      startDate: new Date(2025, 2, 31),
+      endDate: new Date(2025, 2, 31),
+      progress: 0,
+      type: 'milestone',
+      status: 'not-started',
+      priority: 'critical',
+      parent: 'deployment-phase',
+      dependencies: ['go-live'],
+      level: 2,
+    },
+  ];
+};
+
+// Enhanced column configuration
+const sampleColumns: ColumnConfig[] = [
+  { 
+    field: 'name', 
+    title: 'Task / Phase Name', 
+    width: 300, 
+    minWidth: 200, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'text'
+  },
+  { 
+    field: 'assignee', 
+    title: 'Assignee', 
+    width: 150, 
+    minWidth: 120, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true 
+  },
+  { 
+    field: 'startDate', 
+    title: 'Start Date', 
+    width: 120, 
+    minWidth: 100, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'date'
+  },
+  { 
+    field: 'endDate', 
+    title: 'End Date', 
+    width: 120, 
+    minWidth: 100, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'date'
+  },
+  { 
+    field: 'duration', 
+    title: 'Duration', 
+    width: 80, 
+    minWidth: 60, 
+    resizable: true, 
+    sortable: false, 
+    fixed: true,
+    align: 'center'
+  },
+  { 
+    field: 'progress', 
+    title: 'Progress', 
+    width: 120, 
+    minWidth: 100, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'number'
+  },
+  { 
+    field: 'status', 
+    title: 'Status', 
+    width: 100, 
+    minWidth: 80, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'select',
+    editorOptions: [
+      { value: 'not-started', label: 'Not Started' },
+      { value: 'in-progress', label: 'In Progress' },
+      { value: 'completed', label: 'Completed' },
+      { value: 'on-hold', label: 'On Hold' },
+      { value: 'overdue', label: 'Overdue' },
+    ]
+  },
+  { 
+    field: 'priority', 
+    title: 'Priority', 
+    width: 100, 
+    minWidth: 80, 
+    resizable: true, 
+    sortable: true, 
+    fixed: true,
+    editor: 'select',
+    editorOptions: [
+      { value: 'low', label: 'Low' },
+      { value: 'medium', label: 'Medium' },
+      { value: 'high', label: 'High' },
+      { value: 'critical', label: 'Critical' },
+    ]
+  },
+];
+
+const AdvancedGanttDemo: React.FC = () => {
+  const [tasks, setTasks] = useState<GanttTask[]>(generateSampleTasks());
+  const [selectedTasks, setSelectedTasks] = useState<string[]>([]);
+  const themeMode = useAppSelector(state => state.themeReducer.mode);
+
+  const handleTaskUpdate = (taskId: string, updates: Partial<GanttTask>) => {
+    setTasks(prevTasks => 
+      prevTasks.map(task => 
+        task.id === taskId ? { ...task, ...updates } : task
+      )
+    );
+    message.success(`Task "${tasks.find(t => t.id === taskId)?.name}" updated`);
+  };
+
+  const handleTaskMove = (taskId: string, newDates: { start: Date; end: Date }) => {
+    setTasks(prevTasks => 
+      prevTasks.map(task => 
+        task.id === taskId 
+          ? { ...task, startDate: newDates.start, endDate: newDates.end }
+          : task
+      )
+    );
+    message.info(`Task moved: ${newDates.start.toLocaleDateString()} - ${newDates.end.toLocaleDateString()}`);
+  };
+
+  const handleProgressChange = (taskId: string, progress: number) => {
+    setTasks(prevTasks => 
+      prevTasks.map(task => 
+        task.id === taskId ? { ...task, progress } : task
+      )
+    );
+    message.info(`Progress updated: ${Math.round(progress)}%`);
+  };
+
+  const handleSelectionChange = (selection: any) => {
+    setSelectedTasks(selection.selectedTasks);
+  };
+
+  const resetToSampleData = () => {
+    setTasks(generateSampleTasks());
+    setSelectedTasks([]);
+    message.info('Gantt chart reset to sample data');
+  };
+
+  const addSampleTask = () => {
+    const newTask: GanttTask = {
+      id: `task-${Date.now()}`,
+      name: `New Task ${tasks.length + 1}`,
+      startDate: new Date(),
+      endDate: new Date(Date.now() + 7 * 24 * 60 * 60 * 1000), // +7 days
+      progress: 0,
+      type: 'task',
+      status: 'not-started',
+      priority: 'medium',
+      level: 0,
+    };
+    setTasks(prev => [...prev, newTask]);
+    message.success('New task added');
+  };
+
+  const deleteSelectedTasks = () => {
+    if (selectedTasks.length === 0) {
+      message.warning('No tasks selected');
+      return;
+    }
+    
+    setTasks(prev => prev.filter(task => !selectedTasks.includes(task.id)));
+    setSelectedTasks([]);
+    message.success(`${selectedTasks.length} task(s) deleted`);
+  };
+
+  const taskStats = useMemo(() => {
+    const total = tasks.length;
+    const completed = tasks.filter(t => t.status === 'completed').length;
+    const inProgress = tasks.filter(t => t.status === 'in-progress').length;
+    const overdue = tasks.filter(t => t.status === 'overdue').length;
+    const avgProgress = tasks.reduce((sum, t) => sum + t.progress, 0) / total;
+
+    return { total, completed, inProgress, overdue, avgProgress };
+  }, [tasks]);
+
+  return (
+    <div className="advanced-gantt-demo p-6 bg-gray-50 dark:bg-gray-900 min-h-screen">
+      {/* Header */}
+      <div className="bg-white dark:bg-gray-800 border border-gray-200 dark:border-gray-700 rounded-lg shadow-sm mb-4">
+        <div className="p-6">
+          <div className="flex justify-between items-start mb-4">
+            <div>
+              <h1 className="text-3xl font-bold text-gray-900 dark:text-gray-100 mb-2">
+                🚀 Advanced Gantt Chart Demo
+              </h1>
+              <p className="text-gray-600 dark:text-gray-400 mb-4">
+                Professional Gantt chart with draggable tasks, virtual scrolling, holiday markers, 
+                and performance optimizations for modern project management.
+              </p>
+            </div>
+            
+            <div className="flex flex-col items-end space-y-2">
+              <Space>
+                <Button 
+                  onClick={addSampleTask}
+                  type="primary"
+                  className="dark:border-gray-600"
+                >
+                  Add Task
+                </Button>
+                <Button 
+                  onClick={deleteSelectedTasks}
+                  danger
+                  disabled={selectedTasks.length === 0}
+                  className="dark:border-gray-600"
+                >
+                  Delete Selected ({selectedTasks.length})
+                </Button>
+                <Button 
+                  onClick={resetToSampleData}
+                  className="dark:border-gray-600 dark:text-gray-300"
+                >
+                  Reset Data
+                </Button>
+              </Space>
+            </div>
+          </div>
+
+          {/* Project Statistics */}
+          <div className="grid grid-cols-2 md:grid-cols-4 gap-4">
+            <div className="bg-blue-50 dark:bg-blue-900 dark:bg-opacity-20 rounded-lg p-3">
+              <div className="text-blue-600 dark:text-blue-400 text-sm font-medium">Total Tasks</div>
+              <div className="text-2xl font-bold text-blue-700 dark:text-blue-300">{taskStats.total}</div>
+            </div>
+            <div className="bg-green-50 dark:bg-green-900 dark:bg-opacity-20 rounded-lg p-3">
+              <div className="text-green-600 dark:text-green-400 text-sm font-medium">Completed</div>
+              <div className="text-2xl font-bold text-green-700 dark:text-green-300">{taskStats.completed}</div>
+            </div>
+            <div className="bg-yellow-50 dark:bg-yellow-900 dark:bg-opacity-20 rounded-lg p-3">
+              <div className="text-yellow-600 dark:text-yellow-400 text-sm font-medium">In Progress</div>
+              <div className="text-2xl font-bold text-yellow-700 dark:text-yellow-300">{taskStats.inProgress}</div>
+            </div>
+            <div className="bg-purple-50 dark:bg-purple-900 dark:bg-opacity-20 rounded-lg p-3">
+              <div className="text-purple-600 dark:text-purple-400 text-sm font-medium">Avg Progress</div>
+              <div className="text-2xl font-bold text-purple-700 dark:text-purple-300">
+                {Math.round(taskStats.avgProgress)}%
+              </div>
+            </div>
+          </div>
+        </div>
+      </div>
+
+      {/* Gantt Chart */}
+      <div className="bg-white dark:bg-gray-800 border border-gray-200 dark:border-gray-700 rounded-lg shadow-sm" style={{ height: '70vh' }}>
+        <AdvancedGanttChart
+          tasks={tasks}
+          columns={sampleColumns}
+          timelineConfig={{
+            showWeekends: true,
+            showNonWorkingDays: true,
+            holidays: holidayPresets.US,
+            workingDays: workingDayPresets.standard,
+            dayWidth: 30,
+          }}
+          onTaskUpdate={handleTaskUpdate}
+          onTaskMove={handleTaskMove}
+          onProgressChange={handleProgressChange}
+          onSelectionChange={handleSelectionChange}
+          enableDragDrop={true}
+          enableResize={true}
+          enableProgressEdit={true}
+          enableInlineEdit={true}
+          enableVirtualScrolling={true}
+          className="h-full"
+        />
+      </div>
+
+      {/* Feature List */}
+      <div className="bg-white dark:bg-gray-800 border border-gray-200 dark:border-gray-700 rounded-lg shadow-sm mt-4">
+        <div className="p-6">
+          <h3 className="text-lg font-semibold text-gray-900 dark:text-gray-100 mb-3">
+            ✨ Advanced Features Demonstrated
+          </h3>
+          <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-4 text-sm">
+            <div className="space-y-2">
+              <h4 className="font-medium text-gray-900 dark:text-gray-100">Performance & UX</h4>
+              <ul className="space-y-1 text-gray-600 dark:text-gray-400">
+                <li>• Virtual scrolling for 1000+ tasks</li>
+                <li>• Smooth 60fps drag & drop</li>
+                <li>• Debounced updates</li>
+                <li>• Memory-optimized rendering</li>
+                <li>• Responsive design</li>
+              </ul>
+            </div>
+            <div className="space-y-2">
+              <h4 className="font-medium text-gray-900 dark:text-gray-100">Gantt Features</h4>
+              <ul className="space-y-1 text-gray-600 dark:text-gray-400">
+                <li>• Draggable task bars</li>
+                <li>• Resizable task duration</li>
+                <li>• Progress editing</li>
+                <li>• Multi-level hierarchy</li>
+                <li>• Task dependencies</li>
+              </ul>
+            </div>
+            <div className="space-y-2">
+              <h4 className="font-medium text-gray-900 dark:text-gray-100">Timeline & Markers</h4>
+              <ul className="space-y-1 text-gray-600 dark:text-gray-400">
+                <li>• Weekend & holiday markers</li>
+                <li>• Working day indicators</li>
+                <li>• Today line</li>
+                <li>• Multi-tier timeline</li>
+                <li>• Zoom levels (Year/Month/Week/Day)</li>
+              </ul>
+            </div>
+            <div className="space-y-2">
+              <h4 className="font-medium text-gray-900 dark:text-gray-100">Grid Features</h4>
+              <ul className="space-y-1 text-gray-600 dark:text-gray-400">
+                <li>• Fixed columns layout</li>
+                <li>• Inline editing</li>
+                <li>• Column resizing</li>
+                <li>• Multi-select</li>
+                <li>• Hierarchical tree view</li>
+              </ul>
+            </div>
+            <div className="space-y-2">
+              <h4 className="font-medium text-gray-900 dark:text-gray-100">UI/UX</h4>
+              <ul className="space-y-1 text-gray-600 dark:text-gray-400">
+                <li>• Dark/Light theme support</li>
+                <li>• Tailwind CSS styling</li>
+                <li>• Consistent with Worklenz</li>
+                <li>• Accessibility features</li>
+                <li>• Mobile responsive</li>
+              </ul>
+            </div>
+            <div className="space-y-2">
+              <h4 className="font-medium text-gray-900 dark:text-gray-100">Architecture</h4>
+              <ul className="space-y-1 text-gray-600 dark:text-gray-400">
+                <li>• Modern React patterns</li>
+                <li>• TypeScript safety</li>
+                <li>• Optimized performance</li>
+                <li>• Enterprise features</li>
+                <li>• Best practices 2025</li>
+              </ul>
+            </div>
+          </div>
+        </div>
+      </div>
+    </div>
+  );
+};
+
+export default AdvancedGanttDemo;

worklenz-frontend/src/components/advanced-gantt/DraggableTaskBar.tsx

@@ -0,0 +1,304 @@
+import React, { useState, useRef, useCallback, useMemo } from 'react';
+import { GanttTask, DragState } from '../../types/advanced-gantt.types';
+import { useAppSelector } from '../../hooks/useAppSelector';
+import { themeWiseColor } from '../../utils/themeWiseColor';
+import { useDateCalculations } from '../../utils/gantt-performance';
+
+interface DraggableTaskBarProps {
+  task: GanttTask;
+  timelineStart: Date;
+  dayWidth: number;
+  rowHeight: number;
+  index: number;
+  onTaskMove?: (taskId: string, newDates: { start: Date; end: Date }) => void;
+  onTaskResize?: (taskId: string, newDates: { start: Date; end: Date }) => void;
+  onProgressChange?: (taskId: string, progress: number) => void;
+  onTaskClick?: (task: GanttTask) => void;
+  onTaskDoubleClick?: (task: GanttTask) => void;
+  enableDragDrop?: boolean;
+  enableResize?: boolean;
+  enableProgressEdit?: boolean;
+  readOnly?: boolean;
+}
+
+const DraggableTaskBar: React.FC<DraggableTaskBarProps> = ({
+  task,
+  timelineStart,
+  dayWidth,
+  rowHeight,
+  index,
+  onTaskMove,
+  onTaskResize,
+  onProgressChange,
+  onTaskClick,
+  onTaskDoubleClick,
+  enableDragDrop = true,
+  enableResize = true,
+  enableProgressEdit = true,
+  readOnly = false,
+}) => {
+  const [dragState, setDragState] = useState<DragState | null>(null);
+  const [hoverState, setHoverState] = useState<string | null>(null);
+  const taskBarRef = useRef<HTMLDivElement>(null);
+  const themeMode = useAppSelector(state => state.themeReducer.mode);
+  const { getDaysBetween, addDays } = useDateCalculations();
+
+  // Calculate task position and dimensions
+  const taskPosition = useMemo(() => {
+    const startDays = getDaysBetween(timelineStart, task.startDate);
+    const duration = getDaysBetween(task.startDate, task.endDate);
+    
+    return {
+      x: startDays * dayWidth,
+      width: Math.max(dayWidth * 0.5, duration * dayWidth),
+      y: index * rowHeight + 8, // 8px padding
+      height: rowHeight - 16, // 16px total padding
+    };
+  }, [task.startDate, task.endDate, timelineStart, dayWidth, rowHeight, index, getDaysBetween]);
+
+  // Theme-aware colors
+  const colors = useMemo(() => {
+    const baseColor = task.color || getDefaultTaskColor(task.status);
+    return {
+      background: themeWiseColor(baseColor, adjustColorForDarkMode(baseColor), themeMode),
+      border: themeWiseColor(darkenColor(baseColor, 0.2), lightenColor(baseColor, 0.2), themeMode),
+      progress: themeWiseColor('#52c41a', '#34d399', themeMode),
+      text: themeWiseColor('#ffffff', '#f9fafb', themeMode),
+      hover: themeWiseColor(lightenColor(baseColor, 0.1), darkenColor(baseColor, 0.1), themeMode),
+    };
+  }, [task.color, task.status, themeMode]);
+
+  // Mouse event handlers
+  const handleMouseDown = useCallback((e: React.MouseEvent, dragType: DragState['dragType']) => {
+    if (readOnly || !enableDragDrop) return;
+    
+    e.preventDefault();
+    e.stopPropagation();
+
+    const rect = taskBarRef.current?.getBoundingClientRect();
+    if (!rect) return;
+
+    setDragState({
+      isDragging: true,
+      dragType,
+      taskId: task.id,
+      initialPosition: { x: e.clientX, y: e.clientY },
+      currentPosition: { x: e.clientX, y: e.clientY },
+      initialDates: { start: task.startDate, end: task.endDate },
+      initialProgress: task.progress,
+      snapToGrid: true,
+    });
+
+    // Add global mouse event listeners
+    const handleMouseMove = (moveEvent: MouseEvent) => {
+      handleMouseMove_Internal(moveEvent, dragType);
+    };
+
+    const handleMouseUp = () => {
+      handleMouseUp_Internal();
+      document.removeEventListener('mousemove', handleMouseMove);
+      document.removeEventListener('mouseup', handleMouseUp);
+    };
+
+    document.addEventListener('mousemove', handleMouseMove);
+    document.addEventListener('mouseup', handleMouseUp);
+  }, [readOnly, enableDragDrop, task]);
+
+  const handleMouseMove_Internal = useCallback((e: MouseEvent, dragType: DragState['dragType']) => {
+    if (!dragState) return;
+
+    const deltaX = e.clientX - dragState.initialPosition.x;
+    const deltaDays = Math.round(deltaX / dayWidth);
+
+    let newStartDate = task.startDate;
+    let newEndDate = task.endDate;
+
+    switch (dragType) {
+      case 'move':
+        newStartDate = addDays(dragState.initialDates.start, deltaDays);
+        newEndDate = addDays(dragState.initialDates.end, deltaDays);
+        break;
+      
+      case 'resize-start':
+        newStartDate = addDays(dragState.initialDates.start, deltaDays);
+        // Ensure minimum duration
+        if (newStartDate >= newEndDate) {
+          newStartDate = addDays(newEndDate, -1);
+        }
+        break;
+      
+      case 'resize-end':
+        newEndDate = addDays(dragState.initialDates.end, deltaDays);
+        // Ensure minimum duration
+        if (newEndDate <= newStartDate) {
+          newEndDate = addDays(newStartDate, 1);
+        }
+        break;
+      
+      case 'progress':
+        if (enableProgressEdit) {
+          const progressDelta = deltaX / taskPosition.width;
+          const newProgress = Math.max(0, Math.min(100, (dragState.initialProgress || 0) + progressDelta * 100));
+          onProgressChange?.(task.id, newProgress);
+        }
+        return;
+    }
+
+    // Update drag state
+    setDragState(prev => prev ? {
+      ...prev,
+      currentPosition: { x: e.clientX, y: e.clientY },
+    } : null);
+
+    // Call appropriate handler
+    if (dragType === 'move') {
+      onTaskMove?.(task.id, { start: newStartDate, end: newEndDate });
+    } else if (dragType.startsWith('resize')) {
+      onTaskResize?.(task.id, { start: newStartDate, end: newEndDate });
+    }
+  }, [dragState, dayWidth, task, taskPosition.width, enableProgressEdit, onTaskMove, onTaskResize, onProgressChange, addDays]);
+
+  const handleMouseUp_Internal = useCallback(() => {
+    setDragState(null);
+  }, []);
+
+  const handleClick = useCallback((e: React.MouseEvent) => {
+    e.stopPropagation();
+    onTaskClick?.(task);
+  }, [task, onTaskClick]);
+
+  const handleDoubleClick = useCallback((e: React.MouseEvent) => {
+    e.stopPropagation();
+    onTaskDoubleClick?.(task);
+  }, [task, onTaskDoubleClick]);
+
+  // Render task bar with handles
+  const renderTaskBar = () => {
+    const isSelected = false; // TODO: Get from selection state
+    const isDragging = dragState?.isDragging || false;
+
+    return (
+      <div
+        ref={taskBarRef}
+        className={`task-bar relative cursor-pointer select-none transition-all duration-200 ${
+          isDragging ? 'z-10 shadow-lg' : ''
+        } ${isSelected ? 'ring-2 ring-blue-500 ring-opacity-50' : ''}`}
+        style={{
+          position: 'absolute',
+          left: taskPosition.x,
+          top: taskPosition.y,
+          width: taskPosition.width,
+          height: taskPosition.height,
+          backgroundColor: dragState?.isDragging ? colors.hover : colors.background,
+          border: `1px solid ${colors.border}`,
+          borderRadius: '4px',
+          transform: isDragging ? 'translateY(-2px)' : 'none',
+          boxShadow: isDragging ? '0 4px 12px rgba(0,0,0,0.15)' : '0 1px 3px rgba(0,0,0,0.1)',
+        }}
+        onClick={handleClick}
+        onDoubleClick={handleDoubleClick}
+        onMouseEnter={() => setHoverState('task')}
+        onMouseLeave={() => setHoverState(null)}
+        onMouseDown={(e) => handleMouseDown(e, 'move')}
+      >
+        {/* Progress bar */}
+        <div
+          className="progress-bar absolute inset-0 rounded-l"
+          style={{
+            width: `${task.progress}%`,
+            backgroundColor: colors.progress,
+            opacity: 0.7,
+            borderRadius: '3px 0 0 3px',
+          }}
+        />
+
+        {/* Task content */}
+        <div className="task-content relative z-10 h-full flex items-center px-2">
+          <span
+            className="task-name text-xs font-medium truncate"
+            style={{ color: colors.text }}
+          >
+            {task.name}
+          </span>
+          
+          {/* Duration display for smaller tasks */}
+          {taskPosition.width < 100 && (
+            <span
+              className="task-duration text-xs ml-auto"
+              style={{ color: colors.text, opacity: 0.8 }}
+            >
+              {getDaysBetween(task.startDate, task.endDate)}d
+            </span>
+          )}
+        </div>
+
+        {/* Resize handles */}
+        {enableResize && !readOnly && hoverState === 'task' && (
+          <>
+            {/* Left resize handle */}
+            <div
+              className="resize-handle-left absolute left-0 top-0 w-1 h-full cursor-ew-resize bg-white bg-opacity-50 hover:bg-opacity-80"
+              onMouseDown={(e) => handleMouseDown(e, 'resize-start')}
+              onMouseEnter={() => setHoverState('resize-start')}
+            />
+            
+            {/* Right resize handle */}
+            <div
+              className="resize-handle-right absolute right-0 top-0 w-1 h-full cursor-ew-resize bg-white bg-opacity-50 hover:bg-opacity-80"
+              onMouseDown={(e) => handleMouseDown(e, 'resize-end')}
+              onMouseEnter={() => setHoverState('resize-end')}
+            />
+          </>
+        )}
+
+        {/* Progress handle */}
+        {enableProgressEdit && !readOnly && hoverState === 'task' && (
+          <div
+            className="progress-handle absolute top-0 h-full w-1 cursor-ew-resize bg-blue-500 opacity-75"
+            style={{ left: `${task.progress}%` }}
+            onMouseDown={(e) => handleMouseDown(e, 'progress')}
+            onMouseEnter={() => setHoverState('progress')}
+          />
+        )}
+
+        {/* Task type indicator */}
+        {task.type === 'milestone' && (
+          <div
+            className="milestone-indicator absolute -top-1 -right-1 w-3 h-3 transform rotate-45"
+            style={{ backgroundColor: colors.border }}
+          />
+        )}
+      </div>
+    );
+  };
+
+  return renderTaskBar();
+};
+
+// Helper functions
+function getDefaultTaskColor(status: GanttTask['status']): string {
+  switch (status) {
+    case 'completed': return '#52c41a';
+    case 'in-progress': return '#1890ff';
+    case 'overdue': return '#ff4d4f';
+    case 'on-hold': return '#faad14';
+    default: return '#d9d9d9';
+  }
+}
+
+function darkenColor(color: string, amount: number): string {
+  // Simple color darkening - in a real app, use a proper color manipulation library
+  return color;
+}
+
+function lightenColor(color: string, amount: number): string {
+  // Simple color lightening - in a real app, use a proper color manipulation library
+  return color;
+}
+
+function adjustColorForDarkMode(color: string): string {
+  // Adjust color for dark mode - in a real app, use a proper color manipulation library
+  return color;
+}
+
+export default DraggableTaskBar;

worklenz-frontend/src/components/advanced-gantt/GanttGrid.tsx

@@ -0,0 +1,492 @@
+import React, { useMemo, useRef, useState, useCallback } from 'react';
+import { GanttTask, ColumnConfig, SelectionState } from '../../types/advanced-gantt.types';
+import { useAppSelector } from '../../hooks/useAppSelector';
+import { themeWiseColor } from '../../utils/themeWiseColor';
+import { ChevronRightIcon, ChevronDownIcon } from '@heroicons/react/24/outline';
+import { CalendarIcon, UserIcon, FlagIcon } from '@heroicons/react/24/solid';
+
+interface GanttGridProps {
+  tasks: GanttTask[];
+  columns: ColumnConfig[];
+  rowHeight: number;
+  containerHeight: number;
+  selection: SelectionState;
+  enableInlineEdit?: boolean;
+  enableMultiSelect?: boolean;
+  onTaskClick?: (task: GanttTask, event: React.MouseEvent) => void;
+  onTaskDoubleClick?: (task: GanttTask) => void;
+  onTaskExpand?: (taskId: string) => void;
+  onSelectionChange?: (selection: SelectionState) => void;
+  onColumnResize?: (columnField: string, newWidth: number) => void;
+  onTaskUpdate?: (taskId: string, field: string, value: any) => void;
+  className?: string;
+}
+
+const GanttGrid: React.FC<GanttGridProps> = ({
+  tasks,
+  columns,
+  rowHeight,
+  containerHeight,
+  selection,
+  enableInlineEdit = true,
+  enableMultiSelect = true,
+  onTaskClick,
+  onTaskDoubleClick,
+  onTaskExpand,
+  onSelectionChange,
+  onColumnResize,
+  onTaskUpdate,
+  className = '',
+}) => {
+  const [editingCell, setEditingCell] = useState<{ taskId: string; field: string } | null>(null);
+  const [columnWidths, setColumnWidths] = useState<Record<string, number>>(
+    columns.reduce((acc, col) => ({ ...acc, [col.field]: col.width }), {})
+  );
+  const gridRef = useRef<HTMLDivElement>(null);
+  const themeMode = useAppSelector(state => state.themeReducer.mode);
+
+  // Theme-aware colors
+  const colors = useMemo(() => ({
+    background: themeWiseColor('#ffffff', '#1f2937', themeMode),
+    alternateRow: themeWiseColor('#f9fafb', '#374151', themeMode),
+    border: themeWiseColor('#e5e7eb', '#4b5563', themeMode),
+    text: themeWiseColor('#111827', '#f9fafb', themeMode),
+    textSecondary: themeWiseColor('#6b7280', '#d1d5db', themeMode),
+    selected: themeWiseColor('#eff6ff', '#1e3a8a', themeMode),
+    hover: themeWiseColor('#f3f4f6', '#4b5563', themeMode),
+    headerBg: themeWiseColor('#f8f9fa', '#374151', themeMode),
+  }), [themeMode]);
+
+  // Calculate total grid width
+  const totalWidth = useMemo(() => {
+    return columns.reduce((sum, col) => sum + columnWidths[col.field], 0);
+  }, [columns, columnWidths]);
+
+  // Handle column resize
+  const handleColumnResize = useCallback((columnField: string, deltaX: number) => {
+    const column = columns.find(col => col.field === columnField);
+    if (!column) return;
+
+    const currentWidth = columnWidths[columnField];
+    const newWidth = Math.max(column.minWidth || 60, Math.min(column.maxWidth || 400, currentWidth + deltaX));
+    
+    setColumnWidths(prev => ({ ...prev, [columnField]: newWidth }));
+    onColumnResize?.(columnField, newWidth);
+  }, [columns, columnWidths, onColumnResize]);
+
+  // Handle task selection
+  const handleTaskSelection = useCallback((task: GanttTask, event: React.MouseEvent) => {
+    const { ctrlKey, shiftKey } = event;
+    let newSelectedTasks = [...selection.selectedTasks];
+
+    if (shiftKey && enableMultiSelect && selection.selectedTasks.length > 0) {
+      // Range selection
+      const lastSelectedIndex = tasks.findIndex(t => t.id === selection.selectedTasks[selection.selectedTasks.length - 1]);
+      const currentIndex = tasks.findIndex(t => t.id === task.id);
+      const [start, end] = [Math.min(lastSelectedIndex, currentIndex), Math.max(lastSelectedIndex, currentIndex)];
+      
+      newSelectedTasks = tasks.slice(start, end + 1).map(t => t.id);
+    } else if (ctrlKey && enableMultiSelect) {
+      // Multi selection
+      if (newSelectedTasks.includes(task.id)) {
+        newSelectedTasks = newSelectedTasks.filter(id => id !== task.id);
+      } else {
+        newSelectedTasks.push(task.id);
+      }
+    } else {
+      // Single selection
+      newSelectedTasks = [task.id];
+    }
+
+    onSelectionChange?.({
+      ...selection,
+      selectedTasks: newSelectedTasks,
+      focusedTask: task.id,
+    });
+
+    onTaskClick?.(task, event);
+  }, [tasks, selection, enableMultiSelect, onSelectionChange, onTaskClick]);
+
+  // Handle cell editing
+  const handleCellDoubleClick = useCallback((task: GanttTask, column: ColumnConfig) => {
+    if (!enableInlineEdit || !column.editor) return;
+    
+    setEditingCell({ taskId: task.id, field: column.field });
+  }, [enableInlineEdit]);
+
+  const handleCellEditComplete = useCallback((value: any) => {
+    if (!editingCell) return;
+    
+    onTaskUpdate?.(editingCell.taskId, editingCell.field, value);
+    setEditingCell(null);
+  }, [editingCell, onTaskUpdate]);
+
+  // Render cell content
+  const renderCellContent = useCallback((task: GanttTask, column: ColumnConfig) => {
+    const value = task[column.field as keyof GanttTask];
+    const isEditing = editingCell?.taskId === task.id && editingCell?.field === column.field;
+
+    if (isEditing) {
+      return renderCellEditor(value, column, handleCellEditComplete);
+    }
+
+    if (column.renderer) {
+      return column.renderer(value, task);
+    }
+
+    // Default renderers
+    switch (column.field) {
+      case 'name':
+        return (
+          <div className="flex items-center space-x-2">
+            {task.hasChildren && (
+              <button
+                onClick={(e) => {
+                  e.stopPropagation();
+                  onTaskExpand?.(task.id);
+                }}
+                className="p-0.5 hover:bg-gray-200 dark:hover:bg-gray-600 rounded"
+              >
+                {task.isExpanded ? (
+                  <ChevronDownIcon className="w-3 h-3" />
+                ) : (
+                  <ChevronRightIcon className="w-3 h-3" />
+                )}
+              </button>
+            )}
+            <div 
+              className="flex items-center space-x-2"
+              style={{ paddingLeft: `${(task.level || 0) * 16}px` }}
+            >
+              {getTaskTypeIcon(task.type)}
+              <span className="truncate font-medium">{task.name}</span>
+            </div>
+          </div>
+        );
+      
+      case 'startDate':
+      case 'endDate':
+        return (
+          <div className="flex items-center space-x-1">
+            <CalendarIcon className="w-3 h-3 text-gray-400" />
+            <span>{(value as Date)?.toLocaleDateString() || '-'}</span>
+          </div>
+        );
+      
+      case 'assignee':
+        return task.assignee ? (
+          <div className="flex items-center space-x-2">
+            {task.assignee.avatar ? (
+              <img 
+                src={task.assignee.avatar} 
+                alt={task.assignee.name}
+                className="w-6 h-6 rounded-full"
+              />
+            ) : (
+              <UserIcon className="w-6 h-6 text-gray-400" />
+            )}
+            <span className="truncate">{task.assignee.name}</span>
+          </div>
+        ) : (
+          <span className="text-gray-400">Unassigned</span>
+        );
+      
+      case 'progress':
+        return (
+          <div className="flex items-center space-x-2">
+            <div className="flex-1 bg-gray-200 dark:bg-gray-600 rounded-full h-2">
+              <div
+                className="bg-blue-500 h-2 rounded-full transition-all duration-300"
+                style={{ width: `${task.progress}%` }}
+              />
+            </div>
+            <span className="text-xs w-8 text-right">{task.progress}%</span>
+          </div>
+        );
+      
+      case 'status':
+        return (
+          <span className={`px-2 py-1 rounded-full text-xs font-medium ${getStatusColor(task.status)}`}>
+            {task.status.replace('-', ' ')}
+          </span>
+        );
+      
+      case 'priority':
+        return (
+          <div className="flex items-center space-x-1">
+            <FlagIcon className={`w-3 h-3 ${getPriorityColor(task.priority)}`} />
+            <span className="capitalize">{task.priority}</span>
+          </div>
+        );
+      
+      case 'duration':
+        const duration = task.duration || Math.ceil((task.endDate.getTime() - task.startDate.getTime()) / (1000 * 60 * 60 * 24));
+        return <span>{duration}d</span>;
+      
+      default:
+        return <span>{String(value || '')}</span>;
+    }
+  }, [editingCell, onTaskExpand, handleCellEditComplete]);
+
+  // Render header
+  const renderHeader = () => (
+    <div 
+      className="grid-header flex border-b sticky top-0 z-10"
+      style={{ 
+        backgroundColor: colors.headerBg,
+        borderColor: colors.border,
+        height: rowHeight,
+      }}
+    >
+      {columns.map((column, index) => (
+        <div
+          key={column.field}
+          className="column-header flex items-center px-3 py-2 font-medium text-sm border-r relative group"
+          style={{
+            width: columnWidths[column.field],
+            borderColor: colors.border,
+            color: colors.text,
+          }}
+        >
+          <span className="truncate" title={column.title}>
+            {column.title}
+          </span>
+          
+          {/* Resize handle */}
+          {column.resizable && (
+            <ResizeHandle
+              onResize={(deltaX) => handleColumnResize(column.field, deltaX)}
+              className="absolute right-0 top-0 h-full w-1 cursor-col-resize hover:bg-blue-500 opacity-0 group-hover:opacity-100"
+            />
+          )}
+        </div>
+      ))}
+    </div>
+  );
+
+  // Render task rows
+  const renderRows = () => (
+    <div className="grid-body">
+      {tasks.map((task, rowIndex) => {
+        const isSelected = selection.selectedTasks.includes(task.id);
+        const isFocused = selection.focusedTask === task.id;
+        
+        return (
+          <div
+            key={task.id}
+            className={`grid-row flex border-b cursor-pointer hover:bg-opacity-75 ${
+              isSelected ? 'bg-blue-50 dark:bg-blue-900 bg-opacity-50' : 
+              rowIndex % 2 === 0 ? '' : 'bg-gray-50 dark:bg-gray-800 bg-opacity-30'
+            }`}
+            style={{
+              height: rowHeight,
+              borderColor: colors.border,
+              backgroundColor: isSelected ? colors.selected : 
+                              rowIndex % 2 === 0 ? 'transparent' : colors.alternateRow,
+            }}
+            onClick={(e) => handleTaskSelection(task, e)}
+            onDoubleClick={() => onTaskDoubleClick?.(task)}
+          >
+            {columns.map((column) => (
+              <div
+                key={`${task.id}-${column.field}`}
+                className="grid-cell flex items-center px-3 py-1 border-r overflow-hidden"
+                style={{
+                  width: columnWidths[column.field],
+                  borderColor: colors.border,
+                  textAlign: column.align || 'left',
+                }}
+                onDoubleClick={() => handleCellDoubleClick(task, column)}
+              >
+                {renderCellContent(task, column)}
+              </div>
+            ))}
+          </div>
+        );
+      })}
+    </div>
+  );
+
+  return (
+    <div
+      ref={gridRef}
+      className={`gantt-grid border-r ${className}`}
+      style={{
+        width: totalWidth,
+        height: containerHeight,
+        backgroundColor: colors.background,
+        borderColor: colors.border,
+      }}
+    >
+      {renderHeader()}
+      <div 
+        className="grid-content overflow-auto"
+        style={{ height: containerHeight - rowHeight }}
+      >
+        {renderRows()}
+      </div>
+    </div>
+  );
+};
+
+// Resize handle component
+interface ResizeHandleProps {
+  onResize: (deltaX: number) => void;
+  className?: string;
+}
+
+const ResizeHandle: React.FC<ResizeHandleProps> = ({ onResize, className }) => {
+  const [isDragging, setIsDragging] = useState(false);
+  const startXRef = useRef<number>(0);
+
+  const handleMouseDown = useCallback((e: React.MouseEvent) => {
+    e.preventDefault();
+    setIsDragging(true);
+    startXRef.current = e.clientX;
+
+    const handleMouseMove = (moveEvent: MouseEvent) => {
+      const deltaX = moveEvent.clientX - startXRef.current;
+      onResize(deltaX);
+      startXRef.current = moveEvent.clientX;
+    };
+
+    const handleMouseUp = () => {
+      setIsDragging(false);
+      document.removeEventListener('mousemove', handleMouseMove);
+      document.removeEventListener('mouseup', handleMouseUp);
+    };
+
+    document.addEventListener('mousemove', handleMouseMove);
+    document.addEventListener('mouseup', handleMouseUp);
+  }, [onResize]);
+
+  return (
+    <div
+      className={`resize-handle ${className} ${isDragging ? 'bg-blue-500' : ''}`}
+      onMouseDown={handleMouseDown}
+    />
+  );
+};
+
+// Cell editor component
+const renderCellEditor = (value: any, column: ColumnConfig, onComplete: (value: any) => void) => {
+  const [editValue, setEditValue] = useState(value);
+
+  const handleKeyDown = (e: React.KeyboardEvent) => {
+    if (e.key === 'Enter') {
+      onComplete(editValue);
+    } else if (e.key === 'Escape') {
+      onComplete(value); // Cancel editing
+    }
+  };
+
+  const handleBlur = () => {
+    onComplete(editValue);
+  };
+
+  switch (column.editor) {
+    case 'text':
+      return (
+        <input
+          type="text"
+          value={editValue}
+          onChange={(e) => setEditValue(e.target.value)}
+          onKeyDown={handleKeyDown}
+          onBlur={handleBlur}
+          className="w-full px-1 py-0.5 border rounded text-sm"
+          autoFocus
+        />
+      );
+    
+    case 'date':
+      return (
+        <input
+          type="date"
+          value={editValue instanceof Date ? editValue.toISOString().split('T')[0] : editValue}
+          onChange={(e) => setEditValue(new Date(e.target.value))}
+          onKeyDown={handleKeyDown}
+          onBlur={handleBlur}
+          className="w-full px-1 py-0.5 border rounded text-sm"
+          autoFocus
+        />
+      );
+    
+    case 'number':
+      return (
+        <input
+          type="number"
+          value={editValue}
+          onChange={(e) => setEditValue(parseFloat(e.target.value))}
+          onKeyDown={handleKeyDown}
+          onBlur={handleBlur}
+          className="w-full px-1 py-0.5 border rounded text-sm"
+          autoFocus
+        />
+      );
+    
+    case 'select':
+      return (
+        <select
+          value={editValue}
+          onChange={(e) => setEditValue(e.target.value)}
+          onKeyDown={handleKeyDown}
+          onBlur={handleBlur}
+          className="w-full px-1 py-0.5 border rounded text-sm"
+          autoFocus
+        >
+          {column.editorOptions?.map((option: any) => (
+            <option key={option.value} value={option.value}>
+              {option.label}
+            </option>
+          ))}
+        </select>
+      );
+    
+    default:
+      return <span>{String(value)}</span>;
+  }
+};
+
+// Helper functions
+const getTaskTypeIcon = (type: GanttTask['type']) => {
+  switch (type) {
+    case 'project':
+      return <div className="w-3 h-3 bg-blue-500 rounded-sm" />;
+    case 'milestone':
+      return <div className="w-3 h-3 bg-yellow-500 rotate-45" />;
+    default:
+      return <div className="w-3 h-3 bg-gray-400 rounded-full" />;
+  }
+};
+
+const getStatusColor = (status: GanttTask['status']) => {
+  switch (status) {
+    case 'completed':
+      return 'bg-green-100 text-green-800 dark:bg-green-900 dark:text-green-200';
+    case 'in-progress':
+      return 'bg-blue-100 text-blue-800 dark:bg-blue-900 dark:text-blue-200';
+    case 'overdue':
+      return 'bg-red-100 text-red-800 dark:bg-red-900 dark:text-red-200';
+    case 'on-hold':
+      return 'bg-yellow-100 text-yellow-800 dark:bg-yellow-900 dark:text-yellow-200';
+    default:
+      return 'bg-gray-100 text-gray-800 dark:bg-gray-700 dark:text-gray-200';
+  }
+};
+
+const getPriorityColor = (priority: GanttTask['priority']) => {
+  switch (priority) {
+    case 'critical':
+      return 'text-red-600';
+    case 'high':
+      return 'text-orange-500';
+    case 'medium':
+      return 'text-yellow-500';
+    case 'low':
+      return 'text-green-500';
+    default:
+      return 'text-gray-400';
+  }
+};
+
+export default GanttGrid;

worklenz-frontend/src/components/advanced-gantt/TimelineMarkers.tsx

@@ -0,0 +1,295 @@
+import React, { useMemo } from 'react';
+import { Holiday, TimelineConfig } from '../../types/advanced-gantt.types';
+import { useAppSelector } from '../../hooks/useAppSelector';
+import { themeWiseColor } from '../../utils/themeWiseColor';
+import { useDateCalculations } from '../../utils/gantt-performance';
+
+interface TimelineMarkersProps {
+  startDate: Date;
+  endDate: Date;
+  dayWidth: number;
+  containerHeight: number;
+  timelineConfig: TimelineConfig;
+  holidays?: Holiday[];
+  showWeekends?: boolean;
+  showHolidays?: boolean;
+  showToday?: boolean;
+  className?: string;
+}
+
+const TimelineMarkers: React.FC<TimelineMarkersProps> = ({
+  startDate,
+  endDate,
+  dayWidth,
+  containerHeight,
+  timelineConfig,
+  holidays = [],
+  showWeekends = true,
+  showHolidays = true,
+  showToday = true,
+  className = '',
+}) => {
+  const themeMode = useAppSelector(state => state.themeReducer.mode);
+  const { getDaysBetween, isWeekend, isWorkingDay } = useDateCalculations();
+
+  // Generate all dates in the timeline
+  const timelineDates = useMemo(() => {
+    const dates: Date[] = [];
+    const totalDays = getDaysBetween(startDate, endDate);
+    
+    for (let i = 0; i <= totalDays; i++) {
+      const date = new Date(startDate);
+      date.setDate(date.getDate() + i);
+      dates.push(date);
+    }
+    
+    return dates;
+  }, [startDate, endDate, getDaysBetween]);
+
+  // Theme-aware colors
+  const colors = useMemo(() => ({
+    weekend: themeWiseColor('rgba(0, 0, 0, 0.05)', 'rgba(255, 255, 255, 0.05)', themeMode),
+    nonWorkingDay: themeWiseColor('rgba(0, 0, 0, 0.03)', 'rgba(255, 255, 255, 0.03)', themeMode),
+    holiday: themeWiseColor('rgba(255, 107, 107, 0.1)', 'rgba(255, 107, 107, 0.15)', themeMode),
+    today: themeWiseColor('rgba(24, 144, 255, 0.15)', 'rgba(64, 169, 255, 0.2)', themeMode),
+    todayLine: themeWiseColor('#1890ff', '#40a9ff', themeMode),
+    holidayBorder: themeWiseColor('#ff6b6b', '#ff8787', themeMode),
+  }), [themeMode]);
+
+  // Check if a date is a holiday
+  const isHoliday = (date: Date): Holiday | undefined => {
+    return holidays.find(holiday => {
+      if (holiday.recurring) {
+        return holiday.date.getMonth() === date.getMonth() && 
+               holiday.date.getDate() === date.getDate();
+      }
+      return holiday.date.toDateString() === date.toDateString();
+    });
+  };
+
+  // Check if date is today
+  const isToday = (date: Date): boolean => {
+    const today = new Date();
+    return date.toDateString() === today.toDateString();
+  };
+
+  // Render weekend markers
+  const renderWeekendMarkers = () => {
+    if (!showWeekends) return null;
+
+    return timelineDates.map((date, index) => {
+      if (!isWeekend(date)) return null;
+
+      return (
+        <div
+          key={`weekend-${index}`}
+          className="weekend-marker absolute top-0 pointer-events-none"
+          style={{
+            left: index * dayWidth,
+            width: dayWidth,
+            height: containerHeight,
+            backgroundColor: colors.weekend,
+            zIndex: 1,
+          }}
+        />
+      );
+    });
+  };
+
+  // Render non-working day markers
+  const renderNonWorkingDayMarkers = () => {
+    return timelineDates.map((date, index) => {
+      if (isWorkingDay(date, timelineConfig.workingDays)) return null;
+
+      return (
+        <div
+          key={`non-working-${index}`}
+          className="non-working-day-marker absolute top-0 pointer-events-none"
+          style={{
+            left: index * dayWidth,
+            width: dayWidth,
+            height: containerHeight,
+            backgroundColor: colors.nonWorkingDay,
+            zIndex: 1,
+          }}
+        />
+      );
+    });
+  };
+
+  // Render holiday markers
+  const renderHolidayMarkers = () => {
+    if (!showHolidays) return null;
+
+    return timelineDates.map((date, index) => {
+      const holiday = isHoliday(date);
+      if (!holiday) return null;
+
+      const holidayColor = holiday.color || colors.holiday;
+
+      return (
+        <div
+          key={`holiday-${index}`}
+          className="holiday-marker absolute top-0 pointer-events-none group"
+          style={{
+            left: index * dayWidth,
+            width: dayWidth,
+            height: containerHeight,
+            backgroundColor: holidayColor,
+            borderLeft: `2px solid ${colors.holidayBorder}`,
+            zIndex: 2,
+          }}
+        >
+          {/* Holiday tooltip */}
+          <div className="holiday-tooltip absolute top-0 left-1/2 transform -translate-x-1/2 -translate-y-full bg-gray-900 dark:bg-gray-100 text-white dark:text-gray-900 text-xs px-2 py-1 rounded opacity-0 group-hover:opacity-100 transition-opacity duration-200 whitespace-nowrap z-50">
+            <div className="font-medium">{holiday.name}</div>
+            <div className="text-xs opacity-75">{date.toLocaleDateString()}</div>
+            <div className="absolute top-full left-1/2 transform -translate-x-1/2 w-0 h-0 border-l-2 border-r-2 border-t-4 border-transparent border-t-gray-900 dark:border-t-gray-100"></div>
+          </div>
+
+          {/* Holiday icon */}
+          <div className="holiday-icon absolute top-1 left-1 w-3 h-3 rounded-full bg-red-500 opacity-75">
+            <div className="w-full h-full rounded-full animate-pulse"></div>
+          </div>
+        </div>
+      );
+    });
+  };
+
+  // Render today marker
+  const renderTodayMarker = () => {
+    if (!showToday) return null;
+
+    const todayIndex = timelineDates.findIndex(date => isToday(date));
+    if (todayIndex === -1) return null;
+
+    return (
+      <div
+        className="today-marker absolute top-0 pointer-events-none"
+        style={{
+          left: todayIndex * dayWidth,
+          width: dayWidth,
+          height: containerHeight,
+          backgroundColor: colors.today,
+          zIndex: 3,
+        }}
+      >
+        {/* Today line */}
+        <div
+          className="today-line absolute top-0 left-1/2 transform -translate-x-1/2"
+          style={{
+            width: '2px',
+            height: containerHeight,
+            backgroundColor: colors.todayLine,
+            zIndex: 4,
+          }}
+        />
+
+        {/* Today label */}
+        <div className="today-label absolute top-2 left-1/2 transform -translate-x-1/2 bg-blue-500 text-white text-xs px-2 py-1 rounded shadow-sm">
+          Today
+        </div>
+      </div>
+    );
+  };
+
+  // Render time period markers (quarters, months, etc.)
+  const renderTimePeriodMarkers = () => {
+    const markers: React.ReactNode[] = [];
+    const currentDate = new Date(startDate);
+    currentDate.setDate(1); // Start of month
+
+    while (currentDate <= endDate) {
+      const daysSinceStart = getDaysBetween(startDate, currentDate);
+      const isQuarterStart = currentDate.getMonth() % 3 === 0 && currentDate.getDate() === 1;
+      const isYearStart = currentDate.getMonth() === 0 && currentDate.getDate() === 1;
+
+      if (isYearStart) {
+        markers.push(
+          <div
+            key={`year-${currentDate.getTime()}`}
+            className="year-marker absolute top-0 border-l-2 border-blue-600 dark:border-blue-400"
+            style={{
+              left: daysSinceStart * dayWidth,
+              height: containerHeight,
+              zIndex: 5,
+            }}
+          >
+            <div className="year-label absolute top-2 left-1 bg-blue-600 dark:bg-blue-400 text-white text-xs px-1 py-0.5 rounded">
+              {currentDate.getFullYear()}
+            </div>
+          </div>
+        );
+      } else if (isQuarterStart) {
+        markers.push(
+          <div
+            key={`quarter-${currentDate.getTime()}`}
+            className="quarter-marker absolute top-0 border-l border-green-500 dark:border-green-400 opacity-60"
+            style={{
+              left: daysSinceStart * dayWidth,
+              height: containerHeight,
+              zIndex: 4,
+            }}
+          >
+            <div className="quarter-label absolute top-2 left-1 bg-green-500 dark:bg-green-400 text-white text-xs px-1 py-0.5 rounded">
+              Q{Math.floor(currentDate.getMonth() / 3) + 1}
+            </div>
+          </div>
+        );
+      }
+
+      // Move to next month
+      currentDate.setMonth(currentDate.getMonth() + 1);
+    }
+
+    return markers;
+  };
+
+  return (
+    <div className={`timeline-markers absolute inset-0 ${className}`}>
+      {renderNonWorkingDayMarkers()}
+      {renderWeekendMarkers()}
+      {renderHolidayMarkers()}
+      {renderTodayMarker()}
+      {renderTimePeriodMarkers()}
+    </div>
+  );
+};
+
+// Holiday presets for common countries
+export const holidayPresets = {
+  US: [
+    { date: new Date(2024, 0, 1), name: "New Year's Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 0, 15), name: "Martin Luther King Jr. Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 1, 19), name: "Presidents' Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 4, 27), name: "Memorial Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 5, 19), name: "Juneteenth", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 6, 4), name: "Independence Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 8, 2), name: "Labor Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 9, 14), name: "Columbus Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 10, 11), name: "Veterans Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 10, 28), name: "Thanksgiving", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 11, 25), name: "Christmas Day", type: 'national' as const, recurring: true },
+  ],
+  
+  UK: [
+    { date: new Date(2024, 0, 1), name: "New Year's Day", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 2, 29), name: "Good Friday", type: 'religious' as const, recurring: false },
+    { date: new Date(2024, 3, 1), name: "Easter Monday", type: 'religious' as const, recurring: false },
+    { date: new Date(2024, 4, 6), name: "Early May Bank Holiday", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 4, 27), name: "Spring Bank Holiday", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 7, 26), name: "Summer Bank Holiday", type: 'national' as const, recurring: true },
+    { date: new Date(2024, 11, 25), name: "Christmas Day", type: 'religious' as const, recurring: true },
+    { date: new Date(2024, 11, 26), name: "Boxing Day", type: 'national' as const, recurring: true },
+  ],
+};
+
+// Working day presets
+export const workingDayPresets = {
+  standard: [1, 2, 3, 4, 5], // Monday to Friday
+  middle_east: [0, 1, 2, 3, 4], // Sunday to Thursday
+  six_day: [1, 2, 3, 4, 5, 6], // Monday to Saturday
+  four_day: [1, 2, 3, 4], // Monday to Thursday
+};
+
+export default TimelineMarkers;

worklenz-frontend/src/components/advanced-gantt/VirtualScrollContainer.tsx

@@ -0,0 +1,372 @@
+import React, { useRef, useEffect, useState, useCallback, ReactNode } from 'react';
+import { useThrottle, usePerformanceMonitoring } from '../../utils/gantt-performance';
+import { useAppSelector } from '../../hooks/useAppSelector';
+
+interface VirtualScrollContainerProps {
+  items: any[];
+  itemHeight: number;
+  containerHeight: number;
+  containerWidth?: number;
+  overscan?: number;
+  horizontal?: boolean;
+  children: (item: any, index: number, style: React.CSSProperties) => ReactNode;
+  onScroll?: (scrollLeft: number, scrollTop: number) => void;
+  className?: string;
+  style?: React.CSSProperties;
+}
+
+const VirtualScrollContainer: React.FC<VirtualScrollContainerProps> = ({
+  items,
+  itemHeight,
+  containerHeight,
+  containerWidth = 0,
+  overscan = 5,
+  horizontal = false,
+  children,
+  onScroll,
+  className = '',
+  style = {},
+}) => {
+  const containerRef = useRef<HTMLDivElement>(null);
+  const [scrollTop, setScrollTop] = useState(0);
+  const [scrollLeft, setScrollLeft] = useState(0);
+  const { startMeasure, endMeasure, recordMetric } = usePerformanceMonitoring();
+  const themeMode = useAppSelector(state => state.themeReducer.mode);
+
+  // Calculate visible range
+  const totalHeight = items.length * itemHeight;
+  const startIndex = Math.max(0, Math.floor(scrollTop / itemHeight) - overscan);
+  const endIndex = Math.min(
+    items.length - 1,
+    Math.ceil((scrollTop + containerHeight) / itemHeight) + overscan
+  );
+  const visibleItems = items.slice(startIndex, endIndex + 1);
+  const offsetY = startIndex * itemHeight;
+
+  // Throttled scroll handler
+  const throttledScrollHandler = useThrottle(
+    useCallback((event: Event) => {
+      const target = event.target as HTMLDivElement;
+      const newScrollTop = target.scrollTop;
+      const newScrollLeft = target.scrollLeft;
+      
+      setScrollTop(newScrollTop);
+      setScrollLeft(newScrollLeft);
+      onScroll?.(newScrollLeft, newScrollTop);
+    }, [onScroll]),
+    16 // ~60fps
+  );
+
+  useEffect(() => {
+    const container = containerRef.current;
+    if (!container) return;
+
+    container.addEventListener('scroll', throttledScrollHandler, { passive: true });
+    
+    return () => {
+      container.removeEventListener('scroll', throttledScrollHandler);
+    };
+  }, [throttledScrollHandler]);
+
+  // Performance monitoring
+  useEffect(() => {
+    startMeasure('virtualScroll');
+    recordMetric('visibleTaskCount', visibleItems.length);
+    recordMetric('taskCount', items.length);
+    endMeasure('virtualScroll');
+  }, [visibleItems.length, items.length, startMeasure, endMeasure, recordMetric]);
+
+  const renderVisibleItems = () => {
+    return visibleItems.map((item, virtualIndex) => {
+      const actualIndex = startIndex + virtualIndex;
+      const itemStyle: React.CSSProperties = {
+        position: 'absolute',
+        top: horizontal ? 0 : actualIndex * itemHeight,
+        left: horizontal ? actualIndex * itemHeight : 0,
+        height: horizontal ? '100%' : itemHeight,
+        width: horizontal ? itemHeight : '100%',
+        transform: horizontal ? 'none' : `translateY(${offsetY}px)`,
+      };
+
+      return (
+        <div key={item.id || actualIndex} style={itemStyle}>
+          {children(item, actualIndex, itemStyle)}
+        </div>
+      );
+    });
+  };
+
+  return (
+    <div
+      ref={containerRef}
+      className={`virtual-scroll-container overflow-auto ${className}`}
+      style={{
+        height: containerHeight,
+        width: containerWidth || '100%',
+        position: 'relative',
+        ...style,
+      }}
+    >
+      {/* Spacer to maintain scroll height */}
+      <div
+        className="virtual-scroll-spacer"
+        style={{
+          height: horizontal ? '100%' : totalHeight,
+          width: horizontal ? totalHeight : '100%',
+          position: 'relative',
+          pointerEvents: 'none',
+        }}
+      >
+        {/* Visible items container */}
+        <div
+          className="virtual-scroll-content"
+          style={{
+            position: 'absolute',
+            top: 0,
+            left: 0,
+            height: '100%',
+            width: '100%',
+            pointerEvents: 'auto',
+          }}
+        >
+          {renderVisibleItems()}
+        </div>
+      </div>
+    </div>
+  );
+};
+
+// Grid virtual scrolling component for both rows and columns
+interface VirtualGridProps {
+  data: any[][];
+  rowHeight: number;
+  columnWidth: number | number[];
+  containerHeight: number;
+  containerWidth: number;
+  overscan?: number;
+  children: (item: any, rowIndex: number, colIndex: number, style: React.CSSProperties) => ReactNode;
+  onScroll?: (scrollLeft: number, scrollTop: number) => void;
+  className?: string;
+}
+
+export const VirtualGrid: React.FC<VirtualGridProps> = ({
+  data,
+  rowHeight,
+  columnWidth,
+  containerHeight,
+  containerWidth,
+  overscan = 3,
+  children,
+  onScroll,
+  className = '',
+}) => {
+  const containerRef = useRef<HTMLDivElement>(null);
+  const [scrollTop, setScrollTop] = useState(0);
+  const [scrollLeft, setScrollLeft] = useState(0);
+
+  const rowCount = data.length;
+  const colCount = data[0]?.length || 0;
+
+  // Calculate column positions for variable width columns
+  const columnWidths = Array.isArray(columnWidth) ? columnWidth : new Array(colCount).fill(columnWidth);
+  const columnPositions = columnWidths.reduce((acc, width, index) => {
+    acc[index] = index === 0 ? 0 : acc[index - 1] + columnWidths[index - 1];
+    return acc;
+  }, {} as Record<number, number>);
+
+  const totalWidth = columnWidths.reduce((sum, width) => sum + width, 0);
+  const totalHeight = rowCount * rowHeight;
+
+  // Calculate visible ranges
+  const startRowIndex = Math.max(0, Math.floor(scrollTop / rowHeight) - overscan);
+  const endRowIndex = Math.min(rowCount - 1, Math.ceil((scrollTop + containerHeight) / rowHeight) + overscan);
+
+  const startColIndex = Math.max(0, findColumnIndex(scrollLeft) - overscan);
+  const endColIndex = Math.min(colCount - 1, findColumnIndex(scrollLeft + containerWidth) + overscan);
+
+  function findColumnIndex(position: number): number {
+    for (let i = 0; i < colCount; i++) {
+      if (columnPositions[i] <= position && position < columnPositions[i] + columnWidths[i]) {
+        return i;
+      }
+    }
+    return colCount - 1;
+  }
+
+  const throttledScrollHandler = useThrottle(
+    useCallback((event: Event) => {
+      const target = event.target as HTMLDivElement;
+      const newScrollTop = target.scrollTop;
+      const newScrollLeft = target.scrollLeft;
+      
+      setScrollTop(newScrollTop);
+      setScrollLeft(newScrollLeft);
+      onScroll?.(newScrollLeft, newScrollTop);
+    }, [onScroll]),
+    16
+  );
+
+  useEffect(() => {
+    const container = containerRef.current;
+    if (!container) return;
+
+    container.addEventListener('scroll', throttledScrollHandler, { passive: true });
+    
+    return () => {
+      container.removeEventListener('scroll', throttledScrollHandler);
+    };
+  }, [throttledScrollHandler]);
+
+  const renderVisibleCells = () => {
+    const cells: ReactNode[] = [];
+
+    for (let rowIndex = startRowIndex; rowIndex <= endRowIndex; rowIndex++) {
+      for (let colIndex = startColIndex; colIndex <= endColIndex; colIndex++) {
+        const item = data[rowIndex]?.[colIndex];
+        if (!item) continue;
+
+        const cellStyle: React.CSSProperties = {
+          position: 'absolute',
+          top: rowIndex * rowHeight,
+          left: columnPositions[colIndex],
+          height: rowHeight,
+          width: columnWidths[colIndex],
+        };
+
+        cells.push(
+          <div key={`${rowIndex}-${colIndex}`} style={cellStyle}>
+            {children(item, rowIndex, colIndex, cellStyle)}
+          </div>
+        );
+      }
+    }
+
+    return cells;
+  };
+
+  return (
+    <div
+      ref={containerRef}
+      className={`virtual-grid overflow-auto ${className}`}
+      style={{
+        height: containerHeight,
+        width: containerWidth,
+        position: 'relative',
+      }}
+    >
+      <div
+        style={{
+          height: totalHeight,
+          width: totalWidth,
+          position: 'relative',
+        }}
+      >
+        {renderVisibleCells()}
+      </div>
+    </div>
+  );
+};
+
+// Timeline virtual scrolling component
+interface VirtualTimelineProps {
+  startDate: Date;
+  endDate: Date;
+  dayWidth: number;
+  containerWidth: number;
+  containerHeight: number;
+  overscan?: number;
+  children: (date: Date, index: number, style: React.CSSProperties) => ReactNode;
+  onScroll?: (scrollLeft: number) => void;
+  className?: string;
+}
+
+export const VirtualTimeline: React.FC<VirtualTimelineProps> = ({
+  startDate,
+  endDate,
+  dayWidth,
+  containerWidth,
+  containerHeight,
+  overscan = 10,
+  children,
+  onScroll,
+  className = '',
+}) => {
+  const containerRef = useRef<HTMLDivElement>(null);
+  const [scrollLeft, setScrollLeft] = useState(0);
+
+  const totalDays = Math.ceil((endDate.getTime() - startDate.getTime()) / (1000 * 60 * 60 * 24));
+  const totalWidth = totalDays * dayWidth;
+
+  const startDayIndex = Math.max(0, Math.floor(scrollLeft / dayWidth) - overscan);
+  const endDayIndex = Math.min(totalDays - 1, Math.ceil((scrollLeft + containerWidth) / dayWidth) + overscan);
+
+  const throttledScrollHandler = useThrottle(
+    useCallback((event: Event) => {
+      const target = event.target as HTMLDivElement;
+      const newScrollLeft = target.scrollLeft;
+      setScrollLeft(newScrollLeft);
+      onScroll?.(newScrollLeft);
+    }, [onScroll]),
+    16
+  );
+
+  useEffect(() => {
+    const container = containerRef.current;
+    if (!container) return;
+
+    container.addEventListener('scroll', throttledScrollHandler, { passive: true });
+    
+    return () => {
+      container.removeEventListener('scroll', throttledScrollHandler);
+    };
+  }, [throttledScrollHandler]);
+
+  const renderVisibleDays = () => {
+    const days: ReactNode[] = [];
+
+    for (let dayIndex = startDayIndex; dayIndex <= endDayIndex; dayIndex++) {
+      const date = new Date(startDate);
+      date.setDate(date.getDate() + dayIndex);
+
+      const dayStyle: React.CSSProperties = {
+        position: 'absolute',
+        left: dayIndex * dayWidth,
+        top: 0,
+        width: dayWidth,
+        height: '100%',
+      };
+
+      days.push(
+        <div key={dayIndex} style={dayStyle}>
+          {children(date, dayIndex, dayStyle)}
+        </div>
+      );
+    }
+
+    return days;
+  };
+
+  return (
+    <div
+      ref={containerRef}
+      className={`virtual-timeline overflow-x-auto ${className}`}
+      style={{
+        height: containerHeight,
+        width: containerWidth,
+        position: 'relative',
+      }}
+    >
+      <div
+        style={{
+          height: '100%',
+          width: totalWidth,
+          position: 'relative',
+        }}
+      >
+        {renderVisibleDays()}
+      </div>
+    </div>
+  );
+};
+
+export default VirtualScrollContainer;

worklenz-frontend/src/components/advanced-gantt/index.ts

@@ -0,0 +1,17 @@
+// Main Components
+export { default as AdvancedGanttChart } from './AdvancedGanttChart';
+export { default as AdvancedGanttDemo } from './AdvancedGanttDemo';
+
+// Core Components
+export { default as GanttGrid } from './GanttGrid';
+export { default as DraggableTaskBar } from './DraggableTaskBar';
+export { default as TimelineMarkers, holidayPresets, workingDayPresets } from './TimelineMarkers';
+
+// Utility Components
+export { default as VirtualScrollContainer, VirtualGrid, VirtualTimeline } from './VirtualScrollContainer';
+
+// Types
+export * from '../../types/advanced-gantt.types';
+
+// Performance Utilities
+export * from '../../utils/gantt-performance';

worklenz-frontend/src/components/avatars/avatars.tsx

@@ -1,4 +1,4 @@
-import { Avatar, Tooltip } from 'antd';
+import { Avatar, Tooltip } from '@/shared/antd-imports';
 import React, { useCallback, useMemo } from 'react';
 import { InlineMember } from '@/types/teamMembers/inlineMember.types';
 

worklenz-frontend/src/components/board/board-assignee-selector/board-assignee-selector.tsx

@@ -15,7 +15,7 @@ import { useAppSelector } from '@/hooks/useAppSelector';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { toggleProjectMemberDrawer } from '../../../features/projects/singleProject/members/projectMembersSlice';
 import { colors } from '../../../styles/colors';
-import { PlusOutlined, UsergroupAddOutlined } from '@ant-design/icons';
+import { PlusOutlined, UsergroupAddOutlined } from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 import SingleAvatar from '@/components/common/single-avatar/single-avatar';
 import { CheckboxChangeEvent } from 'antd/es/checkbox';

worklenz-frontend/src/components/board/changeCategoryDropdown/ChangeCategoryDropdown.tsx

@@ -1,9 +1,9 @@
-import { Badge, Card, Dropdown, Flex, Menu, MenuProps } from 'antd';
+import { Badge, Card, Dropdown, Flex, Menu, MenuProps } from '@/shared/antd-imports';
 import React from 'react';
 import { TaskStatusType } from '../../../types/task.types';
 import { colors } from '../../../styles/colors';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
-import { RetweetOutlined, RightOutlined } from '@ant-design/icons';
+import { RetweetOutlined, RightOutlined } from '@/shared/antd-imports';
 import './ChangeCategoryDropdown.css';
 import { updateStatusCategory } from '../../../features/projects/status/StatusSlice';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/board/common-members-section/common-members-section.tsx

@@ -1,12 +1,12 @@
 import React, { useEffect, useRef, useState } from 'react';
-import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from 'antd';
+import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from '@/shared/antd-imports';
 import {
   DeleteOutlined,
   EditOutlined,
   LoadingOutlined,
   MoreOutlined,
   PlusOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { setTaskCardDisabled, initializeStatus } from '../../../features/board/create-card.slice';
 import { TaskType } from '../../../types/task.types';
 import TaskCreateCard from '../taskCreateCard/TaskCreateCard';

worklenz-frontend/src/components/board/common-phase-section/common-phase-section.tsx

@@ -1,12 +1,12 @@
 import React, { useEffect, useRef, useState } from 'react';
-import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from 'antd';
+import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from '@/shared/antd-imports';
 import {
   DeleteOutlined,
   EditOutlined,
   LoadingOutlined,
   MoreOutlined,
   PlusOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { setTaskCardDisabled, initializeStatus } from '../../../features/board/create-card.slice';
 import { TaskType } from '../../../types/task.types';
 import TaskCreateCard from '../taskCreateCard/TaskCreateCard';

worklenz-frontend/src/components/board/common-priority-section/common-priority-section.tsx

@@ -1,12 +1,12 @@
 import React, { useEffect, useRef, useState } from 'react';
-import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from 'antd';
+import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from '@/shared/antd-imports';
 import {
   DeleteOutlined,
   EditOutlined,
   LoadingOutlined,
   MoreOutlined,
   PlusOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { setTaskCardDisabled, initializeStatus } from '../../../features/board/create-card.slice';
 import { TaskType } from '../../../types/task.types';
 import TaskCreateCard from '../taskCreateCard/TaskCreateCard';

worklenz-frontend/src/components/board/commonStatusSection/CommonStatusSection.tsx

@@ -1,12 +1,12 @@
 import React, { useEffect, useRef, useState } from 'react';
-import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from 'antd';
+import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from '@/shared/antd-imports';
 import {
   DeleteOutlined,
   EditOutlined,
   LoadingOutlined,
   MoreOutlined,
   PlusOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { setTaskCardDisabled, initializeStatus } from '../../../features/board/create-card.slice';
 import { TaskType } from '../../../types/task.types';
 import TaskCreateCard from '../taskCreateCard/TaskCreateCard';

worklenz-frontend/src/components/board/custom-avatar-group.tsx

@@ -1,4 +1,4 @@
-import { Button, Flex } from 'antd';
+import { Button, Flex } from '@/shared/antd-imports';
 import AddMembersDropdown from '@/components/add-members-dropdown/add-members-dropdown';
 import Avatars from '../avatars/avatars';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';

worklenz-frontend/src/components/board/custom-due-date-picker.tsx

@@ -1,6 +1,6 @@
 import React, { useState, useRef } from 'react';
-import { DatePicker, Button, Flex } from 'antd';
-import { CalendarOutlined } from '@ant-design/icons';
+import { DatePicker, Button, Flex } from '@/shared/antd-imports';
+import { CalendarOutlined } from '@/shared/antd-imports';
 import dayjs, { Dayjs } from 'dayjs';
 import { useSocket } from '@/socket/socketContext';
 import { SocketEvents } from '@/shared/socket-events';

worklenz-frontend/src/components/board/kanban-group/kanban-group.tsx

@@ -1,12 +1,12 @@
 import React, { useEffect, useRef, useState } from 'react';
-import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from 'antd';
+import { Button, Dropdown, Input, InputRef, MenuProps, Typography } from '@/shared/antd-imports';
 import {
   DeleteOutlined,
   EditOutlined,
   LoadingOutlined,
   MoreOutlined,
   PlusOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 import { useDroppable } from '@dnd-kit/core';
 import { SortableContext, verticalListSortingStrategy } from '@dnd-kit/sortable';

worklenz-frontend/src/components/board/subTaskCard/SubTaskCard.tsx

@@ -1,5 +1,5 @@
 import React, { useEffect, useState } from 'react';
-import { Avatar, Col, DatePicker, Divider, Flex, Row, Tooltip, Typography } from 'antd';
+import { Avatar, Col, DatePicker, Divider, Flex, Row, Tooltip, Typography } from '@/shared/antd-imports';
 import StatusDropdown from '../../taskListCommon/statusDropdown/StatusDropdown';
 import dayjs, { Dayjs } from 'dayjs';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/board/taskCard/TaskCard.tsx

@@ -9,7 +9,7 @@ import {
   Dropdown,
   MenuProps,
   Button,
-} from 'antd';
+} from '@/shared/antd-imports';
 import {
   DoubleRightOutlined,
   PauseOutlined,
@@ -20,7 +20,7 @@ import {
   ForkOutlined,
   CaretRightFilled,
   CaretDownFilled,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import './TaskCard.css';
 import dayjs, { Dayjs } from 'dayjs';
 import AddMembersDropdown from '../../add-members-dropdown/add-members-dropdown';

worklenz-frontend/src/components/board/taskCard/priority-section/priority-section.tsx

@@ -1,10 +1,10 @@
-import { Flex, Typography } from 'antd';
+import { Flex, Typography } from '@/shared/antd-imports';
 import './priority-section.css';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useState, useEffect, useMemo } from 'react';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
 import { ITaskPriority } from '@/types/tasks/taskPriority.types';
-import { DoubleLeftOutlined, MinusOutlined, PauseOutlined } from '@ant-design/icons';
+import { DoubleLeftOutlined, MinusOutlined, PauseOutlined } from '@/shared/antd-imports';
 
 type PrioritySectionProps = {
   task: IProjectTask;

worklenz-frontend/src/components/board/taskCreateCard/TaskCreateCard.tsx

@@ -1,4 +1,4 @@
-import { Avatar, Button, DatePicker, Input, InputRef } from 'antd';
+import { Avatar, Button, DatePicker, Input, InputRef } from '@/shared/antd-imports';
 import React, { forwardRef, useEffect, useRef, useState } from 'react';
 import AddMembersDropdown from '../../add-members-dropdown/add-members-dropdown';
 import dayjs, { Dayjs } from 'dayjs';

worklenz-frontend/src/components/board/taskCreateCard/priority-task-create-card.tsx

@@ -1,4 +1,4 @@
-import { Avatar, Button, DatePicker, Input, InputRef } from 'antd';
+import { Avatar, Button, DatePicker, Input, InputRef } from '@/shared/antd-imports';
 import React, { forwardRef, useEffect, useRef, useState } from 'react';
 import AddMembersDropdown from '../../add-members-dropdown/add-members-dropdown';
 import dayjs, { Dayjs } from 'dayjs';

worklenz-frontend/src/components/calendars/homeCalendar/HomeCalendar.tsx

@@ -1,4 +1,4 @@
-import { Calendar } from 'antd';
+import { Calendar } from '@/shared/antd-imports';
 import React, { useEffect } from 'react';
 import type { Dayjs } from 'dayjs';
 import { useAppDispatch } from '@/hooks/useAppDispatch';

worklenz-frontend/src/components/charts/LazyChartComponents.tsx

@@ -1,5 +1,5 @@
 import { lazy, Suspense } from 'react';
-import { Spin } from 'antd';
+import { Spin } from '@/shared/antd-imports';
 
 // Lazy load Chart.js components
 const LazyBarChart = lazy(() => 

worklenz-frontend/src/components/charts/chart-loader.tsx

@@ -1,5 +1,5 @@
 import React, { Suspense } from 'react';
-import { Spin } from 'antd';
+import { Spin } from '@/shared/antd-imports';
 
 // Lazy load chart components to reduce initial bundle size
 const LazyBar = React.lazy(() =>

worklenz-frontend/src/components/common/invite-team-members/invite-team-members.tsx

@@ -1,4 +1,4 @@
-import { AutoComplete, Button, Drawer, Flex, Form, message, Select, Spin, Typography } from 'antd';
+import { AutoComplete, Button, Drawer, Flex, Form, message, Modal, Select, Spin, Typography } from '@/shared/antd-imports';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import {
@@ -11,6 +11,7 @@ import { jobTitlesApiService } from '@/api/settings/job-titles/job-titles.api.se
 import { IJobTitle } from '@/types/job.types';
 import { teamMembersApiService } from '@/api/team-members/teamMembers.api.service';
 import { ITeamMemberCreateRequest } from '@/types/teamMembers/team-member-create-request';
+import { LinkOutlined } from '@ant-design/icons';
 
 interface FormValues {
   email: string[];
@@ -87,23 +88,33 @@ const InviteTeamMembers = () => {
   };
 
   return (
-    <Drawer
+    <Modal
       title={
         <Typography.Text strong style={{ fontSize: 16 }}>
           {t('addMemberDrawerTitle')}
         </Typography.Text>
       }
       open={isDrawerOpen}
-      onClose={handleClose}
+      onCancel={handleClose}
       destroyOnClose
       afterOpenChange={visible => visible && handleSearch('')}
       width={400}
       loading={loading}
       footer={
-        <Flex justify="end">
-          <Button type="primary" onClick={form.submit}>
-            {t('addToTeamButton')}
-          </Button>
+        <Flex justify="space-between">
+          {/* <Button
+            style={{ width: 140, fontSize: 12 }}
+            block
+            icon={<LinkOutlined />}
+            disabled
+          >
+            {t('copyTeamLink')}
+          </Button> */}
+          <Flex justify="end">
+            <Button onClick={form.submit} style={{ fontSize: 12 }}>
+              {t('addToTeamButton')}
+            </Button>
+          </Flex>
         </Flex>
       }
     >
@@ -176,7 +187,7 @@ const InviteTeamMembers = () => {
           />
         </Form.Item>
       </Form>
-    </Drawer>
+    </Modal>
   );
 };
 

worklenz-frontend/src/components/common/people-dropdown/PeopleDropdown.tsx

@@ -1,6 +1,6 @@
 import React, { useState, useRef, useEffect, useMemo, useCallback } from 'react';
 import { createPortal } from 'react-dom';
-import { PlusOutlined, UserAddOutlined } from '@ant-design/icons';
+import { PlusOutlined, UserAddOutlined } from '@/shared/antd-imports';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { toggleProjectMemberDrawer } from '@/features/projects/singleProject/members/projectMembersSlice';

worklenz-frontend/src/components/common/project-status-icon/project-status-icon.tsx

@@ -4,7 +4,7 @@ import Icon, {
   ClockCircleOutlined,
   CloseCircleOutlined,
   StopOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 
 const iconMap = {
   'clock-circle': ClockCircleOutlined,

worklenz-frontend/src/components/common/single-avatar/single-avatar.tsx

@@ -1,5 +1,5 @@
 import { AvatarNamesMap } from '@/shared/constants';
-import { Avatar, Flex, Space } from 'antd';
+import { Avatar, Flex, Space } from '@/shared/antd-imports';
 
 interface SingleAvatarProps {
   avatarUrl?: string;

worklenz-frontend/src/components/common/template-drawer/template-drawer.tsx

@@ -1,4 +1,4 @@
-import type { MenuProps } from 'antd';
+import type { MenuProps } from '@/shared/antd-imports';
 import {
   Empty,
   List,
@@ -10,8 +10,8 @@ import {
   Image,
   Input,
   Flex,
-  Button,
-} from 'antd';
+  theme,
+} from '@/shared/antd-imports';
 import React, { useEffect, useState } from 'react';
 import { useSelector } from 'react-redux';
 import { useTranslation } from 'react-i18next';
@@ -23,7 +23,7 @@ import {
   IWorklenzTemplate,
 } from '@/types/project-templates/project-templates.types';
 import './template-drawer.css';
-import { SearchOutlined } from '@ant-design/icons';
+import { SearchOutlined } from '@/shared/antd-imports';
 import logger from '@/utils/errorLogger';
 
 const { Title, Text } = Typography;
@@ -38,13 +38,12 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
   showBothTabs = false,
   templateSelected = (templateId: string) => {
     if (!templateId) return;
-    templateId;
   },
   selectedTemplateType = (type: 'worklenz' | 'custom') => {
-    type;
   },
 }) => {
   const themeMode = useSelector((state: RootState) => state.themeReducer.mode);
+  const { token } = theme.useToken();
   const { t } = useTranslation('template-drawer');
 
   const [searchQuery, setSearchQuery] = useState('');
@@ -150,7 +149,12 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
                 <Tag
                   key={phase.name}
                   color={phase.color_code}
-                  style={{ color: 'black', marginBottom: '8px' }}
+                  style={{ 
+                    color: token.colorText, 
+                    marginBottom: '8px',
+                    backgroundColor: phase.color_code ? undefined : token.colorBgContainer,
+                    borderColor: phase.color_code ? undefined : token.colorBorder
+                  }}
                 >
                   {phase.name}
                 </Tag>
@@ -172,7 +176,12 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
                 <Tag
                   key={status.name}
                   color={status.color_code}
-                  style={{ color: 'black', marginBottom: '8px' }}
+                  style={{ 
+                    color: token.colorText, 
+                    marginBottom: '8px',
+                    backgroundColor: status.color_code ? undefined : token.colorBgContainer,
+                    borderColor: status.color_code ? undefined : token.colorBorder
+                  }}
                 >
                   {status.name}
                 </Tag>
@@ -194,7 +203,12 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
                 <Tag
                   key={priority.name}
                   color={priority.color_code}
-                  style={{ color: 'black', marginBottom: '8px' }}
+                  style={{ 
+                    color: token.colorText, 
+                    marginBottom: '8px',
+                    backgroundColor: priority.color_code ? undefined : token.colorBgContainer,
+                    borderColor: priority.color_code ? undefined : token.colorBorder
+                  }}
                 >
                   {priority.name}
                 </Tag>
@@ -216,7 +230,12 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
                 <Tag
                   key={label.name}
                   color={label.color_code}
-                  style={{ color: 'black', marginBottom: '8px' }}
+                  style={{ 
+                    color: token.colorText, 
+                    marginBottom: '8px',
+                    backgroundColor: label.color_code ? undefined : token.colorBgContainer,
+                    borderColor: label.color_code ? undefined : token.colorBorder
+                  }}
                 >
                   {label.name}
                 </Tag>
@@ -252,14 +271,24 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
   };
 
   const menuContent = (
-    <div style={{ display: 'flex' }}>
+    <div style={{ display: 'flex', backgroundColor: token.colorBgContainer }}>
       {/* Menu Area */}
-      <div style={{ minWidth: '250px', overflowY: 'auto', height: '100%' }}>
+      <div style={{ 
+        minWidth: '250px', 
+        overflowY: 'auto', 
+        height: '100%',
+        backgroundColor: token.colorBgContainer,
+        borderRight: `1px solid ${token.colorBorder}`
+      }}>
         <Skeleton loading={loadingTemplates} active>
           <Menu
             className="template-menu"
             onClick={({ key }) => handleMenuClick(key)}
-            style={{ width: 256 }}
+            style={{ 
+              width: 256,
+              backgroundColor: token.colorBgContainer,
+              borderColor: token.colorBorder
+            }}
             defaultSelectedKeys={[templates[0]?.id || '']}
             mode="inline"
             items={menuItems}
@@ -273,9 +302,11 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
           flex: 1,
           maxHeight: 'calc(100vh - 200px)',
           padding: '16px',
+          backgroundColor: token.colorBgContainer,
+          color: token.colorText
         }}
       >
-        <Title level={4}>Details</Title>
+        <Title level={4} style={{ color: token.colorText }}>Details</Title>
         <Skeleton loading={loadingSelectedTemplate} active>
           {selectedTemplate?.image_url && (
             <Image preview={false} src={selectedTemplate.image_url} alt={selectedTemplate.name} />
@@ -298,12 +329,17 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
   };
 
   const customTemplatesContent = (
-    <div>
+    <div style={{ backgroundColor: token.colorBgContainer, padding: '16px' }}>
       <Flex justify="space-between" align="center">
         <Input
           placeholder={t('searchTemplates')}
-          suffix={<SearchOutlined />}
-          style={{ maxWidth: '300px' }}
+          suffix={<SearchOutlined style={{ color: token.colorTextTertiary }} />}
+          style={{ 
+            maxWidth: '300px',
+            backgroundColor: token.colorBgContainer,
+            borderColor: token.colorBorder,
+            color: token.colorText
+          }}
           onChange={e => setSearchQuery(e.target.value)}
         />
       </Flex>
@@ -313,10 +349,20 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
         bordered
         dataSource={filteredCustomTemplates}
         loading={loadingCustomTemplates}
+        style={{
+          backgroundColor: token.colorBgContainer,
+          borderColor: token.colorBorder
+        }}
         renderItem={item => (
           <List.Item
             key={item.id}
             onClick={() => handleCustomTemplateClick(item.id || '')}
+            style={{
+              backgroundColor: item.selected ? token.colorPrimaryBg : token.colorBgContainer,
+              borderColor: item.selected ? token.colorPrimary : token.colorBorder,
+              color: token.colorText,
+              cursor: 'pointer'
+            }}
             className={
               item.selected && themeMode === 'dark'
                 ? 'selected-custom-template-dark'
@@ -325,7 +371,7 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
                   : ''
             }
           >
-            {item.name}
+            <span style={{ color: token.colorText }}>{item.name}</span>
           </List.Item>
         )}
       />
@@ -356,18 +402,31 @@ const TemplateDrawer: React.FC<TemplateDrawerProps> = ({
   };
 
   return (
-    <div style={{ height: '100vh', overflow: 'hidden' }}>
+    <div style={{ 
+      height: '100vh', 
+      overflow: 'hidden',
+      backgroundColor: token.colorBgLayout 
+    }}>
       <div
         style={{
           position: 'sticky',
           top: 0,
           zIndex: 100,
-          backgroundColor: themeMode === 'dark' ? '' : '#fff',
+          backgroundColor: token.colorBgContainer,
           overflow: 'hidden',
+          borderBottom: `1px solid ${token.colorBorder}`
         }}
       >
         {showBothTabs ? (
-          <Tabs type="card" items={tabs} onChange={handleTabChange} destroyInactiveTabPane />
+          <Tabs 
+            type="card" 
+            items={tabs} 
+            onChange={handleTabChange} 
+            destroyInactiveTabPane
+            style={{
+              backgroundColor: token.colorBgContainer
+            }}
+          />
         ) : (
           menuContent
         )}

worklenz-frontend/src/components/common/tooltip-wrapper/tooltip-wrapper.tsx

@@ -1,5 +1,5 @@
 import React from 'react';
-import { Tooltip, TooltipProps } from 'antd';
+import { Tooltip, TooltipProps } from '@/shared/antd-imports';
 
 interface TooltipWrapperProps extends Omit<TooltipProps, 'children'> {
   children: React.ReactElement;

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanBoard.tsx

@@ -1,6 +1,6 @@
 import React, { useEffect, useState, useMemo } from 'react';
 import { useSelector, useDispatch } from 'react-redux';
-import { Card, Spin, Empty } from 'antd';
+import { Card, Spin, Empty } from '@/shared/antd-imports';
 import {
   DndContext,
   DragOverlay,

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanBoardNativeDnD/EnhancedKanbanBoardNativeDnD.tsx

@@ -8,7 +8,7 @@ import ImprovedTaskFilters from '../../task-management/improved-task-filters';
 import Card from 'antd/es/card';
 import Spin from 'antd/es/spin';
 import Empty from 'antd/es/empty';
-import { reorderGroups, reorderEnhancedKanbanGroups, reorderTasks, reorderEnhancedKanbanTasks, fetchEnhancedKanbanLabels, fetchEnhancedKanbanGroups, fetchEnhancedKanbanTaskAssignees } from '@/features/enhanced-kanban/enhanced-kanban.slice';
+import { reorderGroups, reorderEnhancedKanbanGroups, reorderTasks, reorderEnhancedKanbanTasks, fetchEnhancedKanbanLabels, fetchEnhancedKanbanGroups, fetchEnhancedKanbanTaskAssignees, updateEnhancedKanbanTaskPriority } from '@/features/enhanced-kanban/enhanced-kanban.slice';
 import { fetchStatusesCategories } from '@/features/taskAttributes/taskStatusSlice';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import KanbanGroup from './KanbanGroup';
@@ -19,11 +19,16 @@ import { useAuthService } from '@/hooks/useAuth';
 import { statusApiService } from '@/api/taskAttributes/status/status.api.service';
 import alertService from '@/services/alerts/alertService';
 import logger from '@/utils/errorLogger';
-import Skeleton from 'antd/es/skeleton/Skeleton';
 import { checkTaskDependencyStatus } from '@/utils/check-task-dependency-status';
 import { useTaskSocketHandlers } from '@/hooks/useTaskSocketHandlers';
+import { phasesApiService } from '@/api/taskAttributes/phases/phases.api.service';
+import { ITaskListGroup } from '@/types/tasks/taskList.types';
+import { fetchPhasesByProjectId, updatePhaseListOrder } from '@/features/projects/singleProject/phase/phases.slice';
+import { useTranslation } from 'react-i18next';
+import { ITaskListPriorityChangeResponse } from '@/types/tasks/task-list-priority.types';
 
 const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ projectId }) => {
+  const { t } = useTranslation('kanban-board');
   const dispatch = useDispatch();
   const authService = useAuthService();
   const { socket } = useSocket();
@@ -35,6 +40,7 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
     loadingGroups,
     error,
   } = useSelector((state: RootState) => state.enhancedKanbanReducer);
+  const { phaseList, loadingPhases } = useAppSelector(state => state.phaseReducer);
   const [draggedGroupId, setDraggedGroupId] = useState<string | null>(null);
   const [draggedTaskId, setDraggedTaskId] = useState<string | null>(null);
   const [draggedTaskGroupId, setDraggedTaskGroupId] = useState<string | null>(null);
@@ -57,6 +63,9 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
     if (!statusCategories.length) {
       dispatch(fetchStatusesCategories() as any);
     }
+    if (groupBy === 'phase' && !phaseList.length) {
+      dispatch(fetchPhasesByProjectId(projectId) as any);
+    }
   }, [dispatch, projectId]);
   // Reset drag state if taskGroups changes (e.g., real-time update)
   useEffect(() => {
@@ -73,6 +82,9 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
     setDraggedGroupId(groupId);
     setDragType('group');
     e.dataTransfer.effectAllowed = 'move';
+    try {
+      e.dataTransfer.setData('text/plain', groupId);
+    } catch {}
   };
   const handleGroupDragOver = (e: React.DragEvent) => {
     if (dragType !== 'group') return;
@@ -91,19 +103,35 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
     reorderedGroups.splice(toIdx, 0, moved);
     dispatch(reorderGroups({ fromIndex: fromIdx, toIndex: toIdx, reorderedGroups }));
     dispatch(reorderEnhancedKanbanGroups({ fromIndex: fromIdx, toIndex: toIdx, reorderedGroups }) as any);
-
     // API call for group order
     try {
-      const columnOrder = reorderedGroups.map(group => group.id);
-      const requestBody = { status_order: columnOrder };
-      const response = await statusApiService.updateStatusOrder(requestBody, projectId);
-      if (!response.done) {
-        // Revert the change if API call fails
-        const revertedGroups = [...reorderedGroups];
-        const [movedBackGroup] = revertedGroups.splice(toIdx, 1);
-        revertedGroups.splice(fromIdx, 0, movedBackGroup);
-        dispatch(reorderGroups({ fromIndex: toIdx, toIndex: fromIdx, reorderedGroups: revertedGroups }));
-        alertService.error('Failed to update column order', 'Please try again');
+      if (groupBy === 'status') {
+        const columnOrder = reorderedGroups.map(group => group.id);
+        const requestBody = { status_order: columnOrder };
+        const response = await statusApiService.updateStatusOrder(requestBody, projectId);
+        if (!response.done) {
+          // Revert the change if API call fails
+          const revertedGroups = [...reorderedGroups];
+          const [movedBackGroup] = revertedGroups.splice(toIdx, 1);
+          revertedGroups.splice(fromIdx, 0, movedBackGroup);
+          dispatch(reorderGroups({ fromIndex: toIdx, toIndex: fromIdx, reorderedGroups: revertedGroups }));
+          alertService.error(t('failedToUpdateColumnOrder'), t('pleaseTryAgain'));
+        }
+      } else if (groupBy === 'phase') {
+        const newPhaseList = [...phaseList];
+        const [movedItem] = newPhaseList.splice(fromIdx, 1);
+        newPhaseList.splice(toIdx, 0, movedItem);
+        dispatch(updatePhaseListOrder(newPhaseList));
+        const requestBody = {
+          from_index: fromIdx,
+          to_index: toIdx,
+          phases: newPhaseList,
+          project_id: projectId,
+        };
+        const response = await phasesApiService.updatePhaseOrder(projectId, requestBody);
+        if (!response.done) {
+          alertService.error(t('failedToUpdatePhaseOrder'), t('pleaseTryAgain'));
+        }
       }
     } catch (error) {
       // Revert the change if API call fails
@@ -111,7 +139,7 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
       const [movedBackGroup] = revertedGroups.splice(toIdx, 1);
       revertedGroups.splice(fromIdx, 0, movedBackGroup);
       dispatch(reorderGroups({ fromIndex: toIdx, toIndex: fromIdx, reorderedGroups: revertedGroups }));
-      alertService.error('Failed to update column order', 'Please try again');
+      alertService.error(t('failedToUpdateColumnOrder'), t('pleaseTryAgain'));
       logger.error('Failed to update column order', error);
     }
 
@@ -119,12 +147,47 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
     setDragType(null);
   };
 
+  // Utility to recalculate all task orders for all groups
+  function getAllTaskUpdates(allGroups: ITaskListGroup[], groupBy: string) {
+    const taskUpdates: Array<{
+      task_id: string | undefined;
+      sort_order: number;
+      status_id?: string;
+      priority_id?: string;
+      phase_id?: string;
+    }> = [];
+    let currentSortOrder = 0;
+    for (const group of allGroups) {
+      for (const task of group.tasks) {
+        const update: {
+          task_id: string | undefined;
+          sort_order: number;
+          status_id?: string;
+          priority_id?: string;
+          phase_id?: string;
+        } = {
+          task_id: task.id,
+          sort_order: currentSortOrder,
+        };
+        if (groupBy === 'status') update.status_id = group.id;
+        else if (groupBy === 'priority') update.priority_id = group.id;
+        else if (groupBy === 'phase' && group.name !== 'Unmapped') update.phase_id = group.id;
+        taskUpdates.push(update);
+        currentSortOrder++;
+      }
+    }
+    return taskUpdates;
+  }
+
   // Task drag handlers
   const handleTaskDragStart = (e: React.DragEvent, taskId: string, groupId: string) => {
     setDraggedTaskId(taskId);
     setDraggedTaskGroupId(groupId);
     setDragType('task');
     e.dataTransfer.effectAllowed = 'move';
+    try {
+      e.dataTransfer.setData('text/plain', taskId);
+    } catch {}
   };
   const handleTaskDragOver = (e: React.DragEvent, groupId: string, taskIdx: number | null) => {
     if (dragType !== 'task') return;
@@ -148,26 +211,28 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
     const targetGroup = taskGroups.find(g => g.id === targetGroupId);
     if (!sourceGroup || !targetGroup) return;
 
-
     const taskIdx = sourceGroup.tasks.findIndex(t => t.id === draggedTaskId);
     if (taskIdx === -1) return;
 
     const movedTask = sourceGroup.tasks[taskIdx];
+    let didStatusChange = false;
     if (groupBy === 'status' && movedTask.id) {
       if (sourceGroup.id !== targetGroup.id) {
         const canContinue = await checkTaskDependencyStatus(movedTask.id, targetGroupId);
         if (!canContinue) {
           alertService.error(
-            'Task is not completed',
-            'Please complete the task dependencies before proceeding'
+            t('taskNotCompleted'),
+            t('completeTaskDependencies')
           );
           return;
         }
+        didStatusChange = true;
       }
     }
     let insertIdx = hoveredTaskIdx;
 
     // Handle same group reordering
+    let newTaskGroups = [...taskGroups];
     if (sourceGroup.id === targetGroup.id) {
       // Create a single updated array for the same group
       const updatedTasks = [...sourceGroup.tasks];
@@ -182,7 +247,6 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
       if (insertIdx > updatedTasks.length) insertIdx = updatedTasks.length;
 
       updatedTasks.splice(insertIdx, 0, movedTask); // Insert at new position
-
       dispatch(reorderTasks({
         activeGroupId: sourceGroup.id,
         overGroupId: targetGroup.id,
@@ -201,6 +265,8 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
         updatedSourceTasks: updatedTasks,
         updatedTargetTasks: updatedTasks,
       }) as any);
+      // Update newTaskGroups for socket emit
+      newTaskGroups = newTaskGroups.map(g => g.id === sourceGroup.id ? { ...g, tasks: updatedTasks } : g);
     } else {
       // Handle cross-group reordering
       const updatedSourceTasks = [...sourceGroup.tasks];
@@ -229,36 +295,63 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
         updatedSourceTasks,
         updatedTargetTasks,
       }) as any);
+      // Update newTaskGroups for socket emit
+      newTaskGroups = newTaskGroups.map(g => {
+        if (g.id === sourceGroup.id) return { ...g, tasks: updatedSourceTasks };
+        if (g.id === targetGroup.id) return { ...g, tasks: updatedTargetTasks };
+        return g;
+      });
     }
 
-    // Socket emit for task order
+    // Socket emit for full task order
     if (socket && projectId && teamId && movedTask) {
-      let toSortOrder = -1;
-      let toLastIndex = false;
-      if (insertIdx === targetGroup.tasks.length) {
-        toSortOrder = -1;
-        toLastIndex = true;
-      } else if (targetGroup.tasks[insertIdx]) {
-        const sortOrder = targetGroup.tasks[insertIdx].sort_order;
-        toSortOrder = typeof sortOrder === 'number' ? sortOrder : 0;
-        toLastIndex = false;
-      } else if (targetGroup.tasks.length > 0) {
-        const lastSortOrder = targetGroup.tasks[targetGroup.tasks.length - 1].sort_order;
-        toSortOrder = typeof lastSortOrder === 'number' ? lastSortOrder : 0;
-        toLastIndex = false;
-      }
+      const taskUpdates = getAllTaskUpdates(newTaskGroups, groupBy);
       socket.emit(SocketEvents.TASK_SORT_ORDER_CHANGE.toString(), {
         project_id: projectId,
-        from_index: movedTask.sort_order ?? 0,
-        to_index: toSortOrder,
-        to_last_index: toLastIndex,
+        group_by: groupBy || 'status',
+        task_updates: taskUpdates,
         from_group: sourceGroup.id,
         to_group: targetGroup.id,
-        group_by: groupBy || 'status',
-        task: movedTask,
         team_id: teamId,
+        from_index: taskIdx,
+        to_index: insertIdx,
+        to_last_index: insertIdx === (targetGroup.id === sourceGroup.id ? (newTaskGroups.find(g => g.id === targetGroup.id)?.tasks.length || 0) - 1 : targetGroup.tasks.length),
+        task: {
+          id: movedTask.id,
+          project_id: movedTask.project_id || projectId,
+          status: movedTask.status || '',
+          priority: movedTask.priority || '',
+        }
       });
 
+      // Emit progress update if status changed
+      if (didStatusChange) {
+        socket.emit(
+          SocketEvents.TASK_STATUS_CHANGE.toString(),
+          JSON.stringify({
+            task_id: movedTask.id,
+            status_id: targetGroupId,
+            parent_task: movedTask.parent_task_id || null,
+            team_id: teamId,
+          })
+        );
+      }
+      if (groupBy === 'priority' && movedTask.id) {
+        socket?.emit(
+          SocketEvents.TASK_PRIORITY_CHANGE.toString(),
+          JSON.stringify({
+            task_id: movedTask.id,
+            priority_id: targetGroupId,
+            team_id: teamId,
+          })
+        );
+        socket?.once(
+          SocketEvents.TASK_PRIORITY_CHANGE.toString(),
+          (data: ITaskListPriorityChangeResponse) => {
+            dispatch(updateEnhancedKanbanTaskPriority(data));
+          }
+        );
+      }
     }
 
     setDraggedTaskId(null);
@@ -279,7 +372,7 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
   if (error) {
     return (
       <Card>
-        <Empty description={`Error loading tasks: ${error}`} image={Empty.PRESENTED_IMAGE_SIMPLE} />
+        <Empty description={`${t('errorLoadingTasks')}: ${error}`} image={Empty.PRESENTED_IMAGE_SIMPLE} />
       </Card>
     );
   }
@@ -287,21 +380,21 @@ const EnhancedKanbanBoardNativeDnD: React.FC<{ projectId: string }> = ({ project
   return (
     <>
       <div className="mb-4">
-        <React.Suspense fallback={<div>Loading filters...</div>}>
+        <React.Suspense fallback={<div>{t('loadingFilters')}</div>}>
           <ImprovedTaskFilters position="board" />
         </React.Suspense>
       </div>
       <div className="enhanced-kanban-board">
         {loadingGroups ? (
-            <div className="flex flex-row gap-2 h-[600px]">
-              <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '60%' }} />
-              <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '100%' }} />
-              <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '80%' }} />
-              <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '40%' }} />
-            </div>
+          <div className="flex flex-row gap-2 h-[600px]">
+            <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '60%' }} />
+            <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '100%' }} />
+            <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '80%' }} />
+            <div className="rounded bg-gray-200 dark:bg-gray-700 animate-pulse w-1/4" style={{ height: '40%' }} />
+          </div>
         ) : taskGroups.length === 0 ? (
           <Card>
-            <Empty description="No tasks found" image={Empty.PRESENTED_IMAGE_SIMPLE} />
+            <Empty description={t('noTasksFound')} image={Empty.PRESENTED_IMAGE_SIMPLE} />
           </Card>
         ) : (
           <div className="kanban-groups-container">

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanBoardNativeDnD/KanbanGroup.tsx

@@ -25,6 +25,7 @@ import {
     IGroupBy,
 } from '@/features/enhanced-kanban/enhanced-kanban.slice';
 import { createPortal } from 'react-dom';
+import { Modal } from 'antd';
 
 // Simple Portal component
 const Portal: React.FC<{ children: React.ReactNode }> = ({ children }) => {
@@ -218,7 +219,42 @@ const KanbanGroup: React.FC<KanbanGroupProps> = memo(({
     };
 
     const handleDelete = () => {
-        setShowDeleteConfirm(true);
+        if (groupBy === IGroupBy.STATUS) {
+            Modal.confirm({
+                title: t('deleteStatusTitle'),
+                content: t('deleteStatusContent'),
+                okText: t('deleteTaskConfirm'),
+                okType: 'danger',
+                cancelText: t('deleteTaskCancel'),
+                centered: true,
+                onOk: async () => {
+                    await handleDeleteSection();
+                },
+            });
+        } else if (groupBy === IGroupBy.PHASE) {
+            Modal.confirm({
+                title: t('deletePhaseTitle'),
+                content: t('deletePhaseContent'),
+                okText: t('deleteTaskConfirm'),
+                okType: 'danger',
+                cancelText: t('deleteTaskCancel'),
+                centered: true,
+                onOk: async () => {
+                    await handleDeleteSection();
+                },
+            });
+        } else {
+            Modal.confirm({
+                title: t('deleteConfirmationTitle'),
+                okText: t('deleteTaskConfirm'),
+                okType: 'danger',
+                cancelText: t('deleteTaskCancel'),
+                centered: true,
+                onOk: async () => {
+                    await handleDeleteSection();
+                },
+            });
+        }
         setShowDropdown(false);
     };
 
@@ -419,56 +455,7 @@ const KanbanGroup: React.FC<KanbanGroupProps> = memo(({
                 </div>
 
                 {/* Simple Delete Confirmation */}
-                {showDeleteConfirm && (
-                    <Portal>
-                        <div 
-                            className="fixed inset-0 bg-black bg-opacity-25 flex items-center justify-center z-[99999]"
-                            onClick={() => setShowDeleteConfirm(false)}
-                        >
-                            <div 
-                                className="bg-white dark:bg-gray-800 rounded-lg shadow-lg border border-gray-200 dark:border-gray-700 max-w-sm w-full mx-4"
-                                onClick={e => e.stopPropagation()}
-                            >
-                                <div className="p-4">
-                                    <div className="flex items-center gap-3 mb-3">
-                                        <div className="flex-shrink-0">
-                                            <svg className="w-5 h-5 text-orange-500" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                                                <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 9v2m0 4h.01m-6.938 4h13.856c1.54 0 2.502-1.667 1.732-2.5L13.732 4c-.77-.833-1.964-.833-2.732 0L3.732 16.5c-.77.833.192 2.5 1.732 2.5z" />
-                                            </svg>
-                                        </div>
-                                        <div>
-                                            <h3 className={`text-base font-medium ${themeMode === 'dark' ? 'text-white' : 'text-gray-900'}`}>
-                                                {t('deleteConfirmationTitle')}
-                                            </h3>
-                                        </div>
-                                    </div>
-                                    <div className="flex justify-end gap-2">
-                                        <button
-                                            type="button"
-                                            className={`px-3 py-1.5 text-sm font-medium rounded border transition-colors ${themeMode === 'dark'
-                                                ? 'border-gray-600 text-gray-300 hover:bg-gray-600'
-                                                : 'border-gray-300 text-gray-700 hover:bg-gray-50'
-                                                }`}
-                                            onClick={() => setShowDeleteConfirm(false)}
-                                        >
-                                            {t('deleteConfirmationCancel')}
-                                        </button>
-                                        <button
-                                            type="button"
-                                            className="px-3 py-1.5 text-sm font-medium text-white bg-red-600 border border-transparent rounded hover:bg-red-700 transition-colors"
-                                            onClick={() => {
-                                                handleDeleteSection();
-                                                setShowDeleteConfirm(false);
-                                            }}
-                                        >
-                                            {t('deleteConfirmationOk')}
-                                        </button>
-                                    </div>
-                                </div>
-                            </div>
-                        </div>
-                    </Portal>
-                )}
+                {/* Portal-based confirmation removed, now handled by Modal.confirm */}
                 <div className="enhanced-kanban-group-tasks">
                     {/* Create card at top */}
                     {showNewCardTop && (

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanBoardNativeDnD/TaskCard.tsx

@@ -14,7 +14,11 @@ import { useSocket } from '@/socket/socketContext';
 import { SocketEvents } from '@/shared/socket-events';
 import { getUserSession } from '@/utils/session-helper';
 import { themeWiseColor } from '@/utils/themeWiseColor';
-import { toggleTaskExpansion, fetchBoardSubTasks } from '@/features/enhanced-kanban/enhanced-kanban.slice';
+import { toggleTaskExpansion, fetchBoardSubTasks, deleteTask as deleteKanbanTask, updateEnhancedKanbanSubtask } from '@/features/enhanced-kanban/enhanced-kanban.slice';
+import TaskProgressCircle from './TaskProgressCircle';
+import { Button, Modal } from 'antd';
+import { DeleteOutlined } from '@ant-design/icons';
+import { tasksApiService } from '@/api/tasks/tasks.api.service';
 
 // Simple Portal component
 const Portal: React.FC<{ children: React.ReactNode }> = ({ children }) => {
@@ -69,7 +73,9 @@ const TaskCard: React.FC<TaskCardProps> = memo(({
         const d = selectedDate || new Date();
         return new Date(d.getFullYear(), d.getMonth(), 1);
     });
-    const [showSubtasks, setShowSubtasks] = useState(false);
+    const [contextMenu, setContextMenu] = useState<{ visible: boolean; x: number; y: number }>({ visible: false, x: 0, y: 0 });
+    const contextMenuRef = useRef<HTMLDivElement>(null);
+    const [selectedTask, setSelectedTask] = useState<IProjectTask | null>(null);
 
     useEffect(() => {
         setSelectedDate(task.end_date ? new Date(task.end_date) : null);
@@ -102,6 +108,21 @@ const TaskCard: React.FC<TaskCardProps> = memo(({
         }
     }, [showDatePicker]);
 
+    // Hide context menu on click elsewhere
+    useEffect(() => {
+        const handleClick = (e: MouseEvent) => {
+            if (contextMenuRef.current && !contextMenuRef.current.contains(e.target as Node)) {
+                setContextMenu({ ...contextMenu, visible: false });
+            }
+        };
+        if (contextMenu.visible) {
+            document.addEventListener('mousedown', handleClick);
+        }
+        return () => {
+            document.removeEventListener('mousedown', handleClick);
+        };
+    }, [contextMenu]);
+
     const handleCardClick = useCallback((e: React.MouseEvent, id: string) => {
         e.stopPropagation();
         dispatch(setSelectedTaskId(id));
@@ -178,6 +199,48 @@ const TaskCard: React.FC<TaskCardProps> = memo(({
         handleSubTaskExpand();
     }, [handleSubTaskExpand]);
 
+    // Delete logic (similar to task-drawer-header)
+    const handleDeleteTask = async (task: IProjectTask | null) => {
+        if (!task || !task.id) return;
+        Modal.confirm({
+            title: t('deleteTaskTitle'),
+            content: t('deleteTaskContent'),
+            okText: t('deleteTaskConfirm'),
+            okType: 'danger',
+            cancelText: t('deleteTaskCancel'),
+            centered: true,
+            onOk: async () => {
+                if (!task.id) return;
+                const res = await tasksApiService.deleteTask(task.id);
+                if (res.done) {
+                    dispatch(setSelectedTaskId(null));
+                    if (task.is_sub_task) {
+                        dispatch(updateEnhancedKanbanSubtask({
+                            sectionId: '',
+                            subtask: { id: task.id , parent_task_id: task.parent_task_id || '', manual_progress: false },
+                            mode: 'delete',
+                        }));
+                    } else {
+                        dispatch(deleteKanbanTask(task.id));
+                    }
+                    dispatch(setShowTaskDrawer(false));
+                    if (task.parent_task_id) {
+                        socket?.emit(
+                            SocketEvents.GET_TASK_PROGRESS.toString(),
+                            task.parent_task_id
+                        );
+                    }
+                }
+                setContextMenu({ visible: false, x: 0, y: 0 });
+                setSelectedTask(null);
+            },
+            onCancel: () => {
+                setContextMenu({ visible: false, x: 0, y: 0 });
+                setSelectedTask(null);
+            },
+        });
+    };
+
     // Calendar rendering helpers
     const year = calendarMonth.getFullYear();
     const month = calendarMonth.getMonth();
@@ -202,7 +265,41 @@ const TaskCard: React.FC<TaskCardProps> = memo(({
 
     return (
         <>
-            <div className="enhanced-kanban-task-card" style={{ background, color, display: 'block' }} >
+            {/* Context menu for delete */}
+            {contextMenu.visible && (
+                <div
+                    ref={contextMenuRef}
+                    style={{
+                        position: 'fixed',
+                        top: contextMenu.y,
+                        left: contextMenu.x,
+                        zIndex: 9999,
+                        background: themeMode === 'dark' ? '#23272f' : '#fff',
+                        borderRadius: 8,
+                        boxShadow: '0 2px 8px rgba(0,0,0,0.12)',
+                        padding: 0,
+                        minWidth: 120,
+                        transition: 'translateY(0)',
+                    }}
+                >
+                    <Button
+                        type="text"
+                        icon={<DeleteOutlined style={{ color: '#ef4444', fontSize: 16 }} />}
+                        style={{ color: '#ef4444', width: '100%', textAlign: 'left', padding: '8px 16px', fontWeight: 500 }}
+                        onClick={() => handleDeleteTask(selectedTask || null)}
+                    >
+                        {t('delete')}
+                    </Button>
+                </div>
+            )}
+            <div
+                className="enhanced-kanban-task-card"
+                style={{ background, color, display: 'block', position: 'relative' }}
+            >
+                {/* Progress circle at top right */}
+                <div style={{ position: 'absolute', top: 6, right: 6, zIndex: 2 }}>
+                    <TaskProgressCircle task={task} size={20} />
+                </div>
                 <div
                     draggable
                     onDragStart={e => onTaskDragStart(e, task.id!, groupId)}
@@ -217,6 +314,11 @@ const TaskCard: React.FC<TaskCardProps> = memo(({
                     onDrop={e => onTaskDrop(e, groupId, idx)}
                     onDragEnd={onDragEnd} // <-- add this
                     onClick={e => handleCardClick(e, task.id!)}
+                    onContextMenu={e => {
+                        e.preventDefault();
+                        setContextMenu({ visible: true, x: e.clientX, y: e.clientY });
+                        setSelectedTask(task);
+                    }}
                 >
                     <div className="task-content">
                         <div className="task_labels" style={{ display: 'flex', gap: 4, marginBottom: 4 }}>
@@ -443,14 +545,21 @@ const TaskCard: React.FC<TaskCardProps> = memo(({
                         {!task.sub_tasks_loading && Array.isArray(task.sub_tasks) && task.sub_tasks.length > 0 && (
                             <ul className="space-y-1">
                                 {task.sub_tasks.map(sub => (
-                                    <li key={sub.id} onClick={e => handleCardClick(e, sub.id!)} className="flex items-center gap-2 px-2 py-1 rounded hover:bg-gray-50 dark:hover:bg-gray-800">
+                                    <li key={sub.id} 
+                                    onClick={e => handleCardClick(e, sub.id!)} 
+                                    className="flex items-center gap-2 px-2 py-1 rounded hover:bg-gray-50 dark:hover:bg-gray-800"
+                                    onContextMenu={e => {
+                                        e.preventDefault();
+                                        setContextMenu({ visible: true, x: e.clientX, y: e.clientY });
+                                        setSelectedTask(sub);
+                                    }}>
                                         {sub.priority_color || sub.priority_color_dark ? (
                                             <span
                                                 className="w-2 h-2 rounded-full inline-block"
                                                 style={{ backgroundColor: themeMode === 'dark' ? (sub.priority_color_dark || sub.priority_color || '#d9d9d9') : (sub.priority_color || '#d9d9d9') }}
                                             ></span>
                                         ) : null}
-                                        <span className="flex-1 truncate text-xs text-gray-800 dark:text-gray-100">{sub.name}</span>
+                                        <span className="flex-1 truncate text-xs text-gray-800 dark:text-gray-100" title={sub.name}>{sub.name}</span>
                                         <span
                                             className="task-due-date ml-2 text-[10px] text-gray-500 dark:text-gray-400"
                                         >

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanBoardNativeDnD/TaskProgressCircle.tsx

@@ -0,0 +1,52 @@
+import { IProjectTask } from "@/types/project/projectTasksViewModel.types";
+
+// Add a simple circular progress component
+const TaskProgressCircle: React.FC<{ task: IProjectTask; size?: number }> = ({ task, size = 28 }) => {
+    const progress = typeof task.complete_ratio === 'number'
+        ? task.complete_ratio
+        : (typeof task.progress === 'number' ? task.progress : 0);
+    const strokeWidth = 1.5;
+    const radius = (size - strokeWidth) / 2;
+    const circumference = 2 * Math.PI * radius;
+    const offset = circumference - (progress / 100) * circumference;
+    return (
+        <svg width={size} height={size} style={{ display: 'block' }}>
+            
+            <circle
+                cx={size / 2}
+                cy={size / 2}
+                r={radius}
+                stroke={progress === 100 ? "#22c55e" : "#3b82f6"}
+                strokeWidth={strokeWidth}
+                fill="none"
+                strokeDasharray={circumference}
+                strokeDashoffset={offset}
+                strokeLinecap="round"
+                style={{ transition: 'stroke-dashoffset 0.3s' }}
+            />
+            {progress === 100 ? (
+                // Green checkmark icon
+                <g>
+                    <circle cx={size / 2} cy={size / 2} r={radius} fill="#22c55e" opacity="0.15" />
+                    <svg x={(size/2)-(size*0.22)} y={(size/2)-(size*0.22)} width={size*0.44} height={size*0.44} viewBox="0 0 24 24">
+                        <path d="M5 13l4 4L19 7" stroke="#22c55e" strokeWidth="2.2" fill="none" strokeLinecap="round" strokeLinejoin="round"/>
+                    </svg>
+                </g>
+            ) : progress > 0 && (
+                <text
+                    x="50%"
+                    y="50%"
+                    textAnchor="middle"
+                    dominantBaseline="central"
+                    fontSize={size * 0.38}
+                    fill="#3b82f6"
+                    fontWeight="bold"
+                >
+                    {Math.round(progress)}
+                </text>
+            )}
+        </svg>
+    );
+};
+
+export default TaskProgressCircle;

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanCreateSection.tsx

@@ -1,9 +1,9 @@
 import React, { useState, useRef, useEffect, useMemo } from 'react';
-import { Button, Flex } from 'antd';
-import { PlusOutlined } from '@ant-design/icons';
+import { Button, Flex } from '@/shared/antd-imports';
+import { PlusOutlined } from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 import { nanoid } from '@reduxjs/toolkit';
-import { DownOutlined } from '@ant-design/icons';
+import { DownOutlined } from '@/shared/antd-imports';
 
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { themeWiseColor } from '@/utils/themeWiseColor';

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanCreateSubtaskCard.tsx

@@ -1,4 +1,4 @@
-import { Flex, Input, InputRef } from 'antd';
+import { Flex, Input, InputRef } from '@/shared/antd-imports';
 import React, { useRef, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanCreateTaskCard.tsx

@@ -1,5 +1,5 @@
 import React, { useRef, useState, useEffect } from 'react';
-import { Button, Flex, Input, InputRef } from 'antd';
+import { Button, Flex, Input, InputRef } from '@/shared/antd-imports';
 import { useTranslation } from 'react-i18next';
 import { nanoid } from '@reduxjs/toolkit';
 import { useAppDispatch } from '@/hooks/useAppDispatch';

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanGroup.tsx

@@ -12,7 +12,7 @@ import EnhancedKanbanTaskCard from './EnhancedKanbanTaskCard';
 import VirtualizedTaskList from './VirtualizedTaskList';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import './EnhancedKanbanGroup.css';
-import { Badge, Flex, InputRef, MenuProps, Popconfirm } from 'antd';
+import { Badge, Flex, InputRef, MenuProps, Popconfirm } from '@/shared/antd-imports';
 import { themeWiseColor } from '@/utils/themeWiseColor';
 import useIsProjectManager from '@/hooks/useIsProjectManager';
 import { useAuthService } from '@/hooks/useAuth';
@@ -25,11 +25,11 @@ import {
   MoreOutlined,
 } from '@ant-design/icons/lib/icons';
 import { colors } from '@/styles/colors';
-import { Input } from 'antd';
-import { Tooltip } from 'antd';
-import { Typography } from 'antd';
-import { Dropdown } from 'antd';
-import { Button } from 'antd';
+import { Input } from '@/shared/antd-imports';
+import { Tooltip } from '@/shared/antd-imports';
+import { Typography } from '@/shared/antd-imports';
+import { Dropdown } from '@/shared/antd-imports';
+import { Button } from '@/shared/antd-imports';
 import { PlusOutlined } from '@ant-design/icons/lib/icons';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanTaskCard.tsx

@@ -15,18 +15,18 @@ import PrioritySection from '../board/taskCard/priority-section/priority-section
 import Typography from 'antd/es/typography';
 import CustomDueDatePicker from '../board/custom-due-date-picker';
 import { themeWiseColor } from '@/utils/themeWiseColor';
-import { ForkOutlined } from '@ant-design/icons';
+import { ForkOutlined } from '@/shared/antd-imports';
 import { Dayjs } from 'dayjs';
 import dayjs from 'dayjs';
-import { CaretDownFilled, CaretRightFilled } from '@ant-design/icons';
+import { CaretDownFilled, CaretRightFilled } from '@/shared/antd-imports';
 import {
   fetchBoardSubTasks,
   toggleTaskExpansion,
 } from '@/features/enhanced-kanban/enhanced-kanban.slice';
-import { Divider } from 'antd';
-import { List } from 'antd';
-import { Skeleton } from 'antd';
-import { PlusOutlined } from '@ant-design/icons';
+import { Divider } from '@/shared/antd-imports';
+import { List } from '@/shared/antd-imports';
+import { Skeleton } from '@/shared/antd-imports';
+import { PlusOutlined } from '@/shared/antd-imports';
 import BoardSubTaskCard from '@/pages/projects/projectView/board/board-section/board-sub-task-card/board-sub-task-card';
 import BoardCreateSubtaskCard from '@/pages/projects/projectView/board/board-section/board-sub-task-card/board-create-sub-task-card';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/enhanced-kanban/PerformanceMonitor.css

@@ -1,101 +0,0 @@
-.performance-monitor {
-  position: fixed;
-  top: 80px;
-  right: 16px;
-  width: 280px;
-  z-index: 1000;
-  background: var(--ant-color-bg-elevated);
-  border: 1px solid var(--ant-color-border);
-  box-shadow: 0 4px 12px var(--ant-color-shadow);
-}
-
-.performance-monitor-header {
-  display: flex;
-  justify-content: space-between;
-  align-items: center;
-  font-weight: 600;
-  color: var(--ant-color-text);
-}
-
-.performance-status {
-  font-size: 10px;
-  font-weight: 600;
-  text-transform: uppercase;
-  letter-spacing: 0.5px;
-}
-
-.performance-metrics {
-  display: grid;
-  grid-template-columns: 1fr 1fr;
-  gap: 12px;
-  margin-bottom: 12px;
-}
-
-.performance-metrics .ant-statistic {
-  text-align: center;
-}
-
-.performance-metrics .ant-statistic-title {
-  font-size: 12px;
-  color: var(--ant-color-text-secondary);
-  margin-bottom: 4px;
-}
-
-.performance-metrics .ant-statistic-content {
-  font-size: 14px;
-  color: var(--ant-color-text);
-}
-
-.virtualization-status {
-  grid-column: 1 / -1;
-  display: flex;
-  justify-content: space-between;
-  align-items: center;
-  padding: 8px 0;
-  border-top: 1px solid var(--ant-color-border);
-}
-
-.status-label {
-  font-size: 12px;
-  color: var(--ant-color-text-secondary);
-  font-weight: 500;
-}
-
-.performance-tips {
-  margin-top: 12px;
-  padding-top: 12px;
-  border-top: 1px solid var(--ant-color-border);
-}
-
-.performance-tips h4 {
-  font-size: 12px;
-  color: var(--ant-color-text);
-  margin-bottom: 8px;
-  font-weight: 600;
-}
-
-.performance-tips ul {
-  margin: 0;
-  padding-left: 16px;
-}
-
-.performance-tips li {
-  font-size: 11px;
-  color: var(--ant-color-text-secondary);
-  margin-bottom: 4px;
-  line-height: 1.4;
-}
-
-/* Responsive design */
-@media (max-width: 768px) {
-  .performance-monitor {
-    position: static;
-    width: 100%;
-    margin-bottom: 16px;
-  }
-
-  .performance-metrics {
-    grid-template-columns: 1fr;
-    gap: 8px;
-  }
-}

worklenz-frontend/src/components/enhanced-kanban/PerformanceMonitor.tsx

@@ -1,108 +0,0 @@
-import React from 'react';
-import { Card, Statistic, Tooltip, Badge } from 'antd';
-import { useSelector } from 'react-redux';
-import { RootState } from '@/app/store';
-import './PerformanceMonitor.css';
-
-const PerformanceMonitor: React.FC = () => {
-  const { performanceMetrics } = useSelector((state: RootState) => state.enhancedKanbanReducer);
-
-  // Only show if there are tasks loaded
-  if (performanceMetrics.totalTasks === 0) {
-    return null;
-  }
-
-  const getPerformanceStatus = () => {
-    if (performanceMetrics.totalTasks > 1000) return 'critical';
-    if (performanceMetrics.totalTasks > 500) return 'warning';
-    if (performanceMetrics.totalTasks > 100) return 'good';
-    return 'excellent';
-  };
-
-  const getStatusColor = (status: string) => {
-    switch (status) {
-      case 'critical':
-        return 'red';
-      case 'warning':
-        return 'orange';
-      case 'good':
-        return 'blue';
-      case 'excellent':
-        return 'green';
-      default:
-        return 'default';
-    }
-  };
-
-  const status = getPerformanceStatus();
-  const statusColor = getStatusColor(status);
-
-  return (
-    <Card
-      size="small"
-      className="performance-monitor"
-      title={
-        <div className="performance-monitor-header">
-          <span>Performance Monitor</span>
-          <Badge
-            status={statusColor as any}
-            text={status.toUpperCase()}
-            className="performance-status"
-          />
-        </div>
-      }
-    >
-      <div className="performance-metrics">
-        <Tooltip title="Total number of tasks across all groups">
-          <Statistic
-            title="Total Tasks"
-            value={performanceMetrics.totalTasks}
-            suffix="tasks"
-            valueStyle={{ fontSize: '16px' }}
-          />
-        </Tooltip>
-
-        <Tooltip title="Largest group by number of tasks">
-          <Statistic
-            title="Largest Group"
-            value={performanceMetrics.largestGroupSize}
-            suffix="tasks"
-            valueStyle={{ fontSize: '16px' }}
-          />
-        </Tooltip>
-
-        <Tooltip title="Average tasks per group">
-          <Statistic
-            title="Average Group"
-            value={Math.round(performanceMetrics.averageGroupSize)}
-            suffix="tasks"
-            valueStyle={{ fontSize: '16px' }}
-          />
-        </Tooltip>
-
-        <Tooltip title="Virtualization is enabled for groups with more than 50 tasks">
-          <div className="virtualization-status">
-            <span className="status-label">Virtualization:</span>
-            <Badge
-              status={performanceMetrics.virtualizationEnabled ? 'success' : 'default'}
-              text={performanceMetrics.virtualizationEnabled ? 'Enabled' : 'Disabled'}
-            />
-          </div>
-        </Tooltip>
-      </div>
-
-      {performanceMetrics.totalTasks > 500 && (
-        <div className="performance-tips">
-          <h4>Performance Tips:</h4>
-          <ul>
-            <li>Use filters to reduce the number of visible tasks</li>
-            <li>Consider grouping by different criteria</li>
-            <li>Virtualization is automatically enabled for large groups</li>
-          </ul>
-        </div>
-      )}
-    </Card>
-  );
-};
-
-export default React.memo(PerformanceMonitor);

worklenz-frontend/src/components/enhanced-kanban/VirtualizedTaskList.css

@@ -1,60 +0,0 @@
-.virtualized-task-list {
-  background: transparent;
-  border-radius: 6px;
-  overflow: hidden;
-}
-
-.virtualized-task-row {
-  padding: 4px 0;
-  display: flex;
-  align-items: stretch;
-}
-
-.virtualized-empty-state {
-  display: flex;
-  align-items: center;
-  justify-content: center;
-  background: var(--ant-color-bg-container);
-  border-radius: 6px;
-  border: 2px dashed var(--ant-color-border);
-}
-
-.empty-message {
-  color: var(--ant-color-text-secondary);
-  font-size: 14px;
-  font-weight: 500;
-}
-
-/* Ensure virtualized list works well with drag and drop */
-.virtualized-task-list .react-window__inner {
-  overflow: visible !important;
-}
-
-/* Performance optimizations */
-.virtualized-task-list * {
-  will-change: transform;
-}
-
-/* Smooth scrolling */
-.virtualized-task-list {
-  scroll-behavior: smooth;
-}
-
-/* Custom scrollbar for better UX */
-.virtualized-task-list::-webkit-scrollbar {
-  width: 6px;
-}
-
-.virtualized-task-list::-webkit-scrollbar-track {
-  background: var(--ant-color-bg-container);
-  border-radius: 3px;
-}
-
-.virtualized-task-list::-webkit-scrollbar-thumb {
-  background: var(--ant-color-border);
-  border-radius: 3px;
-}
-
-.virtualized-task-list::-webkit-scrollbar-thumb:hover {
-  background: var(--ant-color-text-tertiary);
-}

worklenz-frontend/src/components/enhanced-kanban/VirtualizedTaskList.tsx

@@ -1,94 +0,0 @@
-import React, { useMemo, useCallback } from 'react';
-import { FixedSizeList as List } from 'react-window';
-import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
-import EnhancedKanbanTaskCard from './EnhancedKanbanTaskCard';
-import './VirtualizedTaskList.css';
-
-interface VirtualizedTaskListProps {
-  tasks: IProjectTask[];
-  height: number;
-  itemHeight?: number;
-  activeTaskId?: string | null;
-  overId?: string | null;
-  onTaskRender?: (task: IProjectTask, index: number) => void;
-}
-
-const VirtualizedTaskList: React.FC<VirtualizedTaskListProps> = ({
-  tasks,
-  height,
-  itemHeight = 80,
-  activeTaskId,
-  overId,
-  onTaskRender,
-}) => {
-  // Memoize task data to prevent unnecessary re-renders
-  const taskData = useMemo(
-    () => ({
-      tasks,
-      activeTaskId,
-      overId,
-      onTaskRender,
-    }),
-    [tasks, activeTaskId, overId, onTaskRender]
-  );
-
-  // Row renderer for virtualized list
-  const Row = useCallback(
-    ({ index, style }: { index: number; style: React.CSSProperties }) => {
-      const task = tasks[index];
-      if (!task) return null;
-
-      // Call onTaskRender callback if provided
-      onTaskRender?.(task, index);
-
-      return (
-        <EnhancedKanbanTaskCard
-          task={task}
-          isActive={task.id === activeTaskId}
-          isDropTarget={overId === task.id}
-          sectionId={task.status || 'default'}
-        />
-      );
-    },
-    [tasks, activeTaskId, overId, onTaskRender]
-  );
-
-  // Memoize the list component to prevent unnecessary re-renders
-  const VirtualizedList = useMemo(
-    () => (
-      <List
-        height={height}
-        width="100%"
-        itemCount={tasks.length}
-        itemSize={itemHeight}
-        itemData={taskData}
-        overscanCount={10} // Increased overscan for smoother scrolling experience
-        className="virtualized-task-list"
-      >
-        {Row}
-      </List>
-    ),
-    [height, tasks.length, itemHeight, taskData, Row]
-  );
-
-  if (tasks.length === 0) {
-    return (
-      <div className="virtualized-empty-state" style={{ height, display: 'flex', alignItems: 'center', justifyContent: 'center' }}>
-        <div className="empty-message" style={{ 
-          padding: '32px 24px', 
-          color: '#8c8c8c', 
-          fontSize: '14px',
-          backgroundColor: '#fafafa',
-          borderRadius: '8px',
-          border: '1px solid #f0f0f0'
-        }}>
-          No tasks in this group
-        </div>
-      </div>
-    );
-  }
-
-  return VirtualizedList;
-};
-
-export default React.memo(VirtualizedTaskList);

worklenz-frontend/src/components/home-tasks/statusDropdown/home-tasks-status-dropdown.tsx

@@ -1,4 +1,4 @@
-import { Badge, Flex, Select } from 'antd';
+import { Badge, Flex, Select } from '@/shared/antd-imports';
 import './home-tasks-status-dropdown.css';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/home-tasks/taskDatePicker/home-tasks-date-picker.tsx

@@ -1,6 +1,6 @@
 import { useSocket } from '@/socket/socketContext';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
-import { DatePicker } from 'antd';
+import { DatePicker } from '@/shared/antd-imports';
 import dayjs from 'dayjs';
 import calendar from 'dayjs/plugin/calendar';
 import { SocketEvents } from '@/shared/socket-events';

worklenz-frontend/src/components/kanban-board-management-v2/kanbanGroup.tsx

@@ -1,8 +1,8 @@
 import React, { useState } from 'react';
 import { useDroppable } from '@dnd-kit/core';
 import { SortableContext, verticalListSortingStrategy } from '@dnd-kit/sortable';
-import { Button, Typography } from 'antd';
-import { PlusOutlined, MenuOutlined } from '@ant-design/icons';
+import { Button, Typography } from '@/shared/antd-imports';
+import { PlusOutlined, MenuOutlined } from '@/shared/antd-imports';
 import { ITaskListGroup } from '@/types/tasks/taskList.types';
 import { IGroupBy } from '@/features/tasks/tasks.slice';
 import KanbanTaskCard from './kanbanTaskCard';

worklenz-frontend/src/components/kanban-board-management-v2/kanbanTaskCard.tsx

@@ -1,13 +1,13 @@
 import React from 'react';
 import { useSortable } from '@dnd-kit/sortable';
 import { CSS } from '@dnd-kit/utilities';
-import { Avatar, Tag, Progress, Typography, Button, Tooltip, Space } from 'antd';
+import { Avatar, Tag, Progress, Typography, Button, Tooltip, Space } from '@/shared/antd-imports';
 import {
   HolderOutlined,
   MessageOutlined,
   PaperClipOutlined,
   ClockCircleOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
 import { IGroupBy } from '@/features/tasks/tasks.slice';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/kanban-board-management-v2/kanbanTaskListBoard.tsx

@@ -17,18 +17,13 @@ import {
   SortableContext,
   sortableKeyboardCoordinates,
 } from '@dnd-kit/sortable';
-import { Card, Spin, Empty, Flex } from 'antd';
+import { Card, Spin, Empty } from '@/shared/antd-imports';
 import { RootState } from '@/app/store';
-import { IGroupBy, setGroup, fetchTaskGroups, reorderTasks } from '@/features/tasks/tasks.slice';
+import { fetchTaskGroups, reorderTasks } from '@/features/tasks/tasks.slice';
 import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
 import { AppDispatch } from '@/app/store';
-import BoardSectionCard from '@/pages/projects/projectView/board/board-section/board-section-card/board-section-card';
-import BoardCreateSectionCard from '@/pages/projects/projectView/board/board-section/board-section-card/board-create-section-card';
-import { useAuthService } from '@/hooks/useAuth';
+import { useAuthService } from '@/hooks/useAuth'; 
 import useIsProjectManager from '@/hooks/useIsProjectManager';
-import BoardViewTaskCard from '@/pages/projects/projectView/board/board-section/board-task-card/board-view-task-card';
-import TaskGroup from '../task-management/TaskGroup';
-import TaskRow from '../task-management/TaskRow';
 import KanbanGroup from './kanbanGroup';
 import KanbanTaskCard from './kanbanTaskCard';
 import SortableKanbanGroup from './SortableKanbanGroup';

worklenz-frontend/src/components/navbar/notifications/notifications-drawer/notification/notfication-drawer.tsx

@@ -1,4 +1,4 @@
-import { Drawer, Empty, Segmented, Typography, Spin, Button, Flex } from 'antd';
+import { Drawer, Empty, Segmented, Typography, Spin, Button, Flex } from '@/shared/antd-imports';
 import { useEffect, useState } from 'react';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useAppDispatch } from '@/hooks/useAppDispatch';

worklenz-frontend/src/components/navbar/notifications/notifications-drawer/notification/notification-button.tsx

@@ -1,5 +1,5 @@
-import { BellOutlined } from '@ant-design/icons';
-import { Badge, Button, Tooltip } from 'antd';
+import { BellOutlined } from '@/shared/antd-imports';
+import { Badge, Button, Tooltip } from '@/shared/antd-imports';
 import { toggleDrawer } from '@features/navbar/notificationSlice';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { useTranslation } from 'react-i18next';

worklenz-frontend/src/components/navbar/notifications/notifications-drawer/notification/notification-item.tsx

@@ -1,6 +1,6 @@
 import { IWorklenzNotification } from '@/types/notifications/notifications.types';
-import { BankOutlined } from '@ant-design/icons';
-import { Button, Tag, Typography, theme } from 'antd';
+import { BankOutlined } from '@/shared/antd-imports';
+import { Button, Tag, Typography, theme } from '@/shared/antd-imports';
 import DOMPurify from 'dompurify';
 import React, { useState } from 'react';
 import { fromNow } from '@/utils/dateUtils';

worklenz-frontend/src/components/navbar/notifications/notifications-drawer/notification/notification-template.tsx

@@ -1,5 +1,5 @@
-import { Button, Typography, Tag } from 'antd';
-import { BankOutlined } from '@ant-design/icons';
+import { Button, Typography, Tag } from '@/shared/antd-imports';
+import { BankOutlined } from '@/shared/antd-imports';
 import { IWorklenzNotification } from '@/types/notifications/notifications.types';
 import { useNavigate } from 'react-router-dom';
 import { useAppDispatch } from '@/hooks/useAppDispatch';

worklenz-frontend/src/components/navbar/notifications/notifications-drawer/notification/push-notification-template.tsx

@@ -1,8 +1,8 @@
-import { notification } from 'antd';
+import { notification } from '@/shared/antd-imports';
 import { IWorklenzNotification } from '@/types/notifications/notifications.types';
 import { teamsApiService } from '@/api/teams/teams.api.service';
 import { toQueryString } from '@/utils/toQueryString';
-import { BankOutlined } from '@ant-design/icons';
+import { BankOutlined } from '@/shared/antd-imports';
 import './push-notification-template.css';
 
 const PushNotificationTemplate = ({

worklenz-frontend/src/components/project-list/project-group/project-group-list.tsx

@@ -10,12 +10,10 @@ import {
   Tooltip,
   Badge,
   Space,
-  Avatar,
   theme,
   Divider,
-} from 'antd';
+} from '@/shared/antd-imports';
 import {
-  ClockCircleOutlined,
   TeamOutlined,
   CheckCircleOutlined,
   ProjectOutlined,
@@ -23,7 +21,7 @@ import {
   SettingOutlined,
   InboxOutlined,
   MoreOutlined,
-} from '@ant-design/icons';
+} from '@/shared/antd-imports';
 import { ProjectGroupListProps } from '@/types/project/project.types';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import { useAppDispatch } from '@/hooks/useAppDispatch';

worklenz-frontend/src/components/project-list/project-list-table/project-list-actions/project-list-actions.tsx

@@ -11,11 +11,10 @@ import {
 } from '@/features/projects/projectsSlice';
 import { useAppSelector } from '@/hooks/useAppSelector';
 import useIsProjectManager from '@/hooks/useIsProjectManager';
-import { useAuthService } from '@/hooks/useAuth';
 import { IProjectViewModel } from '@/types/project/projectViewModel.types';
 import logger from '@/utils/errorLogger';
-import { SettingOutlined, InboxOutlined } from '@ant-design/icons';
-import { Tooltip, Button, Popconfirm, Space } from 'antd';
+import { SettingOutlined, InboxOutlined } from '@/shared/antd-imports';
+import { Tooltip, Button, Popconfirm, Space } from '@/shared/antd-imports';
 import {
   evt_projects_archive,
   evt_projects_archive_all,

worklenz-frontend/src/components/project-list/project-list-table/project-list-category/project-list-category.tsx

@@ -1,5 +1,5 @@
 import { IProjectViewModel } from '@/types/project/projectViewModel.types';
-import { Tooltip, Tag } from 'antd';
+import { Tooltip, Tag } from '@/shared/antd-imports';
 import { TFunction } from 'i18next';
 import { useAppDispatch } from '@/hooks/useAppDispatch';
 import { setFilteredCategories, setRequestParams } from '@/features/projects/projectsSlice';