Update app.py

app.py

@@ -7,7 +7,17 @@ from omnibin import generate_binary_classification_report, ColorScheme
 # Define results directory
 RESULTS_DIR = "/tmp/results"
 
-def process_csv(csv_file, n_bootstrap=1000, dpi=72, color_scheme=ColorScheme.DEFAULT):
+# Map string color schemes to enum values
+COLOR_SCHEME_MAP = {
+    "DEFAULT": ColorScheme.DEFAULT,
+    "MONOCHROME": ColorScheme.MONOCHROME,
+    "VIBRANT": ColorScheme.VIBRANT
+}
+
+def process_csv(csv_file, n_bootstrap=1000, dpi=72, color_scheme="DEFAULT"):
+    # Convert string color scheme to enum
+    color_scheme_enum = COLOR_SCHEME_MAP[color_scheme]
+    
     # Read the CSV file
     df = pd.read_csv(csv_file.name)
     
@@ -31,7 +41,7 @@ def process_csv(csv_file, n_bootstrap=1000, dpi=72, color_scheme=ColorScheme.DEF
         n_bootstrap=n_bootstrap,
         random_seed=42,
         dpi=dpi,
-        color_scheme=color_scheme
+        color_scheme=color_scheme_enum
     )
     
     # Get paths to individual plots