File size: 2,074 Bytes
62f4c21 |
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 64 65 66 67 68 69 70 71 |
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)
# handle the empty element dataframe returns
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()
|