import gradio as gr
from gradio_molecule3d import Molecule3D
from gradio_cofoldinginput import CofoldingInput

def predict(jobname, inputs, recycling_steps, sampling_steps, diffusion_samples):
    return ""

with gr.Blocks() as blocks:
    gr.Markdown("# Boltz-1")
    with gr.Tab("Main"):
        jobname = gr.Textbox(label="Jobname")
        inp = CofoldingInput(label="Input")
        out = Molecule3D(label="Output")
    with gr.Tab("Settings"):
        recycling_steps =gr.Slider(value=3, minimum=0)
        sampling_steps = gr.Slider(value=200, minimum=0)
        diffusion_samples = gr.Slider(value=1)

    btn = gr.Button("predict")

    btn.click(fn=predict, inputs=[jobname,inp, recycling_steps, sampling_steps, diffusion_samples], outputs=[out],  api_name="predict")

blocks.launch(ssr_mode=False)