|
import gradio as gr |
|
import numpy as np |
|
import re |
|
|
|
|
|
dummy_data = [["Alice", "Springs"], ["Max", "Brown"]] |
|
|
|
def fetch_data(): |
|
"""Fetch data from the data store. |
|
In this case it's just an array. |
|
NOTE: This should happen at page load.""" |
|
return dummy_data |
|
|
|
def create_item(name, surname): |
|
"""Create an item in the data store""" |
|
dummy_data.append([name, surname]) |
|
return dummy_data |
|
|
|
def update_item(out): |
|
"""Update an item in the data store""" |
|
dummy_data.clear() |
|
for item in out.values.tolist(): |
|
dummy_data.append(item) |
|
|
|
if dummy_data[-1] == ["", ""]: |
|
dummy_data.pop(-1) |
|
return dummy_data |
|
|
|
def delete_item(name, surname): |
|
"""Delete an item from the data store""" |
|
if [name, surname] in dummy_data: |
|
dummy_data.remove([name, surname]) |
|
return dummy_data |
|
|
|
def filter_with_prefix(prefix): |
|
"""Return all the item whoes surnames starting with prefix |
|
else return an empty list""" |
|
filtered_data = [] |
|
for idx, lst in enumerate(dummy_data): |
|
if re.search(f"^{prefix}", lst[1], re.IGNORECASE): |
|
filtered_data.append(dummy_data[idx]) |
|
return filtered_data |
|
|
|
with gr.Blocks() as demo: |
|
gr.Markdown("Built by: {}".format(@kysmatter)) |
|
out = gr.Dataframe( |
|
headers=["Name", "Surname"], |
|
datatype=["str", "str"], |
|
col_count=(2, "fixed"), |
|
) |
|
|
|
demo.load(fetch_data, None, out) |
|
|
|
filter_prefix = gr.Text(label="Filter prefix") |
|
with gr.Row(): |
|
with gr.Column(): |
|
name = gr.Text(label="Name") |
|
surname = gr.Text(label="Surname") |
|
with gr.Row(): |
|
create_btn = gr.Button("Create") |
|
update_btn = gr.Button("Update") |
|
delete_btn = gr.Button("Delete") |
|
|
|
create_btn.click(create_item, [name, surname], out) |
|
update_btn.click(update_item, out, out) |
|
delete_btn.click(delete_item, [name, surname], out) |
|
filter_prefix.change(filter_with_prefix, filter_prefix, out) |
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|