UT_Hackathon2 / app.py
ljrmary's picture
Update app.py
e4b11c3
import gradio as gr
import requests
import geopandas as gpd
from shapely.geometry import Point
def process_input(address, selected_option, additional_input, num_units, temporal_split, modal_split, directional_distribution):
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("./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()
output_additional = (
f"Number of Units/Spaces:\n{additional_input}"
if selected_option in ["Off-Street Parking Facility", "Residential"]
else f"Area (in 1000 GSF):\n{additional_input}"
)
output_num_units = f"Number of Units:\n{num_units}"
output_temporal_split = f"Temporal Split:\n{temporal_split}"
output_modal_split = f"Modal Split:\n{modal_split}"
output_directional_distribution = f"Directional Distribution:\n{directional_distribution}"
output_transport_analysis = f"Transport Analysis Needed:\n{transport_analysis_needed}"
return (
output_address,
output_option,
output_additional,
output_num_units,
output_temporal_split,
output_modal_split,
output_directional_distribution,
output_transport_analysis,
f"Zone:\n{zone}",
)
iface = gr.Interface(
fn=process_input,
inputs=[
gr.inputs.Textbox(label="Address", type="text"),
gr.inputs.Radio(
["Office", "Local Retail", "Exhibit Space", "Restaurant"], label="Land Use Selection"
),
gr.inputs.Number(
label="Number of Units/Spaces or Area (in 1000 GSF)", default=1
),
gr.inputs.Textbox(label="Number of Units", type="text"),
gr.inputs.Radio(["morning", "midday", "afternoon"], label="Temporal Split"),
gr.inputs.Radio(["car", "taxi", "bus", "subway", "walking"], label="Modal Split"),
gr.inputs.Dropdown(["inbound", "outbound"], label="Directional Distribution"),
],
outputs=[
gr.outputs.Textbox(label="Address"),
gr.outputs.Textbox(label="Land Use Selection"),
gr.outputs.Textbox(label="Number of Units/Spaces or Area"),
gr.outputs.Textbox(label="Number of Units"),
gr.outputs.Textbox(label="Temporal Split"),
gr.outputs.Textbox(label="Modal Split"),
gr.outputs.Textbox(label="Directional Distribution"),
gr.outputs.Textbox(label="Transport Analysis Needed"),
gr.outputs.Textbox(label="Zone"),
],
)
iface.launch()