import gradio as gr

from foldingdiff import sampling

def sample_at_length(l:int):
    l = int(l)
    s = sampling.sample_simple("wukevin/foldingdiff_cath", n=1, sweep_lengths=(l, l+1))
    return s[0]

interface = gr.Interface(
    fn=sample,
    inputs=gr.Number(value=55, label="Protein backbone length to generate", show_label=True, precision=0),
    outputs=gr.Dataframe(),
)
interface.launch()