Update templates/output.html

templates/output.html

@@ -121,8 +121,8 @@
     <div class="container">
         <h1>Generated Image</h1>
         <div class="image-container">
-            <a href="{{ url_for('static', filename='example.jpeg') }}" target="_blank" onclick="toggleFullScreen()">
-                <img src="{{ url_for('static', filename='example.jpeg') }}" alt="Generated Image">
+            <a href="{{ url_for('static', filename='example.jpeg') }}" target="_blank">
+                <img src="{{ url_for('static', filename='example.jpeg') }}" alt="Generated Image" onclick="toggleFullScreen()">
             </a>
         </div>
         <div class="button-container">
@@ -157,8 +157,8 @@
         }
 
         function toggleFullScreen() {
-            var imageContainer = document.querySelector('.image-container');
-            imageContainer.classList.toggle('fullscreen-image');
+            var bodyElement = document.querySelector('body');
+            bodyElement.classList.toggle('fullscreen-image');
         }
     </script>
     <footer>
@@ -166,4 +166,5 @@
     </footer>
     <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js"></script>
 </body>
-</html>
+  </html>
+