ui bug fix

static/script.js

@@ -778,60 +778,60 @@ function updateHistoryCount() {
         });
         countElement.textContent = totalImages;
     }
+}
+
+// Modal functionality for viewing large images
+let currentModalImage = null;
+
+function openImageModal(imageId, imageSrc, prompt, timestamp) {
+    const modal = document.getElementById('imageModal');
+    const modalImg = document.getElementById('modalImage');
+    const modalCaption = document.getElementById('modalCaption');
+    
+    // Store current image info for use in "Use as Input"
+    currentModalImage = { id: imageId, src: imageSrc };
+    
+    // Set modal content
+    modalImg.src = imageSrc;
+    modalCaption.innerHTML = `
+        <strong>Generated:</strong> ${timestamp}<br>
+        <strong>Prompt:</strong> ${prompt}
+    `;
     
-    // Modal functionality for viewing large images
-    let currentModalImage = null;
+    // Show modal
+    modal.classList.add('show');
     
-    function openImageModal(imageId, imageSrc, prompt, timestamp) {
-        const modal = document.getElementById('imageModal');
-        const modalImg = document.getElementById('modalImage');
-        const modalCaption = document.getElementById('modalCaption');
-        
-        // Store current image info for use in "Use as Input"
-        currentModalImage = { id: imageId, src: imageSrc };
-        
-        // Set modal content
-        modalImg.src = imageSrc;
-        modalCaption.innerHTML = `
-            <strong>Generated:</strong> ${timestamp}<br>
-            <strong>Prompt:</strong> ${prompt}
-        `;
-        
-        // Show modal
-        modal.classList.add('show');
-        
-        // Prevent body scroll when modal is open
-        document.body.style.overflow = 'hidden';
-        
-        // Close modal on escape key
-        document.addEventListener('keydown', handleModalEscape);
-        
-        // Close modal on clicking outside the image
-        modal.onclick = function(event) {
-            if (event.target === modal || event.target === modalImg.parentElement) {
-                closeImageModal();
-            }
-        };
-    }
+    // Prevent body scroll when modal is open
+    document.body.style.overflow = 'hidden';
     
-    function closeImageModal() {
-        const modal = document.getElementById('imageModal');
-        modal.classList.remove('show');
-        document.body.style.overflow = '';
-        document.removeEventListener('keydown', handleModalEscape);
-        currentModalImage = null;
-    }
+    // Close modal on escape key
+    document.addEventListener('keydown', handleModalEscape);
     
-    function handleModalEscape(event) {
-        if (event.key === 'Escape') {
+    // Close modal on clicking outside the image
+    modal.onclick = function(event) {
+        if (event.target === modal || event.target === modalImg.parentElement) {
             closeImageModal();
         }
+    };
+}
+
+function closeImageModal() {
+    const modal = document.getElementById('imageModal');
+    modal.classList.remove('show');
+    document.body.style.overflow = '';
+    document.removeEventListener('keydown', handleModalEscape);
+    currentModalImage = null;
+}
+
+function handleModalEscape(event) {
+    if (event.key === 'Escape') {
+        closeImageModal();
     }
-    
-    function useModalImageAsInput() {
-        if (currentModalImage) {
-            useAsInput(currentModalImage.id, currentModalImage.src);
-            closeImageModal();
-        }
+}
+
+function useModalImageAsInput() {
+    if (currentModalImage) {
+        useAsInput(currentModalImage.id, currentModalImage.src);
+        closeImageModal();
     }
 }