import importlib.metadata
import gradio as gr
from optimum_support import show_is_supported

with gr.Blocks() as app:
    gr.Markdown("# Check if model is supported by optimum-intel[openvino]")
    with gr.Column():
        model_id = gr.Textbox(label="model_id")
        process_button = gr.Button("Check")
        output_text = gr.Markdown(label="result", height=100)
        optimum_intel_version = importlib.metadata.version("optimum-intel")
        gr.Markdown(
            f"Tested with optimum-intel {optimum_intel_version}. For testing purposes only, results may be wrong."
        )

    process_button.click(show_is_supported, inputs=[model_id], outputs=output_text)
    model_id.submit(show_is_supported, inputs=[model_id], outputs=output_text)

app.launch()