Update app.py

app.py

@@ -4,6 +4,8 @@ import openai
 import duckdb
 import gradio as gr
 from functools import lru_cache
+from dotenv import load_dotenv
+from e2b_code_interpreter import Sandbox
 
 # =========================
 # Configuration and Setup
@@ -14,11 +16,8 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 if not openai.api_key:
     raise ValueError("Please set the OPENAI_API_KEY environment variable.")
 
-from dotenv import load_dotenv
 load_dotenv()
-from e2b_code_interpreter import Sandbox
-
-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
@@ -175,7 +174,6 @@ with gr.Blocks(css="""
             ]
             btn_queries = [gr.Button(q, variant="secondary", size="sm") for q in query_buttons]
 
-            # Removed gr.Box and used gr.Group as an alternative
             with gr.Group():
                 query_input = gr.Textbox(
                     label="🔍 Your Query",
@@ -194,6 +192,32 @@ with gr.Blocks(css="""
         with gr.Column(scale=2, min_width=600):
             gr.Markdown("### 📊 Query Results", elem_classes="results")
             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
+            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();
+                            alert("Copied results to clipboard!");
+                        } else {
+                            alert("No results to copy!");
+                        }
+                    }
+                </script>
+            """)
+
+            # Connect JavaScript function to button
+            btn_copy_results.click(
+                None, None, None, _js="copyToClipboard"
+            )
 
     with gr.Tab("📋 Dataset Schema", elem_classes="schema-tab"):
         gr.Markdown("### Dataset Schema")
@@ -204,15 +228,6 @@ with gr.Blocks(css="""
     # =========================
 
     def generate_sql(nl_query):
-        """
-        Generates SQL from a natural language query.
-
-        Args:
-            nl_query (str): The natural language query.
-
-        Returns:
-            tuple: The SQL query, error message (if any), and visibility update.
-        """
         if not nl_query.strip():
             return "", "<p class='error-message'>Please enter a query.</p>", gr.update(visible=True)
         sql_query, error = parse_query(nl_query)
@@ -222,15 +237,6 @@ with gr.Blocks(css="""
             return sql_query, "", gr.update(visible=False)
 
     def execute_query(sql_query):
-        """
-        Executes the generated SQL query.
-
-        Args:
-            sql_query (str): The SQL query to execute.
-
-        Returns:
-            tuple: The result dataframe, error message (if any), and visibility update.
-        """
         if not sql_query.strip():
             return None, "<p class='error-message'>No SQL query to execute.</p>", gr.update(visible=True)
         result_df, error = execute_sql_query(sql_query)
@@ -240,15 +246,6 @@ with gr.Blocks(css="""
             return result_df, "", gr.update(visible=False)
 
     def handle_example_click(example_query):
-        """
-        Handles clicks on example query buttons.
-
-        Args:
-            example_query (str): The example query text.
-
-        Returns:
-            tuple: The SQL query, error message (if any), result dataframe, and visibility update.
-        """
         if example_query.strip().upper().startswith("SELECT"):
             sql_query = example_query
             result_df, error = execute_sql_query(sql_query)
@@ -264,12 +261,9 @@ with gr.Blocks(css="""
             if exec_error:
                 return sql_query, f"<p class='error-message'>{exec_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)
 
-    # =========================
     # Button Click Event Handlers
-    # =========================
-
     btn_generate_sql.click(
         fn=generate_sql,
         inputs=query_input,
@@ -289,7 +283,6 @@ with gr.Blocks(css="""
             outputs=[sql_query_out, error_out, results_out, error_out],
         )
 
-    # Clear error messages when inputs change
     query_input.change(fn=lambda: ("", gr.update(visible=False)), inputs=None, outputs=[error_out])
     sql_query_out.change(fn=lambda: ("", gr.update(visible=False)), inputs=None, outputs=[error_out])
 
@@ -302,5 +295,5 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=False,
-        debug=True,  # Enable debug mode for detailed logs
+        debug=True,
     )