import gradio as gr
from feifeilib.feifeismolvlm import feifeismolvlm

with gr.Blocks() as demo:
    gr.Markdown("Image To Flux Prompt (SmolVLM)")
    with gr.Tab(label="Image To Flux Prompt"):
        input_img = gr.Image(label="Input Picture",height=320,type="filepath")
        submit_btn = gr.Button(value="Submit")
        output_text = gr.Textbox(label="Flux Prompt")


    submit_btn.click(feifeismolvlm, [input_img], [output_text])

demo.launch(server_name="0.0.0.0")