Update app.py

app.py

@@ -28,11 +28,15 @@ def analyze_file(uploaded_file):
 
         # Check if required columns are present
         if 'Gain (dB)' in df.columns and 'Frequency (GHz)' in df.columns:
+            # Handle NaN values by replacing them with the mean of the column
+            df['Gain (dB)'].fillna(df['Gain (dB)'].mean(), inplace=True)  # Replace NaN with the mean of the column
+            df['Frequency (GHz)'].fillna(df['Frequency (GHz)'].mean(), inplace=True)  # Replace NaN with the mean of the column
+            
             # Convert pandas columns to numpy arrays before performing operations
             gain_values = np.array(df['Gain (dB)'])
             freq_values = np.array(df['Frequency (GHz)'])
 
-            # Handle infinite and NaN values by replacing them with the mean of the column or 0
+            # Handle infinite values by replacing them with NaN and then replacing NaNs with 0
             gain_values[np.isinf(gain_values)] = np.nan
             freq_values[np.isinf(freq_values)] = np.nan