Update app.py (#3)

- Update app.py (5526681d03b01090f0a3036f555259ca6c586d5d)


Co-authored-by: Peter Hsieh <peter2520@users.noreply.huggingface.co>

app.py

@@ -1,39 +1,83 @@
 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"]
+    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}"
-    
+    output_transport_analysis = (
+        f"Transport Analysis Needed:\n{transport_analysis_needed}"
+    )
+
     # Replace 'Your Zone Calculation Logic' with the actual zone calculation code
-    zone = 'Your Zone Calculation Logic'
 
     output_zone = f"Zone:\n{zone}"
 
-    return output_address, output_option, output_additional, output_transport_analysis, output_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
+        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")
+        gr.outputs.Textbox(label="Zone"),
     ],
 )