Update app.py

app.py

@@ -4,6 +4,19 @@ import folium
 from datetime import datetime, timedelta
 import tempfile
 import os
+from PIL import Image
+import io
+
+def download_image(url):
+    """Download image from URL and return as PIL Image."""
+    try:
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        image = Image.open(io.BytesIO(response.content))
+        return image
+    except Exception as e:
+        print(f"Error downloading image from {url}: {str(e)}")
+        return None
 
 def get_noaa_forecast(lat, lon):
     """Get NOAA forecast data for given coordinates."""
@@ -36,17 +49,20 @@ def get_noaa_forecast(lat, lon):
 def get_radar_images():
     """Get current radar and forecast radar images."""
     try:
-        # Current radar
-        current_radar = "https://radar.weather.gov/ridge/standard/CONUS_0.gif"
+        # Current radar - National composite
+        current_radar_url = "https://radar.weather.gov/ridge/standard/CONUS_0.gif"
+        current_radar = download_image(current_radar_url)
         
-        # Forecast radar - using NDFD precipitation forecast images
-        # These are probability of precipitation forecasts for different time periods
+        # Get forecast time for NDFD products
         now = datetime.utcnow()
         forecast_date = now.strftime("%Y%m%d")
         
-        # 6-hour and 12-hour precipitation probability forecasts
-        forecast_6hr = f"https://digital.weather.gov/images/conus/MaxPcpn6_{forecast_date}00.png"
-        forecast_12hr = f"https://digital.weather.gov/images/conus/MaxPcpn12_{forecast_date}00.png"
+        # Alternative URLs that might be more reliable
+        forecast_6hr_url = "https://graphical.weather.gov/images/conus/QPF06_conus.png"
+        forecast_12hr_url = "https://graphical.weather.gov/images/conus/QPF12_conus.png"
+        
+        forecast_6hr = download_image(forecast_6hr_url)
+        forecast_12hr = download_image(forecast_12hr_url)
         
         return {
             "current": current_radar,
@@ -54,6 +70,7 @@ def get_radar_images():
             "forecast_12hr": forecast_12hr
         }
     except Exception as e:
+        print(f"Error getting radar images: {str(e)}")
         return None
 
 def create_map():
@@ -130,19 +147,22 @@ with gr.Blocks(title="NOAA Weather Forecast and Radar") as demo:
             current_radar = gr.Image(
                 label="Current Radar",
                 show_label=True,
-                container=True
+                container=True,
+                type="pil"
             )
         with gr.Column():
             forecast_6hr = gr.Image(
                 label="6-Hour Precipitation Forecast",
                 show_label=True,
-                container=True
+                container=True,
+                type="pil"
             )
         with gr.Column():
             forecast_12hr = gr.Image(
                 label="12-Hour Precipitation Forecast",
                 show_label=True,
-                container=True
+                container=True,
+                type="pil"
             )
     
     # Handle submit button click
@@ -166,8 +186,8 @@ with gr.Blocks(title="NOAA Weather Forecast and Radar") as demo:
     
     **Legend:**
     - Current Radar: Shows current precipitation
-    - 6-Hour Forecast: Shows probability of precipitation over next 6 hours
-    - 12-Hour Forecast: Shows probability of precipitation over next 12 hours
+    - 6-Hour Forecast: Shows quantitative precipitation forecast for next 6 hours
+    - 12-Hour Forecast: Shows quantitative precipitation forecast for next 12 hours
     
     **Note**: This app uses the NOAA Weather API and may have occasional delays or service interruptions.
     Data is provided by the National Weather Service (weather.gov).