Update app.py

app.py

@@ -7,6 +7,13 @@ import os
 from PIL import Image
 import io
 
+# Montana Mountain Peaks coordinates
+MONTANA_PEAKS = {
+    "Lone Peak (Big Sky)": (45.27806, -111.45028),  # 45°16′41″N 111°27′01″W
+    "Sacajawea Peak": (45.89583, -110.96861),  # 45°53′45″N 110°58′7″W
+    "Pioneer Mountain": (45.231835, -111.450505)  # 45°13′55″N 111°27′2″W
+}
+
 def download_image(url):
     """Download image from URL and return as PIL Image."""
     try:
@@ -73,9 +80,19 @@ def get_radar_images():
         print(f"Error getting radar images: {str(e)}")
         return None
 
-def create_map():
-    """Create a folium map centered on the US."""
-    m = folium.Map(location=[39.8283, -98.5795], zoom_start=4)
+def create_map(markers=None):
+    """Create a folium map centered on Montana with optional markers."""
+    # Center on Montana
+    m = folium.Map(location=[45.5, -111.0], zoom_start=7)
+    
+    # Add markers for Montana peaks
+    for peak_name, coords in MONTANA_PEAKS.items():
+        folium.Marker(
+            coords,
+            popup=f"{peak_name}<br>Lat: {coords[0]:.4f}, Lon: {coords[1]:.4f}",
+            tooltip=peak_name
+        ).add_to(m)
+    
     m.add_child(folium.ClickForLatLng())  # Enable click events
     return m._repr_html_()
 
@@ -96,8 +113,20 @@ def update_weather(lat, lon):
             return forecast, None, None, None, create_map()
         
         # Create updated map with marker
-        m = folium.Map(location=[lat, lon], zoom_start=8)
-        folium.Marker([lat, lon], popup=f"Lat: {lat:.4f}, Lon: {lon:.4f}").add_to(m)
+        m = folium.Map(location=[lat, lon], zoom_start=9)
+        
+        # Add all Montana peaks
+        for peak_name, coords in MONTANA_PEAKS.items():
+            folium.Marker(
+                coords,
+                popup=f"{peak_name}<br>Lat: {coords[0]:.4f}, Lon: {coords[1]:.4f}",
+                tooltip=peak_name
+            ).add_to(m)
+        
+        # Add current location marker if not a peak
+        if (lat, lon) not in MONTANA_PEAKS.values():
+            folium.Marker([lat, lon], popup=f"Selected Location<br>Lat: {lat:.4f}, Lon: {lon:.4f}").add_to(m)
+        
         m.add_child(folium.ClickForLatLng())  # Enable click events
         
         return (
@@ -112,25 +141,35 @@ def update_weather(lat, lon):
     except Exception as e:
         return f"Error: {str(e)}", None, None, None, create_map()
 
+def peak_button_click(peak_name):
+    """Handle click events for Montana peak buttons."""
+    coords = MONTANA_PEAKS[peak_name]
+    return coords[0], coords[1]
+
 # Create Gradio interface
-with gr.Blocks(title="NOAA Weather Forecast and Radar") as demo:
-    gr.Markdown("# NOAA Weather Forecast and Radar")
+with gr.Blocks(title="Montana Mountain Weather") as demo:
+    gr.Markdown("# Montana Mountain Weather")
     
     with gr.Row():
         with gr.Column(scale=1):
             lat_input = gr.Number(
                 label="Latitude",
-                value=39.8283,
+                value=45.5,
                 minimum=-90,
                 maximum=90
             )
             lon_input = gr.Number(
                 label="Longitude",
-                value=-98.5795,
+                value=-111.0,
                 minimum=-180,
                 maximum=180
             )
-            submit_btn = gr.Button("Get Weather")
+            
+            # Quick access buttons for Montana peaks
+            gr.Markdown("### Quick Access - Montana Peaks")
+            peak_buttons = {name: gr.Button(f"📍 {name}") for name in MONTANA_PEAKS.keys()}
+            
+            submit_btn = gr.Button("Get Weather", variant="primary")
         
         with gr.Column(scale=2):
             map_display = gr.HTML(create_map())
@@ -139,7 +178,7 @@ with gr.Blocks(title="NOAA Weather Forecast and Radar") as demo:
         forecast_output = gr.Textbox(
             label="Forecast",
             lines=12,
-            placeholder="Enter coordinates or click 'Get Weather' to see the forecast..."
+            placeholder="Select a location or mountain peak to see the forecast..."
         )
     
     with gr.Row():
@@ -178,11 +217,35 @@ with gr.Blocks(title="NOAA Weather Forecast and Radar") as demo:
         ]
     )
     
+    # Handle peak button clicks
+    for peak_name, button in peak_buttons.items():
+        button.click(
+            fn=peak_button_click,
+            inputs=[],
+            outputs=[lat_input, lon_input],
+            kwargs={"peak_name": peak_name}
+        ).then(
+            fn=update_weather,
+            inputs=[lat_input, lon_input],
+            outputs=[
+                forecast_output,
+                current_radar,
+                forecast_6hr,
+                forecast_12hr,
+                map_display
+            ]
+        )
+    
     gr.Markdown("""
     ## Instructions
-    1. Enter coordinates manually or click on the map to select a location
-    2. Click "Get Weather" to see the forecast and radar
-    3. View current radar and precipitation forecasts
+    1. Use the quick access buttons to check specific Montana peaks
+    2. Or enter coordinates manually / click on the map
+    3. Click "Get Weather" to see the forecast and radar
+    
+    **Montana Peaks Included:**
+    - Lone Peak (Big Sky): 45°16′41″N 111°27′01″W
+    - Sacajawea Peak: 45°53′45″N 110°58′7″W
+    - Pioneer Mountain: 45°13′55″N 111°27′2″W
     
     **Legend:**
     - Current Radar: Shows current precipitation
@@ -190,7 +253,7 @@ with gr.Blocks(title="NOAA Weather Forecast and Radar") as demo:
     - 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).
+    Mountain weather can change rapidly - always check multiple sources for safety.
     """)
 
 # For Hugging Face Spaces, we need to export the demo