import streamlit as st
import requests
from geopy.geocoders import Nominatim
from dotenv import load_dotenv
import os
import feedparser

load_dotenv()

# Load API keys from environment variables
ORS_API_KEY = os.getenv("ORS_API_KEY")
OPENWEATHER_API_KEY = os.getenv("OPENWEATHER_API_KEY")

# Initialize geolocator
geolocator = Nominatim(user_agent="commute_planner")

# Streamlit UI
st.title("🚗 Free Commute Planner")
st.markdown("Estimate travel time, check weather, AQI, and traffic-related news — no credit card needed!")

# Inputs
source = st.text_input("Enter your current location:")
destination = st.text_input("Enter your destination:")

if st.button("Plan My Commute") and source and destination:
    with st.spinner("Fetching route and environment info..."):

        def get_coords(location):
            try:
                loc = geolocator.geocode(location)
                return (loc.latitude, loc.longitude)
            except:
                return None

        src_coords = get_coords(source)
        dest_coords = get_coords(destination)

        if not src_coords or not dest_coords:
            st.error("Could not find coordinates. Please enter valid locations.")
        else:
            # ✅ Travel Time via OpenRouteService
            ors_url = "https://api.openrouteservice.org/v2/directions/driving-car"
            headers = {
                "Authorization": ORS_API_KEY,
                "Content-Type": "application/json"
            }
            data = {
                "coordinates": [
                    [src_coords[1], src_coords[0]],  # (lon, lat)
                    [dest_coords[1], dest_coords[0]]
                ]
            }
            response = requests.post(ors_url, json=data, headers=headers)

            if response.status_code == 200:
                result = response.json()
                try:
                    duration_sec = result['features'][0]['properties']['segments'][0]['duration']
                    duration_min = round(duration_sec / 60, 2)
                    st.success(f"🕒 Estimated Travel Time: {duration_min} minutes")
                except (KeyError, IndexError):
                    st.error("Received a response but couldn't parse travel time.")
                    st.markdown("**Response from OpenRouteService:**")
                    st.json(result)
            else:
                st.error(f"OpenRouteService API error: {response.status_code}")
                try:
                    st.json(response.json())
                except:
                    st.write(response.text)

            # ✅ Weather from OpenWeatherMap
            weather_url = "https://api.openweathermap.org/data/2.5/weather"
            weather_params = {
                "lat": src_coords[0],
                "lon": src_coords[1],
                "appid": OPENWEATHER_API_KEY,
                "units": "metric"
            }
            weather = requests.get(weather_url, params=weather_params).json()
            temp = weather.get("main", {}).get("temp", "N/A")
            st.info(f"🌡️ Current Temperature: {temp}°C")

            # ✅ Air Quality Index from OpenWeatherMap
            aqi_url = "https://api.openweathermap.org/data/2.5/air_pollution"
            aqi_params = {
                "lat": src_coords[0],
                "lon": src_coords[1],
                "appid": OPENWEATHER_API_KEY
            }
            aqi_data = requests.get(aqi_url, params=aqi_params).json()
            aqi = aqi_data.get("list", [{}])[0].get("main", {}).get("aqi", "N/A")
            aqi_levels = {
                1: "Good",
                2: "Fair",
                3: "Moderate",
                4: "Poor",
                5: "Very Poor"
            }
            st.info(f"🌀 Air Quality Index: {aqi} - {aqi_levels.get(aqi, 'Unknown')}")

            # ✅ Traffic News via RSS Feed
            st.subheader("📰 Traffic-Related News")
            rss_url = "https://www.rssmix.com/u/17853882/rss.xml"  # Replace with your preferred traffic-related feed
            feed = feedparser.parse(rss_url)

            count = 0
            for entry in feed.entries:
                if any(keyword in entry.title.lower() for keyword in ["traffic", "accident", "road", "construction"]):
                    st.markdown(f"[{entry.title}]({entry.link})")
                    count += 1

            if count == 0:
                st.write("No recent traffic-related updates found.")