Update app.py

app.py

@@ -186,5 +186,39 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             outputs=[sql_query_out, error_out, results_out]
         )
 
+    # Add a Gradio File output component for the download functionality
+    download_csv_btn = gr.File(label="📥 Download CSV", visible=False)
+    
+    # Function to save the results to a CSV and return the file path
+    def save_to_csv(results_df):
+        if results_df is None or results_df.empty:
+            return None, "⚠️ No results to download."
+        try:
+            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+            results_df.to_csv(temp_file.name, index=False)
+            return temp_file.name, ""
+        except Exception as e:
+            return None, f"❌ Error generating CSV: {e}"
+    
+    # Add functionality to generate and show the download link for the CSV
+    def generate_download(results_df):
+        file_path, error = save_to_csv(results_df)
+        if error:
+            return None, f"❌ {error}"
+        return file_path, ""
+    
+    # Update the Gradio event handlers
+    btn_execute_query.click(
+        fn=execute_query,
+        inputs=sql_query_out,
+        outputs=[results_out, error_out]
+    )
+    
+    btn_execute_query.click(
+        fn=generate_download,
+        inputs=results_out,
+        outputs=[download_csv_btn, error_out]
+    )
+
 # Launch the Gradio App
 demo.queue().launch()