Update app.py

app.py

@@ -230,13 +230,19 @@ with gr.Blocks(css="""
                             alert("No results to copy!");
                         }
                     }
+
+                    // Attach the copy function to the button
+                    document.addEventListener('DOMContentLoaded', function() {
+                        const copyButton = document.querySelector('.copy-button button');
+                        if (copyButton) {
+                            copyButton.addEventListener('click', copyToClipboard);
+                        }
+                    });
                 </script>
             """)
 
-            # Connect JavaScript function to button
-            btn_copy_results.click(
-                _js="copyToClipboard"
-            )
+            # Include the JavaScript in the app
+            copy_script
 
     # Dataset Schema Tab
     with gr.Tab("📋 Dataset Schema", elem_classes="schema-tab"):
@@ -272,7 +278,7 @@ with gr.Blocks(css="""
             if error:
                 return sql_query, f"<p>{error}</p>", None, gr.update(visible=True)
             else:
-                return sql_query, "", result_df, gr.update(visible(False))
+                return sql_query, "", result_df, gr.update(visible=False)
         else:
             sql_query, error = parse_query(example_query)
             if error: