mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
9b42ba8f10
* changes * changes * revert changes * changes * add changeset * notebooks script * changes * changes --------- Co-authored-by: Ali Abid <aliabid94@gmail.com> Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com> Co-authored-by: Ali Abdalla <ali.si3luwa@gmail.com>
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# type: ignore
|
|
import gradio as gr
|
|
import plotly.graph_objects as go
|
|
from datasets import load_dataset
|
|
|
|
dataset = load_dataset("gradio/NYC-Airbnb-Open-Data", split="train")
|
|
df = dataset.to_pandas()
|
|
|
|
def filter_map(min_price, max_price, boroughs):
|
|
|
|
filtered_df = df[(df['neighbourhood_group'].isin(boroughs)) &
|
|
(df['price'] > min_price) & (df['price'] < max_price)]
|
|
names = filtered_df["name"].tolist()
|
|
prices = filtered_df["price"].tolist()
|
|
text_list = [(names[i], prices[i]) for i in range(0, len(names))]
|
|
fig = go.Figure(go.Scattermapbox(
|
|
customdata=text_list,
|
|
lat=filtered_df['latitude'].tolist(),
|
|
lon=filtered_df['longitude'].tolist(),
|
|
mode='markers',
|
|
marker=go.scattermapbox.Marker(
|
|
size=6
|
|
),
|
|
hoverinfo="text",
|
|
hovertemplate='<b>Name</b>: %{customdata[0]}<br><b>Price</b>: $%{customdata[1]}'
|
|
))
|
|
|
|
fig.update_layout(
|
|
mapbox_style="open-street-map",
|
|
hovermode='closest',
|
|
mapbox=dict(
|
|
bearing=0,
|
|
center=go.layout.mapbox.Center(
|
|
lat=40.67,
|
|
lon=-73.90
|
|
),
|
|
pitch=0,
|
|
zoom=9
|
|
),
|
|
)
|
|
|
|
return fig
|
|
|
|
with gr.Blocks() as demo:
|
|
with gr.Column():
|
|
with gr.Row():
|
|
min_price = gr.Number(value=250, label="Minimum Price")
|
|
max_price = gr.Number(value=1000, label="Maximum Price")
|
|
boroughs = gr.CheckboxGroup(choices=["Queens", "Brooklyn", "Manhattan", "Bronx", "Staten Island"], value=["Queens", "Brooklyn"], label="Select Boroughs:")
|
|
btn = gr.Button(value="Update Filter")
|
|
map = gr.Plot()
|
|
demo.load(filter_map, [min_price, max_price, boroughs], map)
|
|
btn.click(filter_map, [min_price, max_price, boroughs], map)
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|