nakas commited on
Commit
55126fb
·
verified ·
1 Parent(s): 6268d66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -400,7 +400,15 @@ class AccurateAirQualityMapper:
400
  })
401
 
402
  df = pd.DataFrame(table_data)
403
- return df.sort_values('AQI', ascending=False)
 
 
 
 
 
 
 
 
404
 
405
  # Initialize mapper
406
  mapper = AccurateAirQualityMapper()
 
400
  })
401
 
402
  df = pd.DataFrame(table_data)
403
+
404
+ # Convert AQI column to numeric for proper sorting, keeping 'N/A' as 0
405
+ df['AQI_numeric'] = pd.to_numeric(df['AQI'], errors='coerce').fillna(0)
406
+
407
+ # Sort by AQI (air quality first, then meteorological)
408
+ df_sorted = df.sort_values(['AQI_numeric', 'Parameter'], ascending=[False, True])
409
+
410
+ # Drop the helper column
411
+ return df_sorted.drop('AQI_numeric', axis=1)
412
 
413
  # Initialize mapper
414
  mapper = AccurateAirQualityMapper()