Upload folder using huggingface_hub

50a8ea3 verified 2 months ago

20 kB

import React, { useState, useEffect } from 'react';
import { Header } from './components/Header';
import { Footer } from './components/Footer';
import { RulebookSelector } from './components/RulebookSelector';
import { FilingWorkspace } from './components/FilingWorkspace';
import { MyFilingsDashboard } from './components/MyFilingsDashboard';
import { RulebooksModal } from './components/RulebooksModal';
import { AppealTrackerModal } from './components/AppealTrackerModal';
import { RevisionHistoryModal } from './components/RevisionHistoryModal';
import { ShareModal } from './components/ShareModal';
import { PDFExportModal } from './components/PDFExportModal';
import { LegalInfoModal } from './components/LegalInfoModal';
import { GuidedIntakeModal } from './components/GuidedIntakeModal';
import { RtiAuditorModal } from './components/RtiAuditorModal';
import { INITIAL_CASES, INITIAL_RULEBOOKS } from './data/mockData';
import { LegalCase, Rulebook, ExtractedFact } from './types';
import { AIAnalysisResponse } from './services/api';
import { useAuth } from './context/AuthContext';
import {
  subscribeToUserCases,
  saveUserCaseToFirestore,
  deleteUserCaseFromFirestore,
} from './firebase';

export default function App() {
  const { user } = useAuth();
  const [currentView, setCurrentView] = useState<'home' | 'workspace' | 'my-filings'>('home');
  const [cases, setCases] = useState<LegalCase[]>(INITIAL_CASES);
  const [selectedCaseId, setSelectedCaseId] = useState<string>(INITIAL_CASES[0].id);
  const [rulebooks] = useState<Rulebook[]>(INITIAL_RULEBOOKS);
  const [selectedCity, setSelectedCity] = useState<string>('Delhi (NCR)');
  const [toastMessage, setToastMessage] = useState<string | null>(null);

  // Modals state
  const [showRulebooksModal, setShowRulebooksModal] = useState(false);
  const [showAppealTracker, setShowAppealTracker] = useState(false);
  const [showRevisionHistory, setShowRevisionHistory] = useState(false);
  const [showGuidedIntake, setShowGuidedIntake] = useState(false);
  const [guidedIntakeCategory, setGuidedIntakeCategory] = useState<string | undefined>(undefined);
  const [showRtiAuditor, setShowRtiAuditor] = useState(false);
  const [caseToShare, setCaseToShare] = useState<LegalCase | null>(null);
  const [caseToPDF, setCaseToPDF] = useState<LegalCase | null>(null);
  const [legalModalType, setLegalModalType] = useState<
    'disclaimer' | 'accessibility' | 'rulebooks' | 'privacy' | null
  >(null);

  const showToast = (msg: string) => {
    setToastMessage(msg);
    setTimeout(() => setToastMessage(null), 3000);
  };

  // Real-time Firestore synchronization when user is signed in
  useEffect(() => {
    if (!user) {
      // If user logs out or is guest, retain in-memory state
      return;
    }

    // Subscribe to Firestore cases subcollection
    const unsubscribe = subscribeToUserCases(
      user.uid,
      (remoteCases) => {
        setCases(remoteCases);
        if (remoteCases.length > 0) {
          if (!remoteCases.some((c) => c.id === selectedCaseId)) {
            setSelectedCaseId(remoteCases[0].id);
          }
        }
      },
      (err) => {
        console.error('Real-time sync listener error:', err);
      }
    );

    return () => unsubscribe();
  }, [user]);

  const currentCase = cases.find((c) => c.id === selectedCaseId) || cases[0] || INITIAL_CASES[0];

  const handleUpdateCase = (updatedCase: LegalCase) => {
    setCases((prev) => prev.map((c) => (c.id === updatedCase.id ? updatedCase : c)));
    if (user) {
      saveUserCaseToFirestore(user.uid, updatedCase).catch((err) =>
        console.error('Failed to sync updated case to Firestore:', err)
      );
    }
  };

  const handleSelectCase = (caseId: string) => {
    setSelectedCaseId(caseId);
    setCurrentView('workspace');
  };

  const handleLaunchAICase = (aiData: AIAnalysisResponse, rawText: string) => {
    const newCaseId = `case-${Date.now()}`;
    const formattedFacts: ExtractedFact[] = aiData.facts.map((f, idx) => ({
      id: `fact-${Date.now()}-${idx}`,
      label: f.label,
      value: f.value,
    }));

    const formattedPii = (aiData.piiItems || []).map((p, idx) => ({
      id: `pii-${Date.now()}-${idx}`,
      original: p.original,
      type: p.type,
      masked: true,
    }));

    const todayStr = new Intl.DateTimeFormat('en-IN', {
      month: 'long',
      day: 'numeric',
      year: 'numeric',
    }).format(new Date());

    const newCase: LegalCase = {
      id: newCaseId,
      caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
      title: aiData.title,
      category: aiData.category,
      rulebookId: aiData.rulebookId,
      status: 'drafting',
      statusBadge: 'Assessed',
      statusType: 'success',
      deadlineText: `Reply required in ${aiData.daysRemaining} days`,
      filedDate: todayStr,
      description: aiData.formalLetter || aiData.legalDiagnosis,
      formalLetter: aiData.formalLetter,
      rawIntakeText: rawText,
      redactedText: aiData.redactedText,
      piiItems:
        formattedPii.length > 0
          ? formattedPii
          : [{ id: `pii-${Date.now()}-1`, original: 'Citizen Requester', type: 'Identity', masked: true }],
      facts:
        formattedFacts.length > 0
          ? formattedFacts
          : [{ id: `fact-${Date.now()}-1`, label: 'Primary Grievance', value: rawText }],
      officer: {
        name: aiData.officer.name,
        title: aiData.officer.title,
        department: aiData.officer.department,
        avatar: aiData.officer.avatar || 'PO',
        jurisdiction: aiData.officer.jurisdiction || selectedCity,
        email: aiData.officer.email,
        address: aiData.officer.address,
        portalUrl: aiData.officer.portalUrl,
        designationNote: aiData.officer.designationNote,
      },
      requestScope: aiData.requestScope,
      dateRange: aiData.dateRange || 'Past 90 Days',
      statute: aiData.statute,
      score: aiData.initialScore || 85,
      hasHardshipClause: false,
      autoRedactEnabled: true,
      autoEscalateEnabled: true,
      daysRemaining: aiData.daysRemaining || 18,
      revisionCount: 1,
      lastEdited: 'Just now',
      vulnerabilities: aiData.vulnerabilities,
      recommendedFixes: aiData.recommendedFixes,
      appellateStrategy: aiData.appellateStrategy,
      statutoryClauses: aiData.statutoryClauses,
      recommendedExhibits: aiData.recommendedExhibits,
    };

    setCases((prev) => [newCase, ...prev]);
    setSelectedCaseId(newCaseId);

    if (user) {
      saveUserCaseToFirestore(user.uid, newCase)
        .then(() => showToast('Case saved and synced to your Firebase Cloud'))
        .catch((err) => console.error('Cloud save error:', err));
    }

    setCurrentView('workspace');
  };

  const handleLaunchCustomDraft = (draftText: string, caseTitle: string) => {
    const newCaseId = `case-${Date.now()}`;
    const todayStr = new Intl.DateTimeFormat('en-IN', {
      month: 'long',
      day: 'numeric',
      year: 'numeric',
    }).format(new Date());

    const newCase: LegalCase = {
      id: newCaseId,
      caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
      title: caseTitle || 'Right to Information Notice',
      category: 'Public Records (RTI)',
      rulebookId: 'rti',
      status: 'drafting',
      statusBadge: 'RTI Audited',
      statusType: 'success',
      deadlineText: 'Reply required in 30 days',
      filedDate: todayStr,
      description: draftText,
      formalLetter: draftText,
      rawIntakeText: draftText,
      redactedText: draftText,
      piiItems: [],
      facts: [{ id: `fact-1`, label: 'RTI Scope', value: 'Statutory inspection & certified records request.' }],
      officer: {
        name: 'Public Information Officer (PIO)',
        title: 'Designated CPIO / PIO',
        department: 'Public Authority Division',
        avatar: 'PIO',
        jurisdiction: selectedCity,
        email: 'rti.cell@nic.in',
        address: `${selectedCity} Central Government / Civic Complex`,
        portalUrl: 'https://rtionline.gov.in',
      },
      requestScope: 'Certified copies under Section 6(1) and Section 2(j) of RTI Act 2005.',
      dateRange: 'Relevant Financial Year',
      statute: 'Right to Information Act 2005 § 6(1)',
      score: 95,
      hasHardshipClause: false,
      autoRedactEnabled: true,
      autoEscalateEnabled: true,
      daysRemaining: 30,
      revisionCount: 1,
      lastEdited: 'Just now',
    };

    setCases((prev) => [newCase, ...prev]);
    setSelectedCaseId(newCaseId);

    if (user) {
      saveUserCaseToFirestore(user.uid, newCase)
        .then(() => showToast('RTI Filing saved to Firebase Firestore'))
        .catch((err) => console.error('Cloud save error:', err));
    }

    setCurrentView('workspace');
  };

  const handleSaveAICaseToCloud = async (aiData: AIAnalysisResponse, rawText: string) => {
    const newCaseId = `case-${Date.now()}`;
    const formattedFacts: ExtractedFact[] = aiData.facts.map((f, idx) => ({
      id: `fact-${Date.now()}-${idx}`,
      label: f.label,
      value: f.value,
    }));

    const formattedPii = (aiData.piiItems || []).map((p, idx) => ({
      id: `pii-${Date.now()}-${idx}`,
      original: p.original,
      type: p.type,
      masked: true,
    }));

    const todayStr = new Intl.DateTimeFormat('en-IN', {
      month: 'long',
      day: 'numeric',
      year: 'numeric',
    }).format(new Date());

    const newCase: LegalCase = {
      id: newCaseId,
      caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
      title: aiData.title,
      category: aiData.category,
      rulebookId: aiData.rulebookId,
      status: 'drafting',
      statusBadge: 'Assessed',
      statusType: 'success',
      deadlineText: `Reply required in ${aiData.daysRemaining} days`,
      filedDate: todayStr,
      description: aiData.formalLetter || aiData.legalDiagnosis,
      formalLetter: aiData.formalLetter,
      rawIntakeText: rawText,
      redactedText: aiData.redactedText,
      piiItems: formattedPii,
      facts: formattedFacts,
      officer: {
        name: aiData.officer.name,
        title: aiData.officer.title,
        department: aiData.officer.department,
        avatar: aiData.officer.avatar || 'PO',
        jurisdiction: aiData.officer.jurisdiction || selectedCity,
        email: aiData.officer.email,
        address: aiData.officer.address,
        portalUrl: aiData.officer.portalUrl,
        designationNote: aiData.officer.designationNote,
      },
      requestScope: aiData.requestScope,
      dateRange: aiData.dateRange || 'Past 90 Days',
      statute: aiData.statute,
      score: aiData.initialScore || 85,
      hasHardshipClause: false,
      autoRedactEnabled: true,
      autoEscalateEnabled: true,
      daysRemaining: aiData.daysRemaining || 18,
      revisionCount: 1,
      lastEdited: 'Just now',
      vulnerabilities: aiData.vulnerabilities,
      recommendedFixes: aiData.recommendedFixes,
      appellateStrategy: aiData.appellateStrategy,
      statutoryClauses: aiData.statutoryClauses,
      recommendedExhibits: aiData.recommendedExhibits,
    };

    setCases((prev) => [newCase, ...prev]);

    if (user) {
      await saveUserCaseToFirestore(user.uid, newCase);
      showToast('Filing saved to Firebase Firestore Cloud');
    } else {
      showToast('Saved locally. Sign in to enable permanent cloud backup.');
    }
  };

  const handleSelectRulebook = (rulebookId: string, initialPrompt?: string) => {
    let title = 'New Filing Draft';
    let category = 'Public Records (RTI)';
    let scope = 'Certified copies of files, tender details, and official note sheets.';
    let statute = 'Right to Information Act 2005 § 6(1)';

    if (rulebookId === 'consumer') {
      title = 'Consumer Dispute - Defective Product & Deficient Service';
      category = 'Consumer Disputes';
      scope = 'Formal 15-day statutory notice for full refund and compensation.';
      statute = 'Consumer Protection Act 2019 § 35';
    } else if (rulebookId === 'tenancy') {
      title = 'Tenancy Dispute - Security Deposit Recovery';
      category = 'Tenancy Protections';
      scope = 'Statutory notice demanding immediate return of withheld security deposit with interest.';
      statute = 'State Tenancy & Rent Control Act';
    } else if (rulebookId === 'employment') {
      title = 'Wage Recovery & Final Settlement Demand';
      category = 'Labor & Employment Law';
      scope = 'Demand for immediate release of unpaid salary, F&F settlement, and relieving certificate.';
      statute = 'Payment of Wages Act 1936 § 15';
    }

    const newCaseId = `case-${Date.now()}`;
    const newCase: LegalCase = {
      ...currentCase,
      id: newCaseId,
      caseNumber: `#CL-2025-${Math.floor(1000 + Math.random() * 9000)}`,
      title: initialPrompt ? `Case: ${initialPrompt.slice(0, 45)}...` : title,
      category,
      rulebookId,
      requestScope: initialPrompt ? initialPrompt : scope,
      statute,
      filedDate: 'Current Filing',
      status: 'drafting',
      statusBadge: 'In Progress',
      statusType: 'warning',
      score: 85,
      hasHardshipClause: false,
      autoRedactEnabled: true,
      autoEscalateEnabled: false,
      daysRemaining: 30,
      lastEdited: 'Just now',
    };

    setCases((prev) => [newCase, ...prev]);
    setSelectedCaseId(newCaseId);

    if (user) {
      saveUserCaseToFirestore(user.uid, newCase).catch((err) =>
        console.error('Error saving template case to cloud:', err)
      );
    }

    setCurrentView('workspace');
  };

  const handleDeleteCase = (caseId: string) => {
    setCases((prev) => prev.filter((c) => c.id !== caseId));
    if (user) {
      deleteUserCaseFromFirestore(user.uid, caseId)
        .then(() => showToast('Case filing deleted from cloud'))
        .catch((err) => console.error('Error deleting from Firestore:', err));
    } else {
      showToast('Case removed');
    }
  };

  const handleDuplicateCase = (legalCase: LegalCase) => {
    const duplicatedId = `case-${Date.now()}`;
    const duplicatedCase: LegalCase = {
      ...legalCase,
      id: duplicatedId,
      caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
      title: `${legalCase.title} (Copy)`,
      lastEdited: 'Just now',
      revisionCount: 1,
    };

    setCases((prev) => [duplicatedCase, ...prev]);
    if (user) {
      saveUserCaseToFirestore(user.uid, duplicatedCase)
        .then(() => showToast('Case duplicated and saved in Cloud'))
        .catch((err) => console.error('Error saving duplicate to Firestore:', err));
    } else {
      showToast('Case duplicated');
    }
  };

  const handleNewFiling = () => {
    setShowGuidedIntake(true);
  };

  const handleOpenGuidedIntakeWithCategory = (category?: string) => {
    setGuidedIntakeCategory(category);
    setShowGuidedIntake(true);
  };

  const handleDownloadPDF = (legalCase: LegalCase) => {
    setCaseToPDF(legalCase);
  };

  const handleShareCase = (legalCase: LegalCase) => {
    setCaseToShare(legalCase);
  };

  const handleOpenQuickAction = (action: 'appeal' | 'history' | 'rulebooks') => {
    if (action === 'appeal') setShowAppealTracker(true);
    if (action === 'history') setShowRevisionHistory(true);
    if (action === 'rulebooks') setShowRulebooksModal(true);
  };

  return (
    <div className="min-h-screen flex flex-col bg-[#F8FAFC] text-[#191C1E] font-sans antialiased">
      {/* Top Notification Toast */}
      {toastMessage && (
        <div className="fixed top-18 right-6 z-50 bg-[#0F172A] text-white px-4 py-2.5 rounded-xl shadow-xl border border-slate-700 flex items-center gap-2 text-[13px] animate-in fade-in slide-in-from-top-2 duration-200">
          <span className="material-symbols-outlined text-[#006c4a] text-[18px]">check_circle</span>
          <span>{toastMessage}</span>
        </div>
      )}

      {/* TopNavBar Header */}
      <Header
        currentView={currentView}
        onNavigate={setCurrentView}
        selectedCity={selectedCity}
        onSelectCity={setSelectedCity}
        unreadNotifications={2}
        filingsCount={cases.length}
      />

      {/* Main View Router */}
      <main className="flex-grow flex flex-col">
        {currentView === 'home' && (
          <RulebookSelector
            rulebooks={rulebooks}
            selectedCity={selectedCity}
            onSelectRulebook={handleSelectRulebook}
            onLaunchAICase={handleLaunchAICase}
            onSaveAICaseToCloud={handleSaveAICaseToCloud}
            onViewAllRulebooks={() => setShowRulebooksModal(true)}
            onAnalyzePrompt={() => {}}
            onOpenGuidedIntake={handleOpenGuidedIntakeWithCategory}
            onOpenRtiAuditor={() => setShowRtiAuditor(true)}
          />
        )}

        {currentView === 'workspace' && (
          <FilingWorkspace
            currentCase={currentCase}
            onUpdateCase={handleUpdateCase}
            onNavigate={setCurrentView}
            onDownloadPDF={handleDownloadPDF}
          />
        )}

        {currentView === 'my-filings' && (
          <MyFilingsDashboard
            cases={cases}
            onSelectCase={handleSelectCase}
            onNewFiling={handleNewFiling}
            onDeleteCase={handleDeleteCase}
            onDuplicateCase={handleDuplicateCase}
            onDownloadPDF={handleDownloadPDF}
            onShareCase={handleShareCase}
            onOpenQuickAction={handleOpenQuickAction}
          />
        )}
      </main>

      {/* Footer */}
      <Footer
        onOpenLegal={(type) => {
          if (type === 'rulebooks') {
            setShowRulebooksModal(true);
          } else {
            setLegalModalType(type);
          }
        }}
      />

      {/* Modals */}
      {showGuidedIntake && (
        <GuidedIntakeModal
          initialCategory={guidedIntakeCategory}
          selectedCity={selectedCity}
          onClose={() => {
            setShowGuidedIntake(false);
            setGuidedIntakeCategory(undefined);
          }}
          onLaunchCase={handleLaunchAICase}
        />
      )}

      {showRtiAuditor && (
        <RtiAuditorModal
          selectedCity={selectedCity}
          onClose={() => setShowRtiAuditor(false)}
          onLaunchRtiCase={handleLaunchCustomDraft}
        />
      )}

      {showRulebooksModal && (
        <RulebooksModal
          rulebooks={rulebooks}
          onSelectRulebook={handleSelectRulebook}
          onClose={() => setShowRulebooksModal(false)}
        />
      )}

      {showAppealTracker && (
        <AppealTrackerModal
          onClose={() => setShowAppealTracker(false)}
          onOpenWorkspace={() => setCurrentView('workspace')}
        />
      )}

      {showRevisionHistory && (
        <RevisionHistoryModal onClose={() => setShowRevisionHistory(false)} />
      )}

      {caseToShare && (
        <ShareModal
          legalCase={caseToShare}
          onClose={() => setCaseToShare(null)}
        />
      )}

      {caseToPDF && (
        <PDFExportModal
          legalCase={caseToPDF}
          onClose={() => setCaseToPDF(null)}
        />
      )}

      {legalModalType && (
        <LegalInfoModal
          type={legalModalType}
          onClose={() => setLegalModalType(null)}
        />
      )}
    </div>
  );
}

	import React, { useState, useEffect } from 'react';
	import { Header } from './components/Header';
	import { Footer } from './components/Footer';
	import { RulebookSelector } from './components/RulebookSelector';
	import { FilingWorkspace } from './components/FilingWorkspace';
	import { MyFilingsDashboard } from './components/MyFilingsDashboard';
	import { RulebooksModal } from './components/RulebooksModal';
	import { AppealTrackerModal } from './components/AppealTrackerModal';
	import { RevisionHistoryModal } from './components/RevisionHistoryModal';
	import { ShareModal } from './components/ShareModal';
	import { PDFExportModal } from './components/PDFExportModal';
	import { LegalInfoModal } from './components/LegalInfoModal';
	import { GuidedIntakeModal } from './components/GuidedIntakeModal';
	import { RtiAuditorModal } from './components/RtiAuditorModal';
	import { INITIAL_CASES, INITIAL_RULEBOOKS } from './data/mockData';
	import { LegalCase, Rulebook, ExtractedFact } from './types';
	import { AIAnalysisResponse } from './services/api';
	import { useAuth } from './context/AuthContext';
	import {
	subscribeToUserCases,
	saveUserCaseToFirestore,
	deleteUserCaseFromFirestore,
	} from './firebase';

	export default function App() {
	const { user } = useAuth();
	const [currentView, setCurrentView] = useState<'home' \| 'workspace' \| 'my-filings'>('home');
	const [cases, setCases] = useState<LegalCase[]>(INITIAL_CASES);
	const [selectedCaseId, setSelectedCaseId] = useState<string>(INITIAL_CASES[0].id);
	const [rulebooks] = useState<Rulebook[]>(INITIAL_RULEBOOKS);
	const [selectedCity, setSelectedCity] = useState<string>('Delhi (NCR)');
	const [toastMessage, setToastMessage] = useState<string \| null>(null);

	// Modals state
	const [showRulebooksModal, setShowRulebooksModal] = useState(false);
	const [showAppealTracker, setShowAppealTracker] = useState(false);
	const [showRevisionHistory, setShowRevisionHistory] = useState(false);
	const [showGuidedIntake, setShowGuidedIntake] = useState(false);
	const [guidedIntakeCategory, setGuidedIntakeCategory] = useState<string \| undefined>(undefined);
	const [showRtiAuditor, setShowRtiAuditor] = useState(false);
	const [caseToShare, setCaseToShare] = useState<LegalCase \| null>(null);
	const [caseToPDF, setCaseToPDF] = useState<LegalCase \| null>(null);
	const [legalModalType, setLegalModalType] = useState<
	'disclaimer' \| 'accessibility' \| 'rulebooks' \| 'privacy' \| null
	>(null);

	const showToast = (msg: string) => {
	setToastMessage(msg);
	setTimeout(() => setToastMessage(null), 3000);
	};

	// Real-time Firestore synchronization when user is signed in
	useEffect(() => {
	if (!user) {
	// If user logs out or is guest, retain in-memory state
	return;
	}

	// Subscribe to Firestore cases subcollection
	const unsubscribe = subscribeToUserCases(
	user.uid,
	(remoteCases) => {
	setCases(remoteCases);
	if (remoteCases.length > 0) {
	if (!remoteCases.some((c) => c.id === selectedCaseId)) {
	setSelectedCaseId(remoteCases[0].id);
	}
	}
	},
	(err) => {
	console.error('Real-time sync listener error:', err);
	}
	);

	return () => unsubscribe();
	}, [user]);

	const currentCase = cases.find((c) => c.id === selectedCaseId) \|\| cases[0] \|\| INITIAL_CASES[0];

	const handleUpdateCase = (updatedCase: LegalCase) => {
	setCases((prev) => prev.map((c) => (c.id === updatedCase.id ? updatedCase : c)));
	if (user) {
	saveUserCaseToFirestore(user.uid, updatedCase).catch((err) =>
	console.error('Failed to sync updated case to Firestore:', err)
	);
	}
	};

	const handleSelectCase = (caseId: string) => {
	setSelectedCaseId(caseId);
	setCurrentView('workspace');
	};

	const handleLaunchAICase = (aiData: AIAnalysisResponse, rawText: string) => {
	const newCaseId = `case-${Date.now()}`;
	const formattedFacts: ExtractedFact[] = aiData.facts.map((f, idx) => ({
	id: `fact-${Date.now()}-${idx}`,
	label: f.label,
	value: f.value,
	}));

	const formattedPii = (aiData.piiItems \|\| []).map((p, idx) => ({
	id: `pii-${Date.now()}-${idx}`,
	original: p.original,
	type: p.type,
	masked: true,
	}));

	const todayStr = new Intl.DateTimeFormat('en-IN', {
	month: 'long',
	day: 'numeric',
	year: 'numeric',
	}).format(new Date());

	const newCase: LegalCase = {
	id: newCaseId,
	caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
	title: aiData.title,
	category: aiData.category,
	rulebookId: aiData.rulebookId,
	status: 'drafting',
	statusBadge: 'Assessed',
	statusType: 'success',
	deadlineText: `Reply required in ${aiData.daysRemaining} days`,
	filedDate: todayStr,
	description: aiData.formalLetter \|\| aiData.legalDiagnosis,
	formalLetter: aiData.formalLetter,
	rawIntakeText: rawText,
	redactedText: aiData.redactedText,
	piiItems:
	formattedPii.length > 0
	? formattedPii
	: [{ id: `pii-${Date.now()}-1`, original: 'Citizen Requester', type: 'Identity', masked: true }],
	facts:
	formattedFacts.length > 0
	? formattedFacts
	: [{ id: `fact-${Date.now()}-1`, label: 'Primary Grievance', value: rawText }],
	officer: {
	name: aiData.officer.name,
	title: aiData.officer.title,
	department: aiData.officer.department,
	avatar: aiData.officer.avatar \|\| 'PO',
	jurisdiction: aiData.officer.jurisdiction \|\| selectedCity,
	email: aiData.officer.email,
	address: aiData.officer.address,
	portalUrl: aiData.officer.portalUrl,
	designationNote: aiData.officer.designationNote,
	},
	requestScope: aiData.requestScope,
	dateRange: aiData.dateRange \|\| 'Past 90 Days',
	statute: aiData.statute,
	score: aiData.initialScore \|\| 85,
	hasHardshipClause: false,
	autoRedactEnabled: true,
	autoEscalateEnabled: true,
	daysRemaining: aiData.daysRemaining \|\| 18,
	revisionCount: 1,
	lastEdited: 'Just now',
	vulnerabilities: aiData.vulnerabilities,
	recommendedFixes: aiData.recommendedFixes,
	appellateStrategy: aiData.appellateStrategy,
	statutoryClauses: aiData.statutoryClauses,
	recommendedExhibits: aiData.recommendedExhibits,
	};

	setCases((prev) => [newCase, ...prev]);
	setSelectedCaseId(newCaseId);

	if (user) {
	saveUserCaseToFirestore(user.uid, newCase)
	.then(() => showToast('Case saved and synced to your Firebase Cloud'))
	.catch((err) => console.error('Cloud save error:', err));
	}

	setCurrentView('workspace');
	};

	const handleLaunchCustomDraft = (draftText: string, caseTitle: string) => {
	const newCaseId = `case-${Date.now()}`;
	const todayStr = new Intl.DateTimeFormat('en-IN', {
	month: 'long',
	day: 'numeric',
	year: 'numeric',
	}).format(new Date());

	const newCase: LegalCase = {
	id: newCaseId,
	caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
	title: caseTitle \|\| 'Right to Information Notice',
	category: 'Public Records (RTI)',
	rulebookId: 'rti',
	status: 'drafting',
	statusBadge: 'RTI Audited',
	statusType: 'success',
	deadlineText: 'Reply required in 30 days',
	filedDate: todayStr,
	description: draftText,
	formalLetter: draftText,
	rawIntakeText: draftText,
	redactedText: draftText,
	piiItems: [],
	facts: [{ id: `fact-1`, label: 'RTI Scope', value: 'Statutory inspection & certified records request.' }],
	officer: {
	name: 'Public Information Officer (PIO)',
	title: 'Designated CPIO / PIO',
	department: 'Public Authority Division',
	avatar: 'PIO',
	jurisdiction: selectedCity,
	email: 'rti.cell@nic.in',
	address: `${selectedCity} Central Government / Civic Complex`,
	portalUrl: 'https://rtionline.gov.in',
	},
	requestScope: 'Certified copies under Section 6(1) and Section 2(j) of RTI Act 2005.',
	dateRange: 'Relevant Financial Year',
	statute: 'Right to Information Act 2005 § 6(1)',
	score: 95,
	hasHardshipClause: false,
	autoRedactEnabled: true,
	autoEscalateEnabled: true,
	daysRemaining: 30,
	revisionCount: 1,
	lastEdited: 'Just now',
	};

	setCases((prev) => [newCase, ...prev]);
	setSelectedCaseId(newCaseId);

	if (user) {
	saveUserCaseToFirestore(user.uid, newCase)
	.then(() => showToast('RTI Filing saved to Firebase Firestore'))
	.catch((err) => console.error('Cloud save error:', err));
	}

	setCurrentView('workspace');
	};

	const handleSaveAICaseToCloud = async (aiData: AIAnalysisResponse, rawText: string) => {
	const newCaseId = `case-${Date.now()}`;
	const formattedFacts: ExtractedFact[] = aiData.facts.map((f, idx) => ({
	id: `fact-${Date.now()}-${idx}`,
	label: f.label,
	value: f.value,
	}));

	const formattedPii = (aiData.piiItems \|\| []).map((p, idx) => ({
	id: `pii-${Date.now()}-${idx}`,
	original: p.original,
	type: p.type,
	masked: true,
	}));

	const todayStr = new Intl.DateTimeFormat('en-IN', {
	month: 'long',
	day: 'numeric',
	year: 'numeric',
	}).format(new Date());

	const newCase: LegalCase = {
	id: newCaseId,
	caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
	title: aiData.title,
	category: aiData.category,
	rulebookId: aiData.rulebookId,
	status: 'drafting',
	statusBadge: 'Assessed',
	statusType: 'success',
	deadlineText: `Reply required in ${aiData.daysRemaining} days`,
	filedDate: todayStr,
	description: aiData.formalLetter \|\| aiData.legalDiagnosis,
	formalLetter: aiData.formalLetter,
	rawIntakeText: rawText,
	redactedText: aiData.redactedText,
	piiItems: formattedPii,
	facts: formattedFacts,
	officer: {
	name: aiData.officer.name,
	title: aiData.officer.title,
	department: aiData.officer.department,
	avatar: aiData.officer.avatar \|\| 'PO',
	jurisdiction: aiData.officer.jurisdiction \|\| selectedCity,
	email: aiData.officer.email,
	address: aiData.officer.address,
	portalUrl: aiData.officer.portalUrl,
	designationNote: aiData.officer.designationNote,
	},
	requestScope: aiData.requestScope,
	dateRange: aiData.dateRange \|\| 'Past 90 Days',
	statute: aiData.statute,
	score: aiData.initialScore \|\| 85,
	hasHardshipClause: false,
	autoRedactEnabled: true,
	autoEscalateEnabled: true,
	daysRemaining: aiData.daysRemaining \|\| 18,
	revisionCount: 1,
	lastEdited: 'Just now',
	vulnerabilities: aiData.vulnerabilities,
	recommendedFixes: aiData.recommendedFixes,
	appellateStrategy: aiData.appellateStrategy,
	statutoryClauses: aiData.statutoryClauses,
	recommendedExhibits: aiData.recommendedExhibits,
	};

	setCases((prev) => [newCase, ...prev]);

	if (user) {
	await saveUserCaseToFirestore(user.uid, newCase);
	showToast('Filing saved to Firebase Firestore Cloud');
	} else {
	showToast('Saved locally. Sign in to enable permanent cloud backup.');
	}
	};

	const handleSelectRulebook = (rulebookId: string, initialPrompt?: string) => {
	let title = 'New Filing Draft';
	let category = 'Public Records (RTI)';
	let scope = 'Certified copies of files, tender details, and official note sheets.';
	let statute = 'Right to Information Act 2005 § 6(1)';

	if (rulebookId === 'consumer') {
	title = 'Consumer Dispute - Defective Product & Deficient Service';
	category = 'Consumer Disputes';
	scope = 'Formal 15-day statutory notice for full refund and compensation.';
	statute = 'Consumer Protection Act 2019 § 35';
	} else if (rulebookId === 'tenancy') {
	title = 'Tenancy Dispute - Security Deposit Recovery';
	category = 'Tenancy Protections';
	scope = 'Statutory notice demanding immediate return of withheld security deposit with interest.';
	statute = 'State Tenancy & Rent Control Act';
	} else if (rulebookId === 'employment') {
	title = 'Wage Recovery & Final Settlement Demand';
	category = 'Labor & Employment Law';
	scope = 'Demand for immediate release of unpaid salary, F&F settlement, and relieving certificate.';
	statute = 'Payment of Wages Act 1936 § 15';
	}

	const newCaseId = `case-${Date.now()}`;
	const newCase: LegalCase = {
	...currentCase,
	id: newCaseId,
	caseNumber: `#CL-2025-${Math.floor(1000 + Math.random() * 9000)}`,
	title: initialPrompt ? `Case: ${initialPrompt.slice(0, 45)}...` : title,
	category,
	rulebookId,
	requestScope: initialPrompt ? initialPrompt : scope,
	statute,
	filedDate: 'Current Filing',
	status: 'drafting',
	statusBadge: 'In Progress',
	statusType: 'warning',
	score: 85,
	hasHardshipClause: false,
	autoRedactEnabled: true,
	autoEscalateEnabled: false,
	daysRemaining: 30,
	lastEdited: 'Just now',
	};

	setCases((prev) => [newCase, ...prev]);
	setSelectedCaseId(newCaseId);

	if (user) {
	saveUserCaseToFirestore(user.uid, newCase).catch((err) =>
	console.error('Error saving template case to cloud:', err)
	);
	}

	setCurrentView('workspace');
	};

	const handleDeleteCase = (caseId: string) => {
	setCases((prev) => prev.filter((c) => c.id !== caseId));
	if (user) {
	deleteUserCaseFromFirestore(user.uid, caseId)
	.then(() => showToast('Case filing deleted from cloud'))
	.catch((err) => console.error('Error deleting from Firestore:', err));
	} else {
	showToast('Case removed');
	}
	};

	const handleDuplicateCase = (legalCase: LegalCase) => {
	const duplicatedId = `case-${Date.now()}`;
	const duplicatedCase: LegalCase = {
	...legalCase,
	id: duplicatedId,
	caseNumber: `#CL-${new Date().getFullYear()}-${Math.floor(1000 + Math.random() * 9000)}`,
	title: `${legalCase.title} (Copy)`,
	lastEdited: 'Just now',
	revisionCount: 1,
	};

	setCases((prev) => [duplicatedCase, ...prev]);
	if (user) {
	saveUserCaseToFirestore(user.uid, duplicatedCase)
	.then(() => showToast('Case duplicated and saved in Cloud'))
	.catch((err) => console.error('Error saving duplicate to Firestore:', err));
	} else {
	showToast('Case duplicated');
	}
	};

	const handleNewFiling = () => {
	setShowGuidedIntake(true);
	};

	const handleOpenGuidedIntakeWithCategory = (category?: string) => {
	setGuidedIntakeCategory(category);
	setShowGuidedIntake(true);
	};

	const handleDownloadPDF = (legalCase: LegalCase) => {
	setCaseToPDF(legalCase);
	};

	const handleShareCase = (legalCase: LegalCase) => {
	setCaseToShare(legalCase);
	};

	const handleOpenQuickAction = (action: 'appeal' \| 'history' \| 'rulebooks') => {
	if (action === 'appeal') setShowAppealTracker(true);
	if (action === 'history') setShowRevisionHistory(true);
	if (action === 'rulebooks') setShowRulebooksModal(true);
	};

	return (
	<div className="min-h-screen flex flex-col bg-[#F8FAFC] text-[#191C1E] font-sans antialiased">
	{/* Top Notification Toast */}
	{toastMessage && (
	<div className="fixed top-18 right-6 z-50 bg-[#0F172A] text-white px-4 py-2.5 rounded-xl shadow-xl border border-slate-700 flex items-center gap-2 text-[13px] animate-in fade-in slide-in-from-top-2 duration-200">
	<span className="material-symbols-outlined text-[#006c4a] text-[18px]">check_circle</span>
	<span>{toastMessage}</span>
	</div>
	)}

	{/* TopNavBar Header */}
	<Header
	currentView={currentView}
	onNavigate={setCurrentView}
	selectedCity={selectedCity}
	onSelectCity={setSelectedCity}
	unreadNotifications={2}
	filingsCount={cases.length}
	/>

	{/* Main View Router */}
	<main className="flex-grow flex flex-col">
	{currentView === 'home' && (
	<RulebookSelector
	rulebooks={rulebooks}
	selectedCity={selectedCity}
	onSelectRulebook={handleSelectRulebook}
	onLaunchAICase={handleLaunchAICase}
	onSaveAICaseToCloud={handleSaveAICaseToCloud}
	onViewAllRulebooks={() => setShowRulebooksModal(true)}
	onAnalyzePrompt={() => {}}
	onOpenGuidedIntake={handleOpenGuidedIntakeWithCategory}
	onOpenRtiAuditor={() => setShowRtiAuditor(true)}
	/>
	)}

	{currentView === 'workspace' && (
	<FilingWorkspace
	currentCase={currentCase}
	onUpdateCase={handleUpdateCase}
	onNavigate={setCurrentView}
	onDownloadPDF={handleDownloadPDF}
	/>
	)}

	{currentView === 'my-filings' && (
	<MyFilingsDashboard
	cases={cases}
	onSelectCase={handleSelectCase}
	onNewFiling={handleNewFiling}
	onDeleteCase={handleDeleteCase}
	onDuplicateCase={handleDuplicateCase}
	onDownloadPDF={handleDownloadPDF}
	onShareCase={handleShareCase}
	onOpenQuickAction={handleOpenQuickAction}
	/>
	)}
	</main>

	{/* Footer */}
	<Footer
	onOpenLegal={(type) => {
	if (type === 'rulebooks') {
	setShowRulebooksModal(true);
	} else {
	setLegalModalType(type);
	}
	}}
	/>

	{/* Modals */}
	{showGuidedIntake && (
	<GuidedIntakeModal
	initialCategory={guidedIntakeCategory}
	selectedCity={selectedCity}
	onClose={() => {
	setShowGuidedIntake(false);
	setGuidedIntakeCategory(undefined);
	}}
	onLaunchCase={handleLaunchAICase}
	/>
	)}

	{showRtiAuditor && (
	<RtiAuditorModal
	selectedCity={selectedCity}
	onClose={() => setShowRtiAuditor(false)}
	onLaunchRtiCase={handleLaunchCustomDraft}
	/>
	)}

	{showRulebooksModal && (
	<RulebooksModal
	rulebooks={rulebooks}
	onSelectRulebook={handleSelectRulebook}
	onClose={() => setShowRulebooksModal(false)}
	/>
	)}

	{showAppealTracker && (
	<AppealTrackerModal
	onClose={() => setShowAppealTracker(false)}
	onOpenWorkspace={() => setCurrentView('workspace')}
	/>
	)}

	{showRevisionHistory && (
	<RevisionHistoryModal onClose={() => setShowRevisionHistory(false)} />
	)}

	{caseToShare && (
	<ShareModal
	legalCase={caseToShare}
	onClose={() => setCaseToShare(null)}
	/>
	)}

	{caseToPDF && (
	<PDFExportModal
	legalCase={caseToPDF}
	onClose={() => setCaseToPDF(null)}
	/>
	)}

	{legalModalType && (
	<LegalInfoModal
	type={legalModalType}
	onClose={() => setLegalModalType(null)}
	/>
	)}
	</div>
	);
	}