# CUDA_VISIBLE_DEVICES=0 python server.py

import importlib
import gradio as gr


def run(model_id, prompt):
    print(f"{model_id}: {prompt}")
    m = importlib.import_module(model_id)
    print(m)
    image = m.t2i(prompt)
    return image

def app():
    model_id = gr.Textbox(label="model-id")
    prompt = gr.Textbox(label="prompt")
    image = gr.Image(label="output")
    interface = gr.Interface(
        fn=run,
        inputs=[model_id, prompt],
        outputs=image,
    )
    interface.launch()

if __name__ == "__main__":
    app()