Update app.py

update document.querySelectorAll

app.py

@@ -47,31 +47,32 @@ with col2:
 
             # Inject JavaScript untuk notifikasi klik tombol copy
             st.markdown("""
-                <script>
+    <script>
+        setTimeout(() => {
+            const copyBtns = document.querySelectorAll('button[data-testid="stCopyButton"]');
+            copyBtns.forEach(btn => {
+                btn.addEventListener('click', () => {
+                    const notice = document.createElement("div");
+                    notice.innerText = "✅ Prompt copied to clipboard!";
+                    notice.style.position = "fixed";
+                    notice.style.bottom = "30px";
+                    notice.style.right = "30px";
+                    notice.style.backgroundColor = "#4CAF50";
+                    notice.style.color = "white";
+                    notice.style.padding = "10px 16px";
+                    notice.style.borderRadius = "8px";
+                    notice.style.boxShadow = "0 4px 6px rgba(0,0,0,0.2)";
+                    notice.style.zIndex = "9999";
+                    notice.style.fontSize = "14px";
+                    document.body.appendChild(notice);
                     setTimeout(() => {
-                        const copyBtns = window.parent.document.querySelectorAll('[data-testid="stCopyButton"]');
-                        copyBtns.forEach(btn => {
-                            btn.addEventListener('click', () => {
-                                const notice = document.createElement("div");
-                                notice.innerText = "✅ Prompt copied to clipboard!";
-                                notice.style.position = "fixed";
-                                notice.style.bottom = "30px";
-                                notice.style.right = "30px";
-                                notice.style.backgroundColor = "#4CAF50";
-                                notice.style.color = "white";
-                                notice.style.padding = "10px 16px";
-                                notice.style.borderRadius = "8px";
-                                notice.style.boxShadow = "0 4px 6px rgba(0,0,0,0.2)";
-                                notice.style.zIndex = "9999";
-                                notice.style.fontSize = "14px";
-                                document.body.appendChild(notice);
-                                setTimeout(() => {
-                                    notice.remove();
-                                }, 2000);
-                            });
-                        });
-                    }, 1000);
-                </script>
-            """, unsafe_allow_html=True)
+                        notice.remove();
+                    }, 2000);
+                });
+            });
+        }, 1000);
+    </script>
+""", unsafe_allow_html=True)
+
     else:
         st.info("Please upload an image to generate a prompt.")