Upload 3 files

main.py

@@ -10,7 +10,7 @@ import nibabel as nib
 from data_preprocessing import preprocess_fmri_to_fc
 from src.demovae.sklearn import DemoVAE
 from analysis import analyze_fc_patterns
-from visualization import visualize_fc_analysis
+from visualization import plot_fc_matrices
 from config import MODEL_CONFIG, DATASET_CONFIG
 import pandas as pd
 import io
@@ -242,7 +242,7 @@ def run_fc_analysis(data_dir="SreekarB/OSFData",
         
         # Visualize results
         print("Creating visualizations...")
-        fig = visualize_fc_analysis(X[0], reconstructed_fc[0], generated_fc[0], analysis_results)
+        fig = plot_fc_matrices(X[0], reconstructed_fc[0], generated_fc[0])
         
         # If requested, return additional data for accuracy calculations
         if return_data: