import gradio as gr
import os

from PIL import Image
from inference import Inference

inference = Inference()


with gr.Blocks() as block:
    options = gr.Dropdown(choices=["Blip Saffal", "Blip CN"], label="Models", info="Select the model to use..", )
    # need to improve this one...

    txt = gr.Textbox(label="Insert a question...", lines=2)
    txt_3 = gr.Textbox(value="", label="Your answer is here..")
    btn = gr.Button(value="Submit")

    frame_path = os.path.join(os.path.dirname(__file__), "testing.jpg")
    frame = Image.open(frame_path)
    inference.logger.info(f"Image loaded {frame.size}")
    image = gr.Image(type="pil", value=frame)

    btn.click(inference.inference, inputs=[options, image, txt], outputs=[txt_3])

if __name__ == "__main__":
    block.launch()