import gradio as gr
import predictLLaVa

images = gr.Files(label="Image(s)", file_types=list(".png .jpg .jpeg".split()))
inputText = gr.Textbox(lines=1, label="English Instruction:")
languageDropdown = gr.Dropdown(
    choices=["en", "de", "fr", "es", "it", "nl", "ru"],
    label="Output in:",
    multiselect=False,
    value="en"
)

gr.Interface(
    fn=predictLLaVa.process_single_image,
    inputs=[inputText, images, languageDropdown],
    outputs=gr.Textbox(label="Output", show_copy_button=True),
    title="Fluffyvision",
    description="Generate a text description of an image based on an instruction.",
    allow_flagging="never"
).launch(auth=predictLLaVa.check_auth, auth_message="Please enter your username and password.")