Update app.py

app.py

@@ -1,134 +1,291 @@
 import gradio as gr
 import requests
-import os
 import folium
+import os
+import json
+from datetime import datetime, timedelta
+import pandas as pd
 
-# Your Google Pollen API key should be set as a secret in your Hugging Face Space
-API_KEY = os.environ.get("GOOGLE_API_KEY")
-
-# --- Map Function ---
-def create_map_html(latitude, longitude):
-    """Creates a Folium map and returns its HTML representation."""
-    try:
-        lat = float(latitude)
-        lon = float(longitude)
-        # Create a map centered on the coordinates
-        m = folium.Map(location=[lat, lon], zoom_start=12, tiles="OpenStreetMap")
-        # Add a marker to the map
-        folium.Marker([lat, lon], popup="Selected Location").add_to(m)
-        return m._repr_html_()
-    except (ValueError, TypeError):
-        # Return a default map if inputs are invalid or empty
-        m = folium.Map(location=[40.7128, -74.0060], zoom_start=5, tiles="OpenStreetMap")
-        return m._repr_html_()
-
-# --- API Call Function ---
-def get_pollen_data(latitude, longitude):
-    """
-    Fetches and formats all available pollen data from the Google Pollen API.
-    Also returns an updated map.
-    """
-    if not API_KEY:
-        error_message = "Error: GOOGLE_API_KEY not found. Please set it as a secret."
-        map_html = create_map_html(latitude, longitude)
-        return error_message, error_message, error_message, map_html
-
-    if not latitude or not longitude:
-        return "", "", "", create_map_html(latitude, longitude)
-
-    try:
-        lat = float(latitude)
-        lon = float(longitude)
-    except (ValueError, TypeError):
-        error_message = "Error: Invalid latitude or longitude."
-        return error_message, error_message, error_message, create_map_html(latitude, longitude)
+# Get API key from environment
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
-    endpoint = "https://pollen.googleapis.com/v1/forecast:lookup"
-    params = {"key": API_KEY, "location.latitude": lat, "location.longitude": lon, "days": 5, "languageCode": "en"}
+def create_map(lat=40.7128, lon=-74.0060):
+    """Create a Folium map centered at given coordinates"""
+    m = folium.Map(
+        location=[lat, lon],
+        zoom_start=10,
+        tiles="OpenStreetMap"
+    )
+    
+    # Add a marker for the selected location
+    folium.Marker(
+        [lat, lon],
+        popup=f"Selected Location: {lat:.4f}, {lon:.4f}",
+        tooltip="Click to select this location",
+        icon=folium.Icon(color='red', icon='info-sign')
+    ).add_to(m)
+    
+    return m._repr_html_()
 
+def get_pollen_data(lat, lon, days=5):
+    """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."
+    
+    # Google Pollen API endpoint
+    url = "https://pollen.googleapis.com/v1/forecast:lookup"
+    
+    # Calculate date range
+    start_date = datetime.now()
+    end_date = start_date + timedelta(days=days-1)
+    
+    params = {
+        "key": GOOGLE_API_KEY,
+        "location.longitude": lon,
+        "location.latitude": lat,
+        "days": days,
+        "plantsDescription": True
+    }
+    
     try:
-        response = requests.get(endpoint, params=params)
+        response = requests.get(url, params=params)
         response.raise_for_status()
         data = response.json()
-        map_html = create_map_html(latitude, longitude)
-
-        if not data.get('dailyInfo'):
-            no_data_message = "No pollen data found for this location."
-            return no_data_message, no_data_message, no_data_message, map_html
+        return data
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching pollen data: {str(e)}"
+    except json.JSONDecodeError as e:
+        return f"Error parsing response: {str(e)}"
 
-        outputs = {'TREE': "", 'GRASS': "", 'WEED': ""}
-        for day_info in data.get('dailyInfo', []):
-            date = day_info.get('date', {})
-            date_str = f"{date.get('year', 'N/A')}-{date.get('month', 'N/A'):02d}-{date.get('day', 'N/A'):02d}"
+def format_pollen_data(data):
+    """Format pollen data for display"""
+    if isinstance(data, str):  # Error message
+        return data, None
+    
+    if "dailyInfo" not in data:
+        return "No pollen data available for this location.", None
+    
+    # Create formatted output
+    output = []
+    output.append("# 🌸 Pollen Forecast Report\n")
+    
+    # Location info if available
+    if "regionInfo" in data:
+        region = data["regionInfo"]
+        output.append(f"**Location:** {region.get('displayName', 'Unknown')}\n")
+    
+    # Daily pollen data
+    daily_data = []
+    for day_info in data["dailyInfo"]:
+        date = day_info["date"]
+        date_obj = datetime.strptime(date, "%Y-%m-%d")
+        formatted_date = date_obj.strftime("%B %d, %Y")
+        
+        output.append(f"## 📅 {formatted_date}\n")
+        
+        if "pollenTypeInfo" in day_info:
+            pollen_types = day_info["pollenTypeInfo"]
             
-            for cat in outputs.keys():
-                outputs[cat] += f"## Date: {date_str}\n"
-
-            pollen_found_for_day = {'TREE': False, 'GRASS': False, 'WEED': False}
+            # Create a row for the dataframe
+            row_data = {"Date": formatted_date}
             
-            for pollen in day_info.get('pollenTypeInfo', []):
-                plant_type = pollen.get('plantType', 'UNKNOWN')
-                if plant_type not in outputs: continue
+            for pollen in pollen_types:
+                pollen_type = pollen["code"].replace("_", " ").title()
+                index_info = pollen.get("indexInfo", {})
+                index_value = index_info.get("value", "N/A")
+                category = index_info.get("category", "Unknown")
                 
-                pollen_found_for_day[plant_type] = True
+                # Add color coding based on category
+                color_map = {
+                    "VERY_LOW": "🟢",
+                    "LOW": "🟡", 
+                    "MEDIUM": "🟠",
+                    "HIGH": "🔴",
+                    "VERY_HIGH": "🟣"
+                }
+                color = color_map.get(category, "⚪")
                 
-                display_name = pollen.get('displayName', 'N/A')
-                index_info = pollen.get('indexInfo', {})
-                category = index_info.get('category', 'N/A')
-                color = index_info.get('color', '#FFFFFF')
-                color_swatch = f'<span style="display: inline-block; width: 15px; height: 15px; background-color: {color}; border: 1px solid #ccc; vertical-align: middle;"></span>'
+                output.append(f"**{pollen_type}:** {color} {category} (Index: {index_value})")
+                row_data[pollen_type] = f"{category} ({index_value})"
                 
-                formatted_pollen_info = f"### {display_name}\n- **Category:** {color_swatch} {category}\n"
-                outputs[plant_type] += formatted_pollen_info + "\n---\n"
-
-            for p_type, found in pollen_found_for_day.items():
-                if not found:
-                    outputs[p_type] += "- No specific pollen data returned for this category on this day.\n\n---\n"
+                # Add plant descriptions if available
+                if "plantDescription" in pollen:
+                    plants = pollen["plantDescription"]
+                    if "plants" in plants:
+                        plant_list = [plant["displayName"] for plant in plants["plants"][:3]]  # Show top 3
+                        output.append(f"  - *Main sources: {', '.join(plant_list)}*")
+                
+                output.append("")
+            
+            daily_data.append(row_data)
         
-        return outputs['TREE'].strip(), outputs['GRASS'].strip(), outputs['WEED'].strip(), map_html
+        output.append("---\n")
+    
+    # Create DataFrame for tabular view
+    df = None
+    if daily_data:
+        df = pd.DataFrame(daily_data)
+    
+    # Add legend
+    output.append("\n## 📊 Pollen Index Legend")
+    output.append("🟢 Very Low | 🟡 Low | 🟠 Medium | 🔴 High | 🟣 Very High\n")
+    
+    # Add tips
+    output.append("## 💡 Tips")
+    output.append("- Check pollen levels before outdoor activities")
+    output.append("- Take allergy medications during high pollen days")
+    output.append("- Keep windows closed during peak pollen times")
+    output.append("- Shower and change clothes after being outdoors")
+    
+    return "\n".join(output), df
 
-    except Exception as e:
-        error_message = f"An error occurred: {e}"
-        return error_message, error_message, error_message, create_map_html(latitude, longitude)
+def update_location(lat, lon):
+    """Update the map and fetch pollen data for new location"""
+    if lat is None or lon is None:
+        return create_map(), "Please select a location on the map.", None
+    
+    # Create new map
+    new_map = create_map(lat, lon)
+    
+    # Get pollen data
+    pollen_data = get_pollen_data(lat, lon)
+    formatted_data, df = format_pollen_data(pollen_data)
+    
+    return new_map, formatted_data, df
 
-# --- Gradio Interface ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤧 Pollen Forecast Map")
-    gr.Markdown("Enter coordinates and click 'Get Forecast' to see the 5-day pollen forecast and the location on the map.")
+# Create the Gradio interface
+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>Click on the map to select a location and get detailed pollen forecasts powered by Google Pollen API</p>
+    </div>
+    """)
     
     with gr.Row():
         with gr.Column(scale=1):
-            lat_input = gr.Textbox(label="Latitude", value="40.7128")
-            lon_input = gr.Textbox(label="Longitude", value="-74.0060")
-            submit_button = gr.Button("Get Forecast", variant="primary")
-            gr.Examples(
-                examples=[["34.0522", "-118.2437"], ["51.5074", "-0.1278"]],
-                inputs=[lat_input, lon_input],
-                label="Example Locations"
+            gr.HTML("<h3>📍 Location Selection</h3>")
+            
+            # Coordinate inputs
+            lat_input = gr.Number(
+                label="Latitude", 
+                value=40.7128, 
+                precision=6,
+                info="Enter latitude or click on map"
+            )
+            lon_input = gr.Number(
+                label="Longitude", 
+                value=-74.0060, 
+                precision=6,
+                info="Enter longitude or click on map"
             )
+            
+            update_btn = gr.Button("🔄 Update Location", variant="primary")
+            
+            # Preset locations
+            gr.HTML("<h4>📍 Quick Locations</h4>")
+            with gr.Row():
+                nyc_btn = gr.Button("🏙️ NYC", size="sm")
+                la_btn = gr.Button("🌴 LA", size="sm") 
+                chicago_btn = gr.Button("🌬️ Chicago", size="sm")
+                miami_btn = gr.Button("🏖️ Miami", size="sm")
         
         with gr.Column(scale=2):
-            map_output = gr.HTML(label="Location Map")
-
-    with gr.Tabs():
-        with gr.TabItem("🌳 Tree"):
-            tree_output = gr.Markdown()
-        with gr.TabItem("🌱 Grass"):
-            grass_output = gr.Markdown()
-        with gr.TabItem("🌿 Weed"):
-            weed_output = gr.Markdown()
-
-    submit_button.click(
-        fn=get_pollen_data,
+            # Map display
+            map_html = gr.HTML(
+                value=create_map(),
+                label="Interactive Map"
+            )
+    
+    with gr.Row():
+        with gr.Column():
+            # Pollen data output
+            pollen_output = gr.Markdown(
+                value="Select a location to view pollen forecast.",
+                label="Pollen Forecast"
+            )
+            
+            # Data table
+            pollen_table = gr.Dataframe(
+                label="Pollen Data Summary",
+                visible=False
+            )
+    
+    # Event handlers
+    def set_nyc():
+        return 40.7128, -74.0060
+    
+    def set_la():
+        return 34.0522, -118.2437
+    
+    def set_chicago():
+        return 41.8781, -87.6298
+    
+    def set_miami():
+        return 25.7617, -80.1918
+    
+    # Button events
+    update_btn.click(
+        fn=update_location,
         inputs=[lat_input, lon_input],
-        outputs=[tree_output, grass_output, weed_output, map_output]
+        outputs=[map_html, pollen_output, pollen_table]
     )
-    demo.load(
-        fn=get_pollen_data,
+    
+    nyc_btn.click(
+        fn=set_nyc,
+        outputs=[lat_input, lon_input]
+    ).then(
+        fn=update_location,
         inputs=[lat_input, lon_input],
-        outputs=[tree_output, grass_output, weed_output, map_output]
+        outputs=[map_html, pollen_output, pollen_table]
+    )
+    
+    la_btn.click(
+        fn=set_la,
+        outputs=[lat_input, lon_input]
+    ).then(
+        fn=update_location,
+        inputs=[lat_input, lon_input],
+        outputs=[map_html, pollen_output, pollen_table]
+    )
+    
+    chicago_btn.click(
+        fn=set_chicago,
+        outputs=[lat_input, lon_input]
+    ).then(
+        fn=update_location,
+        inputs=[lat_input, lon_input],
+        outputs=[map_html, pollen_output, pollen_table]
+    )
+    
+    miami_btn.click(
+        fn=set_miami,
+        outputs=[lat_input, lon_input]
+    ).then(
+        fn=update_location,
+        inputs=[lat_input, lon_input],
+        outputs=[map_html, pollen_output, pollen_table]
+    )
+    
+    # Auto-update when coordinates change
+    lat_input.change(
+        fn=update_location,
+        inputs=[lat_input, lon_input],
+        outputs=[map_html, pollen_output, pollen_table]
+    )
+    
+    lon_input.change(
+        fn=update_location,
+        inputs=[lat_input, lon_input],
+        outputs=[map_html, pollen_output, pollen_table]
+    )
+    
+    # Load initial data
+    app.load(
+        fn=update_location,
+        inputs=[lat_input, lon_input],
+        outputs=[map_html, pollen_output, pollen_table]
     )
 
 if __name__ == "__main__":
-    demo.launch()
-
+    app.launch()