import gradio as gr
import json

def split(value):
    return value.split(",")

def merge_texts(text1, text2):
    paket_id_by_pr_id = dict(map(split, text1.splitlines()))

    datasets = list(map(split, text2.splitlines()))
    output = { dataset[0]: {"car_id": dataset[1], "client_id": dataset[2], "origin": "team", "packet_id": paket_id_by_pr_id[dataset[0]] }  for dataset in datasets }

    return json.dumps(output, indent=2)

demo = gr.Interface(
    merge_texts,
    [
        gr.Textbox(
            label="[prereserve_id,packet_id]",
            info="""[select id,
                            case when service_id = 16106 then 16 when service_id = 16107 then 17 end as packet_id,
                            status 
                     from prereserves
                     where service_id in (16106, 16107) and status = 'ACTIVE' and client_id = 49795 order by id;]""",
            placeholder="9371,17\n9390,16\n9391,17\n9393,16\n9397,17\n...",
            lines=15,
        ),
        gr.Textbox(
            label="[prereserve_id,car_id,client_id]",
            info="SELECT prereserve_id, car_id, client_id FROM car_prereserves WHERE prereserve_id in(...)",
            placeholder="9371,25668,70633\n9390,45233,103994\n9391,45233,103994\n9393,37033,47747\n9397,12914,12358\n...",
            lines=15,
        ),
    ],
    #"textbox",
    #gr.HTML(),
    #gr.JSON(),
    gr.Code(
        language="json"
    ),
    theme=gr.themes.Base()
)

demo.launch()