import gradio as gr
from func_gradio import run_epcotv2

inputs = [
    gr.Dropdown([i for i in range(1, 23)] + ['X'],label='Chromosome id'),
    gr.Number(label='Interest region on the specified chromosome. Please ensure the region length is 600kb', info='From'),
    gr.Number(info='To', show_label=False),
    gr.CheckboxGroup(
        choices = ["epi","rna","bru","microc","hic","intacthic","rna_strand", "external_tf","tt","groseq","grocap","proseq","netcage","starr"],
        label = "Select the modalities you want to predict"),
    gr.File(label='Processed ATAC-seq file (in .pickle format)')

]

output = gr.File(label="Download Predictions (.pkl)")

demo = gr.Interface(fn=run_epcotv2, inputs=inputs, outputs=output, title = "Predict diverse modalities by EPCOTv2",
description="This is an app to predict diverse genomic and transcriptomic modalities in a small region using EPCOTv2 via ATAC-seq data.")
demo.launch(share=True)