Update app.py

app.py

@@ -41,37 +41,42 @@ def get_snow_forecast(lat, lon):
         
         # Process forecast periods to extract snow data
         snow_data = []
+        
         for period in forecast_data['properties']['periods']:
+            period_data = {
+                'time': period['startTime'],
+                'temperature': period['temperature'],
+                'snowfall': 0  # Default to no snow
+            }
+            
             # Check if snow is mentioned in the forecast
             forecast_text = period['detailedForecast'].lower()
-            if 'snow' in forecast_text:
-                # Try to extract snow amount from the forecast text
-                try:
-                    # Look for patterns like "2 inches of snow" or "snow accumulation of 1 to 3 inches"
-                    import re
-                    snow_patterns = [
-                        r'(\d+(?:\.\d+)?)\s*(?:to\s*\d+(?:\.\d+)?)?\s*inches? of snow',
-                        r'snow accumulation of (\d+(?:\.\d+)?)\s*(?:to\s*\d+(?:\.\d+)?)?\s*inches',
-                    ]
-                    
-                    snow_amount = None
-                    for pattern in snow_patterns:
-                        match = re.search(pattern, forecast_text)
-                        if match:
+            
+            if any(word in forecast_text for word in ['snow', 'flurries', 'wintry mix']):
+                import re
+                # Look for snow amounts in the text
+                amount_patterns = [
+                    r'(\d+(?:\.\d+)?)\s*(?:to\s*\d+(?:\.\d+)?)?\s*inches? of snow',
+                    r'snow accumulation of (\d+(?:\.\d+)?)\s*(?:to\s*\d+(?:\.\d+)?)?\s*inches',
+                    r'accumulation of (\d+(?:\.\d+)?)\s*(?:to\s*\d+(?:\.\d+)?)?\s*inches',
+                ]
+                
+                for pattern in amount_patterns:
+                    match = re.search(pattern, forecast_text)
+                    if match:
+                        try:
                             snow_amount = float(match.group(1))
+                            period_data['snowfall'] = snow_amount / 39.37  # Convert inches to meters
                             break
-                    
-                    if snow_amount is not None:
-                        snow_data.append({
-                            'validTime': period['startTime'],
-                            'value': snow_amount / 39.37,  # Convert inches to meters
-                            'temperature': period['temperature']
-                        })
-                except Exception as e:
-                    print(f"Error parsing snow amount: {str(e)}")
-                    continue
+                        except (ValueError, IndexError):
+                            continue
+            
+            snow_data.append(period_data)
         
-        return {'snowfall': snow_data}
+        return snow_data
+    except Exception as e:
+        print(f"Error in snow forecast: {str(e)}")
+        return []
     except Exception as e:
         print(f"Error fetching snow forecast: {str(e)}")
         return None
@@ -127,10 +132,7 @@ def get_noaa_forecast(lat, lon):
             # Add snow-specific forecasts if available
             if snow_data and 'snowfall' in snow_data:
                 matching_snow = [s for s in snow_data['snowfall'] 
-                               if datetime.fromisoformat(s['validTime'].split('/')[0]) <= 
-                               datetime.fromisoformat(period['startTime']) <= 
-                               datetime.fromisoformat(s['validTime'].split('/')[0]) + 
-                               timedelta(hours=int(s['validTime'].split('/')[-1][1:-1]))]
+                               if s['validTime'] == period['startTime']]
                 
                 if matching_snow:
                     snow_amount = matching_snow[0]['value']  # in meters