Update app.py

app.py

@@ -18,62 +18,42 @@ body {
 """
 st.markdown(custom_css, unsafe_allow_html=True)
 
-# Prescribed limits for AQI, PM2.5, and PM10
+# Prescribed limits for AQI and PM2.5
 PRESCRIBED_LIMITS = {
     'AQI': {
         'good': (0, 50),
         'moderate': (51, 100),
-        'poor': (101, 200),
-        'very_poor': (201, 300),
-        'severe': (301, 500)
+        'unhealthy_sensitive': (101, 150),
+        'unhealthy': (151, 200),
+        'very_unhealthy': (201, 300),
+        'hazardous': (301, 500)
     },
     'PM2.5': {
-        'safe': (0, 30),
-        'moderate': (31, 60),
-        'poor': (61, 90),
-        'very_poor': (91, 120),
-        'severe': (121, 250)
-    },
-    'PM10': {
-        'safe': (0, 50),
-        'moderate': (51, 100),
-        'poor': (101, 250),
-        'very_poor': (251, 350),
-        'severe': (351, 430)
+        'good': (0, 12.0),
+        'moderate': (12.1, 35.4),
+        'unhealthy_sensitive': (35.5, 55.5),
+        'unhealthy': (55.6, 150.4),
+        'very_unhealthy': (150.5, 250.4),
+        'hazardous': (250.5, float('inf'))
     }
 }
 
+# Color codes for different AQI and PM2.5 levels
+COLOR_CODES = {
+    'good': 'green',
+    'moderate': 'yellow',
+    'unhealthy_sensitive': 'orange',
+    'unhealthy': 'red',
+    'very_unhealthy': 'purple',
+    'hazardous': 'maroon'
+}
+
 def check_safety_level(value, parameter):
     for level, limits in PRESCRIBED_LIMITS[parameter].items():
         if limits[0] <= value <= limits[1]:
             return level
     return None
 
-def get_aqi_data():
-    url = 'https://www.aqi.in/dashboard/india/jammu-and-kashmir/srinagar'
-    r = requests.get(url)
-    soup = BeautifulSoup(r.text, 'html.parser')
-    meta_tag = soup.find('meta', {'name': 'description'})
-    line = meta_tag['content']
-
-    # Extract AQI value
-    aqi_start = line.find("AQI) is") + len("AQI) is")
-    aqi_end = line.find(" level")
-    aqi = int(line[aqi_start:aqi_end].strip().split()[0])
-    level = line[aqi_start:aqi_end].strip().split()[1]
-
-    # Extract PM2.5 value
-    pm25_start = line.find("PM2.5 (") + len("PM2.5 (")
-    pm25_end = line.find(" µg/m³)", pm25_start)
-    pm25 = int(line[pm25_start:pm25_end].strip())
-
-    # Extract PM10 value
-    pm10_start = line.find("PM10 (") + len("PM10 (")
-    pm10_end = line.find(" µg/m³)", pm10_start)
-    pm10 = int(line[pm10_start:pm10_end].strip())
-
-    return aqi, level, pm25, pm10
-
 def main():
     # Logo
     logo = 'LOGO.png'
@@ -90,30 +70,19 @@ def main():
     aqi_safety = check_safety_level(aqi, 'AQI')
     if aqi_safety:
         st.write(f"Safety Level: {aqi_safety}")
-        if aqi_safety == 'good':
-            st.markdown('<p style="color:green;">The AQI level is safe.</p>', unsafe_allow_html=True)
-        else:
-            st.markdown('<p style="color:red;">The AQI level is unsafe.</p>', unsafe_allow_html=True)
+        color_code = COLOR_CODES[aqi_safety]
+        st.markdown(f'<p style="color:{color_code};">The AQI level is {aqi_safety}.</p>', unsafe_allow_html=True)
 
     # Display PM2.5 with safety level
     st.write(f"- **PM2.5:** {pm25} µg/m³")
     pm25_safety = check_safety_level(pm25, 'PM2.5')
     if pm25_safety:
         st.write(f"Safety Level: {pm25_safety}")
-        if pm25_safety == 'safe':
-            st.markdown('<p style="color:green;">The PM2.5 level is safe.</p>', unsafe_allow_html=True)
-        else:
-            st.markdown('<p style="color:red;">The PM2.5 level is unsafe.</p>', unsafe_allow_html=True)
+        color_code = COLOR_CODES[pm25_safety]
+        st.markdown(f'<p style="color:{color_code};">The PM2.5 level is {pm25_safety}.</p>', unsafe_allow_html=True)
 
     # Display PM10 with safety level
     st.write(f"- **PM10:** {pm10} µg/m³")
-    pm10_safety = check_safety_level(pm10, 'PM10')
-    if pm10_safety:
-        st.write(f"Safety Level: {pm10_safety}")
-        if pm10_safety == 'safe':
-            st.markdown('<p style="color:green;">The PM10 level is safe.</p>', unsafe_allow_html=True)
-        else:
-            st.markdown('<p style="color:red;">The PM10 level is unsafe.</p>', unsafe_allow_html=True)
 
     # Developer section
     st.markdown('### Developer:')