Update app.py

app.py

@@ -38,6 +38,69 @@ from huggingface_hub import InferenceApi, login, InferenceClient
 import branca.colormap as cm
 from functools import lru_cache
 import time
+import math
+
+# Weather API configuration
+WEATHER_API_KEY = os.getenv("OPENWEATHER_API_KEY")
+WEATHER_API_URL = "https://api.openweathermap.org/data/2.5/weather"
+
+@st.cache_data(ttl=1800)  # Cache for 30 minutes
+def fetch_weather_data(lat, lon):
+    """Fetch weather data for a specific location"""
+    try:
+        params = {
+            'lat': lat,
+            'lon': lon,
+            'appid': WEATHER_API_KEY,
+            'units': 'metric'
+        }
+        response = requests.get(WEATHER_API_URL, params=params, timeout=10)
+        response.raise_for_status()
+        return response.json()
+    except Exception as e:
+        st.warning(f"Could not fetch weather data: {str(e)}")
+        return None
+
+def create_weather_heatmap(weather_data, map_obj):
+    """Create a heatmap layer for weather conditions"""
+    if not weather_data:
+        return
+    
+    # Extract weather data points
+    heat_data = []
+    for data in weather_data:
+        if data and 'main' in data and 'coord' in data:
+            lat = data['coord']['lat']
+            lon = data['coord']['lon']
+            temp = data['main']['temp']
+            humidity = data['main']['humidity']
+            # Use temperature as the weight for the heatmap
+            heat_data.append([lat, lon, temp])
+    
+    if heat_data:
+        # Create heatmap layer
+        plugins.HeatMap(
+            heat_data,
+            min_opacity=0.5,
+            max_val=max([point[2] for point in heat_data]),
+            radius=25,
+            blur=15,
+            gradient={0.4: 'blue', 0.6: 'lime', 0.7: 'yellow', 1: 'red'}
+        ).add_to(map_obj)
+
+def get_weather_grid(lat_min, lat_max, lon_min, lon_max, grid_size=5):
+    """Create a grid of points for weather data collection"""
+    lat_step = (lat_max - lat_min) / grid_size
+    lon_step = (lon_max - lon_min) / grid_size
+    
+    weather_points = []
+    for i in range(grid_size + 1):
+        for j in range(grid_size + 1):
+            lat = lat_min + (i * lat_step)
+            lon = lon_min + (j * lon_step)
+            weather_points.append((lat, lon))
+    
+    return weather_points
 
 # Cache the airport data to avoid reloading it every time
 @st.cache_data(ttl=3600)  # Cache for 1 hour
@@ -380,6 +443,39 @@ def flight_tracking(flight_view_level, country, local_time_zone, flight_info, ai
         tiles='CartoDB dark_matter'
     )
     
+    # Add weather heatmap if API key is available
+    if WEATHER_API_KEY:
+        with st.spinner('Fetching weather data...'):
+            # Get weather grid points
+            weather_points = get_weather_grid(lat_min, lat_max, lon_min, lon_max)
+            
+            # Fetch weather data for all points
+            weather_data = []
+            for lat, lon in weather_points:
+                data = fetch_weather_data(lat, lon)
+                if data:
+                    weather_data.append(data)
+            
+            # Create weather heatmap
+            create_weather_heatmap(weather_data, m)
+            
+            # Add weather legend
+            legend_html = '''
+                <div style="position: fixed; 
+                    bottom: 50px; right: 50px; width: 150px; height: 100px;
+                    background-color: white; padding: 10px; border-radius: 5px;
+                    border: 2px solid grey; z-index: 9999;">
+                    <h4 style="margin: 0 0 5px 0;">Temperature (°C)</h4>
+                    <div style="background: linear-gradient(to right, blue, lime, yellow, red);
+                        height: 20px; width: 100%; margin-bottom: 5px;"></div>
+                    <div style="display: flex; justify-content: space-between;">
+                        <span>Cold</span>
+                        <span>Hot</span>
+                    </div>
+                </div>
+            '''
+            m.get_root().html.add_child(folium.Element(legend_html))
+    
     # Create colormap
     if color == "rainbow":
         colormap = cm.LinearColormap(
@@ -541,6 +637,15 @@ with st.sidebar:
         st.write('Now Airports are Visible')
     else:
         air_port=0
+    
+    # Add weather toggle
+    weather_on = st.toggle('Show Weather Heatmap')
+    if weather_on:
+        if not WEATHER_API_KEY:
+            st.warning("Please set OPENWEATHER_API_KEY environment variable to show weather data")
+        else:
+            st.write(':rainbow[Weather data will be displayed as a heatmap]')
+    
     view = st.slider('Increase Flight Visibility',1,6,2)
     st.write("You Selected:", view)
     cou = st.text_input('Type Country Name', 'north america')
@@ -561,6 +666,7 @@ with st.sidebar:
     elif clr == 'hot':
         st.write('The current color is', "****:red[Hot]****")
     else: None
+
 # with st.spinner('Wait!, We Requesting API Data...'):
 # try:
 flight_tracking(flight_view_level=view, country=cou,flight_info=info,