update

pages/2_Evaluation.py

@@ -37,12 +37,24 @@ def app():
                 divergence_results = calculate_divergences(df)
                 st.write('Divergence Results:', divergence_results)
 
-                # Allow downloading of the evaluation results
-                results_combined = {**test_results, **correlation_results, **divergence_results}
-                results_df = pd.DataFrame.from_dict(results_combined, orient='index', columns=['Value'])
+                # Flatten the results for combining
+                flat_test_results = {f"{key1}_{key2}": value2 for key1, value1 in test_results.items() for key2, value2
+                                     in (value1.items() if isinstance(value1, dict) else {key1: value1}.items())}
+                flat_correlation_results = {f"Correlation_{key1}": value1 for key1, value1 in
+                                            correlation_results.items()}
+                flat_divergence_results = {f"Divergence_{key1}": value1 for key1, value1 in divergence_results.items()}
+
+                # Combine all results
+                results_combined = {**flat_test_results, **flat_correlation_results, **flat_divergence_results}
+
+                # Convert to DataFrame for download
+                results_df = pd.DataFrame(list(results_combined.items()), columns=['Metric', 'Value'])
+
+                st.write('Combined Results:', results_df)
+
                 st.download_button(
                     label="Download Evaluation Results",
-                    data=results_df.to_csv().encode('utf-8'),
+                    data=results_df.to_csv(index=False).encode('utf-8'),
                     file_name='evaluation_results.csv',
                     mime='text/csv',
                 )