Update app.py

app.py

@@ -17,7 +17,7 @@ if not openai.api_key:
     raise ValueError("Please set the OPENAI_API_KEY environment variable.")
 
 load_dotenv()
-sbx = Sandbox()  # By default, the sandbox is alive for 5 minutes
+sbx = Sandbox() # By default, the sandbox is alive for 5 minutes
 
 # Path to your Parquet dataset
 DATASET_PATH = 'hsas.parquet'  # Update with your Parquet file path
@@ -194,18 +194,22 @@ with gr.Blocks(css="""
             results_out = gr.Dataframe(label="Query Results", interactive=False)
             btn_copy_results = gr.Button("Copy to Clipboard", variant="secondary", size="sm")
 
-            # JavaScript for copying to clipboard
+            # JavaScript for copying to clipboard - updated approach
             copy_script = gr.HTML("""
                 <script>
                     function copyToClipboard() {
-                        const resultsTable = document.querySelector(".dataframe-output table");
-                        if (resultsTable) {
-                            const range = document.createRange();
-                            range.selectNode(resultsTable);
-                            window.getSelection().removeAllRanges();
-                            window.getSelection().addRange(range);
-                            document.execCommand("copy");
-                            window.getSelection().removeAllRanges();
+                        const resultsContainer = document.querySelector('.dataframe-output table');
+                        if (resultsContainer) {
+                            const text = Array.from(resultsContainer.rows)
+                                .map(row => Array.from(row.cells)
+                                    .map(cell => cell.innerText).join("\t"))
+                                .join("\n");
+                            const tempTextArea = document.createElement('textarea');
+                            tempTextArea.value = text;
+                            document.body.appendChild(tempTextArea);
+                            tempTextArea.select();
+                            document.execCommand('copy');
+                            document.body.removeChild(tempTextArea);
                             alert("Copied results to clipboard!");
                         } else {
                             alert("No results to copy!");
@@ -216,7 +220,7 @@ with gr.Blocks(css="""
 
             # Connect JavaScript function to button
             btn_copy_results.click(
-                None, None, None, js="copyToClipboard"
+                None, None, None, _js="copyToClipboard"
             )
 
     with gr.Tab("📋 Dataset Schema", elem_classes="schema-tab"):