%%capture
import warnings
warnings.filterwarnings("ignore")

from datetime import datetime
import altair as alt
import branca
import branca.colormap as cm
import calitp_data_analysis.magics
import geopandas as gpd
import pandas as pd
from IPython.display import HTML, Image, Markdown, display, display_html

from shared_utils import webmap_utils
from update_vars import GTFS_DATA_DICT, file_name, analysis_month

from omegaconf import OmegaConf
readable_dict = OmegaConf.load("new_readable.yml")

import google.auth
credentials, project = google.auth.default()
import _new_operator_report_utils as utils

from calitp_data_analysis.gcs_pandas import GCSPandas
from functools import cache

@cache
def gcs_pandas():
    return GCSPandas()

pd.options.display.max_columns = 100
pd.options.display.float_format = "{:.2f}".format
pd.set_option("display.max_rows", None)
pd.set_option("display.max_colwidth", None)

def formatted(number):
    return "{:,}".format(number)

# analysis_name = "Alameda-Contra Costa Transit District"

# Parameters
analysis_name = "Lake Transit Authority"

%%capture_parameters
analysis_name

# Set drop down menu to be on the upper right for the charts
display(
    HTML(
        """
<style>
form.vega-bindings {
  position: absolute;
  right: 0px;
  top: 0px;
}
</style>
"""
    )
)

GCS_PATH = f"{GTFS_DATA_DICT.gcs_paths.DIGEST_GCS}processed/"

analysis_name_edited = analysis_name.replace(" ","_").lower()
operator_webmap_file = f"{analysis_name_edited}_routes"

dt = datetime.strptime(analysis_month, "%Y-%m-%d")
analysis_month_for_filtering = dt.strftime("%m/%Y")

fct_monthly_routes_url = f"{GCS_PATH}{GTFS_DATA_DICT.gtfs_digest_rollup.route_map}_{file_name}.parquet"

schedule_rt_route_direction_summary_url = f"{GCS_PATH}{GTFS_DATA_DICT.gtfs_digest_rollup.schedule_rt_route_direction}_{file_name}.parquet"

operator_hourly_summary_url = f"{GCS_PATH}{GTFS_DATA_DICT.gtfs_digest_rollup.hourly_day_type_summary}_{file_name}.parquet"

ntd_url = f"{GCS_PATH}{GTFS_DATA_DICT.gtfs_digest_rollup.ntd_profile}_{file_name}.parquet"

operator_summary_url = f"{GCS_PATH}{GTFS_DATA_DICT.gtfs_digest_rollup.operator_summary}_{file_name}.parquet"

fct_monthly_route_df = gpd.read_parquet(fct_monthly_routes_url,
                                        filters=[[("Analysis Name", "==", analysis_name)]],
                                        storage_options={"token": credentials.token}).reset_index()

fct_monthly_route_df['Number'] = fct_monthly_route_df.index

operator_hourly_summary_df = gcs_pandas().read_parquet(operator_hourly_summary_url,
                                             filters=[[("Analysis Name", "==", analysis_name),
                                                      ("Departure Hour", "<=", 24)]])

schedule_rt_route_direction_summary_df = gcs_pandas().read_parquet(schedule_rt_route_direction_summary_url,
                                                         filters=[[("Analysis Name", "==", analysis_name)]])

one_month_schedule_rt_route_direction_summary_df = (schedule_rt_route_direction_summary_df.loc[schedule_rt_route_direction_summary_df["Day Type"] == "Weekday"]
    .sort_values(by = ["Date", "Route"], ascending = [False, True])
    .drop_duplicates(subset = ["Route"])
                                                   )

ntd_profile_df =  gcs_pandas().read_parquet(ntd_url, filters=[[("analysis_name", "==", analysis_name)]])

operator_df = gcs_pandas().read_parquet(
    operator_summary_url,
    filters=[
        ("Day Type", "==", "Weekday"),
        ("Analysis Name", "==", analysis_name)],
).drop_duplicates(subset = ["Analysis Name", "Date"])

try:
    service_area = formatted(int(ntd_profile_df.service_area_sq_miles.values[0]))
    service_pop = formatted(int(ntd_profile_df.service_area_pop.values[0]))
except:
    pass

try:
    display(
        Markdown(
            f"""{analysis_name} is headquartered in <b>{ntd_profile_df.hq_county.values[0]}</b> County in the Urbanized Area of <b>{ntd_profile_df.primary_uza_name.values[0]}</b>.<br>
            This operator provides <b>{service_area}</b> square miles of public transit service, which has a service population of <b>{service_pop}</b>.<br>
            This organization is a {ntd_profile_df.reporter_type.values[0]}.<br>
            <b>Data Source</b>: <a href="https://www.transit.dot.gov/ntd/data-product/2022-annual-database-agency-information">National Transit Database</a> Annual Agency Information.
            """
        )
    )
except:
    pass

try:
    n_routes = formatted(one_month_schedule_rt_route_direction_summary_df["Route"].nunique())
    display(
        Markdown(
            f"""{analysis_name} runs <b>{n_routes}</b> unique routes. Below is the breakdown of the routes. Routes can belong to one or more categories.<p>
            Route categories are determined using a approach that looks at GTFS trips data
        alongside National Association of City Transportation Officials (NACTO)'s
        <a href="https://nacto.org/publication/transit-street-design-guide/introduction/service-context/transit-route-types/">Transit Route Types</a> 
        and <a href= "https://nacto.org/publication/transit-street-design-guide/introduction/service-context/transit-frequency-volume/">Frequency and Volume</a>
        guides. Please see the <a href="https://github.com/cal-itp/data-analyses/blob/main/gtfs_digest/methodology.md">methodology docs</a> for more details on this approach.
        """
        )
    )
except:
    display(Markdown(f"""{analysis_name} doesn't have an operator profile."""))

try:
    display(utils.create_route_typology(one_month_schedule_rt_route_direction_summary_df))
except:
    display(Markdown(f"""{analysis_name} doesn't have information on route typologies."""))

try:
    display(utils.create_route_lengths(fct_monthly_route_df))
except:
    display(Markdown(f"""{analysis_name} doesn't have information on route lengths."""))

color_map = cm.linear.Spectral_11.scale(0, len(fct_monthly_route_df))

try:
    operator_map = webmap_utils.set_state_export(
    fct_monthly_route_df.drop_duplicates(subset = ["Route Name"])[["Route Name","Geometry", "Number"]],
    subfolder = "operator_digest/",
    filename=operator_webmap_file,
    map_title=f"Transit Routes for {analysis_name}",
    cmap=color_map,
    color_col="Number",
    overwrite = True
    )
    webmap_utils.render_spa_link(operator_map["spa_link"], text=f"Routes for {analysis_name}")
    webmap_utils.display_spa_map(operator_map["spa_link"])
except:
    display(Markdown(f"""{analysis_name} doesn't have an route geographies."""))

try:
    display(utils.create_hourly_summary(operator_hourly_summary_df, "Weekday"))
    display(utils.create_hourly_summary(operator_hourly_summary_df, "Saturday"))
    display(utils.create_hourly_summary(operator_hourly_summary_df, "Sunday"))
except: 
    display(Markdown(f"""{analysis_name} doesn't have service hours information."""))

try:
    display(utils.create_tu_minute(operator_df))
    display(utils.create_tu_pct(operator_df))
except:
    display(Markdown(f"""{analysis_name} doesn't have trip update information."""))

try:
    display(utils.create_vp_minute(operator_df))
    display(utils.create_vp_pct(operator_df))
except:
    display(Markdown(f"""{analysis_name} doesn't have vehicle positions information."""))

try:
    display(utils.create_scheduled_minutes(schedule_rt_route_direction_summary_df))
    display(utils.create_frequency(schedule_rt_route_direction_summary_df))
    display(utils.create_text_graph(schedule_rt_route_direction_summary_df))
except:
    display(Markdown(f"""{analysis_name} doesn't have detailed route information."""))