App.tsx

import React, { useState } from 'react';
import Sidebar from './components/Sidebar';
import Header from './components/Header';
import DashboardPage from './pages/DashboardPage';
import StudentsPage from './pages/StudentsPage';
import AdmissionsPage from './pages/AdmissionsPage';
import TeachersPage from './pages/TeachersPage';
import GradesPage from './pages/GradesPage';
import FinancePage from './pages/FinancePage';
import AcademicsPage from './pages/AcademicsPage';
import CommunicationPage from './pages/CommunicationPage';
import SettingsPage from './pages/SettingsPage';
import LoginPage from './pages/LoginPage';
import AttendancePage from './pages/AttendancePage';
import TimetablePage from './pages/TimetablePage';
import { Page, User } from './types';
import { PAGES } from './constants';
import AdmissionDetailPage from './pages/AdmissionDetailPage';
import CurriculumPage from './pages/CurriculumPage';
import AIAssistantButton from './components/AIAssistantButton';
import AIAssistant from './components/AIAssistant';
import TeacherDashboardPage from './pages/TeacherDashboardPage';

const App: React.FC = () => {
  const [isLoggedIn, setIsLoggedIn] = useState(false);
  const [user, setUser] = useState<User | null>(null);
  const [currentPage, setCurrentPage] = useState<Page>(PAGES.DASHBOARD);
  const [isSidebarOpen, setSidebarOpen] = useState(false);
  const [isAiAssistantOpen, setIsAiAssistantOpen] = useState(false);

  const handleLogin = (loggedInUser: User) => {
    setUser(loggedInUser);
    setIsLoggedIn(true);
    setCurrentPage(PAGES.DASHBOARD);
  };

  const handleLogout = () => {
    setUser(null);
    setIsLoggedIn(false);
    setCurrentPage(PAGES.DASHBOARD);
  };

  const renderPage = () => {
    if (user?.role === 'Teacher') {
        switch (currentPage) {
            case PAGES.DASHBOARD:
                return <TeacherDashboardPage user={user} />;
            case PAGES.COMMUNICATION:
                return <CommunicationPage />;
            case PAGES.SETTINGS:
                return <SettingsPage />;
            default:
                return <TeacherDashboardPage user={user} />;
        }
    }

    // Admin pages
    switch (currentPage) {
      case PAGES.DASHBOARD:
        return <DashboardPage setCurrentPage={setCurrentPage} />;
      case PAGES.STUDENTS:
        return <StudentsPage />;
      case PAGES.ADMISSIONS:
        return <AdmissionsPage />;
      case PAGES.TEACHERS:
        return <TeachersPage />;
      case PAGES.GRADES:
        return <GradesPage />;
      case PAGES.FINANCE:
        return <FinancePage />;
      case PAGES.ACADEMICS:
        return <AcademicsPage />;
      case PAGES.CURRICULUM:
        return <CurriculumPage />;
      case PAGES.COMMUNICATION:
        return <CommunicationPage />;
      case PAGES.SETTINGS:
        return <SettingsPage />;
      case PAGES.ATTENDANCE:
        return <AttendancePage />;
      case PAGES.TIMETABLE:
        return <TimetablePage />;
      default:
        return <DashboardPage setCurrentPage={setCurrentPage} />;
    }
  };

  if (!isLoggedIn) {
    return <LoginPage onLogin={handleLogin} />;
  }

  return (
    <div className="relative flex h-screen bg-gray-100 font-sans print:h-auto print:bg-white">
      <Sidebar
        user={user}
        currentPage={currentPage}
        setCurrentPage={setCurrentPage}
        isSidebarOpen={isSidebarOpen}
        setSidebarOpen={setSidebarOpen}
      />
      <div className={`flex-1 flex flex-col overflow-hidden print:overflow-visible transition-all duration-300 ease-in-out ${isAiAssistantOpen ? 'mr-0 sm:mr-[24rem] md:mr-[28rem]' : ''}`}>
        <Header
          pageTitle={currentPage}
          user={user}
          onLogout={handleLogout}
          onMenuClick={() => setSidebarOpen(!isSidebarOpen)}
          setCurrentPage={setCurrentPage}
        />
        <main className="flex-1 overflow-x-hidden overflow-y-auto bg-gray-100 p-4 sm:p-6 print:overflow-visible print:p-0 print:bg-white">
          {renderPage()}
        </main>
      </div>
      {!isAiAssistantOpen && <AIAssistantButton onClick={() => setIsAiAssistantOpen(true)} />}
      <AIAssistant isOpen={isAiAssistantOpen} onClose={() => setIsAiAssistantOpen(false)} />
    </div>
  );
};

export default App;