fix: address iframe resizing issue in jupyter notebooks

xability · Oct 17, 2024 · b1dd809 · b1dd809
1 parent a437856
commit b1dd809
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/maidr/core/maidr.py b/maidr/core/maidr.py
@@ -193,24 +193,36 @@ def generate_iframe_script(unique_id: str) -> str:
             resizing_script = f"""
                 function resizeIframe() {{
                     let iframe = document.getElementById('{unique_id}');
+
                     if (iframe && iframe.contentWindow && iframe.contentWindow.document) {{
                         let iframeDocument = iframe.contentWindow.document;
-                        let brailleContainer = iframeDocument.getElementById('braille-div');
+                        let brailleContainer = iframeDocument.getElementById('braille-input');
+
+                        iframe.style.height = 'auto';
+
                         let height = iframeDocument.body.scrollHeight;
                         if (brailleContainer && brailleContainer === iframeDocument.activeElement) {{
-                            height *= 1.35;  # Increase height by 35% if braille-container is in focus
+                            height += 100;
+                        }}else{{
+                            height += 50
                         }}
-                        iframe.style.height = (height + 150) + 'px';
+
+                        iframe.style.height = (height) + 'px';
                         iframe.style.width = iframeDocument.body.scrollWidth + 'px';
                     }}
                 }}
                 let iframe = document.getElementById('{unique_id}');
+                resizeIframe();
                 iframe.onload = function() {{
                     resizeIframe();
                     iframe.contentWindow.addEventListener('resize', resizeIframe);
-                    iframe.contentWindow.document.addEventListener('focusin', resizeIframe);
-                    iframe.contentWindow.document.addEventListener('focusout', resizeIframe);
                 }};
+                iframe.contentWindow.document.addEventListener('focusin', () => {{
+                    resizeIframe();
+                }});
+                iframe.contentWindow.document.addEventListener('focusout', () => {{
+                    resizeIframe();
+                }});
             """
             return resizing_script