import streamlit as st
import folium
from folium.plugins import MarkerCluster
from streamlit_folium import folium_static
import googlemaps
from datetime import datetime
import os

# Initialize Google Maps
gmaps = googlemaps.Client(key=os.getenv('GOOGLE_KEY'))

# Function to fetch directions
def get_directions_and_coords(source, destination):
    now = datetime.now()
    directions_info = gmaps.directions(source, destination, mode='driving', departure_time=now)
    if directions_info:
        steps = directions_info[0]['legs'][0]['steps']
        coords = [(step['start_location']['lat'], step['start_location']['lng']) for step in steps]
        return steps, coords
    else:
        return None, None

# Function to render map with directions
def render_folium_map(coords):
    m = folium.Map(location=[coords[0][0], coords[0][1]], zoom_start=13)
    folium.PolyLine(coords, color="blue", weight=2.5, opacity=1).add_to(m)
    return m

# Function to add medical center paths and annotate distance
def add_medical_center_paths(m, source, med_centers):
    for name, lat, lon, specialty, city in med_centers:
        _, coords = get_directions_and_coords(source, (lat, lon))
        if coords:
            folium.PolyLine(coords, color="red", weight=2.5, opacity=1).add_to(m)
            folium.Marker([lat, lon], popup=name).add_to(m)
            distance_info = gmaps.distance_matrix(source, (lat, lon), mode='driving')
            distance = distance_info['rows'][0]['elements'][0]['distance']['text']
            folium.map.Marker(
                [coords[-1][0], coords[-1][1]],
                icon=folium.DivIcon(
                    icon_size=(150, 36),
                    icon_anchor=(0, 0),
                    html=f'<div style="font-size: 10pt; color : red;">{distance}</div>',
                    )
                ).add_to(m)

# Streamlit UI
st.title('Google Maps and Texas Medical Centers 🤠')
st.sidebar.header('Directions')

source_location = st.sidebar.text_input("Source Location", "Dallas Fort Worth Airport, TX")
destination_location = st.sidebar.text_input("Destination Location", "Corpus Christi Beach, TX")

if st.sidebar.button('Get Directions'):
    steps, coords = get_directions_and_coords(source_location, destination_location)
    if steps and coords:
        st.subheader('Driving Directions:')
        for i, step in enumerate(steps):
            st.write(f"{i+1}. {step['html_instructions']}")
        st.subheader('Route on Map:')
        m1 = render_folium_map(coords)
        folium_static(m1)
    else:
        st.write("No available routes.")

# Top 10 medical centers in Texas
texas_med_centers = [
    ('MD Anderson Cancer Center', 29.7072, -95.3973, 'Cancer specialty', 'Houston'),
    ('Baylor University Medical Center', 32.7894, -96.7805, 'Teaching hospital', 'Dallas'),
    ('Methodist Hospital', 29.7099, -98.5262, 'Community hospital', 'San Antonio'),
    ('Parkland Memorial Hospital', 32.8121, -96.8103, 'Teaching hospital', 'Dallas'),
    ('UT Southwestern Medical Center', 32.8199, -96.8419, 'Teaching hospital', 'Dallas'),
    ('Texas Health Presbyterian Hospital Dallas', 32.8821, -96.7702, 'Community hospital', 'Dallas'),
    ('Memorial Hermann-Texas Medical Center', 29.7096, -95.4017, 'Teaching hospital', 'Houston'),
    ('St. Luke\'s Episcopal Hospital', 29.7096, -95.4019, 'Community hospital', 'Houston'),
    ('Texas Children\'s Hospital', 29.7075, -95.4034, 'Pediatrics', 'Houston'),
    ('Tarrant County Hospital', 32.7571, -97.3311, 'Community hospital', 'Fort Worth')
]

# Annotate distances and paths for each medical center
st.markdown("## 🏥 Texas Medical Centers 🤠")
m2 = folium.Map(location=[31.9686, -99.9018], zoom_start=6)
marker_cluster = MarkerCluster().add_to(m2)
add_medical_center_paths(m2, source_location, texas_med_centers)
folium_static(m2)