Update app.py

app.py

@@ -311,6 +311,23 @@ def create_vibecheck_results_ui():
             outputs=[load_examples_md],
         )
 
+        # Function to convert DataFrame to CSV and return for download
+        def download_dataframe(results_dict):
+            if not results_dict or "df" not in results_dict:
+                return None
+            
+            # Convert DataFrame to CSV
+            csv_file_path = "vibecheck_results.csv"
+            results_dict["df"].to_csv(csv_file_path, index=False)
+            return csv_file_path
+
+        # Connect the download button to the function
+        download_btn.click(
+            fn=download_dataframe,
+            inputs=[load_results_state],
+            outputs=[download_file]
+        )
+
         # Add a footer
         gr.HTML("""
         <div class="footer">