import gradio as gr
from PIL import Image  # For image handling

# Replace with paths or loading functions for your specific models
def load_model_1():
    # ... load your first model 
    return model_1 

def load_model_2():
    # ... load your second model 
    return model_2

def load_model_3():
    # ... load your third model 
    return model_3

def generate_caption(model, image):
    # ... perform inference with your model 
    return caption

models = [load_model_1(), load_model_2(), load_model_3()]

with gr.Blocks() as demo:
    with gr.Row():
        image = gr.Image(label="Upload Chest X-ray")
    with gr.Row():   
        gr.Radio(["Model 1", "Model 2", "Model 3"], label="Select Model")
    with gr.Row():
        caption = gr.Textbox(label="Generated Caption") 

    image.change(
        fn=generate_caption, 
        inputs=[image, gr.inputs.Radio], 
        outputs=caption
    )

demo.launch()