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()