From 70e9d2d3566b9247aed5c76eb992e9f59d4cdd09 Mon Sep 17 00:00:00 2001
From: truskovskiyk <truskovskiyk@gmail.com>
Date: Wed, 25 Dec 2024 22:29:39 +0100
Subject: [PATCH] layout

---
 no-ocr-ui/src/components/Search.tsx | 30 +++++++++++++++++++++++++----
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/no-ocr-ui/src/components/Search.tsx b/no-ocr-ui/src/components/Search.tsx
index 3266492..1b0f94f 100644
--- a/no-ocr-ui/src/components/Search.tsx
+++ b/no-ocr-ui/src/components/Search.tsx
@@ -11,7 +11,9 @@ export default function Search() {
   const [answers, setAnswers] = useState<{ [key: number]: { is_answer: boolean, answer: string } }>({});
 
   const [collections, setCollections] = useState<Collection[]>([]);
-  
+  const [isModalOpen, setIsModalOpen] = useState(false);
+  const [modalImage, setModalImage] = useState('');
+
   useEffect(() => {
     async function fetchCollections() {
       try {
@@ -32,8 +34,8 @@ export default function Search() {
     if (!selectedCollection || !searchQuery) return;
 
     setIsSearching(true);
-    setResults([]); // Clear old results
-    setAnswers({}); // Clear old answers
+    setResults([]);
+    setAnswers({});
 
     try {
       const response = await fetch(`${noOcrApiUrl}/search`, {
@@ -88,6 +90,16 @@ export default function Search() {
     }
   };
 
+  const openModal = (imageBase64: string) => {
+    setModalImage(imageBase64);
+    setIsModalOpen(true);
+  };
+
+  const closeModal = () => {
+    setIsModalOpen(false);
+    setModalImage('');
+  };
+
   return (
     <div className="max-w-4xl mx-auto mt-10 p-6">
       <h1 className="text-3xl font-bold text-gray-900 mb-8">AI Search</h1>
@@ -151,7 +163,8 @@ export default function Search() {
                   <img
                     src={`data:image/jpeg;base64,${result.image_base64}`}
                     alt={`Page ${result.pdf_page} of ${result.pdf_name}`}
-                    className="w-full h-auto rounded-lg border border-blue-300"
+                    className="w-full h-auto rounded-lg border border-blue-300 cursor-pointer"
+                    onClick={() => openModal(result.image_base64)}
                   />
                 </div>
                 <div className="flex flex-col justify-between">
@@ -188,6 +201,15 @@ export default function Search() {
           </div>
         </div>
       )}
+
+      {isModalOpen && (
+        <div className="fixed inset-0 bg-black bg-opacity-50 flex items-center justify-center z-50">
+          <div className="bg-white p-4 rounded-lg shadow-lg">
+            <button onClick={closeModal} className="text-red-500">Close</button>
+            <img src={`data:image/jpeg;base64,${modalImage}`} alt="Enlarged view" className="w-full h-auto mt-4" />
+          </div>
+        </div>
+      )}
     </div>
   );
 }