diff --git a/src/pages/Gallary/ImageGallary.jsx b/src/pages/Gallary/ImageGallary.jsx
index bfa30e79..70f3cf9c 100644
--- a/src/pages/Gallary/ImageGallary.jsx
+++ b/src/pages/Gallary/ImageGallary.jsx
@@ -1,4 +1,3 @@
-// ImageGallery.js
 import React, { useState, useEffect, useRef, useCallback } from "react";
 import "./ImageGallery.css";
 import { ImageGalleryAPI } from "./ImageGalleryAPI";
@@ -7,10 +6,17 @@ import { useSelector } from "react-redux";
 import { useModal } from "./ModalContext";
 import ImagePop from "./ImagePop";
 import Avatar from "../../components/common/Avatar";
-import DateRangePicker from "../../components/common/DateRangePicker"; // Assuming this is the path to your DateRangePicker
+import DateRangePicker from "../../components/common/DateRangePicker";
+import eventBus from "../../services/eventBus";
+
+const PAGE_SIZE = 10;
+const SCROLL_THRESHOLD = 5;
 
 const ImageGallery = () => {
   const [images, setImages] = useState([]);
+  const [allImagesData, setAllImagesData] = useState([]);
+  const [pageNumber, setPageNumber] = useState(1);
+  const [hasMore, setHasMore] = useState(true);
   const selectedProjectId = useSelector((store) => store.localVariables.projectId);
   const { openModal } = useModal();
 
@@ -51,8 +57,10 @@ const ImageGallery = () => {
   const [isFilterPanelOpen, setIsFilterPanelOpen] = useState(false);
   const [hoveredImage, setHoveredImage] = useState(null);
   const [loading, setLoading] = useState(true);
+  const [loadingMore, setLoadingMore] = useState(false);
 
   const imageGroupRefs = useRef({});
+  const loaderRef = useRef(null);
   const filterPanelRef = useRef(null);
   const filterButtonRef = useRef(null);
 
@@ -68,61 +76,159 @@ const ImageGallery = () => {
       }
     };
 
-    document.addEventListener("mousedown", handleClickOutside);
+    if (isFilterPanelOpen) {
+      document.addEventListener("mousedown", handleClickOutside);
+    } else {
+      document.removeEventListener("mousedown", handleClickOutside);
+    }
+
     return () => {
       document.removeEventListener("mousedown", handleClickOutside);
     };
-  }, []);
+  }, [isFilterPanelOpen]);
 
   useEffect(() => {
     if (!selectedProjectId) {
       setImages([]);
+      setAllImagesData([]);
       setLoading(false);
+      setHasMore(false);
       return;
     }
 
+    setImages([]);
+    setPageNumber(1);
+    setHasMore(true);
     setLoading(true);
 
-    ImageGalleryAPI.ImagesGet(selectedProjectId, appliedFilters)
-      .then((res) => {
-        setImages(res.data);
-      })
-      .catch((err) => {
-        console.error("Error fetching images:", err);
-        setImages([]);
-      })
-      .finally(() => {
-        setLoading(false);
-      });
+    setAllImagesData([]);
+    fetchImages(1, appliedFilters, true);
   }, [selectedProjectId, appliedFilters]);
 
-  const getUniqueValuesWithIds = useCallback(
-    (idKey, nameKey) => {
-      const uniqueMap = new Map();
-      images.forEach((img) => {
-        if (img[idKey] && img[nameKey]) {
-          uniqueMap.set(img[idKey], img[nameKey]);
-        }
+  const fetchImages = useCallback(async (page, filters) => {
+    if (!selectedProjectId) return;
+
+    try {
+      if (page === 1) {
+        setLoading(true);
+      } else {
+        setLoadingMore(true);
+      }
+
+      const res = await ImageGalleryAPI.ImagesGet(selectedProjectId, filters, page, PAGE_SIZE);
+      const newBatches = res.data || [];
+      const receivedCount = newBatches.length;
+
+      setImages((prevImages) => {
+        const uniqueNewBatches = newBatches.filter(
+          (newBatch) => !prevImages.some((prevBatch) => prevBatch.batchId === newBatch.batchId)
+        );
+        return [...prevImages, ...uniqueNewBatches];
       });
-      return Array.from(uniqueMap.entries());
-    },
-    [images]
-  );
+
+      setAllImagesData((prevAllImages) => {
+        const uniqueAllImages = newBatches.filter(
+          (newBatch) => !prevAllImages.some((prevBatch) => prevBatch.batchId === newBatch.batchId)
+        );
+        return [...prevAllImages, ...uniqueAllImages];
+      });
+
+      setHasMore(receivedCount === PAGE_SIZE);
+    } catch (err) {
+      console.error("Error fetching images:", err);
+      if (page === 1) {
+        setImages([]);
+        setAllImagesData([]);
+      }
+      setHasMore(false);
+    } finally {
+      setLoading(false);
+      setLoadingMore(false);
+    }
+  }, [selectedProjectId]);
+
+  useEffect(() => {
+    const handleExternalEvent = (data) => {
+      if (selectedProjectId === data.projectId) {
+        setImages([]);
+        setAllImagesData([]);
+        setPageNumber(1);
+        setHasMore(true);
+        fetchImages(1, appliedFilters, true);
+      }
+    };
+
+    eventBus.on("image_gallery", handleExternalEvent);
+
+    return () => {
+      eventBus.off("image_gallery", handleExternalEvent);
+    };
+  }, [appliedFilters, fetchImages, selectedProjectId]);
+
+  useEffect(() => {
+    if (!loaderRef.current) return;
+
+    const observer = new IntersectionObserver(
+      (entries) => {
+        if (entries[0].isIntersecting && hasMore && !loadingMore && !loading) {
+          setPageNumber((prevPageNumber) => prevPageNumber + 1);
+        }
+      },
+      {
+        root: null,
+        rootMargin: "200px",
+        threshold: 0.1,
+      }
+    );
+
+    observer.observe(loaderRef.current);
+
+    return () => {
+      if (loaderRef.current) {
+        observer.unobserve(loaderRef.current);
+      }
+    };
+  }, [hasMore, loadingMore, loading]);
+
+  useEffect(() => {
+    if (pageNumber > 1) {
+      fetchImages(pageNumber, appliedFilters);
+    }
+  }, [pageNumber, fetchImages, appliedFilters]);
+
+  const getUniqueValuesWithIds = useCallback((idKey, nameKey) => {
+    const map = new Map();
+    allImagesData.forEach(batch => {
+      let id;
+      if (idKey === "floorIds") {
+        id = batch.floorIds;
+      } else {
+        id = batch[idKey];
+      }
+      
+      const name = batch[nameKey];
+      
+      if (id && name && !map.has(id)) {
+        map.set(id, name);
+      }
+    });
+    return Array.from(map.entries());
+  }, [allImagesData]);
 
   const getUniqueUploadedByUsers = useCallback(() => {
     const uniqueUsersMap = new Map();
-    images.forEach((img) => {
-      if (img.uploadedBy && img.uploadedBy.id) {
-        const fullName = `${img.uploadedBy.firstName || ""} ${
-          img.uploadedBy.lastName || ""
-        }`.trim();
-        if (fullName) {
-          uniqueUsersMap.set(img.uploadedBy.id, fullName);
+    allImagesData.forEach(batch => {
+      batch.documents.forEach(doc => {
+        if (doc.uploadedBy && doc.uploadedBy.id) {
+          const fullName = `${doc.uploadedBy.firstName || ""} ${doc.uploadedBy.lastName || ""}`.trim();
+          if (fullName) {
+            uniqueUsersMap.set(doc.uploadedBy.id, fullName);
+          }
         }
-      }
+      });
     });
     return Array.from(uniqueUsersMap.entries());
-  }, [images]);
+  }, [allImagesData]);
 
   const buildings = getUniqueValuesWithIds("buildingId", "buildingName");
   const floors = getUniqueValuesWithIds("floorIds", "floorName");
@@ -191,9 +297,35 @@ const ImageGallery = () => {
       startDate: selectedFilters.startDate || null,
       endDate: selectedFilters.endDate || null,
     };
-    setAppliedFilters(payload);
-    setIsFilterPanelOpen(false);
-  }, [selectedFilters]);
+
+    const areFiltersChanged = Object.keys(payload).some(key => {
+      const oldVal = appliedFilters[key];
+      const newVal = payload[key];
+
+      if (Array.isArray(oldVal) && Array.isArray(newVal)) {
+        if (oldVal.length !== newVal.length) return true;
+        const oldSet = new Set(oldVal);
+        const newSet = new Set(newVal);
+        if (oldSet.size !== newSet.size) return true;
+        for (const item of newSet) {
+          if (!oldSet.has(item)) return true;
+        }
+        return false;
+      }
+      if ((oldVal === null && newVal === "") || (oldVal === "" && newVal === null)) {
+          return false;
+      }
+      return oldVal !== newVal;
+    });
+
+    if (areFiltersChanged) {
+      setAppliedFilters(payload);
+      setImages([]);
+      setPageNumber(1);
+      setHasMore(true);
+    }
+    // Removed setIsFilterPanelOpen(false); to keep the drawer open
+  }, [selectedFilters, appliedFilters]);
 
   const handleClearAllFilters = useCallback(() => {
     const initialStateSelected = {
@@ -219,49 +351,11 @@ const ImageGallery = () => {
       endDate: null,
     };
     setAppliedFilters(initialStateApplied);
+    setImages([]);
+    setPageNumber(1);
+    setHasMore(true);
   }, []);
 
-  const filteredImages = images.filter((img) => {
-    const uploadedAtMoment = moment(img.uploadedAt);
-    const startDateMoment = appliedFilters.startDate
-      ? moment(appliedFilters.startDate)
-      : null;
-    const endDateMoment = appliedFilters.endDate
-      ? moment(appliedFilters.endDate)
-      : null;
-
-    const isWithinDateRange =
-      (!startDateMoment || uploadedAtMoment.isSameOrAfter(startDateMoment, "day")) &&
-      (!endDateMoment || uploadedAtMoment.isSameOrBefore(endDateMoment, "day"));
-
-    const passesCategoryFilters =
-      (appliedFilters.buildingIds === null ||
-        appliedFilters.buildingIds.includes(img.buildingId)) &&
-      (appliedFilters.floorIds === null ||
-        appliedFilters.floorIds.includes(img.floorIds)) &&
-      (appliedFilters.activityIds === null ||
-        appliedFilters.activityIds.includes(img.activityId)) &&
-      (appliedFilters.workAreaIds === null ||
-        appliedFilters.workAreaIds.includes(img.workAreaId)) &&
-      (appliedFilters.uploadedByIds === null ||
-        appliedFilters.uploadedByIds.includes(img.uploadedBy?.id)) &&
-      (appliedFilters.workCategoryIds === null ||
-        appliedFilters.workCategoryIds.includes(img.workCategoryId));
-
-    return isWithinDateRange && passesCategoryFilters;
-  });
-
-  const imagesByActivityUser = {};
-  filteredImages.forEach((img) => {
-    const userName = `${img.uploadedBy?.firstName || ""} ${
-      img.uploadedBy?.lastName || ""
-    }`.trim();
-    const workArea = img.workAreaName || "Unknown";
-    const key = `${img.activityName}__${userName}__${workArea}`;
-    if (!imagesByActivityUser[key]) imagesByActivityUser[key] = [];
-    imagesByActivityUser[key].push(img);
-  });
-
   const scrollLeft = useCallback((key) => {
     imageGroupRefs.current[key]?.scrollBy({ left: -200, behavior: "smooth" });
   }, []);
@@ -307,7 +401,15 @@ const ImageGallery = () => {
       {!collapsedFilters[type] && (
         <div className="dropdown-content">
           {type === "dateRange" ? (
-            null
+            <div className="date-range-inputs">
+              <DateRangePicker
+                onRangeChange={setDateRange}
+                defaultStartDate={selectedFilters.startDate || yesterday}
+                defaultEndDate={selectedFilters.endDate || moment().format('YYYY-MM-DD')}
+                startDate={selectedFilters.startDate}
+                endDate={selectedFilters.endDate}
+              />
+            </div>
           ) : (
             items.map((item) => {
               const itemId = item[0];
@@ -334,35 +436,49 @@ const ImageGallery = () => {
   );
 
   return (
-    <div className={`gallery-container ${isFilterPanelOpen ? "filter-panel-open" : ""}`}>
+    <div className={`gallery-container ${isFilterPanelOpen ? "filter-panel-open-end" : ""}`}>
       <div className="main-content">
+        <button
+          className={`filter-button btn-primary ${isFilterPanelOpen ? "closed-icon" : ""}`}
+          onClick={() => setIsFilterPanelOpen(!isFilterPanelOpen)}
+          ref={filterButtonRef}
+        >
+          {isFilterPanelOpen ? (
+            <i className="fa-solid fa-times fs-5"></i>
+          ) : (
+            <><i className="fa-solid fa-filter ms-1 fs-5"></i></>
+          )}
+        </button>
+
         <div className="activity-section">
-          {loading ? (
+          {loading && pageNumber === 1 ? (
             <div className="spinner-container">
               <div className="spinner" />
             </div>
-          ) : Object.entries(imagesByActivityUser).length > 0 ? (
-            Object.entries(imagesByActivityUser).map(([key, imgs]) => {
-              const [activity, userName, workArea] = key.split("__");
-              const { buildingName, floorName, uploadedAt, workCategoryName } = imgs[0];
-              const date = moment(uploadedAt).format("YYYY-MM-DD");
-              const time = moment(uploadedAt).format("hh:mm A");
+          ) : images.length > 0 ? (
+            images.map((batch) => {
+              const firstDoc = batch.documents[0];
+              const userName = `${firstDoc?.uploadedBy?.firstName || ""} ${firstDoc?.uploadedBy?.lastName || ""
+                }`.trim();
+              const date = moment(firstDoc?.uploadedAt).format("DD-MM-YYYY");
+              const time = moment(firstDoc?.uploadedAt).format("hh:mm A");
+
+              const showScrollButtons = batch.documents.length > SCROLL_THRESHOLD;
 
               return (
-                <div key={key} className="grouped-section">
+                <div key={batch.batchId} className="grouped-section">
                   <div className="group-heading">
                     <div className="d-flex flex-column">
                       <div className="d-flex align-items-center mb-1">
                         <Avatar
-                          size="xxs"
-                          firstName={imgs[0].uploadedBy?.firstName}
-                          lastName={imgs[0].uploadedBy?.lastName}
+                          size="xs"
+                          firstName={firstDoc?.uploadedBy?.firstName}
+                          lastName={firstDoc?.uploadedBy?.lastName}
                           className="me-2"
                         />
                         <div className="d-flex flex-column align-items-start">
                           <strong className="user-name-text">
-                            {imgs[0].uploadedBy?.firstName}{" "}
-                            {imgs[0].uploadedBy?.lastName}
+                            {userName}
                           </strong>
                           <span className="me-2">
                             {date} {time}
@@ -373,13 +489,14 @@ const ImageGallery = () => {
 
                     <div className="location-line">
                       <div>
-                        {buildingName} &gt; {floorName} &gt; <strong> {workArea} &gt;{" "}
-                        {activity}</strong>
+                        {batch.buildingName} &gt; {batch.floorName} &gt;{" "}
+                        <strong>{batch.workAreaName || "Unknown"} &gt;{" "}
+                          {batch.activityName}</strong>
                       </div>
-                      {workCategoryName && (
+                      {batch.workCategoryName && (
                         <div className="work-category-display ms-2">
                           <span className="badge bg-label-primary rounded-pill d-flex align-items-center gap-1">
-                            {workCategoryName}
+                            {batch.workCategoryName}
                           </span>
                         </div>
                       )}
@@ -387,34 +504,36 @@ const ImageGallery = () => {
                   </div>
 
                   <div className="image-group-wrapper">
-                    <button
-                      className="scroll-arrow left-arrow"
-                      onClick={() => scrollLeft(key)}
-                    >
-                      &#8249;
-                    </button>
+                    {showScrollButtons && (
+                      <button
+                        className="scroll-arrow left-arrow"
+                        onClick={() => scrollLeft(batch.batchId)}
+                      >
+                        &#8249;
+                      </button>
+                    )}
                     <div
                       className="image-group-horizontal"
-                      ref={(el) => (imageGroupRefs.current[key] = el)}
+                      ref={(el) => (imageGroupRefs.current[batch.batchId] = el)}
                     >
-                      {imgs.map((img, idx) => {
-                        const hoverDate = moment(img.uploadedAt).format("YYYY-MM-DD");
-                        const hoverTime = moment(img.uploadedAt).format("hh:mm A");
+                      {batch.documents.map((doc, idx) => {
+                        const hoverDate = moment(doc.uploadedAt).format("DD-MM-YYYY");
+                        const hoverTime = moment(doc.uploadedAt).format("hh:mm A");
 
                         return (
                           <div
-                            key={img.imageUrl}
+                            key={doc.id}
                             className="image-card"
                             onClick={() =>
-                              openModal(<ImagePop images={imgs} initialIndex={idx} />)
+                              openModal(<ImagePop batch={batch} initialIndex={idx} />)
                             }
-                            onMouseEnter={() => setHoveredImage(img)}
+                            onMouseEnter={() => setHoveredImage(doc)}
                             onMouseLeave={() => setHoveredImage(null)}
                           >
                             <div className="image-wrapper">
-                              <img src={img.imageUrl} alt={`Image ${idx + 1}`} />
+                              <img src={doc.url} alt={`Image ${idx + 1}`} />
                             </div>
-                            {hoveredImage === img && (
+                            {hoveredImage === doc && (
                               <div className="image-hover-description">
                                 <p>
                                   <strong>Date:</strong> {hoverDate}
@@ -423,7 +542,7 @@ const ImageGallery = () => {
                                   <strong>Time:</strong> {hoverTime}
                                 </p>
                                 <p>
-                                  <strong>Activity:</strong> {img.activityName}
+                                  <strong>Activity:</strong> {batch.activityName}
                                 </p>
                               </div>
                             )}
@@ -431,61 +550,53 @@ const ImageGallery = () => {
                         );
                       })}
                     </div>
-                    <button
-                      className="scroll-arrow right-arrow"
-                      onClick={() => scrollRight(key)}
-                    >
-                      &#8250;
-                    </button>
+                    {showScrollButtons && (
+                      <button
+                        className="scroll-arrow right-arrow"
+                        onClick={() => scrollRight(batch.batchId)}
+                      >
+                        &#8250;
+                      </button>
+                    )}
                   </div>
                 </div>
               );
             })
           ) : (
-            <p style={{ textAlign: "center", color: "#777", marginTop: "50px" }}>
+            !loading && <p style={{ textAlign: "center", color: "#777", marginTop: "50px" }}>
               No images match the selected filters.
             </p>
           )}
+
+          <div ref={loaderRef} style={{ height: '50px', margin: '20px 0', display: 'flex', justifyContent: 'center', alignItems: 'center' }}>
+            {loadingMore && hasMore && <div className="spinner" />}
+            {!hasMore && !loading && images.length > 0 && (
+              <p style={{ color: '#aaa' }}>You've reached the end of the images.</p>
+            )}
+          </div>
         </div>
       </div>
 
-      <div className="filter-drawer-wrapper">
-        <button
-          className={`filter-button btn-primary ${isFilterPanelOpen ? "" : "closed-icon"}`}
-          onClick={() => setIsFilterPanelOpen(!isFilterPanelOpen)}
-          ref={filterButtonRef}
-        >
-          {isFilterPanelOpen ? (
-            <>
-              Filter <span className="ms-1">&#10006;</span> {/* Added ms-1 for spacing */}
-            </>
-          ) : (
-            <>
-              <i className="fa-solid fa-filter ms-1 fs-5"></i>
-            </>
-          )}
-        </button>
-        <div className={`filter-panel ${isFilterPanelOpen ? "open" : ""}`} ref={filterPanelRef}>
-          <div className={`dropdown ${collapsedFilters.dateRange ? 'collapsed' : ''}`}>
-            <div className="dropdown-header" onClick={() => toggleCollapse('dateRange')}>
-              <strong>Date Range</strong>
-
-              <span className="collapse-icon">
-                {collapsedFilters.dateRange ? '+' : '-'}
-              </span>
-            </div>
-            {!collapsedFilters.dateRange && (
-              <div className="datepicker" >
-                <DateRangePicker
-                  onRangeChange={setDateRange}
-                  defaultStartDate={selectedFilters.startDate || yesterday}
-                  defaultEndDate={selectedFilters.endDate || moment().format('YYYY-MM-DD')}
-                  startDate={selectedFilters.startDate}
-                  endDate={selectedFilters.endDate}
-                />
-              </div>
-            )}
-          </div>
+      <div
+        className={`offcanvas offcanvas-end ${isFilterPanelOpen ? "show" : ""}`}
+        tabIndex="-1"
+        id="filterOffcanvas"
+        aria-labelledby="filterOffcanvasLabel"
+        ref={filterPanelRef}
+      >
+        <div className="offcanvas-header">
+          <h5 className="offcanvas-title" id="filterOffcanvasLabel">
+            Filters
+          </h5>
+          <button
+            type="button"
+            className="btn-close"
+            onClick={() => setIsFilterPanelOpen(false)}
+            aria-label="Close"
+          ></button>
+        </div>
+        <div className="offcanvas-body d-flex flex-column">
+          {renderFilterCategory("Date Range", [], "dateRange")}
           {renderFilterCategory("Building", buildings, "building")}
           {renderFilterCategory("Floor", floors, "floor")}
           {renderFilterCategory("Work Area", workAreas, "workArea")}
@@ -493,7 +604,7 @@ const ImageGallery = () => {
           {renderFilterCategory("Uploaded By (User)", uploadedByUsers, "uploadedBy")}
           {renderFilterCategory("Work Category", workCategories, "workCategory")}
 
-          <div className="filter-actions">
+          <div className="filter-actions mt-auto">
             <button className="btn btn-secondary btn-xs" onClick={handleClearAllFilters}>
               Clear All
             </button>
diff --git a/src/pages/Gallary/ImageGallery.css b/src/pages/Gallary/ImageGallery.css
index 377fe531..4f7ada69 100644
--- a/src/pages/Gallary/ImageGallery.css
+++ b/src/pages/Gallary/ImageGallery.css
@@ -1,7 +1,5 @@
-/* ImageGallery.css */
 .gallery-container {
   display: grid;
-  grid-template-columns: 1fr 50px;
   gap: 4px;
   padding: 25px;
   font-family: sans-serif;
@@ -69,7 +67,7 @@
   transition: background-color 0.2s ease, box-shadow 0.2s ease, width 0.3s ease-in-out,
     height 0.3s ease-in-out, border-radius 0.3s ease-in-out, padding 0.3s ease-in-out;
   position: absolute;
-  top: 0;
+  top: 250px;
   right: 0;
   height: 40px;
   width: 40px;
@@ -400,6 +398,7 @@
   -webkit-overflow-scrolling: touch;
   scroll-behavior: smooth;
   width: 100%;
+  margin-left: 34px;
 }
 
 .scroll-arrow {
@@ -543,4 +542,4 @@ hr {
 .datepicker {
   margin-right: 135px;
   margin-top: 6px;
-}
\ No newline at end of file
+}
diff --git a/src/pages/Gallary/ImageGalleryAPI.jsx b/src/pages/Gallary/ImageGalleryAPI.jsx
index 13e6a56a..9ca09bfe 100644
--- a/src/pages/Gallary/ImageGalleryAPI.jsx
+++ b/src/pages/Gallary/ImageGalleryAPI.jsx
@@ -1,9 +1,9 @@
 import { api } from "../../utils/axiosClient";
 
 export const ImageGalleryAPI = {
-
-    ImagesGet: (projectId, filter) => {
+    ImagesGet: (projectId, filter, pageNumber, pageSize) => {
         const payloadJsonString = JSON.stringify(filter);
-        return api.get(`/api/image/images/${projectId}?filter=${payloadJsonString}`)
+        // Corrected API endpoint with pagination parameters
+        return api.get(`/api/image/images/${projectId}?filter=${payloadJsonString}&pageNumber=${pageNumber}&pageSize=${pageSize}`);
     },
-}
\ No newline at end of file
+};
\ No newline at end of file
diff --git a/src/pages/Gallary/ImagePop.jsx b/src/pages/Gallary/ImagePop.jsx
index 62f4b8da..ec88e6d4 100644
--- a/src/pages/Gallary/ImagePop.jsx
+++ b/src/pages/Gallary/ImagePop.jsx
@@ -3,32 +3,38 @@ import "./ImagePop.css";
 import { useModal } from "./ModalContext";
 import moment from "moment";
 
-const ImagePop = ({ images, initialIndex = 0 }) => {
+const ImagePop = ({ batch, initialIndex = 0 }) => {
   const { closeModal } = useModal();
-  // State to keep track of the currently displayed image's index
   const [currentIndex, setCurrentIndex] = useState(initialIndex);
 
-  // Effect to update currentIndex if the initialIndex prop changes (e.g., if the modal is reused)
+  // Effect to update currentIndex if the initialIndex prop changes
   useEffect(() => {
     setCurrentIndex(initialIndex);
-  }, [initialIndex, images]);
+  }, [initialIndex, batch]);
 
-  // If no images are provided or the array is empty, don't render
-  if (!images || images.length === 0) return null;
+  // If no batch or documents are provided, don't render
+  if (!batch || !batch.documents || batch.documents.length === 0) return null;
 
-  // Get the current image based on currentIndex
-  const image = images[currentIndex];
+  // Get the current image document from the batch's documents array
+  const image = batch.documents[currentIndex];
 
   // Fallback if for some reason the image at the current index doesn't exist
   if (!image) return null;
 
-  // Format details for display
+  // Format details for display from the individual image document
   const fullName = `${image.uploadedBy?.firstName || ""} ${
     image.uploadedBy?.lastName || ""
   }`.trim();
   const date = moment(image.uploadedAt).format("YYYY-MM-DD");
   const time = moment(image.uploadedAt).format("hh:mm A");
 
+  // Location and category details from the 'batch' object (as previously corrected)
+  const buildingName = batch.buildingName;
+  const floorName = batch.floorName;
+  const workAreaName = batch.workAreaName;
+  const activityName = batch.activityName;  
+  const batchComment = batch.comment;
+
   // Handler for navigating to the previous image
   const handlePrev = () => {
     setCurrentIndex((prevIndex) => Math.max(0, prevIndex - 1));
@@ -37,13 +43,13 @@ const ImagePop = ({ images, initialIndex = 0 }) => {
   // Handler for navigating to the next image
   const handleNext = () => {
     setCurrentIndex((prevIndex) =>
-      Math.min(images.length - 1, prevIndex + 1)
+      Math.min(batch.documents.length - 1, prevIndex + 1)
     );
   };
 
   // Determine if previous/next buttons should be enabled/visible
   const hasPrev = currentIndex > 0;
-  const hasNext = currentIndex < images.length - 1;
+  const hasNext = currentIndex < batch.documents.length - 1;
 
   return (
     <div className="image-modal-overlay">
@@ -61,7 +67,7 @@ const ImagePop = ({ images, initialIndex = 0 }) => {
         )}
 
         {/* The main image display */}
-        <img src={image.imageUrl} alt="Preview" className="modal-image" />
+        <img src={image.url} alt="Preview" className="modal-image" />
 
         {/* Next button, only shown if there's a next image */}
         {hasNext && (
@@ -79,11 +85,12 @@ const ImagePop = ({ images, initialIndex = 0 }) => {
             <strong>📅 Date:</strong> {date} {time}
           </p>
           <p>
-            <strong>🏢 Location:</strong> {image.buildingName} &gt;{" "}
-            {image.floorName} &gt; {image.activityName}
+            <strong>🏢 Location:</strong> {buildingName} &gt; {floorName} &gt;{" "}
+            {workAreaName || "Unknown"} &gt; {activityName}
           </p>
           <p>
-            <strong>📝 Comments:</strong> {image.comment}
+            {/* Display the comment from the batch object */}
+            <strong>📝 Comments:</strong> {batchComment || "N/A"}
           </p>
         </div>
       </div>
diff --git a/src/services/signalRService.js b/src/services/signalRService.js
index 63cabc9c..92c09ec8 100644
--- a/src/services/signalRService.js
+++ b/src/services/signalRService.js
@@ -92,6 +92,13 @@ export function startSignalR(loggedUser) {
         clearCacheKey("AttendanceLogs")
         eventBus.emit("employee", data);
       }
+
+      if (data.keyword == "Task_Report") {
+        eventBus.emit("image_gallery", data);
+      }
+      if (data.keyword == "Task_Comment") {
+        eventBus.emit("image_gallery", data);
+      }
     }
   });