import gradio as gr
import requests
import geopandas as gpd
from shapely.geometry import Point


def process_input(address, selected_option, additional_input):
    transport_analysis_needed = selected_option in [
        "Residential",
        "Office",
        "Community Facility",
    ]

    output_address = f"You entered the address:\n{address}"
    output_option = f"Selected option:\n{selected_option}"

    response = requests.get(
        f"https://geosearch.planninglabs.nyc/v2/autocomplete?text={address}"
    )
    data = response.json()
    x = data["features"][0]["geometry"]["coordinates"]

    # Load the GeoJSON file into a GeoDataFrame
    geodata = gpd.read_file("/content/zone_data.geojson")

    # Create a Point for the given coordinates
    location_point = Point(x[0], x[1])

    # Find the zone that the location point is in
    zone = geodata[geodata.geometry.contains(location_point)]["id"].values.item()

    if selected_option in ["Off-Street Parking Facility", "Residential"]:
        output_additional = f"Number of Units/Spaces:\n{additional_input}"
    else:
        output_additional = f"Area (in 1000 GSF):\n{additional_input}"

    output_transport_analysis = (
        f"Transport Analysis Needed:\n{transport_analysis_needed}"
    )

    # Replace 'Your Zone Calculation Logic' with the actual zone calculation code

    output_zone = f"Zone:\n{zone}"

    return (
        output_address,
        output_option,
        output_additional,
        output_transport_analysis,
        output_zone,
    )


iface = gr.Interface(
    fn=process_input,
    inputs=[
        gr.inputs.Textbox(label="Enter your address"),
        gr.inputs.Radio(
            [
                "Residential",
                "Office",
                "Regional Retail",
                "Local Retail",
                "Sit Down/High Turnover Restaurant",
                "Fast Food/without Drive Through",
                "Community Facility",
                "Off-Street Parking Facility",
            ],
            label="Select an option",
        ),
        gr.inputs.Number(
            label="Number of Units/Spaces or Area (in 1000 GSF)", default=1
        ),  # Default value is 1
    ],
    outputs=[
        gr.outputs.Textbox(label="Address"),
        gr.outputs.Textbox(label="Selected Option"),
        gr.outputs.Textbox(label="Number of Units/Spaces or Area"),
        gr.outputs.Textbox(label="Transport Analysis Needed"),
        gr.outputs.Textbox(label="Zone"),
    ],
)

iface.launch()