Replace all files except .github/workflows

Rezasz · Rezasz · commit 970f7ea3dcc2 · 2024-12-02T02:00:20.000+04:00
diff --git a/index.html b/index.html
@@ -4,10 +4,10 @@
     <meta charset="UTF-8" />
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Vite + React + TS</title>
+    <title>Kanz.AI</title>
   </head>
   <body>
     <div id="root"></div>
     <script type="module" src="/src/main.tsx"></script>
   </body>
-</html>
+</html>
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -15,7 +15,8 @@
     "react-dom": "^18.2.0",
     "react-router-dom": "^6.20.0",
     "react-markdown": "^8.0.7",
-    "firebase": "^10.7.1"
+    "firebase": "^10.7.1",
+    "react-hot-toast": "^2.4.1"
   },
   "devDependencies": {
     "@types/react": "^18.2.37",
diff --git a/src/components/DocumentList.tsx b/src/components/DocumentList.tsx
@@ -1,21 +1,26 @@
 import React from 'react';
 import { Download, Trash2, FileText } from 'lucide-react';
 import type { Document } from '../types/document';
+import { getDocumentUrl } from '../utils/documentStorage';
 
 interface DocumentListProps {
   documents: Document[];
   isAuthenticated: boolean;
-  onDelete: (docId: string) => Promise<void>;
+  onDelete: (docId: string) => void;
 }
 
-const DocumentList: React.FC<DocumentListProps> = ({ documents, isAuthenticated, onDelete }) => {
+const DocumentList: React.FC<DocumentListProps> = ({ 
+  documents, 
+  isAuthenticated, 
+  onDelete
+}) => {
   if (documents.length === 0) {
     return (
       <div className="text-center py-12">
         <FileText className="mx-auto h-12 w-12 text-gray-400" />
         <h3 className="mt-2 text-sm font-medium text-gray-900">No documents found</h3>
         <p className="mt-1 text-sm text-gray-500">
-          Upload some documents to get started
+          {isAuthenticated ? 'Upload some documents to get started' : 'Check back later for new documents'}
         </p>
       </div>
     );
@@ -35,17 +40,18 @@ const DocumentList: React.FC<DocumentListProps> = ({ documents, isAuthenticated,
             </div>
             <div className="flex items-center space-x-4">
               <a
-                href={doc.pdfUrl}
-                target="_blank"
-                rel="noopener noreferrer"
+                href={getDocumentUrl(doc.id) || '#'}
+                download={doc.fileName}
                 className="inline-flex items-center text-blue-600 hover:text-blue-700"
+                title="Download document"
               >
                 <Download className="h-5 w-5" />
               </a>
               {isAuthenticated && (
                 <button
                   onClick={() => onDelete(doc.id)}
                   className="text-red-600 hover:text-red-700"
+                  title="Delete document"
                 >
                   <Trash2 className="h-5 w-5" />
                 </button>
diff --git a/src/components/DocumentUploadForm.tsx b/src/components/DocumentUploadForm.tsx
@@ -14,10 +14,20 @@ const DocumentUploadForm: React.FC<DocumentUploadFormProps> = ({ onUpload, isUpl
   const handleSubmit = async (e: React.FormEvent) => {
     e.preventDefault();
     if (!file || !title || !brief) return;
+    
+    if (!file.type.includes('pdf')) {
+      alert('Please upload a PDF file');
+      return;
+    }
+    
     await onUpload(title, brief, file);
     setTitle('');
     setBrief('');
     setFile(null);
+    
+    // Reset file input
+    const fileInput = document.querySelector('input[type="file"]') as HTMLInputElement;
+    if (fileInput) fileInput.value = '';
   };
 
   return (
diff --git a/src/pages/KnowledgeBase.tsx b/src/pages/KnowledgeBase.tsx
@@ -1,9 +1,10 @@
 import React, { useState, useEffect } from 'react';
 import { useAuth } from '../contexts/AuthContext';
 import { Search } from 'lucide-react';
+import { Toaster, toast } from 'react-hot-toast';
 import DocumentUploadForm from '../components/DocumentUploadForm';
 import DocumentList from '../components/DocumentList';
-import { uploadDocument, fetchDocuments, deleteDocument } from '../utils/storage';
+import { saveDocument, getDocuments, deleteDocument } from '../utils/documentStorage';
 import type { Document } from '../types/document';
 
 const KnowledgeBase = () => {
@@ -16,32 +17,36 @@ const KnowledgeBase = () => {
     loadDocuments();
   }, []);
 
-  const loadDocuments = async () => {
-    try {
-      const docs = await fetchDocuments();
-      setDocuments(docs);
-    } catch (error) {
-      console.error('Error fetching documents:', error);
-    }
+  const loadDocuments = () => {
+    const docs = getDocuments();
+    setDocuments(docs);
   };
 
   const handleUpload = async (title: string, brief: string, file: File) => {
     setIsUploading(true);
     try {
-      await uploadDocument(title, brief, file);
-      await loadDocuments();
+      await saveDocument(title, brief, file);
+      loadDocuments();
+      toast.success('Document uploaded successfully!');
     } catch (error) {
       console.error('Error uploading document:', error);
+      toast.error('Failed to upload document. Please try again.');
+    } finally {
+      setIsUploading(false);
     }
-    setIsUploading(false);
   };
 
-  const handleDelete = async (docId: string) => {
+  const handleDelete = (docId: string) => {
+    const confirmed = window.confirm('Are you sure you want to delete this document?');
+    if (!confirmed) return;
+
     try {
-      await deleteDocument(docId);
-      await loadDocuments();
+      deleteDocument(docId);
+      loadDocuments();
+      toast.success('Document deleted successfully!');
     } catch (error) {
       console.error('Error deleting document:', error);
+      toast.error('Failed to delete document. Please try again.');
     }
   };
 
@@ -52,6 +57,8 @@ const KnowledgeBase = () => {
 
   return (
     <div>
+      <Toaster position="top-right" />
+      
       {/* Hero Section */}
       <section className="bg-gradient-to-r from-blue-600 to-blue-800 text-white py-20">
         <div className="max-w-7xl mx-auto px-4 sm:px-6 lg:px-8">
diff --git a/src/types/document.ts b/src/types/document.ts
@@ -2,7 +2,7 @@ export interface Document {
   id: string;
   title: string;
   brief: string;
-  pdfUrl: string;
+  fileName: string;
   uploadDate: string;
   uploadedBy: string;
 }
diff --git a/src/utils/documentStorage.ts b/src/utils/documentStorage.ts
@@ -0,0 +1,48 @@
+import type { Document } from '../types/document';
+
+const STORAGE_KEY = 'documents';
+
+export const saveDocument = async (title: string, brief: string, file: File): Promise<void> => {
+  const documents = getDocuments();
+  const id = crypto.randomUUID();
+  const timestamp = new Date().toISOString();
+  
+  // Create a blob URL for the file
+  const fileUrl = URL.createObjectURL(file);
+  
+  const newDocument: Document = {
+    id,
+    title,
+    brief,
+    fileName: file.name,
+    uploadDate: timestamp,
+    uploadedBy: 'admin'
+  };
+
+  // Store the blob URL in sessionStorage to persist across page reloads
+  sessionStorage.setItem(`file_${id}`, fileUrl);
+  
+  documents.push(newDocument);
+  localStorage.setItem(STORAGE_KEY, JSON.stringify(documents));
+};
+
+export const getDocuments = (): Document[] => {
+  const stored = localStorage.getItem(STORAGE_KEY);
+  return stored ? JSON.parse(stored) : [];
+};
+
+export const deleteDocument = (id: string): void => {
+  const documents = getDocuments().filter(doc => doc.id !== id);
+  localStorage.setItem(STORAGE_KEY, JSON.stringify(documents));
+  
+  // Clean up the blob URL
+  const fileUrl = sessionStorage.getItem(`file_${id}`);
+  if (fileUrl) {
+    URL.revokeObjectURL(fileUrl);
+    sessionStorage.removeItem(`file_${id}`);
+  }
+};
+
+export const getDocumentUrl = (id: string): string | null => {
+  return sessionStorage.getItem(`file_${id}`);
+};
diff --git a/src/utils/storage.ts b/src/utils/storage.ts
@@ -2,35 +2,73 @@ import { ref, uploadBytes, getDownloadURL, listAll, deleteObject } from 'firebas
 import { storage } from '../config/firebase';
 import type { Document } from '../types/document';
 
-export const uploadDocument = async (title: string, brief: string, file: File) => {
-  const fileName = `${title}__${brief}__${new Date().toISOString()}__admin`;
+const MAX_RETRIES = 3;
+const RETRY_DELAY = 1000;
+
+const sleep = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));
+
+const retry = async <T>(
+  operation: () => Promise<T>,
+  retries = MAX_RETRIES,
+  delay = RETRY_DELAY
+): Promise<T> => {
+  try {
+    return await operation();
+  } catch (error) {
+    if (retries > 0) {
+      await sleep(delay);
+      return retry(operation, retries - 1, delay * 2);
+    }
+    throw error;
+  }
+};
+
+export const uploadDocument = async (title: string, brief: string, file: File): Promise<void> => {
+  const sanitizedTitle = title.replace(/[^a-zA-Z0-9-_]/g, '_');
+  const timestamp = new Date().toISOString();
+  const fileName = `${sanitizedTitle}__${timestamp}__admin.pdf`;
   const storageRef = ref(storage, `documents/${fileName}`);
-  await uploadBytes(storageRef, file);
+  
+  const metadata = {
+    customMetadata: {
+      title,
+      brief,
+      uploadDate: timestamp,
+      uploadedBy: 'admin'
+    }
+  };
+
+  await retry(() => uploadBytes(storageRef, file, metadata));
 };
 
 export const fetchDocuments = async (): Promise<Document[]> => {
   const storageRef = ref(storage, 'documents');
-  const result = await listAll(storageRef);
+  
+  const result = await retry(() => listAll(storageRef));
   
   const docs = await Promise.all(
     result.items.map(async (item) => {
-      const url = await getDownloadURL(item);
-      const metadata = item.name.split('__');
+      const url = await retry(() => getDownloadURL(item));
+      const [fileName] = item.name.split('__');
+      const metadata = await retry(() => item.getMetadata());
+      
       return {
         id: item.name,
-        title: metadata[0],
-        brief: metadata[1],
+        title: metadata.customMetadata?.title || fileName,
+        brief: metadata.customMetadata?.brief || '',
         pdfUrl: url,
-        uploadDate: metadata[2],
-        uploadedBy: metadata[3]
+        uploadDate: metadata.customMetadata?.uploadDate || metadata.timeCreated,
+        uploadedBy: metadata.customMetadata?.uploadedBy || 'admin'
       };
     })
   );
   
-  return docs;
+  return docs.sort((a, b) => 
+    new Date(b.uploadDate).getTime() - new Date(a.uploadDate).getTime()
+  );
 };
 
-export const deleteDocument = async (docId: string) => {
+export const deleteDocument = async (docId: string): Promise<void> => {
   const docRef = ref(storage, `documents/${docId}`);
-  await deleteObject(docRef);
+  await retry(() => deleteObject(docRef));
 };

Original file line number	Diff line number	Diff line change
`@@ -2,7 +2,7 @@ export interface Document {`
`2`	`2`	`id: string;`
`3`	`3`	`title: string;`
`4`	`4`	`brief: string;`
`5`		`- pdfUrl: string;`
	`5`	`+ fileName: string;`
`6`	`6`	`uploadDate: string;`
`7`	`7`	`uploadedBy: string;`
`8`	`8`	`}`