# Edit Anything trained with Stable Diffusion + ControlNet + SAM  + BLIP2
import os
import gradio as gr
from diffusers.utils import load_image
from editany_lora import EditAnythingLoraModel, config_dict
from editany_demo import create_demo_template
from huggingface_hub import hf_hub_download, snapshot_download


def create_demo(process, process_image_click=None):

    examples = None
    INFO = f"""
    ## EditAnything https://github.com/sail-sg/EditAnything
    """
    WARNING_INFO = None

    demo = create_demo_template(
        process,
        process_image_click,
        examples=examples,
        INFO=INFO,
        WARNING_INFO=WARNING_INFO,
        enable_auto_prompt_default=True,
    )
    return demo


if __name__ == "__main__":
    model = EditAnythingLoraModel(
        base_model_path="runwayml/stable-diffusion-v1-5",
        controlmodel_name='LAION Pretrained(v0-4)-SD15',
        lora_model_path=None, use_blip=True, extra_inpaint=True,
    )
    demo = create_demo(model.process, model.process_image_click)
    demo.queue().launch(server_name="0.0.0.0")