Update app.py

app.py

@@ -53,7 +53,21 @@ def geocode_address(address):
     except Exception as e:
         return None, None, f"Geocoding error: {str(e)}"
 
-def get_pollen_data(lat, lon, days=5):
+def get_pollen_category_from_index(index_value):
+    """Convert index value to category when API doesn't provide it"""
+    if not isinstance(index_value, (int, float)):
+        return "UNKNOWN"
+    
+    if index_value <= 2.4:
+        return "VERY_LOW"
+    elif index_value <= 4.8:
+        return "LOW"
+    elif index_value <= 7.2:
+        return "MEDIUM"
+    elif index_value <= 9.6:
+        return "HIGH"
+    else:
+        return "VERY_HIGH"
     """Fetch pollen data from Google Pollen API"""
     if not GOOGLE_API_KEY:
         return "Error: Google API key not found. Please set GOOGLE_API_KEY as a secret."
@@ -83,6 +97,16 @@ def get_pollen_data(lat, lon, days=5):
         if "dailyInfo" in data and len(data["dailyInfo"]) > 0:
             print(f"First day info keys: {data['dailyInfo'][0].keys()}")
             print(f"Date structure: {data['dailyInfo'][0].get('date', 'No date field')}")
+            
+            # Debug pollen info structure
+            if "pollenTypeInfo" in data["dailyInfo"][0]:
+                for pollen in data["dailyInfo"][0]["pollenTypeInfo"]:
+                    print(f"Pollen type: {pollen.get('code', 'Unknown')}")
+                    if "indexInfo" in pollen:
+                        index_info = pollen["indexInfo"]
+                        print(f"  Index info: {index_info}")
+                        print(f"  Category: '{index_info.get('category', 'Missing')}'")
+                        print(f"  Value: {index_info.get('value', 'Missing')}")
         
         return data
     except requests.exceptions.RequestException as e:
@@ -177,7 +201,18 @@ def format_pollen_data(data):
                 pollen_type = pollen["code"].replace("_", " ").title()
                 index_info = pollen.get("indexInfo", {})
                 index_value = index_info.get("value", 0)
-                category = index_info.get("category", "Unknown")
+                api_category = index_info.get("category", "")
+                
+                # Debug: Print what we're getting from the API
+                print(f"Processing {pollen_type}: index={index_value}, api_category='{api_category}'")
+                
+                # Use fallback if API category is missing or unknown
+                if api_category and api_category in ["VERY_LOW", "LOW", "MEDIUM", "HIGH", "VERY_HIGH"]:
+                    category = api_category
+                else:
+                    # Use our fallback function
+                    category = get_pollen_category_from_index(index_value)
+                    print(f"  Using fallback category: {category}")
                 
                 if isinstance(index_value, (int, float)) and index_value > max_index:
                     max_index = index_value
@@ -189,10 +224,11 @@ def format_pollen_data(data):
                     "LOW": {"emoji": "🟡", "description": "Good", "advice": "Most people will be fine outdoors"},
                     "MEDIUM": {"emoji": "🟠", "description": "Moderate", "advice": "Consider taking allergy medication"},
                     "HIGH": {"emoji": "🔴", "description": "Poor", "advice": "Limit outdoor activities if allergic"},
-                    "VERY_HIGH": {"emoji": "🟣", "description": "Very Poor", "advice": "Stay indoors if possible"}
+                    "VERY_HIGH": {"emoji": "🟣", "description": "Very Poor", "advice": "Stay indoors if possible"},
+                    "UNKNOWN": {"emoji": "⚪", "description": "Data Unavailable", "advice": "Monitor symptoms and check again later"}
                 }
                 
-                severity = severity_map.get(category, {"emoji": "⚪", "description": "Unknown", "advice": "Monitor symptoms"})
+                severity = severity_map.get(category, severity_map["UNKNOWN"])
                 
                 # More specific pollen type descriptions
                 pollen_descriptions = {
@@ -203,11 +239,26 @@ def format_pollen_data(data):
                 
                 pollen_desc = pollen_descriptions.get(pollen_type, f"🌸 {pollen_type}")
                 
+                # Convert index to approximate grains per cubic meter
+                grains_per_m3 = "Unknown"
+                if isinstance(index_value, (int, float)):
+                    if index_value <= 2:
+                        grains_per_m3 = f"~{int(index_value * 25)} grains/m³"
+                    elif index_value <= 5:
+                        grains_per_m3 = f"~{int(50 + (index_value-2) * 33)} grains/m³"
+                    elif index_value <= 7:
+                        grains_per_m3 = f"~{int(150 + (index_value-5) * 175)} grains/m³"
+                    elif index_value <= 9:
+                        grains_per_m3 = f"~{int(500 + (index_value-7) * 250)} grains/m³"
+                    else:
+                        grains_per_m3 = f"~{int(1000 + (index_value-10) * 500)}+ grains/m³"
+                
                 output.append(f"### {pollen_desc}")
                 output.append(f"**Level:** {severity['emoji']} {severity['description']} (Index: {index_value}/10)")
+                output.append(f"**Scientific Measurement:** {grains_per_m3}")
                 output.append(f"**What this means:** {severity['advice']}")
                 
-                row_data[pollen_type] = f"{severity['description']} ({index_value})"
+                row_data[pollen_type] = f"{severity['description']} ({index_value}) - {grains_per_m3}"
                 
                 # Add plant descriptions if available
                 if "plantDescription" in pollen:
@@ -244,15 +295,22 @@ def format_pollen_data(data):
     
     # Add comprehensive advice section
     output.append("\n## 💡 Practical Tips for Pollen Season")
-    output.append("### 🏠 **Indoor Protection:**")
-    output.append("- Keep windows and doors closed during high pollen days")
+    
+    output.append("### 🔬 **Understanding the Numbers:**")
+    output.append("- **Under 100 grains/m³:** Like having a few specks of dust in a large room")
+    output.append("- **100-500 grains/m³:** Similar to light sawdust in the air - noticeable to sensitive people")
+    output.append("- **500-1000 grains/m³:** Like fine powder in the air - most people will feel it")
+    output.append("- **Over 1000 grains/m³:** Heavy particulate load - breathing through a pollen cloud")
+    
+    output.append("\n### 🏠 **Indoor Protection:**")
+    output.append("- Keep windows and doors closed during high pollen days (500+ grains/m³)")
     output.append("- Use air conditioning with clean HEPA filters")
     output.append("- Consider an air purifier for your bedroom")
     
     output.append("\n### 🚶 **When Going Outside:**")
     output.append("- Check this forecast before planning outdoor activities")
     output.append("- Wear wraparound sunglasses to protect your eyes")
-    output.append("- Consider wearing a mask during very high pollen days")
+    output.append("- Consider wearing a mask during very high pollen days (1000+ grains/m³)")
     output.append("- Plan outdoor activities for late evening when pollen counts are lower")
     
     output.append("\n### 🚿 **After Being Outdoors:**")
@@ -286,8 +344,14 @@ with gr.Blocks(title="🌸 Pollen Forecast Map", theme=gr.themes.Soft()) as app:
     gr.HTML("""
     <div style="text-align: center; padding: 20px;">
         <h1>🌸 Pollen Forecast Map</h1>
-        <p>Get detailed, easy-to-understand pollen forecasts for any location worldwide!</p>
-        <p><strong>💡 How to use:</strong> Search by address, use preset cities, or enter coordinates manually</p>
+        <p>Get detailed, scientifically-accurate pollen forecasts with actual particle concentrations!</p>
+        <div style="background-color: #e8f5e8; padding: 15px; border-radius: 10px; margin: 15px 0;">
+            <strong>🎯 How to Use:</strong>
+            <br>• <strong>Search by Address:</strong> Type any city, address, or landmark in the search box
+            <br>• <strong>Use Quick Cities:</strong> Click preset buttons for major cities worldwide  
+            <br>• <strong>Enter Coordinates:</strong> Input precise latitude/longitude values
+            <br>• <strong>Map Note:</strong> The map displays your location but doesn't support clicking to select new points
+        </div>
     </div>
     """)
     
@@ -335,9 +399,17 @@ with gr.Blocks(title="🌸 Pollen Forecast Map", theme=gr.themes.Soft()) as app:
         
         with gr.Column(scale=2):
             # Map display
+            gr.HTML("<h3>🗺️ Location Map</h3>")
+            gr.HTML("""
+            <div style="background-color: #f0f8ff; padding: 10px; border-radius: 5px; margin-bottom: 10px;">
+                <strong>📍 Note:</strong> The map shows your selected location but doesn't support clicking to select new points. 
+                Use the address search or coordinate inputs on the left to change locations.
+            </div>
+            """)
+            
             map_html = gr.HTML(
                 value=create_map(),
-                label="📍 Current Location"
+                label="Current Location"
             )
             
             # Location info display