Spaces:
Sleeping
Sleeping
File size: 1,823 Bytes
8f8897d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
from datasets import load_dataset
import pandas as pd
import gradio as gr
# Load the Pokémon Cards dataset from Hugging Face
dataset = load_dataset("TheFusion21/PokemonCards")
df = pd.DataFrame(dataset["train"])
display_columns = ["name", "hp"] + [col for col in df.columns if col not in ["image_url", "name", "hp"]]
display_df = df[display_columns]
def style_df(df):
hp_min = df['hp'].min()
hp_max = df['hp'].max()
def color_hp(val):
norm_val = (val - hp_min) / (hp_max - hp_min)
r = int(255 * (1 - norm_val))
g = int(255 * norm_val)
return f'background-color: rgba({r}, {g}, 0, 0.2)'
styled = df.style.applymap(color_hp, subset=['hp'])
return styled
# Function to filter data based on user input
def filter_cards(set_name=None):
filtered_df = df.copy()
if set_name and set_name != "All":
filtered_df = filtered_df[filtered_df["set_name"] == set_name]
return style_df(filtered_df[display_columns])
def update_display(evt: gr.SelectData):
selected_data = df.iloc[evt.index[0]]
return selected_data["image_url"]
# Gradio interface
with gr.Blocks() as demo:
gr.Markdown("## Pokémon Cards Explorer")
with gr.Row():
set_filter = gr.Dropdown(
choices=["All"] + df["set_name"].unique().tolist(),
label="Filter by Set Name"
)
filtered_table = gr.DataFrame(style_df(display_df), show_fullscreen_button=True, show_search="search", max_chars=20, pinned_columns=1)
card_image = gr.Image(label="Card Image", height=200)
filter_button = gr.Button("Apply Filters")
filter_button.click(
filter_cards,
inputs=[set_filter],
outputs=filtered_table
)
filtered_table.select(
update_display,
None,
card_image
)
demo.launch()
|