UPDATE: pass document not documentId to editor

Author: yosr-nabli

client/src/components/DocumentArchive.tsx

@@ -171,26 +171,20 @@ const DocumentArchive = () => {
               <div className="flex justify-between text-sm text-muted-foreground mb-4">
                 <span>{formatDate(doc.uploadDate)}</span>
                 <span
-                  className={`px-2 py-0.5 rounded-full transition-all duration-200 ${
-                    doc.status === 'Anonymized'
+                  className={`px-2 py-0.5 rounded-full transition-all duration-200 ${doc.status === 'Anonymized'
                       ? 'bg-green-100 text-green-800 group-hover:bg-green-200'
                       : 'bg-gray-100 text-gray-800 group-hover:bg-gray-200'
-                  }`}
+                    }`}
                 >
                   {doc.status}
                 </span>
               </div>
               <Link
-                to={`/editor?id=${
-                  doc.documentType === 'anonymized'
-                    ? doc.originalDocumentId
-                    : doc.id
-                }`}
+                to="/editor"
+                state={{ document: doc }}
               >
                 <Button className="w-full transition-all duration-200 group-hover:bg-primary/90">
-                  {doc.documentType === 'anonymized'
-                    ? 'View Anonymized'
-                    : 'Open Document'}
+                  {doc.documentType === 'anonymized' ? 'View Anonymized' : 'Open Document'}
                 </Button>
               </Link>
             </div>

client/src/components/DocumentEditor.tsx

@@ -6,13 +6,14 @@ import AnonymizationPanel from './AnonymizationPanel';
 import SummarizationPanel from './SummarizationPanel';
 import EditAnonymizedDialog from './EditAnonymizedDialog';
 import { DocumentContent } from '@/types/documentContent';
+import {Document} from '@/types/document';
 import { toast } from 'sonner';
 
 type DocumentEditorProps = {
-  documentId?: string | null;
+  document?: Document | any;
 };
 
-const DocumentEditor = ({ documentId }: DocumentEditorProps) => {
+const DocumentEditor = ({ document }: DocumentEditorProps) => {
   const [activeTab, setActiveTab] = useState('anonymize');
   const [documentData, setDocumentData] = useState<DocumentContent>();
   const [selectedText, setSelectedText] = useState<string>('');
@@ -62,8 +63,18 @@ const DocumentEditor = ({ documentId }: DocumentEditorProps) => {
   }, [documentData]);
 
   useEffect(() => {
-    console.log('Fetching document with ID:', documentId);
+    console.log('Fetching document with ID:', document);
 
+    const documentInfo = sessionStorage.getItem('currentDocument');
+    if (document) {
+    const realContent: DocumentContent = {
+      title: document.fileName?.replace(/\.pdf$/i, '') || 'Untitled',
+      paragraph: document.documentText || '',
+      sensitive: [],
+      summary: '',
+    };
+    setDocumentData(realContent);
+  } else {
     const documentInfo = sessionStorage.getItem('currentDocument');
     if (documentInfo) {
       const parsedInfo = JSON.parse(documentInfo);
@@ -73,10 +84,10 @@ const DocumentEditor = ({ documentId }: DocumentEditorProps) => {
         sensitive: [],
         summary: '',
       };
-
       setDocumentData(realContent);
     }
-  }, [documentId]);
+  }
+}, [document]);
 
   if (!documentData) {
     return <div className="p-4 text-muted-foreground">Loading document...</div>;

client/src/pages/Editor.tsx

@@ -1,14 +1,24 @@
 import { useState, useEffect } from 'react';
-import { useSearchParams, useNavigate } from 'react-router-dom';
+import { useSearchParams, useNavigate, useLocation } from 'react-router-dom';
 import { Button } from '@/components/ui/button';
 import { ArrowLeft } from 'lucide-react';
 import DocumentEditor from '@/components/DocumentEditor';
 import Navbar from '@/components/Navbar';
+import { Document } from '@/types/document';
+
+type DocumentState = {
+  document: Document; 
+};
 
 const Editor = () => {
   const [searchParams] = useSearchParams();
   const documentId = searchParams.get('id');
   const navigate = useNavigate();
+  const location = useLocation();
+  const state = location.state as DocumentState | undefined;
+  const document = state?.document;
+
+
 
   return (
     <div className="min-h-screen flex flex-col">
@@ -31,7 +41,7 @@ const Editor = () => {
             </p>
           </div>
 
-          <DocumentEditor documentId={documentId} />
+          <DocumentEditor document={document} />
         </div>
       </main>
     </div>

client/src/pages/Index.tsx

@@ -70,7 +70,7 @@ const Index = () => {
       if (response.status == "PROCESSED") {
         toast.success("File uploaded successfully!");
         sessionStorage.setItem("currentDocument", JSON.stringify(response));
-        navigate(`/editor`);
+        navigate(`/editor`, { state: { document: response } });
         setSelectedFile(null);
         if (fileInputRef.current) {
           fileInputRef.current.value = "";