Update app.py

app.py

@@ -48,6 +48,8 @@ class AirQualityMapper:
         if not api_key or api_key.strip() == "":
             return [], "❌ Please enter a valid AirNow API key"
         
+        print(f"Using API key: {api_key[:8]}..." if len(api_key) > 8 else "API key too short")
+        
         try:
             # Get data for major US cities and regions
             # We'll use a comprehensive list of state capitals and major cities
@@ -72,8 +74,9 @@ class AirQualityMapper:
             ]
             
             all_data = []
+            successful_requests = 0
             
-            for zipcode, state in locations:
+            for zipcode, state in locations[:10]:  # Start with first 10 locations for testing
                 try:
                     # Current observations endpoint
                     url = f"{self.base_url}/aq/observation/zipCode/current/"
@@ -84,24 +87,35 @@ class AirQualityMapper:
                         "API_KEY": api_key
                     }
                     
+                    print(f"Fetching data for {zipcode} ({state})...")
                     response = requests.get(url, params=params, timeout=10)
                     
+                    print(f"Response status: {response.status_code}")
+                    
                     if response.status_code == 200:
                         data = response.json()
+                        print(f"Received {len(data) if data else 0} records for {state}")
                         if data:  # If data is not empty
                             for observation in data:
                                 observation['source_state'] = state
                                 observation['source_zipcode'] = zipcode
                             all_data.extend(data)
+                            successful_requests += 1
+                    else:
+                        print(f"Error response for {zipcode}: {response.text[:200]}")
                     
                     # Add delay to respect rate limits
                     time.sleep(0.5)
                     
                 except requests.exceptions.RequestException as e:
+                    print(f"Request error for {zipcode}: {str(e)}")
                     continue  # Skip this location and continue with others
             
+            print(f"Total successful requests: {successful_requests}")
+            print(f"Total data points collected: {len(all_data)}")
+            
             if not all_data:
-                return [], "⚠️ No air quality data found. Please check your API key or try again later."
+                return [], f"⚠️ No air quality data found after {successful_requests} successful API calls. Please check your API key or try again later."
             
             # Remove duplicates based on reporting area
             seen_areas = set()
@@ -112,9 +126,10 @@ class AirQualityMapper:
                     seen_areas.add(area_key)
                     unique_data.append(item)
             
-            return unique_data, f"✅ Successfully loaded {len(unique_data)} monitoring locations"
+            return unique_data, f"✅ Successfully loaded {len(unique_data)} monitoring locations from {successful_requests} API calls"
             
         except Exception as e:
+            print(f"General error: {str(e)}")
             return [], f"❌ Error fetching data: {str(e)}"
     
     def create_map(self, data: List[Dict]) -> str:
@@ -203,7 +218,7 @@ class AirQualityMapper:
                     font-size:14px; padding: 10px">
         <h4>AQI Legend</h4>
         <p><i class="fa fa-circle" style="color:green"></i> Good (0-50)</p>
-        <p><i class="fa fa-circle" style="color:yellow"></i> Moderate (51-100)</p>
+        <p><i class="fa fa-circle" style="color:orange"></i> Moderate (51-100)</p>
         <p><i class="fa fa-circle" style="color:orange"></i> Unhealthy for Sensitive (101-150)</p>
         <p><i class="fa fa-circle" style="color:red"></i> Unhealthy (151-200)</p>
         <p><i class="fa fa-circle" style="color:purple"></i> Very Unhealthy (201-300)</p>
@@ -240,6 +255,12 @@ class AirQualityMapper:
 # Initialize the mapper
 mapper = AirQualityMapper()
 
+# Check environment variable on startup
+env_api_key = os.getenv('AIRNOW_API_KEY')
+print(f"Environment variable AIRNOW_API_KEY: {'SET' if env_api_key else 'NOT SET'}")
+if env_api_key:
+    print(f"API key starts with: {env_api_key[:8]}...")
+
 def update_map(api_key: str):
     """Update the map with fresh air quality data"""
     # Check for environment variable first, then use provided key
@@ -248,7 +269,7 @@ def update_map(api_key: str):
         api_key = env_api_key
         
     if not api_key.strip():
-        return "Please enter your AirNow API key above or set AIRNOW_API_KEY environment variable.", pd.DataFrame()
+        return "Please enter your AirNow API key above or set AIRNOW_API_KEY environment variable.", pd.DataFrame(), "❌ No API key provided"
     
     # Fetch data
     data, status = mapper.fetch_airnow_data(api_key)
@@ -259,7 +280,7 @@ def update_map(api_key: str):
     # Create data table
     df = mapper.create_data_table(data)
     
-    return map_html, df
+    return map_html, df, status
 
 # Create Gradio interface
 with gr.Blocks(title="AirNow Air Quality Sensor Map", theme=gr.themes.Soft()) as demo:
@@ -341,12 +362,9 @@ with gr.Blocks(title="AirNow Air Quality Sensor Map", theme=gr.themes.Soft()) as
     load_button.click(
         fn=update_map,
         inputs=[api_key_input],
-        outputs=[map_output, data_table]
-    ).then(
-        fn=lambda: "Map updated with latest air quality data! 🌍",
-        outputs=[status_text]
+        outputs=[map_output, data_table, status_text]
     )
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()