Update app.py

app.py

@@ -19,17 +19,21 @@ def calculate_vif(df):
 # Function to load and process data (including VIF and PCA)
 def process_data(file, scaler_option):
     df = pd.read_csv(file)
-    features = ['RI4', 'RI5', 'RI7', 'RI9']
-    df_selected = df[features].fillna(df.mean())
+    
+    # Select only numeric columns for VIF calculation
+    df_numeric = df.select_dtypes(include=[np.number])
+    
+    # Handle missing values by filling them with the mean of the respective columns
+    df_numeric = df_numeric.fillna(df_numeric.mean())
 
     # Calculate VIF and filter features with VIF < 10
-    selected_features = calculate_vif(df_selected)
-    
+    selected_features = calculate_vif(df_numeric)
+
     if not selected_features:
         st.error("No features with VIF < 10 found. Please review the data.")
-        return None
+        return None, None
 
-    df_filtered = df[selected_features]
+    df_filtered = df_numeric[selected_features]
 
     # Apply chosen scaler
     if scaler_option == 'StandardScaler':